| // 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 private(@_QFtestEt1_private_i32 %4 -> %arg0, @_QFtestEt2_private_i32 %3 -> %arg1, @_QFtestEi_private_i32 %5 -> %arg2 : !llvm.ptr, !llvm.ptr, !llvm.ptr) { |
| omp.loop_nest (%arg3) : i32 = (%0) to (%1) inclusive step (%0) { |
| llvm.store %arg3, %arg2 : i32, !llvm.ptr |
| omp.yield |
| } |
| } |
| 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.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 %[[VAL_16]], i32 0, i32 0 |
| // CHECK: %[[VAL_18:.*]] = getelementptr { i64, i64, i64 }, ptr %[[VAL_16]], i32 0, i32 1 |
| // CHECK: %[[VAL_19:.*]] = getelementptr { i64, i64, i64 }, ptr %[[VAL_16]], i32 0, i32 2 |
| // 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.region: ; preds = %[[VAL_35]] |
| // CHECK: br label %[[VAL_38:.*]] |
| // CHECK: omp_loop.preheader: ; preds = %[[VAL_37]] |
| // CHECK: %[[VAL_39:.*]] = sub i64 %[[VAL_27]], %[[VAL_25]] |
| // CHECK: %[[VAL_40:.*]] = sdiv i64 %[[VAL_39]], 1 |
| // 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.cont: ; preds = %[[VAL_52]] |
| // 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.cont3: ; 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]] |
| // CHECK: ret void |
| |
| // CHECK-NOT: define internal void @omp_taskloop_dup |
