tlc-pack / relax

Apache License 2.0
193 stars 58 forks source link

[IR][ASTPrinter] Tweaks to AST printer's handling of struct info #330

Closed slyubomirsky closed 1 year ago

slyubomirsky commented 1 year ago

Many thanks to @Hzfengsy and @tqchen for pushing through the transition to StructInfo. This PR makes a few tweaks to the AST printer as a result of these changes. In particular, it adds unit tests for printing StructInfo, eliminates outdated comments, prints ndim for ShapeType, corrects a typo (TupleType instead of TupleStructInfo), and also distinguishes between the case of omitting the params in FuncStructInfo (as would happen with opaque FuncStructInfo) and having zero params.