mirror of
https://github.com/tile-ai/tilelang.git
synced 2026-10-26 07:22:54 +08:00
1774a1aae7
* [Feature] Add 1D TMA support - Check the contiguous conditions of 1D TMA copy - Add new interface and params order of `tma_load` and `tma_store` call - Add 1D `tma_store` interface in sm90 template - Add elementwise kernel for 1D TMA example * [Lint] * [BugFix] Add conditions for 1D TMA copy on non-swizzle shared tensors * [Lint] * [BugFix] 1D TMA load * [README] Update GDN README for clarity and add acknowledgements (#758) - Improved formatting and clarity of the GDN kernel implementation description. - Updated requirement section to list dependencies in a clearer format. - Added an acknowledgements section to credit the developers and the Xiaomi LLM-Core Team for their contributions. * cutlass v4.2.0 supporting cuda 13 (#760) * [Lint] * [Lint] * [MXFP4] Add test for bf16&mxfp4 gemm * [BugFix] * [Lint] --------- Co-authored-by: Yu Cheng <54519279+chengyupku@users.noreply.github.com> Co-authored-by: Johnny <johnnync13@gmail.com>
16 lines
310 B
Python
16 lines
310 B
Python
import tilelang.testing
|
|
import example_elementwise_add
|
|
import example_elementwise_add_tma_1d
|
|
|
|
|
|
def test_example_elementwise_add():
|
|
example_elementwise_add.main()
|
|
|
|
|
|
def test_example_elementwise_add_tma_1d():
|
|
example_elementwise_add_tma_1d.main()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
tilelang.testing.main()
|