• Lei Wang's avatar
    [Typo] Replace `kernel.func` with `kernel` in mla benchmark scripts (#354) · 6d44c465
    Lei Wang authored
    * [Refactor] Update import structure in benchmark_mla.py
    
    - Moved the import of `flash_mla` functions to the `run_flash_mla` function for better encapsulation.
    - Added a comment for `flashinfer` installation to clarify dependencies.
    - Cleaned up unused imports to enhance code readability.
    
    * lint fix
    6d44c465
benchmark_mla.py 20.8 KB