• Yu Cheng's avatar
    [Refactor] Update main function structure in example scripts and add tests (#475) · 73ae8087
    Yu Cheng authored
    * [Refactor] Update example_mla_decode.py and add tests for block_sparse_attn_tilelang
    
    * Refactor example_mla_decode.py to define a main function for better structure and clarity.
    * Introduce test_example_mla_decode.py to validate the functionality of example_mla_decode.
    * Refactor block_sparse_attn_tilelang.py to define a main function and add test_block_sparse_attn_tilelang.py for testing.
    * Ensure all new test files are integrated with tilelang testing framework.
    
    * [Test] Enhance test_example_mla_decode with argument mocking
    
    * Update test_example_mla_decode.py to mock sys.argv for better test isolation.
    * Ensure the main function of example_mla_decode is called with the correct arguments during testing.
    73ae8087
example_mla_decode.py 14 KB