add ifdef __ARM_FEATURE_FP16_VECTOR_ARITHMETIC

ztxz16 · Aug 21, 2024 · 10f8461 · 10f8461
1 parent 29b8fa2
commit 10f8461
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/devices/cpu/linear.cpp b/src/devices/cpu/linear.cpp
@@ -52,7 +52,7 @@ namespace fastllm {
 
     void MatMulFloat16Float16(uint16_t *inputData, uint16_t *weightData, float *biasData, uint16_t *outputData, 
                             int n, int m, int k, int st, int end) {
-#ifdef __aarch64__
+#ifdef __ARM_FEATURE_FP16_VECTOR_ARITHMETIC
         if (n > 1 && n < 8) {
             const int BLOCKA = 7;
             float16x8_t va[BLOCKA], vb, vc[BLOCKA];