This is an archive of the discontinued LLVM Phabricator instance.

[mlir][sparse][gpu][nvidia] add pruning step and check to 2:4 matrix multiplication
ClosedPublic

Authored by aartbik on Jul 14 2023, 11:18 AM.

Details

Summary

(1) without the check, the results may silently be wrong, so check is needed
(2) add pruning step to guarantee 2:4 property

Note, in the longer run, we may want to split out the pruning step somehow,
or make it optional.

Diff Detail

Event Timeline

aartbik created this revision.Jul 14 2023, 11:18 AM
Herald added a project: Restricted Project. · View Herald Transcript
aartbik requested review of this revision.Jul 14 2023, 11:18 AM
K-Wu accepted this revision.Jul 14 2023, 11:22 AM
This revision is now accepted and ready to land.Jul 14 2023, 11:22 AM
This revision was landed with ongoing or failed builds.Jul 14 2023, 12:08 PM
This revision was automatically updated to reflect the committed changes.