[OpenMP] Make reduction functions SPMD compatible
authorJoseph Huber <jhuber6@vols.utk.edu>
Mon, 6 Dec 2021 16:14:54 +0000 (11:14 -0500)
committerJoseph Huber <jhuber6@vols.utk.edu>
Mon, 6 Dec 2021 17:32:02 +0000 (12:32 -0500)
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

index e20540a..f289e3e 100644 (file)
@@ -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: