v0.5.3
What's Changed
Optimizer and rewriter
- Fix Onnx 23 Rotary Fusion by @gramalingam in #2576
- Record names of contributing values in the constant folding pass by @justinchuby in #2575
- Merge output shape with input shape instead of override by @wodesuck in #2578
- Extend utilities for checking a scalar value by @gramalingam in #2587
- Merge input and output shape when removing identity by @wodesuck in #2588
- Add NaN handling in softmax pattern in SDPA fusion by @gramalingam in #2593
- Fix collapse slices rewrite rules to handle unknown dims by @justinchuby in #2583
- Expose the should_fold option to optimize() by @justinchuby in #2594
Torch Lib
- [torchlib] Add trace_only flag to aten_copy, aten_tril, aten_triu by @justinchuby in #2572
- [torchlib] Support integers in logical_and/or ops and update other logical ops by @justinchuby in #2582
- [torchlib] Add back operator and/or by @justinchuby in #2590
- Improve aten_floor_divide for int inputs by @justinchuby in #2592
Other Changes
- Remove usages of ir.Input in test by @justinchuby in #2591
New Contributors
Full Changelog: v0.5.2...v0.5.3