mirror of
https://github.com/tile-ai/tilelang.git
synced 2026-10-21 06:40:23 +08:00
2905fde91d
* [Refactor] Enhance SetMaxNRegCollector and SetMaxNRegInjector for improved register management * Refactor SetMaxNRegCollector to return a structured result, including a flag for preserving explicit register settings. * Update SetMaxNRegInjector to utilize the new result structure, ensuring proper handling of register hints. * Modify logic to conditionally inject register allocation based on the presence of SIMT copies and explicit settings. * Improve test cases to validate the injection of set_max_nreg calls and ensure correct behavior in various scenarios. * lint * [Refactor] Update register allocation logic to honor explicit hints * Modify the register allocation logic to ensure that explicit user hints are respected even when SIMT copy-like statements are present. * Adjust conditions for injecting register allocation to improve handling of register hints, enhancing overall register management. * Ensure default behavior remains conservative while allowing for user-defined configurations. * Fix warp-specialized reg hint annotation --------- Co-authored-by: wutong.1109 <wutong.1109@bytedance.com> Co-authored-by: LeiWang1999 <leiwang1999@outlook.com>