add riscv zba extension limit

alex-spacemit · alex-spacemit · commit 2ad4450e5f63 · 2025-08-14T06:57:14.000Z
Change-Id: I321eb200f859751727afe5cae13074dfce2bb0ce
diff --git a/ggml/src/ggml-cpu/vec.cpp b/ggml/src/ggml-cpu/vec.cpp
@@ -460,7 +460,7 @@ ggml_float ggml_vec_soft_max_f32(const int n, float * y, const float * x, float
         vst1q_f32(y + i, val);
         sum += (ggml_float)vaddvq_f32(val);
     }
-#elif defined(__riscv) && defined(__riscv_v)
+#elif defined(__riscv) && defined(__riscv_v) && defined(__riscv_zba)
     int N = n;
     i += n;
     float* src = const_cast<float*>(reinterpret_cast<const float*>(x));

Original file line number	Diff line number	Diff line change
`@@ -460,7 +460,7 @@ ggml_float ggml_vec_soft_max_f32(const int n, float * y, const float * x, float`
`460`	`460`	`vst1q_f32(y + i, val);`
`461`	`461`	`sum += (ggml_float)vaddvq_f32(val);`
`462`	`462`	`}`
`463`		`-#elif defined(__riscv) && defined(__riscv_v)`
	`463`	`+#elif defined(__riscv) && defined(__riscv_v) && defined(__riscv_zba)`
`464`	`464`	`int N = n;`
`465`	`465`	`i += n;`
`466`	`466`	`float* src = const_cast<float>(reinterpret_cast<const float>(x));`