File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed
Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -12066,7 +12066,7 @@ static void ggml_compute_forward_gla_f32(
1206612066 #define GGML_F32X_STORE GGML_F32x4_STORE
1206712067 #define GGML_F32X_MUL GGML_F32x4_MUL
1206812068 #define GGML_F32X_FMA GGML_F32x4_FMA
12069- #define WKV_VECTOR_SIZE 4
12069+ #define GLA_VECTOR_SIZE 4
1207012070 #endif
1207112071
1207212072 #ifdef GLA_VECTOR_SIZE
@@ -12121,7 +12121,7 @@ static void ggml_compute_forward_gla_f32(
1212112121 }
1212212122
1212312123 // Handle remaining elements, this will not be used.
12124- for (int64_t j = vec_count * WKV_VECTOR_SIZE ; j < head_size ; j ++ ) {
12124+ for (int64_t j = vec_count * GLA_VECTOR_SIZE ; j < head_size ; j ++ ) {
1212512125 size_t t_h_j_offset = t_h_offset + j ;
1212612126 size_t h_2d_i_j_offset = h_2d_i_offset + j ;
1212712127 float v_val = v [t_h_j_offset ];
You can’t perform that action at this time.
0 commit comments