Changes

Summary

  1. [mlir][sparse] Expose SparseTensor passes as enums instead of opaque numbers for vectorization and parallelization options. (details)
Commit 30ceb783e232ff625cb25b9a264ae60519241f30 by nick.kreeger
[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
The file was modifiedmlir/test/Integration/Dialect/SparseTensor/CPU/sparse_quantized_matmul.mlir (diff)
The file was modifiedmlir/test/Dialect/SparseTensor/sparse_vector_peeled.mlir (diff)
The file was modifiedmlir/benchmark/python/common.py (diff)
The file was modifiedmlir/test/Integration/Dialect/SparseTensor/CPU/sparse_out_simple.mlir (diff)
The file was modifiedmlir/test/Integration/Dialect/SparseTensor/CPU/sparse_scale.mlir (diff)
The file was modifiedmlir/lib/Dialect/SparseTensor/Transforms/SparseTensorPasses.cpp (diff)
The file was modifiedmlir/test/Integration/Dialect/SparseTensor/CPU/sparse_matvec.mlir (diff)
The file was modifiedmlir/include/mlir/Dialect/SparseTensor/Pipelines/Passes.h (diff)
The file was modifiedmlir/test/Dialect/SparseTensor/sparse_parallel.mlir (diff)
The file was modifiedmlir/test/Integration/Dialect/SparseTensor/CPU/sparse_sampled_mm_fusion.mlir (diff)
The file was modifiedmlir/test/Integration/Dialect/SparseTensor/CPU/sparse_cast.mlir (diff)
The file was modifiedmlir/test/Integration/Dialect/SparseTensor/CPU/sparse_spmm.mlir (diff)
The file was modifiedmlir/test/Integration/Dialect/SparseTensor/CPU/sparse_reductions.mlir (diff)
The file was modifiedmlir/test/Integration/Dialect/SparseTensor/python/test_SpMM.py (diff)
The file was modifiedmlir/test/Integration/Dialect/SparseTensor/CPU/sparse_filter_conv2d.mlir (diff)
The file was modifiedmlir/test/Dialect/SparseTensor/sparse_vector.mlir (diff)
The file was modifiedmlir/test/Dialect/SparseTensor/sparse_vector_index.mlir (diff)
The file was modifiedmlir/test/Integration/Dialect/SparseTensor/CPU/sparse_mttkrp.mlir (diff)
The file was modifiedmlir/test/Integration/Dialect/SparseTensor/CPU/sparse_index_dense.mlir (diff)
The file was modifiedmlir/include/mlir/Dialect/SparseTensor/Transforms/Passes.td (diff)
The file was modifiedmlir/test/Integration/Dialect/SparseTensor/python/test_SDDMM.py (diff)
The file was modifiedmlir/test/Dialect/SparseTensor/sparse_vector_chain.mlir (diff)
The file was modifiedmlir/test/Integration/Dialect/SparseTensor/python/test_stress.py (diff)
The file was modifiedmlir/test/Integration/Dialect/SparseTensor/CPU/sparse_sampled_matmul.mlir (diff)
The file was modifiedmlir/test/Integration/Dialect/SparseTensor/CPU/sparse_sum.mlir (diff)
The file was modifiedmlir/include/mlir/Dialect/SparseTensor/Transforms/Passes.h (diff)
The file was modifiedmlir/test/Integration/Dialect/SparseTensor/CPU/sparse_flatten.mlir (diff)