diff --git a/ggml/src/ggml-vulkan/vulkan-shaders/gated_delta_net_chunk_inter.comp b/ggml/src/ggml-vulkan/vulkan-shaders/gated_delta_net_chunk_inter.comp index 11cd0e18a8..0aa54e718f 100644 --- a/ggml/src/ggml-vulkan/vulkan-shaders/gated_delta_net_chunk_inter.comp +++ b/ggml/src/ggml-vulkan/vulkan-shaders/gated_delta_net_chunk_inter.comp @@ -60,7 +60,7 @@ void main() { float state[S_V]; [[unroll]] for (uint i = 0; i < S_V; i++) { - state[i] = state_in[state_base + i * S_V + col]; + state[i] = state_in[state_base + col * S_V + i]; } for (uint c = 0; c < n_chunks; c++) { @@ -121,6 +121,6 @@ void main() { // Write final state to dst at s_off [[unroll]] for (uint i = 0; i < S_V; i++) { - final_out[s_off + state_base + i * S_V + col] = state[i]; + final_out[s_off + state_base + col * S_V + i] = state[i]; } }