1// RUN: mlir-opt -split-input-file -shape-bufferize <%s | FileCheck %s
2
3// -----
4// Check that shape.assuming returns a memref.
5//
6// CHECK-LABEL: @shape_assuming_returns_memref
7func @shape_assuming_returns_memref() {
8  %0 = shape.const_witness true
9  // CHECK: shape.assuming %{{.*}} -> (memref<2xf16>) {
10  %1 = shape.assuming %0 -> (tensor<2xf16>) {
11    %2 = "test.source"() : () -> (tensor<2xf16>)
12    shape.assuming_yield %2 : tensor<2xf16>
13  }
14  "test.sink"(%1) : (tensor<2xf16>) -> ()
15  return
16}
17