search

tlc-pack / relax

Apache License 2.0
195 stars 58 forks source link

[Op] Enforce int64 output shape in CallTIR #322

Closed MasterJH5574 closed 1 year ago

MasterJH5574 commented 1 year ago

This PR enforces CallTIR always generating output ShapeExpr in dtype int64, and enables the dtype check accordingly.

cc @Hzfengsy