[ROCm] Fix build failure. (#3630)

Signed-off-by: Rohit Santhanam <rohit.santhanam@amd.com>
horovod · Aug 3, 2022 · 94ec15c · 94ec15c
1 parent 8f450ab
commit 94ec15c
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