xref: /llvm-project/mlir/test/Transforms/buffer-results-to-out-params-add-result-attr.mlir (revision e6048b728d3170651828ad2dd7ed5ad0bc5e4f06)
1// RUN: mlir-opt -p 'builtin.module(buffer-results-to-out-params{add-result-attr})' -split-input-file -verify-diagnostics %s | FileCheck %s
2
3// CHECK-LABEL: @basic({{.*}}: memref<f32> {bufferize.result})
4func.func @basic() -> (memref<f32>) {
5  %0 = "test.source"() : () -> (memref<f32>)
6  return %0 : memref<f32>
7}
8
9// -----
10
11// CHECK-LABEL: multiple_results
12// CHECK-SAME:  memref<1xf32> {bufferize.result}
13// CHECK-SAME:  memref<2xf32> {bufferize.result}
14func.func @multiple_results() -> (memref<1xf32>, memref<2xf32>) {
15  %0, %1 = "test.source"() : () -> (memref<1xf32>, memref<2xf32>)
16  return %0, %1 : memref<1xf32>, memref<2xf32>
17}
18