Skip to content

Commit be17331

Browse files
committed
vulkan: fix bug in coopmat1 mul_mat_id
1 parent 5e220c3 commit be17331

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

ggml/src/ggml-vulkan/vulkan-shaders/mul_mm.comp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -777,7 +777,7 @@ void main() {
777777
[[unroll]] for (uint cm_col = 0; cm_col < cms_per_col; cm_col++) {
778778
coopMatStore(sums[cm_col * cms_per_row + cm_row], coopmat_stage, warp_i * TM * TN, TM, gl_CooperativeMatrixLayoutColumnMajor);
779779

780-
[[unroll]] for (uint col = 0; col < BN; col += storestride) {
780+
[[unroll]] for (uint col = 0; col < TN; col += storestride) {
781781
const uint row_i = dc + cm_col * TN + col + store_c;
782782
if (row_i >= _ne1) break;
783783

0 commit comments

Comments
 (0)