BUG, SIMD: Fix detecting NEON/ASIMD on aarch64

numpy · Jun 14, 2022 · 3b792f1 · 3b792f1
1 parent 5c598ed
commit 3b792f1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/numpy/distutils/checks/cpu_asimd.c b/numpy/distutils/checks/cpu_asimd.c
@@ -14,7 +14,7 @@ int main(int argc, char **argv)
     ret += (int)vgetq_lane_f32(vrndq_f32(v1), 0);
 #ifdef __aarch64__
     {
-        double *src2 = (float*)argv[argc-1];
+        double *src2 = (double*)argv[argc-1];
         float64x2_t vd1 = vdupq_n_f64(src2[0]), vd2 = vdupq_n_f64(src2[1]);
         /* MAXMIN */
         ret += (int)vgetq_lane_f64(vmaxnmq_f64(vd1, vd2), 0);