"vscode:/vscode.git/clone" did not exist on "803f056a307366dbef613defae9000131c5391c7"
test_example_cast.py 445 Bytes
Newer Older
1
2
3
4
5
6
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():
7
8
    example_group_per_split_token_cast_to_fp8.main(
        M=1024, N=1024, BG=2, blk_m=4, batch_sizes=[128, 896])
9
10
11


def test_example_per_token_cast_to_fp8():
12
    example_per_token_cast_to_fp8.main(M=2048, N=512, blk_m=8)
13
14
15
16


if __name__ == "__main__":
    tilelang.testing.main()