Merge pull request #187 from aaronenyeshi/fix-hip-vdi-hsa-ext

Fix HIP-Clang build with HSA headers
ROCm · Apr 3, 2020 · fa52d4f · fa52d4f
2 parents 3cbe5c8 + a95090d
commit fa52d4f
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git 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
@@ -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)
 {