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

[Matthias Springer]

Author: Matthias Springer
Date: 2022-01-31T05:02:11+09:00
New Revision: dab72a35e9e319151576f9f6539b52634ceb6f5c

URL: https://github.com/llvm/llvm-project/commit/dab72a35e9e319151576f9f6539b52634ceb6f5c
DIFF: https://github.com/llvm/llvm-project/commit/dab72a35e9e319151576f9f6539b52634ceb6f5c.diff

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

Added: 
    

Modified: 
    mlir/benchmark/python/common.py
    mlir/docs/Bufferization.md
    mlir/test/Integration/Dialect/Vector/CPU/AMX/test-mulf-full.mlir
    mlir/test/Integration/Dialect/Vector/CPU/AMX/test-muli-full.mlir

Removed: 
    


################################################################################
diff  --git a/mlir/benchmark/python/common.py b/mlir/benchmark/python/common.py
index 305d8a5c3896c..8c7e8e2f3150d 100644
--- a/mlir/benchmark/python/common.py
+++ b/mlir/benchmark/python/common.py
@@ -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}},"

diff  --git a/mlir/docs/Bufferization.md b/mlir/docs/Bufferization.md
index b7e9665acd7d5..1bd0188f97229 100644
--- a/mlir/docs/Bufferization.md
+++ b/mlir/docs/Bufferization.md
@@ -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

diff  --git a/mlir/test/Integration/Dialect/Vector/CPU/AMX/test-mulf-full.mlir b/mlir/test/Integration/Dialect/Vector/CPU/AMX/test-mulf-full.mlir
index 37c68416e544f..422b619145dee 100644
--- a/mlir/test/Integration/Dialect/Vector/CPU/AMX/test-mulf-full.mlir
+++ b/mlir/test/Integration/Dialect/Vector/CPU/AMX/test-mulf-full.mlir
@@ -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" \

diff  --git a/mlir/test/Integration/Dialect/Vector/CPU/AMX/test-muli-full.mlir b/mlir/test/Integration/Dialect/Vector/CPU/AMX/test-muli-full.mlir
index 241e08f6208b4..1517358c0f265 100644
--- a/mlir/test/Integration/Dialect/Vector/CPU/AMX/test-muli-full.mlir
+++ b/mlir/test/Integration/Dialect/Vector/CPU/AMX/test-muli-full.mlir
@@ -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" \