// RUN: mlir-translate -mlir-to-llvmir %s | FileCheck %s

llvm.func @cancel_distribute_parallel_do(%lb : i32, %ub : i32, %step : i32) {
  omp.teams {
    omp.parallel {
      omp.distribute {
        omp.wsloop {
          omp.loop_nest (%iv) : i32 = (%lb) to (%ub) step (%step) {
            omp.cancel cancellation_construct_type(loop)
            omp.yield
          }
        } {omp.composite}
      } {omp.composite}
      omp.terminator
    } {omp.composite}
    omp.terminator
  }
  llvm.return
}
// CHECK-LABEL: define internal void @cancel_distribute_parallel_do..omp_par
// [...]
// CHECK:       omp_loop.cond:
// CHECK:         %[[VAL_102:.*]] = icmp ult i32 %{{.*}}, %{{.*}}
// CHECK:         br i1 %[[VAL_102]], label %omp_loop.body, label %omp_loop.exit
// CHECK:       omp_loop.exit:
// CHECK:         call void @__kmpc_for_static_fini(
// CHECK:         %[[VAL_106:.*]] = call i32 @__kmpc_global_thread_num(ptr @1)
// CHECK:         call void @__kmpc_barrier(ptr @3, i32 %[[VAL_106]])
// CHECK:         br label %omp_loop.after
// CHECK:       omp_loop.after:
// CHECK:         br label %omp.region.cont6
// CHECK:       omp.region.cont6:
// CHECK:         br label %omp.region.cont4
// CHECK:       omp.region.cont4:
// CHECK:         br label %omp.par.exit.exitStub
// CHECK:       omp_loop.body:
// CHECK:         %[[VAL_111:.*]] = add i32 %{{.*}}, %{{.*}}
// CHECK:         %[[VAL_112:.*]] = mul i32 %[[VAL_111]], %{{.*}}
// CHECK:         %[[VAL_113:.*]] = add i32 %[[VAL_112]], %{{.*}}
// CHECK:         br label %omp.loop_nest.region
// CHECK:       omp.loop_nest.region:
// CHECK:         %[[VAL_115:.*]] = call i32 @__kmpc_global_thread_num(ptr @1)
// CHECK:         %[[VAL_116:.*]] = call i32 @__kmpc_cancel(ptr @1, i32 %[[VAL_115]], i32 2)
// CHECK:         %[[VAL_117:.*]] = icmp eq i32 %[[VAL_116]], 0
// CHECK:         br i1 %[[VAL_117]], label %omp.loop_nest.region.split, label %omp.loop_nest.region.cncl
// CHECK:       omp.loop_nest.region.cncl:
// CHECK:         br label %omp_loop.exit
// CHECK:       omp.loop_nest.region.split:
// CHECK:         br label %omp.region.cont7
// CHECK:       omp.region.cont7:
// CHECK:         br label %omp_loop.inc
// CHECK:       omp_loop.inc:
// CHECK:         %[[VAL_100:.*]] = add nuw i32 %{{.*}}, 1
// CHECK:         br label %omp_loop.header
// CHECK:       omp.par.exit.exitStub:
// CHECK:         ret void