80 lines · plain
1// RUN: mlir-translate -mlir-to-llvmir %s | FileCheck %s2 3// Only check the overall shape of the code and the presence of relevant4// runtime calls. Actual IR checking is done at the OpenMPIRBuilder level.5 6omp.declare_reduction @add_reduction_i32 : i32 init {7^bb0(%arg0: i32):8 %0 = llvm.mlir.constant(0 : i32) : i329 omp.yield(%0 : i32)10} combiner {11^bb0(%arg0: i32, %arg1: i32):12 %0 = llvm.add %arg0, %arg1 : i3213 omp.yield(%0 : i32)14}15llvm.func @simple_teams_only_reduction_() attributes {fir.internal_name = "_QPsimple_teams_only_reduction", frame_pointer = #llvm.framePointerKind<all>, target_cpu = "x86-64"} {16 %0 = llvm.mlir.constant(1 : i64) : i6417 %1 = llvm.alloca %0 x i32 {bindc_name = "sum"} : (i64) -> !llvm.ptr18 %2 = llvm.mlir.constant(1 : i64) : i6419 %3 = llvm.alloca %2 x i32 {bindc_name = "index_"} : (i64) -> !llvm.ptr20 %4 = llvm.mlir.constant(0 : index) : i6421 %5 = llvm.mlir.constant(10000 : index) : i6422 %6 = llvm.mlir.constant(1 : index) : i6423 %7 = llvm.mlir.constant(0 : i32) : i3224 %8 = llvm.mlir.constant(1 : i64) : i6425 %9 = llvm.mlir.constant(1 : i64) : i6426 llvm.store %7, %1 : i32, !llvm.ptr27 omp.teams reduction(@add_reduction_i32 %1 -> %arg0 : !llvm.ptr) {28 %10 = llvm.trunc %6 : i64 to i3229 llvm.br ^bb1(%10, %5 : i32, i64)30 ^bb1(%11: i32, %12: i64): // 2 preds: ^bb0, ^bb231 %13 = llvm.icmp "sgt" %12, %4 : i6432 llvm.cond_br %13, ^bb2, ^bb333 ^bb2: // pred: ^bb134 llvm.store %11, %3 : i32, !llvm.ptr35 %14 = llvm.load %arg0 : !llvm.ptr -> i3236 %15 = llvm.load %3 : !llvm.ptr -> i3237 %16 = llvm.add %14, %15 : i3238 llvm.store %16, %arg0 : i32, !llvm.ptr39 %17 = llvm.load %3 : !llvm.ptr -> i3240 %18 = llvm.add %17, %10 overflow<nsw> : i3241 %19 = llvm.sub %12, %6 : i6442 llvm.br ^bb1(%18, %19 : i32, i64)43 ^bb3: // pred: ^bb144 llvm.store %11, %3 : i32, !llvm.ptr45 omp.terminator46 }47 llvm.return48}49 50// Allocate reduction array51// CHECK: %[[REDARRAY:[A-Za-z_.][A-Za-z0-9_.]*]] = alloca [1 x ptr], align 852// Call to outlined function53// CHECK: call void (ptr, i32, ptr, ...) @__kmpc_fork_teams54// CHECK-SAME: @[[OUTLINED:[A-Za-z_.][A-Za-z0-9_.]*]]55// Outlined function.56 57// Private reduction variable and its initialization.58 59// Call to the reduction function.60// CHECK: call i32 @__kmpc_reduce61// Check that the reduction array is passed in.62// CHECK-SAME: %[[REDARRAY]]63// CHECK-SAME: @[[REDFUNC:[A-Za-z_.][A-Za-z0-9_.]*]]64 65// CHECK: [[FINALIZE:.+]]:66// CHECK: call void @__kmpc_barrier67 68// Non atomic version69// CHECK: call void @__kmpc_end_reduce70// CHECK: br label %[[FINALIZE]]71 72// Atomic version not generated73// CHECK: unreachable74 75// CHECK: define internal void @[[OUTLINED]]76 77// Reduction function.78// CHECK: define internal void @[[REDFUNC]]79// CHECK: add i3280