[Mlir-commits] [mlir] 180c9f9 - [mlir][sparse] enable scalar test

[Aart Bik]

Author: Aart Bik
Date: 2022-02-25T15:05:25-08:00
New Revision: 180c9f9efe586af9ba4204b19b0250ecf323f162

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

LOG: [mlir][sparse] enable scalar test

Removed TODO now that we support scalars properly

Reviewed By: bixia

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

Added: 
    

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

Removed: 
    


################################################################################
diff  --git a/mlir/test/Integration/Dialect/SparseTensor/taco/test_Tensor.py b/mlir/test/Integration/Dialect/SparseTensor/taco/test_Tensor.py
index 08710cd429c89..5b8b687562de7 100644
--- a/mlir/test/Integration/Dialect/SparseTensor/taco/test_Tensor.py
+++ b/mlir/test/Integration/Dialect/SparseTensor/taco/test_Tensor.py
@@ -25,17 +25,15 @@
 S.insert([4, 4, 4], 4.0)
 S.insert([7, 7, 7], 5.0)
 
-# TODO: make this work:
-# X[i, j, k] = alpha[0] * S[i, j, k]
-X[i, j, k] = S[i, j, k]
+X[i, j, k] = alpha[0] * S[i, j, k]
 
 expected = """; extended FROSTT format
 3 4
 8 8 8
-1 1 1 2
-2 2 2 3
-5 5 5 4
-8 8 8 5
+1 1 1 84
+2 2 2 126
+5 5 5 168
+8 8 8 210
 """
 
 # Force evaluation of the kernel by writing out X.