[ROCm] Fix build failure. (horovod#3630)

Signed-off-by: Rohit Santhanam <rohit.santhanam@amd.com> Signed-off-by: Lee Yang <leey@nvidia.com>
leewyang · Aug 5, 2022 · d818fe1 · d818fe1
1 parent 5b22a73
commit d818fe1
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/horovod/common/controller.cc b/horovod/common/controller.cc
@@ -29,6 +29,8 @@
 
 #if HAVE_CUDA
 #include "ops/cuda/cuda_kernels.h"
+#elif HAVE_ROCM
+#include "ops/rocm/hip_kernels.h"
 #endif