message: [ROCm] Adding ROCm support for reduction gpu kernels
diff: @@ -16,7 +16,7 @@ limitations under the License.
  #ifndef TENSORFLOW_CORE_KERNELS_REDUCTION_GPU_KERNELS_CU_H_
  #define TENSORFLOW_CORE_KERNELS_REDUCTION_GPU_KERNELS_CU_H_
  -#if GOOGLE_CUDA
  +#if GOOGLE_CUDA || TENSORFLOW_USE_ROCM
  #define EIGEN_USE_GPU
  @@ -1071,6 +1071,6 @@ struct ReduceFunctor<GPUDevice, Eigen::internal::OrReducer> {
  } // namespace functor
  } // namespace tensorflow
  -#endif // GOOGLE_CUDA
  +#endif // GOOGLE_CUDA || TENSORFLOW_USE_ROCM
  #endif // TENSORFLOW_CORE_KERNELS_REDUCTION_GPU_KERNELS_CU_H_
  