-
Notifications
You must be signed in to change notification settings - Fork 29
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
Update on naive_attn module #21
Conversation
To fix, https://ontrack-internal.amd.com/browse/SWDEV-469079 CP https://github.com/ROCm/vllm/pull/16/files to perf_benchmark_navi branch
This pull request has been automatically marked as stale because it has not had any activity within 90 days. It will be automatically closed if no further activity occurs within 30 days. Leave a comment if you feel this pull request should remain open. Thank you! |
This pull request has been automatically closed due to inactivity. Please feel free to reopen if you intend to continue working on it. Thank you! |
Hi,
Fixed some syntax errors in use_naive_attn module in rocm_flash_attn.py
Also, allowed navi3x to use use_triton_flash_attn.
https://github.com/ROCm/vllm/blob/perf_benchmark_navi/vllm/attention/backends/rocm_flash_attn.py
https://github.com/ROCm/vllm/blob/perf_benchmark_navi/vllm/attention/ops/triton_flash_attention.py
Regards,
Seungrok