]> git.djapps.eu Git - pkg/ggml/sources/whisper.cpp/commit
vulkan: add POOL_1D op (llama/25431)
authorAnand Patil <redacted>
Fri, 31 Jul 2026 14:48:58 +0000 (09:48 -0500)
committerGeorgi Gerganov <redacted>
Tue, 4 Aug 2026 10:37:47 +0000 (13:37 +0300)
commitcb625868e8486f15412f563953fdf9a2507305d2
treea15b2eebf3b20ccb2dcea2ea8529672a21f0d518
parent153a336f660f6472b3f28bf8dccd105304893b22
vulkan: add POOL_1D op (llama/25431)

* vulkan : add pool1d push constants and pipeline field

Declared data structures needed for POOL1D OP, which are the vk_op_pool1d_push_constants struct and pipeline_pool1d_f32 field.

* vulkan : add pool1d compute shader

Added pool1d.comp for Vulkan backend mirroring the existing pool2d shader.

* vulkan : add full GGML_OP_POOL_1D support

Added pipeline creation and op dispatch for 1D pooling in the Vulkan backend.

* vulkan : fix pool1d shader logic

Registered pool1d_f32 in vulkan-shaders-gen.cpp and fixed tensor dimension indices and avg pool scale.

* vulkan : fix pool1d end boundary crash and expand test coverage

Fixed an issue where the shader crashed when the end boundary was negative when k0 < p0. Also, added more test cases related to this fix.
ggml/src/ggml-vulkan/ggml-vulkan.cpp
ggml/src/ggml-vulkan/vulkan-shaders/pool1d.comp [new file with mode: 0644]
ggml/src/ggml-vulkan/vulkan-shaders/vulkan-shaders-gen.cpp