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
84 lines
2.6 KiB
MLIR
84 lines
2.6 KiB
MLIR
// RUN: mlir-opt -split-input-file --verify-roundtrip %s | FileCheck %s
|
|
|
|
llvm.func @ib0(%dest : !llvm.ptr, %arg0 : i32, %arg1 : i32) -> i32 {
|
|
llvm.indirectbr %dest : !llvm.ptr, [
|
|
^head(%arg0 : i32),
|
|
^tail(%arg1, %arg0 : i32, i32)
|
|
]
|
|
^head(%r0 : i32):
|
|
llvm.return %r0 : i32
|
|
^tail(%r1 : i32, %r2 : i32):
|
|
%r = llvm.add %r1, %r2 : i32
|
|
llvm.return %r : i32
|
|
}
|
|
|
|
// CHECK: llvm.func @ib0(%[[Addr:.*]]: !llvm.ptr, %[[A0:.*]]: i32, %[[A1:.*]]: i32) -> i32 {
|
|
// CHECK: llvm.indirectbr %[[Addr]] : !llvm.ptr, [
|
|
// CHECK: ^bb1(%[[A0:.*]] : i32)
|
|
// CHECK: ^bb2(%[[A1:.*]], %[[A0]] : i32, i32)
|
|
// CHECK: ]
|
|
// CHECK: ^bb1(%[[Op0:.*]]: i32):
|
|
// CHECK: llvm.return %[[Op0]] : i32
|
|
// CHECK: ^bb2(%[[Op1:.*]]: i32, %[[Op2:.*]]: i32):
|
|
// CHECK: %[[Op3:.*]] = llvm.add %[[Op1]], %[[Op2]] : i32
|
|
// CHECK: llvm.return %[[Op3]] : i32
|
|
// CHECK: }
|
|
|
|
// -----
|
|
|
|
llvm.func @ib1(%dest : !llvm.ptr) {
|
|
llvm.indirectbr %dest : !llvm.ptr, []
|
|
}
|
|
|
|
// CHECK: llvm.func @ib1(%[[Addr:.*]]: !llvm.ptr) {
|
|
// CHECK: llvm.indirectbr %[[Addr]] : !llvm.ptr, []
|
|
// CHECK: }
|
|
|
|
// -----
|
|
|
|
// CHECK: llvm.func @test_indirectbr_phi(
|
|
// CHECK-SAME: %arg0: !llvm.ptr, %arg1: !llvm.ptr, %arg2: !llvm.ptr, %arg3: i32) -> i32 {
|
|
llvm.func @callee(!llvm.ptr, i32, i32) -> i32
|
|
llvm.func @test_indirectbr_phi(%arg0: !llvm.ptr, %arg1: !llvm.ptr, %arg2: !llvm.ptr, %arg3: i32) -> i32 {
|
|
%0 = llvm.mlir.undef : i1
|
|
%1 = llvm.mlir.addressof @test_indirectbr_phi : !llvm.ptr
|
|
%2 = llvm.blockaddress <function = @test_indirectbr_phi, tag = <id = 1>> : !llvm.ptr
|
|
// CHECK: %[[ONE:.*]] = llvm.mlir.constant(1 : i32) : i32
|
|
%3 = llvm.mlir.constant(1 : i32) : i32
|
|
// CHECK: %[[TWO:.*]] = llvm.mlir.constant(2 : i32) : i32
|
|
%4 = llvm.mlir.constant(2 : i32) : i32
|
|
%5 = llvm.select %0, %2, %arg0 : i1, !llvm.ptr
|
|
// CHECK: llvm.indirectbr {{.*}} : !llvm.ptr, [
|
|
// CHECK: ^[[HEAD_BB:.*]],
|
|
// CHECK: ^[[TAIL_BB:.*]](%[[ONE]] : i32)
|
|
// CHECK: ]
|
|
llvm.indirectbr %5 : !llvm.ptr, [
|
|
^bb1,
|
|
^bb2(%3 : i32)
|
|
]
|
|
^bb1:
|
|
// CHECK: ^[[HEAD_BB]]:
|
|
// CHECK: llvm.indirectbr {{.*}} : !llvm.ptr, [
|
|
// CHECK: ^[[TAIL_BB]](%[[TWO]] : i32),
|
|
// CHECK: ^[[END_BB:.*]]
|
|
// CHECK: ]
|
|
%6 = llvm.select %0, %2, %arg0 : i1, !llvm.ptr
|
|
llvm.indirectbr %6 : !llvm.ptr, [
|
|
^bb2(%4 : i32),
|
|
^bb3
|
|
]
|
|
^bb2(%7: i32):
|
|
// CHECK: ^[[TAIL_BB]](%[[BLOCK_ARG:.*]]: i32):
|
|
// CHECK: {{.*}} = llvm.call @callee({{.*}}, %[[BLOCK_ARG]])
|
|
// CHECK: llvm.return
|
|
%8 = llvm.call @callee(%arg1, %arg3, %7) : (!llvm.ptr, i32, i32) -> i32
|
|
llvm.return %8 : i32
|
|
^bb3:
|
|
// CHECK: ^[[END_BB]]:
|
|
// CHECK: llvm.blocktag
|
|
// CHECK: llvm.return
|
|
// CHECK: }
|
|
llvm.blocktag <id = 1>
|
|
llvm.return %arg3 : i32
|
|
}
|