From a95090d981834584ba129a6d2a37c92ba10816b0 Mon Sep 17 00:00:00 2001
From: Aaron Enye Shi <enye.shi@gmail.com>
Date: Fri, 3 Apr 2020 17:58:23 -0400
Subject: [PATCH] Fix HIP-Clang build with HSA headers

HIP-Clang does not include these HSA headers, and they need to be explicitly added in RCCL.
---
 src/graph/xml.cc                      | 4 ++++
 tools/TransferBench/TransferBench.cpp | 4 ++++
 2 files changed, 8 insertions(+)

diff --git a/src/graph/xml.cc b/src/graph/xml.cc
index a2b3a77d38..900a1ddfc1 100644
--- a/src/graph/xml.cc
+++ b/src/graph/xml.cc
@@ -13,6 +13,10 @@
 #include "core.h"
 #include "nvmlwrap.h"
 #include "xml.h"
+#if defined(__HIP_PLATFORM_HCC__) || defined(__HCC__) || defined(__HIPCC__)
+#include <hsa/hsa.h>
+#include <hsa/hsa_ext_amd.h>
+#endif
 
 /*******************/
 /* XML File Parser */
diff --git a/tools/TransferBench/TransferBench.cpp b/tools/TransferBench/TransferBench.cpp
index 1b41413cd3..447eb6ca96 100644
--- a/tools/TransferBench/TransferBench.cpp
+++ b/tools/TransferBench/TransferBench.cpp
@@ -34,6 +34,10 @@ THE SOFTWARE.
 #include <hip/hip_runtime.h>
 #include "copy_kernel.h"
 #include "TransferBench.hpp"
+#if defined(__HIP_PLATFORM_HCC__) || defined(__HCC__) || defined(__HIPCC__)
+#include <hsa/hsa.h>
+#include <hsa/hsa_ext_amd.h>
+#endif
 
 int main(int argc, char **argv)
 {