spectrometerHBH opened a new pull request, #19988: URL: https://github.com/apache/tvm/pull/19988
## Summary - Add predicate support for SM100 block-scaled `tcgen05` MMA and commit operations. - Preserve and reuse hoisted shared-memory descriptors in grouped GEMM pipelines. - Add configurable TMA L2 promotion and include it in tensor-map caching and encoding. - Prune compile-time-only tile-scheduler branches and keep pipeline state warp-uniform. ## Motivation The downstream grouped FP8 GEMM needs predicated warp-specialized MMA issue, explicit TMA cache policy, and lower-overhead persistent scheduling. Without these changes, the generated SASS performs redundant descriptor and scheduler work on the critical producer paths. ## Validation - `pre-commit run` - `python -m pytest tests/python/tirx/operator/tile_primitive/cuda/copy_async/test_tma.py -k test_copy_tma_codegen` — 54 passed - `python -m pytest tests/python/tirx/operator/tile_primitive/cuda/gemm_async/test_gemm_async.py -k test_gemm_smem_desc_hoist_vs_recompute` — 2 passed - `python -m pytest tests/python/tirx/codegen/test_codegen_blackwell.py -k test_tcgen05_block_scaled_mma_predicate_codegen` — 1 passed -- 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]
