mirror of
https://github.com/tile-ai/tilelang.git
synced 2026-10-25 07:14:16 +08:00
Da1L8-Xanddaniu.ld
6ed02aee47
* feat(autotuner): support per-config pass_configs with code deduplication - Extract _merge_pass_configs_into_compile_args helper to deduplicate pass_configs merge logic across grouped_compile.py and tuner.py - Replace json.dumps uniformity check with direct equality comparison - Extract _make_jit_compile_func factory to unify lazy/eager jit_compile - Fix decorator-mode cache isolation: use compile() directly when per_config_pass_configs is present to avoid JIT cache key collision - Add example and test for per-config pass_configs tuning * docs: add docstrings to meet coverage threshold * refactor(autotuner): move pass_configs resolution to tuner so grouped compile handles uniform configs * fix(autotuner): define _PASS_CONFIGS_KEY constant and fix jit_compile race condition - Replace all magic string "__pass_configs__" with _PASS_CONFIGS_KEY constant - Move _pass_configs_lock to cover both branches in jit_compile to prevent concurrent reads of temporarily mutated pass_configs * refactor(autotuner): introduce type aliases for compile-unit bookkeeping - Add ConfigArg, UnitItem, UnitResult, CompileUnit type aliases - Replace all long tuple type annotations with aliases - Expand example configs to include num_stages=[1,2] and enable_grouped_compile --------- Co-authored-by: daniu.ld <daniu.ld@alibaba-inc.com>