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

Multi-node patterns with Add, Sub, Mul, Div #135

Merged
merged 6 commits into from
Jul 9, 2024
Merged

Conversation

slyalin
Copy link
Owner

@slyalin slyalin commented Jul 8, 2024

Implemented generic partitioner to fuse multiple nodes into a single MLIR op. Applied it for element-wise ops: Add, Sub, Mul and Div.
Automatic detection of implicit broadcast based on dimension symbolic tracking: nodes with implicit broadcasts are not included in partitions (ugly failed before). Will be extended with support of implicit broadcasts.
Minor code reorgs.

@slyalin slyalin merged commit 9663602 into mlir Jul 9, 2024
14 of 30 checks passed
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.

1 participant