-
Notifications
You must be signed in to change notification settings - Fork 47
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
Conversation
There was a problem hiding this 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.
let const_val = input.expand.as_const().unwrap(); | ||
let var = Variable::constant(const_val.cast_to(Out::as_elem(scope))); | ||
ExpandElement::Plain(var).into() |
There was a problem hiding this comment.
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.
There was a problem hiding this comment.
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).
Adds plane inclusive/exclusive sum and prod.
Testing
Adds new tests for the new ops.