60 lines · plain
1// RUN: mlir-opt --split-input-file --convert-cf-to-spirv --verify-diagnostics %s | FileCheck %s2 3//===----------------------------------------------------------------------===//4// cf.br, cf.cond_br5//===----------------------------------------------------------------------===//6 7module attributes {8 spirv.target_env = #spirv.target_env<#spirv.vce<v1.0, [], []>, #spirv.resource_limits<>>9} {10 11// CHECK-LABEL: func @simple_loop12func.func @simple_loop(%begin: i32, %end: i32, %step: i32) {13// CHECK-NEXT: spirv.Branch ^bb114 cf.br ^bb115 16// CHECK-NEXT: ^bb1: // pred: ^bb017// CHECK-NEXT: spirv.Branch ^bb2({{.*}} : i32)18^bb1: // pred: ^bb019 cf.br ^bb2(%begin : i32)20 21// CHECK: ^bb2({{.*}}: i32): // 2 preds: ^bb1, ^bb322// CHECK: spirv.BranchConditional {{.*}}, ^bb3, ^bb423^bb2(%0: i32): // 2 preds: ^bb1, ^bb324 %1 = arith.cmpi slt, %0, %end : i3225 cf.cond_br %1, ^bb3, ^bb426 27// CHECK: ^bb3: // pred: ^bb228// CHECK: spirv.Branch ^bb2({{.*}} : i32)29^bb3: // pred: ^bb230 %2 = arith.addi %0, %step : i3231 cf.br ^bb2(%2 : i32)32 33// CHECK: ^bb4: // pred: ^bb234^bb4: // pred: ^bb235 return36}37 38}39 40// -----41 42// Handle blocks whose arguments require type conversion.43 44// CHECK-LABEL: func.func @main_graph45func.func @main_graph(%arg0: index) {46 %c3 = arith.constant 1 : index47// CHECK: spirv.Branch ^bb1({{.*}} : i32)48 cf.br ^bb1(%arg0 : index)49// CHECK: ^bb1({{.*}}: i32): // 2 preds: ^bb0, ^bb250^bb1(%0: index): // 2 preds: ^bb0, ^bb251 %1 = arith.cmpi slt, %0, %c3 : index52// CHECK: spirv.BranchConditional {{.*}}, ^bb2, ^bb353 cf.cond_br %1, ^bb2, ^bb354^bb2: // pred: ^bb155// CHECK: spirv.Branch ^bb1({{.*}} : i32)56 cf.br ^bb1(%c3 : index)57^bb3: // pred: ^bb158 return59}60