mlc-ai / relax

Apache License 2.0
140 stars 70 forks source link

[Analysis] Reshape TIR detection with iter-map-simplify #251

Closed MasterJH5574 closed 1 year ago

MasterJH5574 commented 1 year ago

This PR supports the reshape PrimFunc detection analysis with iter-map-simplify, in order to better support TIR functions with dynamic variables, which is not well handled before.

Meanwhie, this PR also keeps the existing analysis logic, so that we are still able to cover the cases like reshape with raggedness.

Unit tests for reshape are added to ensure correctness.