84 lines · plain
1// RUN: mlir-opt -split-input-file --verify-roundtrip %s | FileCheck %s2 3llvm.func @ib0(%dest : !llvm.ptr, %arg0 : i32, %arg1 : i32) -> i32 {4 llvm.indirectbr %dest : !llvm.ptr, [5 ^head(%arg0 : i32),6 ^tail(%arg1, %arg0 : i32, i32)7 ]8^head(%r0 : i32):9 llvm.return %r0 : i3210^tail(%r1 : i32, %r2 : i32):11 %r = llvm.add %r1, %r2 : i3212 llvm.return %r : i3213}14 15// CHECK: llvm.func @ib0(%[[Addr:.*]]: !llvm.ptr, %[[A0:.*]]: i32, %[[A1:.*]]: i32) -> i32 {16// CHECK: llvm.indirectbr %[[Addr]] : !llvm.ptr, [17// CHECK: ^bb1(%[[A0:.*]] : i32)18// CHECK: ^bb2(%[[A1:.*]], %[[A0]] : i32, i32)19// CHECK: ]20// CHECK: ^bb1(%[[Op0:.*]]: i32):21// CHECK: llvm.return %[[Op0]] : i3222// CHECK: ^bb2(%[[Op1:.*]]: i32, %[[Op2:.*]]: i32):23// CHECK: %[[Op3:.*]] = llvm.add %[[Op1]], %[[Op2]] : i3224// CHECK: llvm.return %[[Op3]] : i3225// CHECK: }26 27// -----28 29llvm.func @ib1(%dest : !llvm.ptr) {30 llvm.indirectbr %dest : !llvm.ptr, []31}32 33// CHECK: llvm.func @ib1(%[[Addr:.*]]: !llvm.ptr) {34// CHECK: llvm.indirectbr %[[Addr]] : !llvm.ptr, []35// CHECK: }36 37// -----38 39// CHECK: llvm.func @test_indirectbr_phi(40// CHECK-SAME: %arg0: !llvm.ptr, %arg1: !llvm.ptr, %arg2: !llvm.ptr, %arg3: i32) -> i32 {41llvm.func @callee(!llvm.ptr, i32, i32) -> i3242llvm.func @test_indirectbr_phi(%arg0: !llvm.ptr, %arg1: !llvm.ptr, %arg2: !llvm.ptr, %arg3: i32) -> i32 {43 %0 = llvm.mlir.undef : i144 %1 = llvm.mlir.addressof @test_indirectbr_phi : !llvm.ptr45 %2 = llvm.blockaddress <function = @test_indirectbr_phi, tag = <id = 1>> : !llvm.ptr46 // CHECK: %[[ONE:.*]] = llvm.mlir.constant(1 : i32) : i3247 %3 = llvm.mlir.constant(1 : i32) : i3248 // CHECK: %[[TWO:.*]] = llvm.mlir.constant(2 : i32) : i3249 %4 = llvm.mlir.constant(2 : i32) : i3250 %5 = llvm.select %0, %2, %arg0 : i1, !llvm.ptr51 // CHECK: llvm.indirectbr {{.*}} : !llvm.ptr, [52 // CHECK: ^[[HEAD_BB:.*]],53 // CHECK: ^[[TAIL_BB:.*]](%[[ONE]] : i32)54 // CHECK: ]55 llvm.indirectbr %5 : !llvm.ptr, [56 ^bb1,57 ^bb2(%3 : i32)58 ]59^bb1:60 // CHECK: ^[[HEAD_BB]]:61 // CHECK: llvm.indirectbr {{.*}} : !llvm.ptr, [62 // CHECK: ^[[TAIL_BB]](%[[TWO]] : i32),63 // CHECK: ^[[END_BB:.*]]64 // CHECK: ]65 %6 = llvm.select %0, %2, %arg0 : i1, !llvm.ptr66 llvm.indirectbr %6 : !llvm.ptr, [67 ^bb2(%4 : i32),68 ^bb369 ]70^bb2(%7: i32):71 // CHECK: ^[[TAIL_BB]](%[[BLOCK_ARG:.*]]: i32):72 // CHECK: {{.*}} = llvm.call @callee({{.*}}, %[[BLOCK_ARG]])73 // CHECK: llvm.return74 %8 = llvm.call @callee(%arg1, %arg3, %7) : (!llvm.ptr, i32, i32) -> i3275 llvm.return %8 : i3276^bb3:77 // CHECK: ^[[END_BB]]:78 // CHECK: llvm.blocktag79 // CHECK: llvm.return80 // CHECK: }81 llvm.blocktag <id = 1>82 llvm.return %arg3 : i3283}84