]> git.djapps.eu Git - pkg/ggml/sources/whisper.cpp/commit
vulkan: Flash Attention DP4A shader for quantized KV cache (llama/20797)
authorRuben Ortlam <redacted>
Mon, 13 Apr 2026 12:21:31 +0000 (14:21 +0200)
committerGeorgi Gerganov <redacted>
Thu, 30 Apr 2026 08:29:07 +0000 (11:29 +0300)
commit0f99a47177a887b3771c876fcd5b8de4711c9fe3
treef5022b109f5aacf3ea3e297cbeda162ba7c4f795
parentd9ed371c2c50bdaef8134a05694b826e1cb7f7c6
vulkan: Flash Attention DP4A shader for quantized KV cache (llama/20797)

* use integer dot product for quantized KV flash attention

* small improvements

* fix SHMEM_STAGING indexing

* add missing KV type quants

* fixes

* add supported quants to FA tests

* readd fast paths for <8bit quants

* fix mmq gate and shmem checks
ggml/src/ggml-vulkan/ggml-vulkan.cpp
ggml/src/ggml-vulkan/vulkan-shaders/flash_attn.comp
ggml/src/ggml-vulkan/vulkan-shaders/flash_attn_base.glsl
ggml/src/ggml-vulkan/vulkan-shaders/flash_attn_mmq_funcs.glsl [new file with mode: 0644]
ggml/src/ggml-vulkan/vulkan-shaders/mul_mmq_shmem_types.glsl
ggml/src/ggml-vulkan/vulkan-shaders/types.glsl
ggml/src/ggml-vulkan/vulkan-shaders/vulkan-shaders-gen.cpp