mirror of
https://github.com/tile-ai/tilelang.git
synced 2026-10-21 06:40:23 +08:00
Lei Wang
6fd0bd31fe
* refactor params of ptx_cp_aync * Enhancement: Introduce ptx_cp_async intrinsic and improve async copy handling - Added the `ptx_cp_async` intrinsic for asynchronous memory copy operations in TileLang, supporting both predicated and non-predicated versions. - Updated the code generation for `ptx_cp_async` to handle offsets correctly and added inline functions for simplified usage. - Enhanced the `inject_ptx_async_copy` transformation to support predicated vectorized copies. - Updated example scripts to disable caching and print kernel sources for better debugging and performance insights. * Remove tilelang.disable_cache() calls from example scripts for cleaner execution * Refactor comments and formatting in CUDA code generation and PTX async copy transformation - Updated comments for clarity in `codegen_cuda.cc` regarding `ptx_cp_async` arguments. - Improved formatting in `inject_ptx_async_copy.cc` for better readability of access pointer creation and logging. - Added a blank line in `op.py` for consistency in the `ptx_cp_async` function definition. * Remove unused ptx_cp_async handling from CUDA code generation - Eliminated the implementation for `ptx_cp_async` and `tl::ptx_cp_async` from `codegen_cuda.cc`, streamlining the code by removing unnecessary checks and assembly generation. - This change focuses on cleaning up the codebase and improving maintainability by removing deprecated functionality.