tracel-ai / cubecl

Multi-platform high-performance compute language extension for Rust.
https://burn.dev
Apache License 2.0
535 stars 21 forks source link

Add if as a value expression #120

Closed wingertge closed 4 days ago

wingertge commented 5 days ago

Simulates if as a value expression by assigning the return value to an outer variable. As a result, this only works with primitives right now. Also fixes some minor issues I encountered porting my deformable_conv2d implementation.

Testing

All tests pass, new test added for this feature.