search

tlc-pack / relax

Apache License 2.0
193 stars 58 forks source link

[Utils] Introduce Auto Arg Converter for Relax Ops and Functions #381

Closed Hzfengsy closed 1 year ago

Hzfengsy commented 1 year ago

This PR introduces a new decorator to automatically convert the arguments to the expr according to the Relax Expr, including:

  1. convert PrimExpr to relax.PrimValue;
  2. convert tvm.String or str to relax.StringImm;
  3. convert tuple/list of PrimExpr to relax.ShapeExpr;
  4. convert tuple/list of Expr to relax.Tuple.

cc @tqchen