38 lines · plain
1// RUN: mlir-opt %s -test-tensorlike-bufferlike -split-input-file | FileCheck %s2 3// CHECK: func.func @builtin_unranked4// CHECK-SAME: {found = {operand_0 = "is_tensor_like", result_0 = "is_buffer_like"}}5func.func @builtin_unranked(%t: tensor<*xf32>) -> (memref<*xf32>)6{7 %0 = bufferization.to_buffer %t : tensor<*xf32> to memref<*xf32>8 return %0 : memref<*xf32>9}10 11// -----12 13// CHECK: func.func @builtin_ranked14// CHECK-SAME: {found = {operand_0 = "is_tensor_like", result_0 = "is_buffer_like"}}15func.func @builtin_ranked(%t: tensor<42xf32>) -> (memref<42xf32>)16{17 %0 = bufferization.to_buffer %t : tensor<42xf32> to memref<42xf32>18 return %0 : memref<42xf32>19}20 21// -----22 23// CHECK: func.func @custom_tensor24// CHECK-SAME: {found = {operand_0 = "is_tensor_like"}}25func.func @custom_tensor(%t: !test.test_tensor<[42], f32>) -> ()26{27 return28}29 30// -----31 32// CHECK: func.func @custom_memref33// CHECK-SAME: {found = {operand_0 = "is_buffer_like"}}34func.func @custom_memref(%t: !test.test_memref<[42], f32>) -> ()35{36 return37}38