Skip to content

fix: correct error messages and outdated function references#131

Open
yurekami wants to merge 1 commit intodeepseek-ai:mainfrom
yurekami:fix/docstring-function-name
Open

fix: correct error messages and outdated function references#131
yurekami wants to merge 1 commit intodeepseek-ai:mainfrom
yurekami:fix/docstring-function-name

Conversation

@yurekami
Copy link
Copy Markdown

Summary

This PR fixes incorrect error messages and outdated documentation.

Fixes

  1. Outdated function name in docstring (flash_mla/flash_mla_interface.py)

    • The get_mla_metadata docstring referenced the old function name flash_mla_with_kvcache_sm90
    • Updated to reference the current function name flash_mla_with_kvcache
  2. Wrong value in error message (csrc/pybind.cpp:230)

    • Error message said: "Only head_size_v == 576 is supported"
    • But the actual check is: head_size_v == 512
    • Fixed message to correctly say: "Only head_size_v == 512 is supported"

Test plan

  • Verify error messages are accurate
  • Verify docstring references correct function name

🤖 Generated with Claude Code

Documentation fixes:
- Update `get_mla_metadata` docstring to reference `flash_mla_with_kvcache`
  instead of the old name `flash_mla_with_kvcache_sm90`

Error message fixes:
- Fix typo in head_size_v error message: said "576" but should be "512"
  (the check validates head_size_v == 512)

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant