From d483824fc8edc3463970beeb7b6c9cb3991bd510 Mon Sep 17 00:00:00 2001 From: Jon Chesterfield Date: Thu, 20 Jul 2023 13:43:17 +0100 Subject: [PATCH] [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 --- libc/utils/gpu/loader/amdgpu/Loader.cpp | 14 ++++++++++++-- 1 file changed, 12 insertions(+), 2 deletions(-) diff --git a/libc/utils/gpu/loader/amdgpu/Loader.cpp b/libc/utils/gpu/loader/amdgpu/Loader.cpp index b49d576..c80c4e6 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 -#include +#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 #include -- 2.7.4