jhuber6 updated this revision to Diff 371153. jhuber6 added a comment. Adding `no_openmp` assumption as well.
Repository: rG LLVM Github Monorepo CHANGES SINCE LAST ACTION https://reviews.llvm.org/D108958/new/ https://reviews.llvm.org/D108958 Files: clang/lib/Headers/__clang_cuda_libdevice_declares.h Index: clang/lib/Headers/__clang_cuda_libdevice_declares.h =================================================================== --- clang/lib/Headers/__clang_cuda_libdevice_declares.h +++ clang/lib/Headers/__clang_cuda_libdevice_declares.h @@ -16,6 +16,7 @@ #if defined(__OPENMP_NVPTX__) #define __DEVICE__ +#pragma omp begin assumes ext_spmd_amenable no_openmp #elif defined(__CUDA__) #define __DEVICE__ __device__ #endif @@ -456,6 +457,11 @@ __DEVICE__ float __nv_y1f(float __a); __DEVICE__ float __nv_ynf(int __a, float __b); __DEVICE__ double __nv_yn(int __a, double __b); + +#if defined(__OPENMP_NVPTX__) +#pragma omp end assumes ext_spmd_amenable, omp_no_openmp +#endif + #if defined(__cplusplus) } // extern "C" #endif
Index: clang/lib/Headers/__clang_cuda_libdevice_declares.h =================================================================== --- clang/lib/Headers/__clang_cuda_libdevice_declares.h +++ clang/lib/Headers/__clang_cuda_libdevice_declares.h @@ -16,6 +16,7 @@ #if defined(__OPENMP_NVPTX__) #define __DEVICE__ +#pragma omp begin assumes ext_spmd_amenable no_openmp #elif defined(__CUDA__) #define __DEVICE__ __device__ #endif @@ -456,6 +457,11 @@ __DEVICE__ float __nv_y1f(float __a); __DEVICE__ float __nv_ynf(int __a, float __b); __DEVICE__ double __nv_yn(int __a, double __b); + +#if defined(__OPENMP_NVPTX__) +#pragma omp end assumes ext_spmd_amenable, omp_no_openmp +#endif + #if defined(__cplusplus) } // extern "C" #endif
_______________________________________________ cfe-commits mailing list cfe-commits@lists.llvm.org https://lists.llvm.org/cgi-bin/mailman/listinfo/cfe-commits