8e72fbd616
This unit attribute indicates to the bufferization that the resulting buffer will not be written to by another op. Differential Revision: https://reviews.llvm.org/D154967
101 lines
3.8 KiB
MLIR
101 lines
3.8 KiB
MLIR
// RUN: mlir-opt %s -one-shot-bufferize="test-analysis-only" \
|
|
// RUN: -allow-unregistered-dialect -split-input-file | FileCheck %s
|
|
|
|
// RUN: mlir-opt %s -one-shot-bufferize="test-analysis-only dump-alias-sets" \
|
|
// RUN: -allow-unregistered-dialect -split-input-file | \
|
|
// RUN: FileCheck %s --check-prefix=CHECK-ALIAS-SETS
|
|
|
|
// CHECK-LABEL: func @unknown_op_aliasing(
|
|
func.func @unknown_op_aliasing(%f: f32, %f2: f32, %pos: index) -> f32 {
|
|
// CHECK-ALIAS-SETS: %[[empty:.*]] = tensor.empty
|
|
|
|
%0 = tensor.empty() : tensor<10xf32>
|
|
// CHECK: linalg.fill {__inplace_operands_attr__ = ["none", "true"]}
|
|
// CHECK-ALIAS-SETS: %[[fill1:.*]] = linalg.fill
|
|
%1 = linalg.fill ins(%f : f32) outs(%0 : tensor<10xf32>) -> tensor<10xf32>
|
|
|
|
// Something must bufferize out-of-place because the op may return an alias
|
|
// of %1.
|
|
// CHECK: "dummy.dummy_op"(%{{.*}}) {__inplace_operands_attr__ = ["false"]}
|
|
%alias = "dummy.dummy_op"(%1) : (tensor<10xf32>) -> (tensor<10xf32>)
|
|
|
|
// CHECK: linalg.fill {__inplace_operands_attr__ = ["none", "true"]}
|
|
// CHECK-ALIAS-SETS: %[[fill2:.*]] = linalg.fill {__alias_set_attr__ = [
|
|
// CHECK-ALIAS-SETS-SAME: ["%[[fill2]]", "%[[fill1]]", "%[[empty]]"]]
|
|
%2 = linalg.fill ins(%f2 : f32) outs(%1 : tensor<10xf32>) -> tensor<10xf32>
|
|
%3 = tensor.extract %alias[%pos] : tensor<10xf32>
|
|
return %3 : f32
|
|
}
|
|
|
|
// -----
|
|
|
|
// CHECK-LABEL: func @unknown_op_writing(
|
|
func.func @unknown_op_writing(%f: f32, %f2: f32, %pos: index) -> f32 {
|
|
%0 = tensor.empty() : tensor<10xf32>
|
|
// CHECK: linalg.fill {__inplace_operands_attr__ = ["none", "true"]}
|
|
%1 = linalg.fill ins(%f : f32) outs(%0 : tensor<10xf32>) -> tensor<10xf32>
|
|
|
|
// The op may bufferize to a memory write, so it must bufferize out-of-place.
|
|
// CHECK: "dummy.dummy_op"(%{{.*}}) {__inplace_operands_attr__ = ["false"]}
|
|
"dummy.dummy_op"(%1) : (tensor<10xf32>) -> ()
|
|
|
|
%3 = tensor.extract %1[%pos] : tensor<10xf32>
|
|
return %3 : f32
|
|
}
|
|
|
|
// -----
|
|
|
|
// CHECK-LABEL: func @read_of_undef_is_not_a_conflict(
|
|
func.func @read_of_undef_is_not_a_conflict(%f: f32, %idx: index) -> f32 {
|
|
%0 = tensor.empty() : tensor<10xf32>
|
|
// This can be in-place because the read below does reads undefined data.
|
|
// CHECK: tensor.insert {{.*}} {__inplace_operands_attr__ = ["none", "true", "none"]}
|
|
%1 = tensor.insert %f into %0[%idx] : tensor<10xf32>
|
|
%2 = tensor.extract %0[%idx] : tensor<10xf32>
|
|
return %2 : f32
|
|
}
|
|
|
|
// -----
|
|
|
|
// CHECK-LABEL: func @read_of_alloc_tensor_is_not_a_conflict(
|
|
func.func @read_of_alloc_tensor_is_not_a_conflict(%f: f32, %idx: index) -> f32 {
|
|
%0 = bufferization.alloc_tensor() : tensor<10xf32>
|
|
// This can be in-place because the read below does reads undefined data.
|
|
// CHECK: tensor.insert {{.*}} {__inplace_operands_attr__ = ["none", "true", "none"]}
|
|
%1 = tensor.insert %f into %0[%idx] : tensor<10xf32>
|
|
%2 = tensor.extract %0[%idx] : tensor<10xf32>
|
|
return %2 : f32
|
|
}
|
|
|
|
// -----
|
|
|
|
// CHECK-LABEL: func @to_memref_not_read_only(
|
|
func.func @to_memref_not_read_only(%idx : index, %f: f32) -> f32 {
|
|
%t = tensor.generate {
|
|
^bb0(%i : index):
|
|
tensor.yield %f : f32
|
|
} : tensor<5xf32>
|
|
// Some op may write into the result of to_memref later.
|
|
// CHECK: bufferization.to_memref
|
|
// CHECK-SAME: {__inplace_operands_attr__ = ["false"]}
|
|
%m = bufferization.to_memref %t : memref<5xf32>
|
|
%2 = tensor.extract %t[%idx] : tensor<5xf32>
|
|
return %2 : f32
|
|
}
|
|
|
|
// -----
|
|
|
|
// CHECK-LABEL: func @to_memref_read_only(
|
|
func.func @to_memref_read_only(%idx : index, %f: f32) -> f32 {
|
|
%t = tensor.generate {
|
|
^bb0(%i : index):
|
|
tensor.yield %f : f32
|
|
} : tensor<5xf32>
|
|
// Some op may write into the result of to_memref later.
|
|
// CHECK: bufferization.to_memref
|
|
// CHECK-SAME: {__inplace_operands_attr__ = ["true"]}
|
|
%m = bufferization.to_memref %t {read_only} : memref<5xf32>
|
|
%2 = tensor.extract %t[%idx] : tensor<5xf32>
|
|
return %2 : f32
|
|
}
|