26 lines · plain
1// RUN: mlir-opt %s -test-tensor-copy-insertion="must-infer-memory-space" -split-input-file | FileCheck %s2 3// CHECK-LABEL: func @unknown_op_copy4func.func @unknown_op_copy() -> (tensor<10xf32>, tensor<10xf32>) {5 %c0 = arith.constant 0 : index6 %cst = arith.constant 0.0 : f327 // CHECK: %[[dummy:.*]] = "test.dummy_op"() : () -> tensor<10xf32>8 %t = "test.dummy_op"() : () -> tensor<10xf32>9 // CHECK: %[[copy:.*]] = bufferization.alloc_tensor() copy(%[[dummy]]) : tensor<10xf32>10 %s = tensor.insert %cst into %t[%c0] : tensor<10xf32>11 return %s, %t : tensor<10xf32>, tensor<10xf32>12}13 14// -----15 16// CHECK-LABEL: func @alloc_tensor_copy17func.func @alloc_tensor_copy() -> (tensor<10xf32>, tensor<10xf32>) {18 %c0 = arith.constant 0 : index19 %cst = arith.constant 0.0 : f3220 // CHECK: bufferization.alloc_tensor() {memory_space = 1 : ui64} : tensor<10xf32>21 %t = bufferization.alloc_tensor() {memory_space = 1 : ui64} : tensor<10xf32>22 // CHECK: bufferization.alloc_tensor() {memory_space = 1 : ui64} : tensor<10xf32>23 %s = tensor.insert %cst into %t[%c0] : tensor<10xf32>24 return %s, %t : tensor<10xf32>, tensor<10xf32>25}26