mirror of
https://github.com/tile-ai/tilelang.git
synced 2026-10-30 08:03:09 +08:00
* [Refactor] Enhance variable substitution in device function generation - Updated `wrapBodyWithHostSideAssumes` to accept a mapping of host-side variable names to device-side variables, ensuring consistent variable references during SSA conversion. - Introduced a substitution mechanism for old parameter variables to new ones in the device function, preventing variable sharing between host and device. - Adjusted buffer declarations to utilize new variable mappings, enhancing the robustness of the device function generation process. * lint fix * test fix * lint fix --------- Co-authored-by: Zhiwen Mo <zm125@ic.ac.uk>