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

Add inclusive/exclusive sum and prod #461

Merged
merged 6 commits into from
Feb 3, 2025

Conversation

wingertge
Copy link
Contributor

Adds plane inclusive/exclusive sum and prod.

Testing

Adds new tests for the new ops.

Copy link
Member

@nathanielsimard nathanielsimard left a comment

Choose a reason for hiding this comment

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

Just a few comments.

Comment on lines +385 to +387
let const_val = input.expand.as_const().unwrap();
let var = Variable::constant(const_val.cast_to(Out::as_elem(scope)));
ExpandElement::Plain(var).into()
Copy link
Member

Choose a reason for hiding this comment

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

Nice fix! Forcing the constant is the right move here.

Copy link
Contributor Author

Choose a reason for hiding this comment

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

It was causing issues because I implemented the IntExpand dynamic element, and that was causing this code to emit i64 constants in WGPU (which isn't supported).

@nathanielsimard nathanielsimard merged commit ff94be8 into tracel-ai:main Feb 3, 2025
5 checks passed
@wingertge wingertge deleted the feat/inclusive-add branch February 3, 2025 15:33
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