Skip to content

Commit b4e6cf4

Browse files
committed
WKV7 Vulkan bugfix
Signed-off-by: Molly Sophia <[email protected]>
1 parent d11f487 commit b4e6cf4

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

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

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -57,7 +57,7 @@ void main() {
5757
A_TYPE sa = 0.0;
5858
[[unroll]] for (uint j = 0; j < head_size; j += 4) {
5959
vec4 s_vec = vec4(state[j], state[j+1], state[j+2], state[j+3]);
60-
vec4 a_vec = vec4(a[j], a[j+1], a[j+2], a[j+3]);
60+
vec4 a_vec = vec4(_a[j], _a[j+1], _a[j+2], _a[j+3]);
6161
sa += dot(s_vec, a_vec);
6262
}
6363

0 commit comments

Comments
 (0)