tlc-pack / relax

Apache License 2.0
193 stars 58 forks source link

[FIX] Fix ShapeExpr/Tuple getitem #357

Closed Ubospica closed 1 year ago

Ubospica commented 1 year ago

Currently ShapeExpr checks index should be non-negative in __getitem__, which is unnecessary and effects the lowering process of several Ops that receives negative index. This PR removes this check.

MasterJH5574 commented 1 year ago

Thanks for the fix! Could you add a test for Tuple as well please?