ExternalVendorCode/whisper.cpp
1
0
Fork 0
mirror of https://github.com/ggerganov/whisper.cpp.git synced 2025-04-19 00:27:10 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

vulkan: fix bug in coopmat1 mul_mat_id (llama/12316)

Browse Source 
* tests: run mul_mat_id with a larger N

* vulkan: fix bug in coopmat1 mul_mat_id
This commit is contained in:
Jeff Bolz 2025-03-12 00:59:19 -05:00 committed by Georgi Gerganov
parent 394fae57c3
commit 08f32992d0
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
ggml/src/ggml-vulkan/vulkan-shaders/mul_mm.comp
View File

@ -777,7 +777,7 @@ void main() {
[[unroll]] for (uint cm_col = 0; cm_col < cms_per_col; cm_col++) {
coopMatStore(sums[cm_col * cms_per_row + cm_row], coopmat_stage, warp_i * TM * TN, TM, gl_CooperativeMatrixLayoutColumnMajor);
[[unroll]] for (uint col = 0; col < BN; col += storestride) {
[[unroll]] for (uint col = 0; col < TN; col += storestride) {
const uint row_i = dc + cm_col * TN + col + store_c;
if (row_i >= _ne1) break;