Skip to content

Fix store lowering#304

Open
zoecarver wants to merge 2 commits intomainfrom
zoecarver/store-lowering
Open

Fix store lowering#304
zoecarver wants to merge 2 commits intomainfrom
zoecarver/store-lowering

Conversation

@zoecarver
Copy link
Contributor

Add a frontend store op and lower it all the way to pack tile; a few related fixes.

This is a temporary fix, Boyana may have a cleaner approach, using this branch to unblock claude skills.

@zoecarver zoecarver requested a review from a team as a code owner February 2, 2026 20:08
Copy link
Contributor

@brnorris03 brnorris03 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Would prefer to have MLIR lit tests (as is normally done for any new ops), otherwise lgtm.

Comment on lines +845 to +849
for (Value yielded : yieldOp.getValues()) {
auto arg = dyn_cast<BlockArgument>(yielded);
if (!arg || !isTileValue(arg)) {
continue;
}
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Could consider marking ttl.compute (e.g., computeOp->setDiscardableAttr("ttl.passthrough", rewriter.getUnitAttr()); when creating pass through computes and check that attribute instead of checking yielded values.

Comment on lines +902 to +906
if (getAttachedCB(input)) {
return op.emitError(
"passthrough tensor_store should be lowered to ComputeOp first; "
"ensure convert-ttl-to-compute runs before this pass");
}
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

negative test for this error?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants