mirror of
https://github.com/tile-ai/tilelang.git
synced 2026-10-21 06:40:23 +08:00
Lei WangandSiriusNEO
f862dc38a8
* Add language dialect overlays * Refactor language dialects into static backend layers * Make language facade a static re-export and extract sparse layout leaf Replace the runtime-activated language facade with a static `from tilelang.cuda.language import *` re-export: drop the .pyi stubs + generator + py.typed, the globals()-based __all__ scraping, and the _activate_cuda_facade() setattr. Backend dialects (cuda/metal/rocm) now build on tilelang.language.common with per-backend tir overlays. Import-time dtype defaults in the mma/wgmma/mfma macro generators are pinned to the dtypes leaf so they no longer touch the half-initialized facade during bootstrap. Metal generators import tilelang.metal.language (not the cuda facade) so the eager builder's real getattr resolves the simdgroup ops; a lazy __getattr__ breaks the metal.language <-> intrinsics cycle. Extract the 2:4 sparsity layout metadata into tilelang/cuda/intrinsics/sparse_layout.py (a dtypes-only leaf) so mma_sp imports it directly and tilelang.utils.sparse drops out of the bootstrap closure, letting its @tilelang.jit compressor go back to a module-level decorator. Exports unchanged: common=558, cuda/facade=683. * Remove unwired BackendModule scaffolding The BackendModule manifest and per-backend backend.py registrations were included by mistake: nothing imports the per-backend backend.py files, the ensure_*_loaded lazy-loaders have no call sites, and the mechanism duplicates the register_lazy_execution_backends wiring already in tilelang/backend/__init__.py. Drop the dead code and its unit test. * Import tvm.arith in metal macro generator The simdgroup stride check uses arith.Analyzer but the module was missing the tvm.arith import. * Use string dtype defaults in CUDA macro generators Default dtype arguments are evaluated at import time, when the language facade is only partially initialized, so they cannot reference T.*. They previously used the tilelang.language.dtypes leaf; since every consumer treats these as strings (annotated str, wrapped via DataType before use), plain string literals are equivalent and drop the _dtypes dependency. * Drop MPSIntrinEmitter from metal language facade The emitter is only consumed directly from tilelang.metal.intrinsics, never via the language surface, and exposing it here required a lazy __getattr__ to break the metal.language <-> metal.intrinsics import cycle. Removing it from __all__ eliminates the cycle and the deferred import. * Restore dtype forwarding for ptx_ldmatrix T.ptx_ldmatrix previously accepted a dtype= keyword via _dtype_forward (from the old language/ast/ir.py path). The CUDA dialect alias dropped the wrapper while every neighboring ptx_* operator kept it, changing the public call contract. Wrap it to match. * Re-export tir Select/min/max wrappers and dedup metal __all__ Add Select/min/max to tilelang.language.tir.ir __all__ so the common facade re-exports the documented wrappers instead of the raw TVM symbols. Build metal __all__ via dict.fromkeys to match the sibling dialects and satisfy PLE0604. --------- Co-authored-by: SiriusNEO <chaofan@deepseek.com>