[Openmp-commits] [llvm] [openmp] Reapply "[OpenMP][OMPX] Add shfl_down_sync (#93311)" (PR #94139)

Shilei Tian via Openmp-commits openmp-commits at lists.llvm.org
Mon Jun 3 08:14:42 PDT 2024


================
@@ -9,6 +9,12 @@
 #ifndef __OMPX_H
 #define __OMPX_H
 
+#ifdef __AMDGCN_WAVEFRONT_SIZE
+#define __WARP_SIZE __AMDGCN_WAVEFRONT_SIZE
+#else
+#define __WARP_SIZE 32
----------------
shiltian wrote:

For NVIDIA GPU it is 32. For AMDGPU, it is `__AMDGCN_WAVEFRONT_SIZE`.

https://github.com/llvm/llvm-project/pull/94139


More information about the Openmp-commits mailing list