Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Remove thread shape analysis #334

Merged
merged 2 commits into from
Dec 15, 2024
Merged

Remove thread shape analysis #334

merged 2 commits into from
Dec 15, 2024

Conversation

harsh-nod
Copy link
Contributor

No description provided.

Signed-off-by: Harsh Menon <harsh@nod-labs.com>
continue
if lhs_dim != rhs_dim:
# If the dimensions don't agree, we can still do this broadcast only if
# this has a reduce op in its backward slice along the broadcasting dimension,
Copy link
Contributor

Choose a reason for hiding this comment

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

Can w get an IR/visual example for this :)

Copy link
Contributor Author

Choose a reason for hiding this comment

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

Sure will add it into the comments.

Copy link
Contributor Author

@harsh-nod harsh-nod Dec 15, 2024

Choose a reason for hiding this comment

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

okay I simplified this a bit. Can you take another look? I don't need to check for read/reduce ops in backward slice. Its enough to just do checks on the shapes.

Signed-off-by: Harsh Menon <harsh@nod-labs.com>
@harsh-nod harsh-nod enabled auto-merge (squash) December 15, 2024 06:55
@harsh-nod harsh-nod merged commit 91d8f59 into iree-org:main Dec 15, 2024
10 checks passed
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