Skip to content

Conversation

@victor-eds
Copy link
Contributor

Use UnitAttr in tt.reshape definition to make working with them and creating them easier.

allow_reorder and efficient_layout are now UnitAttr, allowing dropping the additional constructor.

Use `UnitAttr` in `tt.reshape` definition to make working with them and creating them easier.

`allow_reorder` and `efficient_layout` are now `UnitAttr`, allowing dropping the additional constructor.

Signed-off-by: victor-eds <[email protected]>
@victor-eds victor-eds requested review from a team, etiotto and whitneywhtsang October 16, 2024 09:33
@victor-eds victor-eds self-assigned this Oct 16, 2024
@victor-eds
Copy link
Contributor Author

Noticed this flaw when working with the operation. Will make working with it cleaner. Worth upstreaming.

@victor-eds
Copy link
Contributor Author

victor-eds commented Oct 16, 2024

Now we get the very convenient:

static ReshapeOp build(/*...,*/ Type type, Value src, bool allowReorder = false, bool efficientLayout = false);

@victor-eds
Copy link
Contributor Author

Make code needed for #2266 simpler.

@vlad-penkin vlad-penkin linked an issue Oct 16, 2024 that may be closed by this pull request
Copy link
Contributor

@FMarno FMarno left a comment

Choose a reason for hiding this comment

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

Good Idea

@victor-eds
Copy link
Contributor Author

If the Intel side is happy with this, I'll push a PR upstream. This isn't blocking us, but it may make working with this op easier.

Copy link
Contributor

@whitneywhtsang whitneywhtsang left a comment

Choose a reason for hiding this comment

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

LGTM, let's upstream it.

@victor-eds victor-eds marked this pull request as draft October 17, 2024 15:15
@victor-eds
Copy link
Contributor Author

triton-lang/triton#4947

@victor-eds
Copy link
Contributor Author

Closing as upstream PR is up for review.

@victor-eds victor-eds closed this Oct 18, 2024
@victor-eds victor-eds deleted the improve-reshape-definition branch October 18, 2024 11:38
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Port "sub-group transpose reduction" to default path

4 participants