[mlir][bufferize] Clean up remaining references to `tensor-constant-bufferize`

This commit is contained in:
Matthias Springer 2022-01-31 05:01:24 +09:00
parent d1aed486ef
commit dab72a35e9
4 changed files with 6 additions and 6 deletions

View File

@ -28,7 +28,7 @@ def setup_passes(mlir_module):
f"(linalg-bufferize,convert-linalg-to-loops,convert-vector-to-scf),"
f"convert-scf-to-std,"
f"func-bufferize,"
f"tensor-constant-bufferize,"
f"arith-bufferize,"
f"builtin.func(tensor-bufferize,std-bufferize,finalizing-bufferize),"
f"convert-vector-to-llvm"
f"{{reassociate-fp-reductions=1 enable-index-optimizations=1}},"

View File

@ -235,10 +235,10 @@ which helps with this in general.
subdivisions.
- `arith-bufferize`
([code](https://github.com/llvm/llvm-project/blob/bc8acf2ce8ad6e8c9b1d97b2e02d3f4ad26e1d9d/mlir/lib/Dialect/StandardOps/Transforms/TensorConstantBufferize.cpp#L1),
[test](https://github.com/llvm/llvm-project/blob/bc8acf2ce8ad6e8c9b1d97b2e02d3f4ad26e1d9d/mlir/test/Dialect/Standard/tensor-constant-bufferize.mlir#L1))
([code](https://github.com/llvm/llvm-project/blob/446425f89871aa7849c5615e6b695ebd10c9b34a/mlir/lib/Dialect/Arithmetic/Transforms/Bufferize.cpp),
[test](https://github.com/llvm/llvm-project/blob/d1aed486efc6d35a81ca4acbabb4203c4b91cda9/mlir/test/Dialect/Arithmetic/bufferize.mlir))
- Bufferizes only `arith.constant` ops of `tensor` type.
- Bufferizes only `arith` ops of `tensor` type.
- This is an example of setting up the legality so that only a subset of
`arith.constant` ops get bufferized.
- This is an example of a pass that is not split along dialect

View File

@ -1,5 +1,5 @@
// RUN: mlir-opt %s -convert-vector-to-scf -lower-affine -convert-scf-to-std \
// RUN: -tensor-constant-bufferize -convert-vector-to-llvm="enable-amx" \
// RUN: -arith-bufferize -convert-vector-to-llvm="enable-amx" \
// RUN: -convert-memref-to-llvm -convert-std-to-llvm -reconcile-unrealized-casts | \
// RUN: mlir-translate -mlir-to-llvmir | \
// RUN: %lli --entry-function=entry --mattr="+amx-tile,+amx-int8,+amx-bf16" \

View File

@ -1,5 +1,5 @@
// RUN: mlir-opt %s -convert-vector-to-scf -lower-affine -convert-scf-to-std \
// RUN: -tensor-constant-bufferize -convert-vector-to-llvm="enable-amx" \
// RUN: -arith-bufferize -convert-vector-to-llvm="enable-amx" \
// RUN: -convert-memref-to-llvm -convert-std-to-llvm -reconcile-unrealized-casts | \
// RUN: mlir-translate -mlir-to-llvmir | \
// RUN: %lli --entry-function=entry --mattr="+amx-tile,+amx-int8,+amx-bf16" \