mirror of
https://github.com/tile-ai/tilelang.git
synced 2026-11-09 09:46:50 +08:00
Yuqi DongandLei Wang
4ca6c13117
* [CI]:Reduce test shapes to avoid OOM errors during CI. * rabbit * Increase number of processes for pytest from 2 to 4 --------- Co-authored-by: Lei Wang <34334180+LeiWang1999@users.noreply.github.com>
16 lines
413 B
Python
16 lines
413 B
Python
import tilelang.testing
|
|
import example_group_per_split_token_cast_to_fp8
|
|
import example_per_token_cast_to_fp8
|
|
|
|
|
|
def test_example_group_per_split_token_cast_to_fp8():
|
|
example_group_per_split_token_cast_to_fp8.main(M=8192, N=2048, BG=2, blk_m=8)
|
|
|
|
|
|
def test_example_per_token_cast_to_fp8():
|
|
example_per_token_cast_to_fp8.main(M=8192, N=2048, blk_m=8)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
tilelang.testing.main()
|