Skip to content

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