58 lines · plain
1// RUN: mlir-translate -mlir-to-llvmir %s | FileCheck %s2 3llvm.func @cancel_distribute_parallel_do(%lb : i32, %ub : i32, %step : i32) {4 omp.teams {5 omp.parallel {6 omp.distribute {7 omp.wsloop {8 omp.loop_nest (%iv) : i32 = (%lb) to (%ub) step (%step) {9 omp.cancel cancellation_construct_type(loop)10 omp.yield11 }12 } {omp.composite}13 } {omp.composite}14 omp.terminator15 } {omp.composite}16 omp.terminator17 }18 llvm.return19}20// CHECK-LABEL: define internal void @cancel_distribute_parallel_do..omp_par21// [...]22// CHECK: omp_loop.cond:23// CHECK: %[[VAL_102:.*]] = icmp ult i32 %{{.*}}, %{{.*}}24// CHECK: br i1 %[[VAL_102]], label %omp_loop.body, label %omp_loop.exit25// CHECK: omp_loop.exit:26// CHECK: call void @__kmpc_for_static_fini(27// CHECK: %[[VAL_106:.*]] = call i32 @__kmpc_global_thread_num(ptr @1)28// CHECK: call void @__kmpc_barrier(ptr @2, i32 %[[VAL_106]])29// CHECK: br label %omp_loop.after30// CHECK: omp_loop.after:31// CHECK: br label %omp.region.cont632// CHECK: omp.region.cont6:33// CHECK: br label %omp.region.cont434// CHECK: omp.region.cont4:35// CHECK: br label %omp.par.exit.exitStub36// CHECK: omp_loop.body:37// CHECK: %[[VAL_111:.*]] = add i32 %{{.*}}, %{{.*}}38// CHECK: %[[VAL_112:.*]] = mul i32 %[[VAL_111]], %{{.*}}39// CHECK: %[[VAL_113:.*]] = add i32 %[[VAL_112]], %{{.*}}40// CHECK: br label %omp.loop_nest.region41// CHECK: omp.loop_nest.region:42// CHECK: %[[VAL_115:.*]] = call i32 @__kmpc_global_thread_num(ptr @1)43// CHECK: %[[VAL_116:.*]] = call i32 @__kmpc_cancel(ptr @1, i32 %[[VAL_115]], i32 2)44// CHECK: %[[VAL_117:.*]] = icmp eq i32 %[[VAL_116]], 045// CHECK: br i1 %[[VAL_117]], label %omp.loop_nest.region.split, label %omp.loop_nest.region.cncl46// CHECK: omp.loop_nest.region.cncl:47// CHECK: br label %omp_loop.exit48// CHECK: omp.loop_nest.region.split:49// CHECK: br label %omp.region.cont750// CHECK: omp.region.cont7:51// CHECK: br label %omp_loop.inc52// CHECK: omp_loop.inc:53// CHECK: %[[VAL_100:.*]] = add nuw i32 %{{.*}}, 154// CHECK: br label %omp_loop.header55// CHECK: omp.par.exit.exitStub:56// CHECK: ret void57 58