Page MenuHomePhabricator

[mlir][sparse][taco] Support reduction to scalar tensors.
ClosedPublic

Authored by bixia on Feb 25 2022, 9:39 AM.

Details

Summary

The PyTACO DSL doesn't support reduction to scalars. This change
enhances the MLIR-PyTACO implementation to support reduction to scalars.

Extend an existing test to show the syntax of reduction to scalars and
two methods to retrieve the scalar values.

Diff Detail

Event Timeline

bixia created this revision.Feb 25 2022, 9:39 AM
bixia requested review of this revision.Feb 25 2022, 9:39 AM
aartbik accepted this revision.Feb 25 2022, 1:34 PM

Nice!

This revision is now accepted and ready to land.Feb 25 2022, 1:34 PM
This revision was automatically updated to reflect the committed changes.