| // RUN: mlir-translate -mlir-to-llvmir %s | FileCheck %s | |
| // Regression check for a taskloop with private variables but none of the | |
| // private variables go into the context struct. | |
| omp.private {type = private} @_QFtestEi_private_i32 : i32 | |
| omp.private {type = private} @_QFtestEt2_private_i32 : i32 | |
| omp.private {type = private} @_QFtestEt1_private_i32 : i32 | |
| llvm.func @_QPtest() { | |
| %0 = llvm.mlir.constant(1 : i32) : i32 | |
| %1 = llvm.mlir.constant(20 : i32) : i32 | |
| %2 = llvm.mlir.constant(1 : i64) : i64 | |
| %3 = llvm.alloca %2 x i32 {bindc_name = "t2"} : (i64) -> !llvm.ptr | |
| %4 = llvm.alloca %2 x i32 {bindc_name = "t1"} : (i64) -> !llvm.ptr | |
| %5 = llvm.alloca %2 x i32 {bindc_name = "i"} : (i64) -> !llvm.ptr | |
| omp.taskloop.context private(@_QFtestEt1_private_i32 %4 -> %arg0, @_QFtestEt2_private_i32 %3 -> %arg1, @_QFtestEi_private_i32 %5 -> %arg2 : !llvm.ptr, !llvm.ptr, !llvm.ptr) { | |
| omp.taskloop.wrapper { | |
| omp.loop_nest (%arg3) : i32 = (%0) to (%1) inclusive step (%0) { | |
| llvm.store %arg3, %arg2 : i32, !llvm.ptr | |
| omp.yield | |
| } | |
| } | |
| omp.terminator | |
| } {omp.combined} | |
| llvm.return | |
| } | |
| // CHECK-LABEL: define void @_QPtest() { | |
| // No task context structure: | |
| // CHECK: %[[STRUCTARG:.*]] = alloca { i64, i64, i64 }, align 8 | |
| // CHECK: %[[VAL_0:.*]] = alloca i32, i64 1, align 4 | |
| // CHECK: %[[VAL_1:.*]] = alloca i32, i64 1, align 4 | |
| // CHECK: %[[VAL_2:.*]] = alloca i32, i64 1, align 4 | |
| // CHECK: br label %[[VAL_3:.*]] | |
| // CHECK: entry: ; preds = %[[VAL_4:.*]] | |
| // CHECK: br label %[[VAL_5:.*]] | |
| // CHECK: omp.private.init: ; preds = %[[VAL_3]] | |
| // CHECK: br label %[[VAL_7:.*]] | |
| // CHECK: omp.private.copy: ; preds = %[[VAL_5]] | |
| // CHECK: br label %[[VAL_8:.*]] | |
| // CHECK: omp.taskloop.wrapper.start: ; preds = %[[VAL_7]] | |
| // CHECK: br label %[[VAL_9:.*]] | |
| // CHECK: codeRepl: ; preds = %[[VAL_8]] | |
| // CHECK: %[[VAL_10:.*]] = getelementptr { i64, i64, i64 }, ptr %[[STRUCTARG]], i32 0, i32 0 | |
| // CHECK: store i64 1, ptr %[[VAL_10]], align 4 | |
| // CHECK: %[[VAL_11:.*]] = getelementptr { i64, i64, i64 }, ptr %[[STRUCTARG]], i32 0, i32 1 | |
| // CHECK: store i64 20, ptr %[[VAL_11]], align 4 | |
| // CHECK: %[[VAL_12:.*]] = getelementptr { i64, i64, i64 }, ptr %[[STRUCTARG]], i32 0, i32 2 | |
| // CHECK: store i64 1, ptr %[[VAL_12]], align 4 | |
| // CHECK: %[[VAL_14:.*]] = call i32 @__kmpc_global_thread_num(ptr @1) | |
| // CHECK: call void @__kmpc_taskgroup(ptr @1, i32 %[[VAL_14]]) | |
| // CHECK: %[[VAL_15:.*]] = call ptr @__kmpc_omp_task_alloc(ptr @1, i32 %[[VAL_14]], i32 1, i64 40, i64 24, ptr @_QPtest..omp_par) | |
| // CHECK: %[[VAL_16:.*]] = load ptr, ptr %[[VAL_15]], align 8 | |
| // CHECK: call void @llvm.memcpy.p0.p0.i64(ptr align 1 %[[VAL_16]], ptr align 1 %[[STRUCTARG]], i64 24, i1 false) | |
| // CHECK: %[[VAL_17:.*]] = getelementptr { i64, i64, i64, ptr }, ptr %[[VAL_16]], i32 0, i32 0 | |
| // CHECK: %[[VAL_18:.*]] = getelementptr { i64, i64, i64, ptr }, ptr %[[VAL_16]], i32 0, i32 1 | |
| // CHECK: %[[VAL_19:.*]] = getelementptr { i64, i64, i64, ptr }, ptr %[[VAL_16]], i32 0, i32 2 | |
| // CHECK-NOT: %[[VAL_INVALID:.*]] = getelementptr { i64, i64, i64, ptr }, ptr %[[VAL_16]], i32 0, i32 3 | |
| // CHECK: %[[VAL_20:.*]] = load i64, ptr %[[VAL_19]], align 4 | |
| // CHECK: call void @__kmpc_taskloop(ptr @1, i32 %[[VAL_14]], ptr %[[VAL_15]], i32 1, ptr %[[VAL_17]], ptr %[[VAL_18]], i64 %[[VAL_20]], i32 1, i32 0, i64 0, ptr null) | |
| // CHECK: call void @__kmpc_end_taskgroup(ptr @1, i32 %[[VAL_14]]) | |
| // CHECK: br label %[[VAL_21:.*]] | |
| // CHECK: taskloop.exit: ; preds = %[[VAL_9]] | |
| // CHECK: ret void | |
| // CHECK-LABEL: define internal void @_QPtest..omp_par | |
| // CHECK: taskloop.alloca: | |
| // CHECK: %[[VAL_22:.*]] = load ptr, ptr %[[VAL_23:.*]], align 8 | |
| // CHECK: %[[VAL_24:.*]] = getelementptr { i64, i64, i64 }, ptr %[[VAL_22]], i32 0, i32 0 | |
| // CHECK: %[[VAL_25:.*]] = load i64, ptr %[[VAL_24]], align 4 | |
| // CHECK: %[[VAL_26:.*]] = getelementptr { i64, i64, i64 }, ptr %[[VAL_22]], i32 0, i32 1 | |
| // CHECK: %[[VAL_27:.*]] = load i64, ptr %[[VAL_26]], align 4 | |
| // CHECK: %[[VAL_28:.*]] = getelementptr { i64, i64, i64 }, ptr %[[VAL_22]], i32 0, i32 2 | |
| // CHECK: %[[VAL_29:.*]] = load i64, ptr %[[VAL_28]], align 4 | |
| // CHECK: %[[VAL_32:.*]] = alloca i32, align 4 | |
| // CHECK: %[[VAL_33:.*]] = alloca i32, align 4 | |
| // CHECK: %[[VAL_34:.*]] = alloca i32, align 4 | |
| // CHECK: br label %[[VAL_35:.*]] | |
| // CHECK: taskloop.body: ; preds = %[[VAL_36:.*]] | |
| // CHECK: br label %[[VAL_37:.*]] | |
| // CHECK: omp.taskloop.context.region: ; preds = %[[VAL_35]] | |
| // CHECK: br label %[[VAL_38_1:.*]] | |
| // CHECK: omp.taskloop.wrapper.region: ; preds = %[[VAL_37]] | |
| // CHECK: br label %[[VAL_38:.*]] | |
| // CHECK: omp_loop.preheader: ; preds = %[[VAL_38_1]] | |
| // CHECK: %[[VAL_39:.*]] = sub i64 %[[VAL_27]], %[[VAL_25]] | |
| // CHECK: %[[VAL_40:.*]] = sdiv i64 %[[VAL_39]], %[[VAL_29]] | |
| // CHECK: %[[VAL_41:.*]] = add i64 %[[VAL_40]], 1 | |
| // CHECK: %[[VAL_42:.*]] = trunc i64 %[[VAL_41]] to i32 | |
| // CHECK: %[[VAL_43:.*]] = trunc i64 %[[VAL_25]] to i32 | |
| // CHECK: br label %[[VAL_44:.*]] | |
| // CHECK: omp_loop.header: ; preds = %[[VAL_45:.*]], %[[VAL_38]] | |
| // CHECK: %[[VAL_46:.*]] = phi i32 [ 0, %[[VAL_38]] ], [ %[[VAL_47:.*]], %[[VAL_45]] ] | |
| // CHECK: br label %[[VAL_48:.*]] | |
| // CHECK: omp_loop.cond: ; preds = %[[VAL_44]] | |
| // CHECK: %[[VAL_49:.*]] = icmp ult i32 %[[VAL_46]], %[[VAL_42]] | |
| // CHECK: br i1 %[[VAL_49]], label %[[VAL_50:.*]], label %[[VAL_51:.*]] | |
| // CHECK: omp_loop.exit: ; preds = %[[VAL_48]] | |
| // CHECK: br label %[[VAL_52:.*]] | |
| // CHECK: omp_loop.after: ; preds = %[[VAL_51]] | |
| // CHECK: br label %[[VAL_53:.*]] | |
| // CHECK: omp.region.cont3: ; preds = %[[VAL_52]] | |
| // CHECK: br label %[[VAL_53_1:.*]] | |
| // CHECK: omp.region.cont: ; preds = %[[VAL_53_1]] | |
| // CHECK: br label %[[VAL_54:.*]] | |
| // CHECK: omp_loop.body: ; preds = %[[VAL_48]] | |
| // CHECK: %[[VAL_55:.*]] = mul i32 %[[VAL_46]], 1 | |
| // CHECK: %[[VAL_56:.*]] = add i32 %[[VAL_55]], %[[VAL_43]] | |
| // CHECK: br label %[[VAL_57:.*]] | |
| // CHECK: omp.loop_nest.region: ; preds = %[[VAL_50]] | |
| // CHECK: store i32 %[[VAL_56]], ptr %[[VAL_34]], align 4 | |
| // CHECK: br label %[[VAL_58:.*]] | |
| // CHECK: omp.region.cont4: ; preds = %[[VAL_57]] | |
| // CHECK: br label %[[VAL_45]] | |
| // CHECK: omp_loop.inc: ; preds = %[[VAL_58]] | |
| // CHECK: %[[VAL_47]] = add nuw i32 %[[VAL_46]], 1 | |
| // CHECK: br label %[[VAL_44]] | |
| // CHECK: taskloop.exit.exitStub: ; preds = %[[VAL_53_1]] | |
| // CHECK: ret void | |
| // CHECK-NOT: define internal void @omp_taskloop_dup |