Fix HIP-Clang build with HSA headers

HIP-Clang does not include these HSA headers, and they need to be explicitly added in RCCL.
2020-04-03 17:58:23 -04:00
parent 3cbe5c8a40
commit a95090d981
2 changed files with 8 additions and 0 deletions
@@ -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)
 {