From 1224fa6da5e18b4db806d06c4620117416dab9cd Mon Sep 17 00:00:00 2001 From: "Yaxun (Sam) Liu" Date: Mon, 23 Dec 2019 08:54:54 -0500 Subject: [PATCH] Add macro __HIP_ENABLE_CUDA_WRAPPER_FOR_OPENMP__ (#1761) This is to allow force enable cuda wrapper for OpenMP for flexibility [ROCm/clr commit: 35988729f35d2241f0163aa5b2dec823b6810c89] --- projects/clr/hipamd/include/hip/hcc_detail/hip_runtime.h | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/projects/clr/hipamd/include/hip/hcc_detail/hip_runtime.h b/projects/clr/hipamd/include/hip/hcc_detail/hip_runtime.h index fbba1419fa..4240dd7259 100644 --- a/projects/clr/hipamd/include/hip/hcc_detail/hip_runtime.h +++ b/projects/clr/hipamd/include/hip/hcc_detail/hip_runtime.h @@ -506,7 +506,7 @@ hc_get_workitem_absolute_id(int dim) #endif // Support std::complex. -#ifndef _OPENMP +#if !_OPENMP || __HIP_ENABLE_CUDA_WRAPPER_FOR_OPENMP__ #pragma push_macro("__CUDA__") #define __CUDA__ #include <__clang_cuda_math_forward_declares.h> @@ -516,7 +516,7 @@ hc_get_workitem_absolute_id(int dim) #include #undef __CUDA__ #pragma pop_macro("__CUDA__") -#endif // ndef _OPENMP +#endif // !_OPENMP || __HIP_ENABLE_CUDA_WRAPPER_FOR_OPENMP__ #endif // defined(__clang__) && defined(__HIP__)