This is an archive of the discontinued LLVM Phabricator instance.

Differential D120590

[mlir][sparse] enable scalar test
ClosedPublic

Authored by aartbik on Feb 25 2022, 1:19 PM.

Details

Reviewers
bixia
Commits
rG180c9f9efe58: [mlir][sparse] enable scalar test
Summary

Removed TODO now that we support scalars properly

Diff Detail

Event Timeline

aartbik created this revision.Feb 25 2022, 1:19 PM
Herald added subscribers: sdasgup3, wenzhicui, wrengr and 18 others. · View Herald TranscriptFeb 25 2022, 1:19 PM
aartbik requested review of this revision.Feb 25 2022, 1:19 PM
Herald added a project: Restricted Project. · View Herald TranscriptFeb 25 2022, 1:19 PM
Herald added subscribers: stephenneuendorffer, nicolasvasilache. · View Herald Transcript
Harbormaster completed remote builds in B151539: Diff 411509.Feb 25 2022, 1:46 PM
aartbik updated this revision to Diff 411520.Feb 25 2022, 1:53 PM

rebased main

Harbormaster completed remote builds in B151546: Diff 411520.Feb 25 2022, 2:10 PM
bixia accepted this revision.Feb 25 2022, 2:20 PM
This revision is now accepted and ready to land.Feb 25 2022, 2:20 PM
Closed by commit rG180c9f9efe58: [mlir][sparse] enable scalar test (authored by aartbik). · Explain WhyFeb 25 2022, 3:05 PM
This revision was automatically updated to reflect the committed changes.
aartbik added a commit: rG180c9f9efe58: [mlir][sparse] enable scalar test.

Revision Contents

PathSize
mlir/
test/
Integration/
Dialect/
SparseTensor/
taco/
12 lines

Diff 411520

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

Show All 19 LinesS = pt.tensor([8, 8, 8],
pt.format([pt.compressed, pt.compressed, pt.compressed])) pt.format([pt.compressed, pt.compressed, pt.compressed]))
X = pt.tensor([8, 8, 8], X = pt.tensor([8, 8, 8],
pt.format([pt.compressed, pt.compressed, pt.compressed])) pt.format([pt.compressed, pt.compressed, pt.compressed]))
S.insert([0, 0, 0], 2.0) S.insert([0, 0, 0], 2.0)
S.insert([1, 1, 1], 3.0) S.insert([1, 1, 1], 3.0)
S.insert([4, 4, 4], 4.0) S.insert([4, 4, 4], 4.0)
S.insert([7, 7, 7], 5.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] = alpha[0] * S[i, j, k]
X[i, j, k] = S[i, j, k]
expected = """; extended FROSTT format expected = """; extended FROSTT format
3 4 3 4
8 8 8 8 8 8
1 1 1 2 1 1 1 84
2 2 2 3 2 2 2 126
5 5 5 4 5 5 5 168
8 8 8 5 8 8 8 210
""" """
# Force evaluation of the kernel by writing out X. # Force evaluation of the kernel by writing out X.
with tempfile.TemporaryDirectory() as test_dir: with tempfile.TemporaryDirectory() as test_dir:
x_file = os.path.join(test_dir, 'X.tns') x_file = os.path.join(test_dir, 'X.tns')
pt.write(x_file, X) pt.write(x_file, X)
# #
# CHECK: Compare result True # CHECK: Compare result True
# #
x_data = utils.file_as_string(x_file) x_data = utils.file_as_string(x_file)
print(f'Compare result {x_data == expected}') print(f'Compare result {x_data == expected}')