mirror of
https://github.com/tile-ai/tilelang.git
synced 2026-10-15 05:39:41 +08:00
Lei Wang
4fb96a0f56
* [Enhancement] Update global load/store functions for CUDA compatibility (#1652) Refactor the `ld_global_256` and `st_global_256` functions to support both CUDA versions above 12.9 and earlier versions. This change ensures that 256-bit loads and stores are handled correctly across different CUDA versions, improving performance and compatibility. The implementation now uses two 128-bit loads/stores for older versions, enhancing the robustness of the codebase. * Update comments in global load/store functions for CUDA compatibility Clarified comments in `ld_global_256` and `st_global_256` functions to indicate that the fallback for CUDA versions below 12.9 may have performance regressions. This change enhances code readability and provides better context for developers working with different CUDA versions. * Update submodule and enhance LetStmt handling in inject_pipeline.cc - Updated the TVM submodule to the latest commit. - Improved the handling of LetStmt in the inject_pipeline.cc file to account for transitive dependencies on loop variables, ensuring correct variable substitution in rewritten blocks. - Adjusted test_tilelang_issue_1263.py to remove unnecessary jit decorator and updated the kernel compilation process with specific pass configurations. * lint fix * revert tvm * remove unused test * test fix
11 lines
185 B
Python
11 lines
185 B
Python
import tilelang.testing
|
|
import example_elementwise_add
|
|
|
|
|
|
def test_example_elementwise_add():
|
|
example_elementwise_add.main()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
tilelang.testing.main()
|