Skip to content

[Fix] fix the flash_attn_v1 when Br!=Bc #68

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 3 commits into from
Apr 12, 2025

Conversation

zeroRains
Copy link
Contributor

修复flash_attn_v1.cu中,当Br!=Bc时,共享内存空间大小的设置,以及后面取出Qi的索引和处理方式

@AndSonder
Copy link
Collaborator

@zeroRains 请使用 pre-commit format md 文件
image

@zeroRains
Copy link
Contributor Author

@zeroRains 请使用 pre-commit format md 文件 image

Done

@AndSonder AndSonder changed the title fix the flash_attn_v1 when Br!=Bc [Fix] fix the flash_attn_v1 when Br!=Bc Apr 12, 2025
Copy link
Collaborator

@AndSonder AndSonder left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM, Great work!

@AndSonder AndSonder merged commit 919a26d into PaddleJitLab:develop Apr 12, 2025
2 checks passed
@zeroRains zeroRains deleted the fa1 branch April 12, 2025 16:21
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.

2 participants