fix MUSA build

MollySophia · MollySophia · commit 35a8f06817a9 · 2025-03-17T10:38:26.000+08:00
Signed-off-by: Molly Sophia &lt;mollysophia379@gmail.com&gt;
diff --git a/ggml/src/ggml-cuda/wkv.cu b/ggml/src/ggml-cuda/wkv.cu
@@ -79,7 +79,9 @@ static __global__ void rwkv_wkv7_f32(const int B, const int T, const int C, cons
     float state[head_size];
     __shared__ float _r[head_size], _w[head_size], _k[head_size], _a[head_size], _b[head_size];
 
+#ifndef GGML_USE_MUSA
     #pragma unroll
+#endif
     for (int i = 0; i < head_size; i++) {
         state[i] = s[batch_i * state_size + head_i * head_size * head_size + tid * head_size + i];
     }

Original file line number	Diff line number	Diff line change
`@@ -79,7 +79,9 @@ static __global__ void rwkv_wkv7_f32(const int B, const int T, const int C, cons`
`79`	`79`	`float state[head_size];`
`80`	`80`	`__shared__ float _r[head_size], _w[head_size], _k[head_size], _a[head_size], _b[head_size];`
`81`	`81`
	`82`	`+#ifndef GGML_USE_MUSA`
`82`	`83`	`#pragma unroll`
	`84`	`+#endif`
`83`	`85`	`for (int i = 0; i < head_size; i++) {`
`84`	`86`	`state[i] = s[batch_i * state_size + head_i * head_size * head_size + tid * head_size + i];`
`85`	`87`	`}`