[libc-commits] [libc] d483824 - [libc][amdgpu] Tolerate different install directories for hsa.h

[Jon Chesterfield via libc-commits]

Author: Jon Chesterfield
Date: 2023-07-20T13:43:17+01:00
New Revision: d483824fc8edc3463970beeb7b6c9cb3991bd510

URL: https://github.com/llvm/llvm-project/commit/d483824fc8edc3463970beeb7b6c9cb3991bd510
DIFF: https://github.com/llvm/llvm-project/commit/d483824fc8edc3463970beeb7b6c9cb3991bd510.diff

LOG: [libc][amdgpu] Tolerate different install directories for hsa.h

HSA headers might be under a hsa/ directory or might not.
This scheme matches the one used by the openmp amdgpu plugin.

Reviewed By: jhuber6, jplehr

Differential Revision: https://reviews.llvm.org/D155812

Added: 
    

Modified: 
    libc/utils/gpu/loader/amdgpu/Loader.cpp

Removed: 
    


################################################################################
diff  --git a/libc/utils/gpu/loader/amdgpu/Loader.cpp b/libc/utils/gpu/loader/amdgpu/Loader.cpp
index b49d576a7f63e4..c80c4e6c8ac0f0 100644
--- a/libc/utils/gpu/loader/amdgpu/Loader.cpp
+++ b/libc/utils/gpu/loader/amdgpu/Loader.cpp
@@ -15,8 +15,18 @@
 
 #include "Loader.h"
 
-#include <hsa/hsa.h>
-#include <hsa/hsa_ext_amd.h>
+#if defined(__has_include)
+#if __has_include("hsa/hsa.h")
+#include "hsa/hsa.h"
+#include "hsa/hsa_ext_amd.h"
+#elif __has_include("hsa.h")
+#include "hsa.h"
+#include "hsa_ext_amd.h"
+#endif
+#else
+#include "hsa/hsa.h"
+#include "hsa/hsa_ext_amd.h"
+#endif
 
 #include <cstdio>
 #include <cstdlib>