[Mlir-commits] [mlir] 0daa72a - [mlir][sparse][pytaco] enable dimension ordering syntax

[Aart Bik]

Author: Aart Bik
Date: 2022-01-28T13:33:04-08:00
New Revision: 0daa72a505a910cfa417069327c176fc914f6e29

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

LOG: [mlir][sparse][pytaco] enable dimension ordering syntax

Reviewed By: bixia

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

Added: 
    

Modified: 
    mlir/test/Integration/Dialect/SparseTensor/taco/test_SpMM.py

Removed: 
    


################################################################################
diff  --git a/mlir/test/Integration/Dialect/SparseTensor/taco/test_SpMM.py b/mlir/test/Integration/Dialect/SparseTensor/taco/test_SpMM.py
index edee224fc8ce3..872c73681c7d0 100644
--- a/mlir/test/Integration/Dialect/SparseTensor/taco/test_SpMM.py
+++ b/mlir/test/Integration/Dialect/SparseTensor/taco/test_SpMM.py
@@ -12,10 +12,7 @@
 from tools import mlir_pytaco_api as pt
 
 # Define the CSR format.
-#
-# TODO: accept "csr = pt.format([pt.dense, pt.compressed], [0, 1])"
-#
-csr = pt.format([pt.dense, pt.compressed])
+csr = pt.format([pt.dense, pt.compressed], [0, 1])
 
 # Read matrices A and B from file, infer size of output matrix C.
 A = pt.read(os.path.join(_SCRIPT_PATH, "data/A.mtx"), csr)