tqchen opened a new pull request, #19891: URL: https://github.com/apache/tvm/pull/19891
This PR continues the Relax IR unification work by phasing out the `PrimValue` wrapper and the Relax-specific leaf/base expression layers. Summary: - Remove `LeafExpr` / `LeafExprNode` and use direct expression-node checks where needed. - Converge Relax expression typing onto the shared IR `Expr` base. - Remove the `PrimValue` node wrapper while keeping `relax.prim_value` / `R.prim_value` as conversion helpers that return `PrimExpr` values unchanged. - Register direct `PrimExpr` handling through exact concrete node dispatch, aligned with the `tirx` expression visitor list and excluding arith iter-map intermediate nodes. - Inline the private Python primitive conversion helper into public `relax.prim_value`. Validation from the branch handoff: - `/home/scratch.tianqic_sw/x86-ws/miniforge3/bin/ninja -C build -j24` - `git diff --check upstream/main...HEAD` - `git diff --check` - Ruff check and format check on touched Python files - Focused pytest: `204 passed, 1 xfailed` for `tests/python/relax/test_expr.py`, `tests/python/relax/test_expr_functor.py`, `tests/python/relax/test_tvmscript_parser.py`, and `tests/python/relax/test_tvmscript_printer_relax.py` Task: tqchen/tdev#205 -- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. To unsubscribe, e-mail: [email protected] For queries about this service, please contact Infrastructure at: [email protected] --------------------------------------------------------------------- To unsubscribe, e-mail: [email protected] For additional commands, e-mail: [email protected]
