Fix HIP-Clang build with HSA headers

HIP-Clang does not include these HSA headers, and they need to be explicitly added in RCCL.


[ROCm/rccl commit: a95090d981]
This commit is contained in:
Aaron Enye Shi
2020-04-03 17:58:23 -04:00
parent 857fcc5015
commit bfbfe370c3
2 changed files with 8 additions and 0 deletions
+4
View File
@@ -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 */
@@ -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)
{