From 94ec15c3b09d0c211b1624b729b6e1e3c3d329fd Mon Sep 17 00:00:00 2001
From: rsanthanam-amd <70280935+rsanthanam-amd@users.noreply.github.com>
Date: Wed, 3 Aug 2022 12:31:33 -0500
Subject: [PATCH] [ROCm] Fix build failure. (#3630)

Signed-off-by: Rohit Santhanam <rohit.santhanam@amd.com>
---
 horovod/common/controller.cc | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/horovod/common/controller.cc b/horovod/common/controller.cc
index c844467bae..6d68f89a0e 100644
--- 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