From 744aa09f5218f7db206af2e1945e23f90df46f1f Mon Sep 17 00:00:00 2001 From: Joseph Huber Date: Mon, 6 Dec 2021 11:14:54 -0500 Subject: [PATCH] [OpenMP] Make reduction functions SPMD compatible Reduction functions were guarded before which was wrong, these are SPMD compatible. Reviewed By: jdoerfert Differential Revision: https://reviews.llvm.org/D115159 --- llvm/lib/Transforms/IPO/OpenMPOpt.cpp | 3 +++ 1 file changed, 3 insertions(+) diff --git a/llvm/lib/Transforms/IPO/OpenMPOpt.cpp b/llvm/lib/Transforms/IPO/OpenMPOpt.cpp index e20540a90899..f289e3ecc979 100644 --- a/llvm/lib/Transforms/IPO/OpenMPOpt.cpp +++ b/llvm/lib/Transforms/IPO/OpenMPOpt.cpp @@ -3964,6 +3964,9 @@ struct AAKernelInfoCallSite : AAKernelInfo { case OMPRTL___kmpc_master: case OMPRTL___kmpc_end_master: case OMPRTL___kmpc_barrier: + case OMPRTL___kmpc_nvptx_parallel_reduce_nowait_v2: + case OMPRTL___kmpc_nvptx_teams_reduce_nowait_v2: + case OMPRTL___kmpc_nvptx_end_reduce_nowait: break; case OMPRTL___kmpc_distribute_static_init_4: case OMPRTL___kmpc_distribute_static_init_4u: