diff options
author | Nick Kreeger <nick.kreeger@gmail.com> | 2022-09-03 15:45:49 -0500 |
---|---|---|
committer | Nick Kreeger <nick.kreeger@gmail.com> | 2022-09-03 15:45:49 -0500 |
commit | ef25b5d93d0b5621eb5d0482abd30a4e127e9223 (patch) | |
tree | 918b9b6750cccd1da41cee2d6848807c03ab528b /mlir/benchmark | |
parent | 9905dae5e18cd55ee6bb8678c95fed940ded1ef9 (diff) | |
download | llvm-ef25b5d93d0b5621eb5d0482abd30a4e127e9223.tar.gz |
[mlir][sparse] Expose SparseTensor passes as enums instead of opaque
numbers for vectorization and parallelization options.
The SparseTensor passes currently use opaque numbers for the CLI,
despite using an enum internally. This patch exposes the enums instead
of numbered items that are matched back to the enum.
Fixes https://github.com/llvm/llvm-project/issues/53389
Differential Revision: https://reviews.llvm.org/D123876
Please also see:
https://reviews.llvm.org/D118379
https://reviews.llvm.org/D117919
Diffstat (limited to 'mlir/benchmark')
-rw-r--r-- | mlir/benchmark/python/common.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/mlir/benchmark/python/common.py b/mlir/benchmark/python/common.py index b2269251480b..3634641074f3 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=0" - " vectorization-strategy=0 vl=1 enable-simd-index32=False" + "parallelization-strategy=none" + " vectorization-strategy=none vl=1 enable-simd-index32=False" ) pipeline = f"sparse-compiler{{{opt}}}" PassManager.parse(pipeline).run(mlir_module) |