mlir/test/Transforms/buffer-results-to-out-params-add-result-attr.mlir

   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})
   4 func.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}
  14 func.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 }