87 lines · plain
1; RUN: mlir-translate --import-llvm %s -split-input-file | FileCheck %s2 3; CHECK: llvm.func @basic(%[[arg0:.*]]: !llvm.ptr)4define i32 @basic(ptr %dst) {5 ; CHECK: llvm.indirectbr %[[arg0]] : !llvm.ptr, [6 ; CHECK: ^[[bb1:.*]],7 ; CHECK: ^[[bb2:.*]]8 ; CHECK: ]9 indirectbr ptr %dst, [label %bb1, label %bb2]10bb1:11 ; CHECK: ^[[bb1]]:12 ; CHECK: llvm.return13 ret i32 014bb2:15 ; CHECK: ^[[bb2]]:16 ; CHECK: llvm.return17 ret i32 118}19 20; // -----21 22; CHECK: llvm.mlir.global external @addr()23@addr = global ptr null24 25; CHECK-LABEL: llvm.func @test_indirectbr() {26define void @test_indirectbr() {27 ; CHECK: %[[BA:.*]] = llvm.blockaddress <function = @test_indirectbr, tag = <id = 1>> : !llvm.ptr28 ; CHECK: {{.*}} = llvm.mlir.addressof @addr : !llvm.ptr29 ; CHECK: llvm.store %[[BA]], {{.*}} : !llvm.ptr, !llvm.ptr30 store ptr blockaddress(@test_indirectbr, %block), ptr @addr31 ; CHECK: %[[TARGET_ADDR:.*]] = llvm.load {{.*}} : !llvm.ptr -> !llvm.ptr32 %val = load ptr, ptr @addr33 ; CHECK: llvm.indirectbr %[[TARGET_ADDR]] : !llvm.ptr, [34 ; CHECK: ^[[TARGET_BB:.*]]35 ; CHECK: ]36 indirectbr ptr %val, [label %block]37 ; CHECK: ^[[TARGET_BB]]:38 ; CHECK: llvm.blocktag <id = 1>39 ; CHECK: llvm.return40 ; CHECK: }41block:42 ret void43}44 45; // -----46 47; CHECK: llvm.func @callee(!llvm.ptr, i32, i32) -> i3248declare i32 @callee(ptr %a, i32 %v, i32 %p)49 50; CHECK: llvm.func @test_indirectbr_phi(51; CHECK-SAME: %arg0: !llvm.ptr, %arg1: !llvm.ptr, %arg2: !llvm.ptr, %arg3: i32) -> i32 {52define i32 @test_indirectbr_phi(ptr %address, ptr %a, ptr %b, i32 %v) {53entry:54 ; CHECK: %[[ONE:.*]] = llvm.mlir.constant(1 : i32) : i3255 ; CHECK: %[[TWO:.*]] = llvm.mlir.constant(2 : i32) : i3256 %dest = select i1 poison, ptr blockaddress(@test_indirectbr_phi, %end), ptr %address57 ; CHECK: llvm.indirectbr {{.*}} : !llvm.ptr, [58 ; CHECK: ^[[HEAD_BB:.*]],59 ; CHECK: ^[[TAIL_BB:.*]](%[[ONE]] : i32)60 ; CHECK: ]61 indirectbr ptr %dest, [label %head, label %tail]62 63head:64 ; CHECK: ^[[HEAD_BB]]:65 ; CHECK: llvm.indirectbr {{.*}} : !llvm.ptr, [66 ; CHECK: ^[[TAIL_BB]](%[[TWO]] : i32),67 ; CHECK: ^[[END_BB:.*]]68 ; CHECK: ]69 %dest2 = select i1 poison, ptr blockaddress(@test_indirectbr_phi, %end), ptr %address70 indirectbr ptr %dest2, [label %tail, label %end]71 72tail:73 ; CHECK: ^[[TAIL_BB]](%[[BLOCK_ARG:.*]]: i32):74 ; CHECK: {{.*}} = llvm.call @callee({{.*}}, %[[BLOCK_ARG]])75 ; CHECK: llvm.return76 %p = phi i32 [1, %entry], [2, %head]77 %r = call i32 @callee(ptr %a, i32 %v, i32 %p)78 ret i32 %r79 80end:81 ; CHECK: ^[[END_BB]]:82 ; CHECK: llvm.blocktag83 ; CHECK: llvm.return84 ; CHECK: }85 ret i32 %v86}87