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
308 lines
9.8 KiB
MLIR
308 lines
9.8 KiB
MLIR
// RUN: mlir-opt %s | FileCheck %s --enable-var-scope
|
|
// RUN: mlir-opt %s | mlir-opt | FileCheck %s --enable-var-scope
|
|
|
|
|
|
// CHECK-LABEL: @omp_canonloop_raw(
|
|
// CHECK-SAME: %[[tc:.+]]: i32)
|
|
func.func @omp_canonloop_raw(%tc : i32) -> () {
|
|
// CHECK: omp.canonical_loop %iv : i32 in range(%[[tc]]) {
|
|
"omp.canonical_loop" (%tc) ({
|
|
^bb0(%iv: i32):
|
|
// CHECK-NEXT: = llvm.add %iv, %iv : i32
|
|
%newval = llvm.add %iv, %iv : i32
|
|
// CHECK-NEXT: omp.terminator
|
|
omp.terminator
|
|
// CHECK-NEXT: }
|
|
}) : (i32) -> ()
|
|
// CHECK-NEXT: return
|
|
return
|
|
}
|
|
|
|
|
|
// CHECK-LABEL: @omp_canonloop_sequential_raw(
|
|
// CHECK-SAME: %[[tc:.+]]: i32)
|
|
func.func @omp_canonloop_sequential_raw(%tc : i32) -> () {
|
|
// CHECK-NEXT: %canonloop_s0 = omp.new_cli
|
|
%canonloop_s0 = "omp.new_cli" () : () -> (!omp.cli)
|
|
// CHECK-NEXT: omp.canonical_loop(%canonloop_s0) %iv_s0 : i32 in range(%[[tc]]) {
|
|
"omp.canonical_loop" (%tc, %canonloop_s0) ({
|
|
^bb_first(%iv_first: i32):
|
|
// CHECK-NEXT: = llvm.add %iv_s0, %iv_s0 : i32
|
|
%newval = llvm.add %iv_first, %iv_first : i32
|
|
// CHECK-NEXT: omp.terminator
|
|
omp.terminator
|
|
// CHECK-NEXT: }
|
|
}) : (i32, !omp.cli) -> ()
|
|
|
|
// CHECK-NEXT: %canonloop_s1 = omp.new_cli
|
|
%canonloop_s1 = "omp.new_cli" () : () -> (!omp.cli)
|
|
// CHECK-NEXT: omp.canonical_loop(%canonloop_s1) %iv_s1 : i32 in range(%[[tc]]) {
|
|
"omp.canonical_loop" (%tc, %canonloop_s1) ({
|
|
^bb_second(%iv_second: i32):
|
|
// CHECK: omp.terminator
|
|
omp.terminator
|
|
// CHECK-NEXT: }
|
|
}) : (i32, !omp.cli) -> ()
|
|
|
|
// CHECK-NEXT: return
|
|
return
|
|
}
|
|
|
|
|
|
// CHECK-LABEL: @omp_nested_canonloop_raw(
|
|
// CHECK-SAME: %[[tc_outer:.+]]: i32, %[[tc_inner:.+]]: i32)
|
|
func.func @omp_nested_canonloop_raw(%tc_outer : i32, %tc_inner : i32) -> () {
|
|
// CHECK-NEXT: %canonloop = omp.new_cli
|
|
%outer = "omp.new_cli" () : () -> (!omp.cli)
|
|
// CHECK-NEXT: %canonloop_d1 = omp.new_cli
|
|
%inner = "omp.new_cli" () : () -> (!omp.cli)
|
|
// CHECK-NEXT: omp.canonical_loop(%canonloop) %iv : i32 in range(%[[tc_outer]]) {
|
|
"omp.canonical_loop" (%tc_outer, %outer) ({
|
|
^bb_outer(%iv_outer: i32):
|
|
// CHECK-NEXT: omp.canonical_loop(%canonloop_d1) %iv_d1 : i32 in range(%[[tc_inner]]) {
|
|
"omp.canonical_loop" (%tc_inner, %inner) ({
|
|
^bb_inner(%iv_inner: i32):
|
|
// CHECK-NEXT: = llvm.add %iv, %iv_d1 : i32
|
|
%newval = llvm.add %iv_outer, %iv_inner: i32
|
|
// CHECK-NEXT: omp.terminator
|
|
omp.terminator
|
|
}) : (i32, !omp.cli) -> ()
|
|
// CHECK: omp.terminator
|
|
omp.terminator
|
|
}) : (i32, !omp.cli) -> ()
|
|
return
|
|
}
|
|
|
|
|
|
// CHECK-LABEL: @omp_canonloop_pretty(
|
|
// CHECK-SAME: %[[tc:.+]]: i32)
|
|
func.func @omp_canonloop_pretty(%tc : i32) -> () {
|
|
// CHECK-NEXT: omp.canonical_loop %iv : i32 in range(%[[tc]]) {
|
|
omp.canonical_loop %iv : i32 in range(%tc) {
|
|
// CHECK-NEXT: llvm.add %iv, %iv : i32
|
|
%newval = llvm.add %iv, %iv: i32
|
|
// CHECK-NEXT: omp.terminator
|
|
omp.terminator
|
|
}
|
|
return
|
|
}
|
|
|
|
|
|
// CHECK-LABEL: @omp_canonloop_constant_pretty()
|
|
func.func @omp_canonloop_constant_pretty() -> () {
|
|
// CHECK-NEXT: %[[tc:.+]] = llvm.mlir.constant(4 : i32) : i32
|
|
%tc = llvm.mlir.constant(4 : i32) : i32
|
|
// CHECK-NEXT: omp.canonical_loop %iv : i32 in range(%[[tc]]) {
|
|
omp.canonical_loop %iv : i32 in range(%tc) {
|
|
// CHECK-NEXT: llvm.add %iv, %iv : i32
|
|
%newval = llvm.add %iv, %iv: i32
|
|
// CHECK-NEXT: omp.terminator
|
|
omp.terminator
|
|
}
|
|
return
|
|
}
|
|
|
|
|
|
// CHECK-LABEL: @omp_canonloop_sequential_pretty(
|
|
// CHECK-SAME: %[[tc:.+]]: i32)
|
|
func.func @omp_canonloop_sequential_pretty(%tc : i32) -> () {
|
|
// CHECK-NEXT: %canonloop_s0 = omp.new_cli
|
|
%canonloop_s0 = omp.new_cli
|
|
// CHECK-NEXT: omp.canonical_loop(%canonloop_s0) %iv_s0 : i32 in range(%[[tc]]) {
|
|
omp.canonical_loop(%canonloop_s0) %iv_s0 : i32 in range(%tc) {
|
|
// CHECK-NEXT: omp.terminator
|
|
omp.terminator
|
|
}
|
|
|
|
// CHECK: %canonloop_s1 = omp.new_cli
|
|
%canonloop_s1 = omp.new_cli
|
|
// CHECK-NEXT: omp.canonical_loop(%canonloop_s1) %iv_s1 : i32 in range(%[[tc]]) {
|
|
omp.canonical_loop(%canonloop_s1) %iv_s1 : i32 in range(%tc) {
|
|
// CHECK-NEXT: omp.terminator
|
|
omp.terminator
|
|
}
|
|
|
|
// CHECK: %canonloop_s2 = omp.new_cli
|
|
%canonloop_s2 = omp.new_cli
|
|
// CHECK-NEXT: omp.canonical_loop(%canonloop_s2) %iv_s2 : i32 in range(%[[tc]]) {
|
|
omp.canonical_loop(%canonloop_s2) %iv_s2 : i32 in range(%tc) {
|
|
// CHECK-NEXT: omp.terminator
|
|
omp.terminator
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
|
|
// CHECK-LABEL: @omp_canonloop_2d_nested_pretty(
|
|
// CHECK-SAME: %[[tc:.+]]: i32)
|
|
func.func @omp_canonloop_2d_nested_pretty(%tc : i32) -> () {
|
|
// CHECK-NEXT: %canonloop = omp.new_cli
|
|
%canonloop = omp.new_cli
|
|
// CHECK-NEXT: %canonloop_d1 = omp.new_cli
|
|
%canonloop_d1 = omp.new_cli
|
|
// CHECK-NEXT: omp.canonical_loop(%canonloop) %iv : i32 in range(%[[tc]]) {
|
|
omp.canonical_loop(%canonloop) %iv : i32 in range(%tc) {
|
|
// CHECK-NEXT: omp.canonical_loop(%canonloop_d1) %iv_d1 : i32 in range(%[[tc]]) {
|
|
omp.canonical_loop(%canonloop_d1) %iv_d1 : i32 in range(%tc) {
|
|
// CHECK: omp.terminator
|
|
omp.terminator
|
|
}
|
|
// CHECK: omp.terminator
|
|
omp.terminator
|
|
}
|
|
return
|
|
}
|
|
|
|
|
|
// CHECK-LABEL: @omp_canonloop_3d_nested_pretty(
|
|
// CHECK-SAME: %[[tc:.+]]: i32)
|
|
func.func @omp_canonloop_3d_nested_pretty(%tc : i32) -> () {
|
|
// CHECK: %canonloop = omp.new_cli
|
|
%canonloop = omp.new_cli
|
|
// CHECK: %canonloop_d1 = omp.new_cli
|
|
%canonloop_d1 = omp.new_cli
|
|
// CHECK: %canonloop_d2 = omp.new_cli
|
|
%canonloop_d2 = omp.new_cli
|
|
// CHECK-NEXT: omp.canonical_loop(%canonloop) %iv : i32 in range(%[[tc]]) {
|
|
omp.canonical_loop(%canonloop) %iv : i32 in range(%tc) {
|
|
// CHECK-NEXT: omp.canonical_loop(%canonloop_d1) %iv_d1 : i32 in range(%[[tc]]) {
|
|
omp.canonical_loop(%canonloop_d1) %iv_1d : i32 in range(%tc) {
|
|
// CHECK-NEXT: omp.canonical_loop(%canonloop_d2) %iv_d2 : i32 in range(%[[tc]]) {
|
|
omp.canonical_loop(%canonloop_d2) %iv_d2 : i32 in range(%tc) {
|
|
// CHECK-NEXT: omp.terminator
|
|
omp.terminator
|
|
// CHECK-NEXT: }
|
|
}
|
|
// CHECK-NEXT: omp.terminator
|
|
omp.terminator
|
|
// CHECK-NEXT: }
|
|
}
|
|
// CHECK-NEXT: omp.terminator
|
|
omp.terminator
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
|
|
// CHECK-LABEL: @omp_canonloop_sequential_nested_pretty(
|
|
// CHECK-SAME: %[[tc:.+]]: i32)
|
|
func.func @omp_canonloop_sequential_nested_pretty(%tc : i32) -> () {
|
|
// CHECK-NEXT: %canonloop_s0 = omp.new_cli
|
|
%canonloop_s0 = omp.new_cli
|
|
// CHECK-NEXT: %canonloop_s0_d1 = omp.new_cli
|
|
%canonloop_s0_d1 = omp.new_cli
|
|
// CHECK-NEXT: omp.canonical_loop(%canonloop_s0) %iv_s0 : i32 in range(%[[tc]]) {
|
|
omp.canonical_loop(%canonloop_s0) %iv_s0 : i32 in range(%tc) {
|
|
// CHECK-NEXT: omp.canonical_loop(%canonloop_s0_d1) %iv_s0_d1 : i32 in range(%[[tc]]) {
|
|
omp.canonical_loop(%canonloop_s0_d1) %iv_s0_d1 : i32 in range(%tc) {
|
|
// CHECK-NEXT: omp.terminator
|
|
omp.terminator
|
|
// CHECK-NEXT: }
|
|
}
|
|
// CHECK-NEXT: omp.terminator
|
|
omp.terminator
|
|
// CHECK-NEXT: }
|
|
}
|
|
|
|
// CHECK-NEXT: %canonloop_s1 = omp.new_cli
|
|
%canonloop_s1 = omp.new_cli
|
|
// CHECK-NEXT: %canonloop_s1_d1 = omp.new_cli
|
|
%canonloop_s1_d1 = omp.new_cli
|
|
// CHECK-NEXT: omp.canonical_loop(%canonloop_s1) %iv_s1 : i32 in range(%[[tc]]) {
|
|
omp.canonical_loop(%canonloop_s1) %iv_s1 : i32 in range(%tc) {
|
|
// CHECK-NEXT: omp.canonical_loop(%canonloop_s1_d1) %iv_s1_d1 : i32 in range(%[[tc]]) {
|
|
omp.canonical_loop(%canonloop_s1_d1) %iv_s1d1 : i32 in range(%tc) {
|
|
// CHECK-NEXT: omp.terminator
|
|
omp.terminator
|
|
// CHECK-NEXT: }
|
|
}
|
|
// CHECK-NEXT: omp.terminator
|
|
omp.terminator
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
|
|
// CHECK-LABEL: @omp_newcli_unused(
|
|
// CHECK-SAME: )
|
|
func.func @omp_newcli_unused() -> () {
|
|
// CHECK-NEXT: %cli = omp.new_cli
|
|
%cli = omp.new_cli
|
|
// CHECK-NEXT: return
|
|
return
|
|
}
|
|
|
|
|
|
// CHECK-LABEL: @omp_canonloop_multiregion_isolatedfromabove(
|
|
func.func @omp_canonloop_multiregion_isolatedfromabove() -> () {
|
|
omp.private {type = firstprivate} @x.privatizer : !llvm.ptr init {
|
|
^bb0(%arg0: !llvm.ptr, %arg1: !llvm.ptr):
|
|
%c42_i32 = arith.constant 42: i32
|
|
// CHECK: omp.canonical_loop %iv : i32 in range(%c42_i32) {
|
|
omp.canonical_loop %iv1 : i32 in range(%c42_i32) {
|
|
omp.terminator
|
|
}
|
|
// CHECK: omp.yield
|
|
omp.yield(%arg0 : !llvm.ptr)
|
|
} copy {
|
|
^bb0(%arg0: !llvm.ptr, %arg1: !llvm.ptr):
|
|
%c42_i32 = arith.constant 42: i32
|
|
// CHECK: omp.canonical_loop %iv : i32 in range(%c42_i32) {
|
|
omp.canonical_loop %iv : i32 in range(%c42_i32) {
|
|
// CHECK: omp.canonical_loop %iv_d1 : i32 in range(%c42_i32) {
|
|
omp.canonical_loop %iv_d1 : i32 in range(%c42_i32) {
|
|
omp.terminator
|
|
}
|
|
omp.terminator
|
|
}
|
|
// CHECK: omp.yield
|
|
omp.yield(%arg0 : !llvm.ptr)
|
|
} dealloc {
|
|
^bb0(%arg0: !llvm.ptr):
|
|
%c42_i32 = arith.constant 42: i32
|
|
// CHECK: omp.canonical_loop %iv_s0 : i32 in range(%c42_i32) {
|
|
omp.canonical_loop %iv_s0 : i32 in range(%c42_i32) {
|
|
omp.terminator
|
|
}
|
|
// CHECK: omp.canonical_loop %iv_s1 : i32 in range(%c42_i32) {
|
|
omp.canonical_loop %iv_s1 : i32 in range(%c42_i32) {
|
|
omp.terminator
|
|
}
|
|
// CHECK: omp.yield
|
|
omp.yield
|
|
}
|
|
|
|
// CHECK: return
|
|
return
|
|
}
|
|
|
|
|
|
// CHECK-LABEL: @omp_canonloop_multiregion(
|
|
func.func @omp_canonloop_multiregion(%c : i1) -> () {
|
|
%c42_i32 = arith.constant 42: i32
|
|
%canonloop1 = omp.new_cli
|
|
%canonloop2 = omp.new_cli
|
|
%canonloop3 = omp.new_cli
|
|
scf.if %c {
|
|
// CHECK: omp.canonical_loop(%canonloop_r0) %iv_r0 : i32 in range(%c42_i32) {
|
|
omp.canonical_loop(%canonloop1) %iv1 : i32 in range(%c42_i32) {
|
|
omp.terminator
|
|
}
|
|
} else {
|
|
// CHECK: omp.canonical_loop(%canonloop_r1_s0) %iv_r1_s0 : i32 in range(%c42_i32) {
|
|
omp.canonical_loop(%canonloop2) %iv2 : i32 in range(%c42_i32) {
|
|
omp.terminator
|
|
}
|
|
// CHECK: omp.canonical_loop(%canonloop_r1_s1) %iv_r1_s1 : i32 in range(%c42_i32) {
|
|
omp.canonical_loop(%canonloop3) %iv3 : i32 in range(%c42_i32) {
|
|
omp.terminator
|
|
}
|
|
}
|
|
|
|
// CHECK: return
|
|
return
|
|
}
|