summaryrefslogtreecommitdiff
path: root/mlir/benchmark/python
diff options
context:
space:
mode:
authorNick Kreeger <nick.kreeger@gmail.com>2022-09-03 15:47:40 -0500
committerNick Kreeger <nick.kreeger@gmail.com>2022-09-03 15:47:40 -0500
commit91470d635290928e2bdaab82d52ebfd263eae267 (patch)
tree56f85c024d3f3323466086d127ba9e2ded24b89b /mlir/benchmark/python
parentef25b5d93d0b5621eb5d0482abd30a4e127e9223 (diff)
Revert "[mlir][sparse] Expose SparseTensor passes as enums instead of opaque"
This reverts commit ef25b5d93d0b5621eb5d0482abd30a4e127e9223.
Diffstat (limited to 'mlir/benchmark/python')
-rw-r--r--mlir/benchmark/python/common.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/mlir/benchmark/python/common.py b/mlir/benchmark/python/common.py
index 3634641074f3..b2269251480b 100644
--- a/mlir/benchmark/python/common.py
+++ b/mlir/benchmark/python/common.py
@@ -13,8 +13,8 @@ def setup_passes(mlir_module):
"""Setup pass pipeline parameters for benchmark functions.
"""
opt = (
- "parallelization-strategy=none"
- " vectorization-strategy=none vl=1 enable-simd-index32=False"
+ "parallelization-strategy=0"
+ " vectorization-strategy=0 vl=1 enable-simd-index32=False"
)
pipeline = f"sparse-compiler{{{opt}}}"
PassManager.parse(pipeline).run(mlir_module)