adapt to GPU cluster