| // RUN: mlir-translate -mlir-to-llvmir %s | FileCheck %s | |
| omp.private {type = private} @_QFtestEi_private_i32 : i32 | |
| omp.private {type = firstprivate} @_QFtestEa_firstprivate_i32 : i32 copy { | |
| ^bb0(%arg0: !llvm.ptr, %arg1: !llvm.ptr): | |
| %0 = llvm.load %arg0 : !llvm.ptr -> i32 | |
| llvm.store %0, %arg1 : i32, !llvm.ptr | |
| omp.yield(%arg1 : !llvm.ptr) | |
| } | |
| llvm.func @_QPtest() { | |
| %0 = llvm.mlir.constant(1 : i64) : i64 | |
| %1 = llvm.alloca %0 x i32 {bindc_name = "i"} : (i64) -> !llvm.ptr | |
| %3 = llvm.alloca %0 x i32 {bindc_name = "a"} : (i64) -> !llvm.ptr | |
| %6 = llvm.mlir.constant(20 : i32) : i32 | |
| llvm.store %6, %3 : i32, !llvm.ptr | |
| %7 = llvm.mlir.constant(1 : i32) : i32 | |
| %8 = llvm.mlir.constant(5 : i32) : i32 | |
| %9 = llvm.mlir.constant(1 : i32) : i32 | |
| omp.taskloop.context private(@_QFtestEa_firstprivate_i32 %3 -> %arg0, @_QFtestEi_private_i32 %1 -> %arg1 : !llvm.ptr, !llvm.ptr) { | |
| omp.taskloop.wrapper { | |
| omp.loop_nest (%arg2) : i32 = (%7) to (%8) inclusive step (%9) { | |
| llvm.store %arg2, %arg1 : i32, !llvm.ptr | |
| %10 = llvm.load %arg0 : !llvm.ptr -> i32 | |
| %11 = llvm.mlir.constant(1 : i32) : i32 | |
| %12 = llvm.add %10, %11 : i32 | |
| llvm.store %12, %arg0 : i32, !llvm.ptr | |
| omp.yield | |
| } | |
| } | |
| omp.terminator | |
| } {omp.combined} | |
| llvm.return | |
| } | |
| // CHECK-LABEL: define void @_QPtest() { | |
| // CHECK: %[[STRUCTARG:.*]] = alloca { i64, i64, i64, ptr }, align 8 | |
| // CHECK: %[[VAL_0:.*]] = alloca i32, i64 1, align 4 | |
| // CHECK: %[[VAL_1:.*]] = alloca i32, i64 1, align 4 | |
| // CHECK: store i32 20, ptr %[[VAL_1]], align 4 | |
| // CHECK: br label %[[VAL_2:.*]] | |
| // CHECK: entry: ; preds = %[[VAL_3:.*]] | |
| // CHECK: br label %[[VAL_4:.*]] | |
| // CHECK: omp.private.init: ; preds = %[[VAL_2]] | |
| // CHECK: %[[VAL_5:.*]] = tail call ptr @malloc(i64 4) | |
| // CHECK: %[[VAL_6:.*]] = getelementptr { i32 }, ptr %[[VAL_5]], i32 0, i32 0 | |
| // CHECK: br label %[[VAL_7:.*]] | |
| // CHECK: omp.private.copy: ; preds = %[[VAL_4]] | |
| // CHECK: br label %[[VAL_8:.*]] | |
| // CHECK: omp.private.copy1: ; preds = %[[VAL_7]] | |
| // CHECK: %[[VAL_9:.*]] = load i32, ptr %[[VAL_1]], align 4 | |
| // CHECK: store i32 %[[VAL_9]], ptr %[[VAL_6]], align 4 | |
| // CHECK: br label %[[VAL_10:.*]] | |
| // CHECK: omp.taskloop.wrapper.start: ; preds = %[[VAL_8]] | |
| // CHECK: br label %[[VAL_11:.*]] | |
| // CHECK: codeRepl: ; preds = %[[VAL_10]] | |
| // CHECK: %[[VAL_12:.*]] = getelementptr { i64, i64, i64, ptr }, ptr %[[STRUCTARG]], i32 0, i32 0 | |
| // CHECK: store i64 1, ptr %[[VAL_12]], align 4 | |
| // CHECK: %[[VAL_13:.*]] = getelementptr { i64, i64, i64, ptr }, ptr %[[STRUCTARG]], i32 0, i32 1 | |
| // CHECK: store i64 5, ptr %[[VAL_13]], align 4 | |
| // CHECK: %[[VAL_14:.*]] = getelementptr { i64, i64, i64, ptr }, ptr %[[STRUCTARG]], i32 0, i32 2 | |
| // CHECK: store i64 1, ptr %[[VAL_14]], align 4 | |
| // CHECK: %[[VAL_15:.*]] = getelementptr { i64, i64, i64, ptr }, ptr %[[STRUCTARG]], i32 0, i32 3 | |
| // CHECK: store ptr %[[VAL_5]], ptr %[[VAL_15]], align 8 | |
| // CHECK: %[[VAL_16:.*]] = call i32 @__kmpc_global_thread_num(ptr @1) | |
| // CHECK: call void @__kmpc_taskgroup(ptr @1, i32 %[[VAL_16]]) | |
| // CHECK: %[[VAL_17:.*]] = call ptr @__kmpc_omp_task_alloc(ptr @1, i32 %[[VAL_16]], i32 1, i64 40, i64 32, ptr @_QPtest..omp_par) | |
| // CHECK: %[[VAL_18:.*]] = load ptr, ptr %[[VAL_17]], align 8 | |
| // CHECK: call void @llvm.memcpy.p0.p0.i64(ptr align 1 %[[VAL_18]], ptr align 1 %[[STRUCTARG]], i64 32, i1 false) | |
| // CHECK: %[[VAL_19:.*]] = getelementptr { i64, i64, i64, ptr }, ptr %[[VAL_18]], i32 0, i32 0 | |
| // CHECK: %[[VAL_20:.*]] = getelementptr { i64, i64, i64, ptr }, ptr %[[VAL_18]], i32 0, i32 1 | |
| // CHECK: %[[VAL_21:.*]] = getelementptr { i64, i64, i64, ptr }, ptr %[[VAL_18]], i32 0, i32 2 | |
| // CHECK: %[[VAL_22:.*]] = load i64, ptr %[[VAL_21]], align 4 | |
| // CHECK: call void @__kmpc_taskloop(ptr @1, i32 %[[VAL_16]], ptr %[[VAL_17]], i32 1, ptr %[[VAL_19]], ptr %[[VAL_20]], i64 %[[VAL_22]], i32 1, i32 0, i64 0, ptr @omp_taskloop_dup) | |
| // CHECK: call void @__kmpc_end_taskgroup(ptr @1, i32 %[[VAL_16]]) | |
| // CHECK: br label %[[VAL_23:.*]] | |
| // CHECK: taskloop.exit: ; preds = %[[VAL_11]] | |
| // CHECK: ret void | |
| // CHECK-LABEL: define internal void @_QPtest..omp_par( | |
| // CHECK: taskloop.alloca: | |
| // CHECK: %[[VAL_24:.*]] = load ptr, ptr %[[VAL_25:.*]], align 8 | |
| // CHECK: %[[VAL_26:.*]] = getelementptr { i64, i64, i64, ptr }, ptr %[[VAL_24]], i32 0, i32 0 | |
| // CHECK: %[[VAL_27:.*]] = load i64, ptr %[[VAL_26]], align 4 | |
| // CHECK: %[[VAL_28:.*]] = getelementptr { i64, i64, i64, ptr }, ptr %[[VAL_24]], i32 0, i32 1 | |
| // CHECK: %[[VAL_29:.*]] = load i64, ptr %[[VAL_28]], align 4 | |
| // CHECK: %[[VAL_30:.*]] = getelementptr { i64, i64, i64, ptr }, ptr %[[VAL_24]], i32 0, i32 2 | |
| // CHECK: %[[VAL_31:.*]] = load i64, ptr %[[VAL_30]], align 4 | |
| // CHECK: %[[VAL_32:.*]] = getelementptr { i64, i64, i64, ptr }, ptr %[[VAL_24]], i32 0, i32 3 | |
| // CHECK: %[[VAL_33:.*]] = load ptr, ptr %[[VAL_32]], align 8, !align !1 | |
| // CHECK: %[[VAL_34:.*]] = alloca i32, align 4 | |
| // CHECK: br label %[[VAL_35:.*]] | |
| // CHECK: taskloop.body: ; preds = %[[VAL_36:.*]] | |
| // CHECK: %[[VAL_37:.*]] = getelementptr { i32 }, ptr %[[VAL_33]], i32 0, i32 0 | |
| // CHECK: br label %[[VAL_38:.*]] | |
| // CHECK: omp.taskloop.context.region: ; preds = %[[VAL_35]] | |
| // CHECK: br label %[[VAL_38_1:.*]] | |
| // CHECK: omp.taskloop.wrapper.region: ; preds = %[[VAL_38]] | |
| // CHECK: br label %[[VAL_39:.*]] | |
| // CHECK: omp_loop.preheader: ; preds = %[[VAL_38_1]] | |
| // CHECK: %[[VAL_40:.*]] = sub i64 %[[VAL_29]], %[[VAL_27]] | |
| // CHECK: %[[VAL_41:.*]] = sdiv i64 %[[VAL_40]], %[[VAL_31]] | |
| // CHECK: %[[VAL_42:.*]] = add i64 %[[VAL_41]], 1 | |
| // CHECK: %[[VAL_43:.*]] = trunc i64 %[[VAL_42]] to i32 | |
| // CHECK: %[[VAL_44:.*]] = trunc i64 %[[VAL_27]] to i32 | |
| // CHECK: br label %[[VAL_45:.*]] | |
| // CHECK: omp_loop.header: ; preds = %[[VAL_46:.*]], %[[VAL_39]] | |
| // CHECK: %[[VAL_47:.*]] = phi i32 [ 0, %[[VAL_39]] ], [ %[[VAL_48:.*]], %[[VAL_46]] ] | |
| // CHECK: br label %[[VAL_49:.*]] | |
| // CHECK: omp_loop.cond: ; preds = %[[VAL_45]] | |
| // CHECK: %[[VAL_50:.*]] = icmp ult i32 %[[VAL_47]], %[[VAL_43]] | |
| // CHECK: br i1 %[[VAL_50]], label %[[VAL_51:.*]], label %[[VAL_52:.*]] | |
| // CHECK: omp_loop.exit: ; preds = %[[VAL_49]] | |
| // CHECK: br label %[[VAL_53:.*]] | |
| // CHECK: omp_loop.after: ; preds = %[[VAL_52]] | |
| // CHECK: br label %[[VAL_54:.*]] | |
| // CHECK: omp.region.cont2: ; preds = %[[VAL_53]] | |
| // CHECK: br label %[[VAL_54_1:.*]] | |
| // CHECK: omp.region.cont: ; preds = %[[VAL_54]] | |
| // CHECK: tail call void @free(ptr %[[VAL_33]]) | |
| // CHECK: br label %[[VAL_55:.*]] | |
| // CHECK: omp_loop.body: ; preds = %[[VAL_49]] | |
| // CHECK: %[[VAL_56:.*]] = mul i32 %[[VAL_47]], 1 | |
| // CHECK: %[[VAL_57:.*]] = add i32 %[[VAL_56]], %[[VAL_44]] | |
| // CHECK: br label %[[VAL_58:.*]] | |
| // CHECK: omp.loop_nest.region: ; preds = %[[VAL_51]] | |
| // CHECK: store i32 %[[VAL_57]], ptr %[[VAL_34]], align 4 | |
| // CHECK: %[[VAL_59:.*]] = load i32, ptr %[[VAL_37]], align 4 | |
| // CHECK: %[[VAL_60:.*]] = add i32 %[[VAL_59]], 1 | |
| // CHECK: store i32 %[[VAL_60]], ptr %[[VAL_37]], align 4 | |
| // CHECK: br label %[[VAL_61:.*]] | |
| // CHECK: omp.region.cont3: ; preds = %[[VAL_58]] | |
| // CHECK: br label %[[VAL_46]] | |
| // CHECK: omp_loop.inc: ; preds = %[[VAL_61]] | |
| // CHECK: %[[VAL_48]] = add nuw i32 %[[VAL_47]], 1 | |
| // CHECK: br label %[[VAL_45]] | |
| // CHECK: taskloop.exit.exitStub: ; preds = %[[VAL_54_1]] | |
| // CHECK: ret void | |
| // CHECK-LABEL: define internal void @omp_taskloop_dup( | |
| // CHECK: entry: | |
| // CHECK: %[[VAL_62:.*]] = getelementptr { %[[VAL_63:.*]], { i64, i64, i64, ptr } }, ptr %[[VAL_64:.*]], i32 0, i32 1 | |
| // CHECK: %[[VAL_65:.*]] = getelementptr { i64, i64, i64, ptr }, ptr %[[VAL_62]], i32 0, i32 3 | |
| // CHECK: %[[VAL_66:.*]] = getelementptr { %[[VAL_63]], { i64, i64, i64, ptr } }, ptr %[[VAL_67:.*]], i32 0, i32 1 | |
| // CHECK: %[[VAL_68:.*]] = getelementptr { i64, i64, i64, ptr }, ptr %[[VAL_66]], i32 0, i32 3 | |
| // CHECK: %[[VAL_69:.*]] = load ptr, ptr %[[VAL_68]], align 8 | |
| // CHECK: %[[VAL_70:.*]] = tail call ptr @malloc(i64 4) | |
| // CHECK: store ptr %[[VAL_70]], ptr %[[VAL_65]], align 8 | |
| // CHECK: %[[VAL_71:.*]] = getelementptr { i32 }, ptr %[[VAL_69]], i32 0, i32 0 | |
| // CHECK: %[[VAL_72:.*]] = getelementptr { i32 }, ptr %[[VAL_70]], i32 0, i32 0 | |
| // CHECK: br label %[[VAL_73:.*]] | |
| // CHECK: omp.private.copy: ; preds = %[[VAL_74:.*]] | |
| // CHECK: %[[VAL_75:.*]] = load i32, ptr %[[VAL_71]], align 4 | |
| // CHECK: store i32 %[[VAL_75]], ptr %[[VAL_72]], align 4 | |
| // CHECK: ret void |