Skip to content

Commit a04b329

Browse files
jeffbolznvggerganov
authored andcommitted
vulkan: scalar flash attention implementation (llama/13324)
* vulkan: scalar flash attention implementation * vulkan: always use fp32 for scalar flash attention * vulkan: use vector loads in scalar flash attention shader * vulkan: remove PV matrix, helps with register usage * vulkan: reduce register usage in scalar FA, but perf may be slightly worse * vulkan: load each Q value once. optimize O reduction. more tuning * vulkan: support q4_0/q8_0 KV in scalar FA * CI: increase timeout to accommodate newly-supported tests * vulkan: for scalar FA, select between 1 and 8 rows * vulkan: avoid using Float16 capability in scalar FA
1 parent 45d8b23 commit a04b329

File tree

3 files changed

+645
-93
lines changed

3 files changed

+645
-93
lines changed

0 commit comments

Comments
 (0)