[OpenMP][FIX] Repair accidental replacement of _shfl_sync with _shfl
This was broken accidentally in D95752.
Reviewed By: ye-luo
Differential Revision: https://reviews.llvm.org/D98677
GitOrigin-RevId: 0a954a528b8702681118cfb773d651c1860aee36
diff --git a/libomptarget/deviceRTLs/common/include/target/shuffle.h b/libomptarget/deviceRTLs/common/include/target/shuffle.h
index e274aa9..5841b11 100644
--- a/libomptarget/deviceRTLs/common/include/target/shuffle.h
+++ b/libomptarget/deviceRTLs/common/include/target/shuffle.h
@@ -85,13 +85,13 @@
inline int32_t __kmpc_impl_shfl_sync(uint64_t Mask, int32_t Var,
int32_t SrcLane) {
- return __nvvm_shfl_idx_i32(Var, SrcLane, 0x1f);
+ return __nvvm_shfl_sync_idx_i32(Mask, Var, SrcLane, 0x1f);
}
inline int32_t __kmpc_impl_shfl_down_sync(uint64_t Mask, int32_t Var,
uint32_t Delta, int32_t Width) {
int32_t T = ((GetWarpSize() - Width) << 8) | 0x1f;
- return __nvvm_shfl_down_i32(Var, Delta, T);
+ return __nvvm_shfl_sync_down_i32(Mask, Var, Delta, T);
}
#pragma omp end declare variant