• Lei Wang's avatar
    [Example] Add sparse mla examples (#896) · 65ac7454
    Lei Wang authored
    * Update README.md to include directory structure and file descriptions for deepseek_v32 example
    
    * Refactor and clean up deepseek_v32 example scripts
    
    - Removed unused imports and functions from `fp8_mqa_logits.py` to streamline the code.
    - Improved formatting and readability in `sparse_mla_fwd_pipelined.py` and `sparse_mla_fwd.py` by adjusting function signatures and indentation.
    - Added `# ruff: noqa` comments to suppress linting warnings in multiple files.
    - Enhanced the `generate_random_cu_seqlens` function in `utils.py` for better clarity and organization.
    - Updated print statements for consistency in output formatting.
    65ac7454
fp8_mqa_logits.py 10.7 KB