[Mlir-commits] [mlir] f30ca85 - [mlir][sparse] reduce the input size in stress_test.py to make it finish in time.

[Peiming Liu]

Author: Peiming Liu
Date: 2023-02-16T01:18:03Z
New Revision: f30ca850ef8b3bc12c86b6c10093fb8fd8fae7d0

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

LOG: [mlir][sparse] reduce the input size in stress_test.py to make it finish in time.

Reviewed By: aartbik

Differential Revision: https://reviews.llvm.org/D144150

Added: 
    

Modified: 
    mlir/test/Integration/Dialect/SparseTensor/python/test_stress.py

Removed: 
    


################################################################################
diff  --git a/mlir/test/Integration/Dialect/SparseTensor/python/test_stress.py b/mlir/test/Integration/Dialect/SparseTensor/python/test_stress.py
index 0efe1c5e6ac13..3a04e5b9ab5ca 100644
--- a/mlir/test/Integration/Dialect/SparseTensor/python/test_stress.py
+++ b/mlir/test/Integration/Dialect/SparseTensor/python/test_stress.py
@@ -197,7 +197,7 @@ def main():
     f64 = ir.F64Type.get()
     # Be careful about increasing this because
     #     len(types) = 1 + len(level_choices)^rank * rank! * len(bitwidths)^2
-    shape = range(2, 6)
+    shape = range(2, 3)
     rank = len(shape)
     # All combinations.
     # TODO: add singleton here too; which requires updating how `np_arg0`