derek.holloway
7502bf64a4
CodeQL / Analyze (actions) (push) Canceled after 0s
CodeQL / Analyze (python) (push) Canceled after 0s
pre-commit / pre-commit (push) Canceled after 0s
Multi-Arch CI / setup (push) Canceled after 0s
Publish build_manylinux_x86_64 images / publish_build_manylinux_x86_64 (push) Canceled after 0s
Setup (Multi-Arch) / setup (push) Canceled after 0s
Multi-Arch CI / Linux::${{ fromJSON(needs.setup.outputs.linux_build_config || '{}').build_variant_label || 'skip' }} (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Publish no_rocm_image_ubi10 image / publish_no_rocm_image_ubi10 (push) Canceled after 0s
Multi-Arch CI / Windows::${{ fromJSON(needs.setup.outputs.windows_build_config || '{}').build_variant_label || 'skip' }} (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Multi-Arch CI / Manifest Diff (push) Canceled after 0s
TheRock Manifest Diff Report / Generate Manifest Diff Report (push) Canceled after 0s
Multi-Arch CI / CI Summary (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04 (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04_ocl_rt images / publish_no_rocm_image_ubuntu24_04_ocl_rt (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_rocgdb (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04_rocgdb images / publish_no_rocm_image_ubuntu24_04_rocgdb (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_ocl_rt (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Unit Tests / Unit Tests :: ubuntu-24.04 (push) Canceled after 0s
Unit Tests / Unit Tests :: windows-2022 (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_media (push) Canceled after 0s
Unit Tests / Unit Tests Summary (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_openmpi (push) Canceled after 0s
158 lines
9.7 KiB
MLIR
158 lines
9.7 KiB
MLIR
// 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
|
|
}
|
|
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 ptrtoint (ptr getelementptr ({ i32 }, ptr null, i32 1) to i64))
|
|
// 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 ptrtoint (ptr getelementptr ({ i32 }, ptr null, i32 1) to i64))
|
|
// 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
|