xref: /llvm-project/mlir/test/Dialect/ControlFlow/one-shot-bufferize-invalid.mlir (revision 6bf043e7433680c6f4e36393734ef83699b30f14)
1// RUN: mlir-opt -one-shot-bufferize="bufferize-function-boundaries" -split-input-file %s -verify-diagnostics
2
3// expected-error @below{{failed to bufferize op}}
4// expected-error @below{{incoming operands of block argument have inconsistent memory spaces}}
5func.func @inconsistent_memory_space() -> tensor<5xf32> {
6  %0 = bufferization.alloc_tensor() {memory_space = 0 : ui64} : tensor<5xf32>
7  cf.br ^bb1(%0: tensor<5xf32>)
8^bb1(%arg1: tensor<5xf32>):
9  func.return %arg1 : tensor<5xf32>
10^bb2():
11  %1 = bufferization.alloc_tensor() {memory_space = 1 : ui64} : tensor<5xf32>
12  cf.br ^bb1(%1: tensor<5xf32>)
13}
14
15// -----
16
17// expected-error @below{{failed to bufferize op}}
18// expected-error @below{{could not infer buffer type of block argument}}
19func.func @cannot_infer_type() {
20  return
21  // The type of the block argument cannot be inferred.
22^bb1(%t: tensor<5xf32>):
23  cf.br ^bb1(%t: tensor<5xf32>)
24}
25