// 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 } 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