Closed Ubospica closed 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.
__getitem__
Thanks for the fix! Could you add a test for Tuple as well please?
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.