]> git.djapps.eu Git - pkg/ggml/sources/ggml/commit
Vulkan: VK_KHR_cooperative_matrix support to speed up prompt processing (llama/10597)
author0cc4m <redacted>
Sat, 7 Dec 2024 09:24:15 +0000 (10:24 +0100)
committerGeorgi Gerganov <redacted>
Tue, 10 Dec 2024 16:33:03 +0000 (18:33 +0200)
commit72f04e91375ca8a4350ba5afe739308b3b20ba18
treeacfce258119152b33d8ed1e8e8423bf97a91cc72
parentf1e5a9ddc9702631c3e4f3a6b1183536d6db7de4
Vulkan: VK_KHR_cooperative_matrix support to speed up prompt processing (llama/10597)

* Vulkan: Implement VK_KHR_cooperative_matrix support in the matrix matrix multiplication shader

* Improve performance with better q4_k and q5_k dequant and store unrolling

* Add Vulkan MUL_MAT and MUL_MAT_ID accumulator precision selection

* Rework mulmat shader selection and compilation logic, avoid compiling shaders that won't get used by device

* Vulkan: Implement accumulator switch for specific mul mat mat shaders

* Vulkan: Unroll more loops for more mul mat mat performance

* Vulkan: Add VK_AMD_shader_core_properties2 support to read Compute Unit count for split_k logic

* Disable coopmat support on AMD proprietary driver

* Remove redundant checks

* Add environment variable GGML_VK_DISABLE_COOPMAT to disable VK_KHR_cooperative_matrix support

* Fix rebase typo

* Fix coopmat2 MUL_MAT_ID pipeline selection
src/ggml-vulkan/ggml-vulkan.cpp
src/ggml-vulkan/vulkan-shaders/mul_mm.comp
src/ggml-vulkan/vulkan-shaders/vulkan-shaders-gen.cpp