Skip to content

Commit 4fac91e

Browse files
committed
adapt arm platform
1 parent bfdf4b5 commit 4fac91e

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

kernel/simd/intrin_neon.h

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@
88
* Data Type
99
***************************/
1010
typedef float32x4_t v_f32;
11-
#if NPY_SIMD_F64
11+
#if V_SIMD_F64
1212
typedef float64x2_t v_f64;
1313
#endif
1414
#define v_nlanes_f32 4
@@ -33,7 +33,7 @@ typedef float32x4_t v_f32;
3333
#endif
3434

3535
// FUSED F64
36-
#if NPY_SIMD_F64
36+
#if V_SIMD_F64
3737
BLAS_FINLINE v_f64 v_muladd_f64(v_f64 a, v_f64 b, v_f64 c)
3838
{ return vfmaq_f64(c, a, b); }
3939
#endif
@@ -45,7 +45,7 @@ BLAS_FINLINE float v_sum_f32(float32x4_t a)
4545
return vget_lane_f32(vpadd_f32(r, r), 0);
4646
}
4747

48-
#if NPY_SIMD_F64
48+
#if V_SIMD_F64
4949
BLAS_FINLINE double v_sum_f64(float64x2_t a)
5050
{
5151
return vget_lane_f64(vget_low_f64(a) + vget_high_f64(a), 0);
@@ -60,7 +60,7 @@ BLAS_FINLINE float v_sum_f32(float32x4_t a)
6060
#define v_storeu_f32 vst1q_f32
6161
#define v_setall_f32(VAL) vdupq_n_f32(VAL)
6262
#define v_zero_f32() vdupq_n_f32(0.0f)
63-
#if NPY_SIMD_F64
63+
#if V_SIMD_F64
6464
#define v_loadu_f64(a) vld1q_f64((const double*)a)
6565
#define v_storeu_f64 vst1q_f64
6666
#define v_setall_f64 vdupq_n_f64

0 commit comments

Comments
 (0)