From 180c9f9efe586af9ba4204b19b0250ecf323f162 Mon Sep 17 00:00:00 2001 From: Aart Bik Date: Fri, 25 Feb 2022 13:18:50 -0800 Subject: [PATCH] [mlir][sparse] enable scalar test Removed TODO now that we support scalars properly Reviewed By: bixia Differential Revision: https://reviews.llvm.org/D120590 --- .../Dialect/SparseTensor/taco/test_Tensor.py | 12 +++++------- 1 file changed, 5 insertions(+), 7 deletions(-) diff --git a/mlir/test/Integration/Dialect/SparseTensor/taco/test_Tensor.py b/mlir/test/Integration/Dialect/SparseTensor/taco/test_Tensor.py index 08710cd429c8..5b8b687562de 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([1, 1, 1], 3.0) 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.