tqchen opened a new pull request, #19884: URL: https://github.com/apache/tvm/pull/19884
This PR follows up on the PrimType refactor by cleaning up a few residual type-boundary cases. Summary: - Use expression PrimType access for Metal TVMScript shuffle helpers. - Extend namespace cleanup coverage for Metal shuffle up/down helpers. - Simplify narrow backend Python raw dtype boundaries while leaving true raw ABI/codegen boundaries intact. - Inline `PrimType::StorageBytes()` in the header and remove the exported out-of-line symbol. Validation: - `git diff --check` - `pre-commit run --files python/tvm/backend/metal/script.py tests/python/tirx/test_op_namespace_cleanup.py python/tvm/backend/cuda/operator/tile_primitive/elementwise/smem.py python/tvm/backend/trn/transform/naive_allocator.py` - `pre-commit run --files include/tvm/ir/base_expr.h src/ir/type.cc` - `cmake --build build --target tvm_runtime tvm_compiler -j$(nproc)` - `python -m pytest tests/python/tirx/test_op_namespace_cleanup.py -xvs -rs` (passes with 10 skips due the suite's sm_100a device gate) - checked `libtvm_compiler.so` no longer exports `tvm::PrimType::StorageBytes` -- 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]
