v0.2.3

Summary

This bug fix release includes multiple important fixes for torchlib and enhancement to the rewriter rules. It also contains fixes for the pass infrastructure in ir and addition of convenience constructors (#2126). Common graph transformation passes are now exposed in onnxscript.ir.passes.common.

What's Changed

• [torchlib] Fix index_put: handle None cases by @AyoubMDL in #2061
• [IR] Add support for quant_parameter_tensor_names field by @justinchuby in #2080
• Add unique op by @a-gardner1 in #1547
• Add Op (Slice - complex) | feat torchlib by @titaiwangms in #2089
• Fix Op (Slice complex) | improve dim expression by @titaiwangms in #2094
• Generalize MHA pattern by @gramalingam in #2092
• [IR] Fix deserialize_node by @Johansmm in #2098
• Fusion for partial rotary embedding by @gramalingam in #2095
• [torchlib] Fix layer norm dtype by @justinchuby in #2100
• [torchlib] Register aten::__lshift__ and __rshift__ by @justinchuby in #2102
• Remove the experimental IR graph builder by @justinchuby in #2104
• [CI] Fix test errors on windows by @justinchuby in #2103
• A couple of extensions to MHA fusion by @gramalingam in #2106
• Make test test_smollm 20% faster by @xadupre in #2107
• Fix Op (convolution) | add nd support to convolution by @titaiwangms in #2108
• Add Op (aten::masked_scatter) | feat (torchlib) by @titaiwangms in #2112
• Enable version converter for torch>=2.6 by @shubhambhokare1 in #2111
• Turn constant folder and dce into passes by @justinchuby in #2109
• Add SDPA fusion unit test case by @gramalingam in #2116
• Extend sdpa tests by @shubhambhokare1 in #2118
• [IR] Create a shape inference pass using onnx shape inference by @justinchuby in #2117
• Fix DORT CI according to torch-nightly changes by @justinchuby in #2125
• [rewriter] Enable llama rule sets by @titaiwangms in #2124
• [IR] Convenience constructor for Node by @justinchuby in #2126
• Add spda rules and tests for unmasked patterns by @shubhambhokare1 in #2123
• [IR] Handle ONNX custom types in DataType.from_numpy by @justinchuby in #2131
• Add Gelu Tanh fusion rule by @gramalingam in #2132
• Cleanup ort transformer fusions by @gramalingam in #2115
• Use self._use_mask in sdpa rewrite call by @shubhambhokare1 in #2135
• [rewriter | torchlib] respect ops order in torchscript graph by @titaiwangms in #2134
• [IR] Improve pass infra by @justinchuby in #2120
• A couple of ort fusion fixes by @gramalingam in #2136
• Rotary embedding needs function extraction by @gramalingam in #2139
• [Passes] Consolidate DCE passes into common passes by @justinchuby in #2143
• Make the return type of rewrite check functions a MatchResult object by @shubhambhokare1 in #2138
• Optimize away zero-length concat operands by @gramalingam in #2150
• Fix include_self for scatter_reduce by @xadupre in #2090
• [torchlib] Fix aten_div rounding_mode by @justinchuby in #2147
• Remove warning messages by @gramalingam in #2151

New Contributors

• @AyoubMDL made their first contribution in #2061
• @a-gardner1 made their first contribution in #1547

Full Changelog: v0.2.2...v0.2.3.1