| // RUN: iree-opt --pass-pipeline="builtin.module(func.func(iree-preprocessing-make-single-dispatch-for-function))" %s | iree-run-mlir --Xcompiler,iree-hal-target-backends=llvm-cpu --input="1" - |
| |
| func.func @simple_test_with_cfg(%arg0 : i8) -> tensor<2x4xf32> { |
| %c0_i8 = arith.constant 0 : i8 |
| %cond = arith.cmpi eq, %arg0, %c0_i8 : i8 |
| cf.cond_br %cond, ^bb1, ^bb2 |
| ^bb1: |
| %0 = tensor.empty() : tensor<2x4xf32> |
| return %0 : tensor<2x4xf32> |
| ^bb2: |
| %1 = arith.constant dense<1.0> : tensor<2x4xf32> |
| return %1 : tensor<2x4xf32> |
| } |
| // CHECK-LABEL: EXEC @simple_test_with_cfg |
| // CHECK: 2x4xf32=[1 1 1 1][1 1 1 1] |
