Page MenuHomePhabricator

[mlir][sparse] parameterize MTTKRP kernel
ClosedPublic

Authored by aartbik on Fri, Jan 14, 11:25 AM.

Details

Summary

Rather than hardcoding all constants, we now use the input tensor to drive the
code setup. Of course, we still need to hardcode dim-2 of A and the final
verification in CHECK is input dependent, but overall this sets a slightly
better example of tensor setup in general.

Diff Detail

Event Timeline

aartbik created this revision.Fri, Jan 14, 11:25 AM
aartbik requested review of this revision.Fri, Jan 14, 11:25 AM
bixia accepted this revision.Fri, Jan 14, 4:15 PM
This revision is now accepted and ready to land.Fri, Jan 14, 4:15 PM
This revision was automatically updated to reflect the committed changes.