Skip to content

Commit 180c9f9

Browse files
committed
[mlir][sparse] enable scalar test
Removed TODO now that we support scalars properly Reviewed By: bixia Differential Revision: https://reviews.llvm.org/D120590
1 parent f58fb8a commit 180c9f9

File tree

1 file changed

+5
-7
lines changed

1 file changed

+5
-7
lines changed

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

Lines changed: 5 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -25,17 +25,15 @@
2525
S.insert([4, 4, 4], 4.0)
2626
S.insert([7, 7, 7], 5.0)
2727

28-
# TODO: make this work:
29-
# X[i, j, k] = alpha[0] * S[i, j, k]
30-
X[i, j, k] = S[i, j, k]
28+
X[i, j, k] = alpha[0] * S[i, j, k]
3129

3230
expected = """; extended FROSTT format
3331
3 4
3432
8 8 8
35-
1 1 1 2
36-
2 2 2 3
37-
5 5 5 4
38-
8 8 8 5
33+
1 1 1 84
34+
2 2 2 126
35+
5 5 5 168
36+
8 8 8 210
3937
"""
4038

4139
# Force evaluation of the kernel by writing out X.

0 commit comments

Comments
 (0)