8f91b108df
As part of the work on transitioning bufferization dialect, ops, and
associated logic to operate on newly added type interfaces (see
00eaff3e9c), rename the
bufferization.to_memref to highlight the generic nature of the op.
Bufferization process produces buffers while memref is a builtin type
rather than a generic term.
Preserve the current API (to_buffer still produces a memref), however,
as the new type interfaces are not used yet.
38 lines
1.0 KiB
MLIR
38 lines
1.0 KiB
MLIR
// RUN: mlir-opt %s -test-tensorlike-bufferlike -split-input-file | FileCheck %s
|
|
|
|
// CHECK: func.func @builtin_unranked
|
|
// CHECK-SAME: {found = {operand_0 = "is_tensor_like", result_0 = "is_buffer_like"}}
|
|
func.func @builtin_unranked(%t: tensor<*xf32>) -> (memref<*xf32>)
|
|
{
|
|
%0 = bufferization.to_buffer %t : tensor<*xf32> to memref<*xf32>
|
|
return %0 : memref<*xf32>
|
|
}
|
|
|
|
// -----
|
|
|
|
// CHECK: func.func @builtin_ranked
|
|
// CHECK-SAME: {found = {operand_0 = "is_tensor_like", result_0 = "is_buffer_like"}}
|
|
func.func @builtin_ranked(%t: tensor<42xf32>) -> (memref<42xf32>)
|
|
{
|
|
%0 = bufferization.to_buffer %t : tensor<42xf32> to memref<42xf32>
|
|
return %0 : memref<42xf32>
|
|
}
|
|
|
|
// -----
|
|
|
|
// CHECK: func.func @custom_tensor
|
|
// CHECK-SAME: {found = {operand_0 = "is_tensor_like"}}
|
|
func.func @custom_tensor(%t: !test.test_tensor<[42], f32>) -> ()
|
|
{
|
|
return
|
|
}
|
|
|
|
// -----
|
|
|
|
// CHECK: func.func @custom_memref
|
|
// CHECK-SAME: {found = {operand_0 = "is_buffer_like"}}
|
|
func.func @custom_memref(%t: !test.test_memref<[42], f32>) -> ()
|
|
{
|
|
return
|
|
}
|