message: Disabling non-GPU subtests on the ROCm platform
diff: @@ -315,7 +315,7 @@ TEST_F(DeviceResDistTest, Workers2Devices2) {
  ValidateCollectiveParams(num_workers, num_devices);
  }
  -#ifndef GOOGLE_CUDA
  +#if !GOOGLE_CUDA && !TENSORFLOW_USE_ROCM
  namespace {
  // A mock NcclReducer for testing group runtime details initialization with CPU
  // builds. The only meaningful function in this class is
  