[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