Disable neon when enable gpu.

8 years ago · 4ed5f7dbb9
parent 688305f82d
commit 4ed5f7dbb9
1 changed files with 1 additions and 1 deletions
--- a/paddle/cuda/include/hl_matrix_type.cuh
+++ b/paddle/cuda/include/hl_matrix_type.cuh
@ -38,7 +38,7 @@ typedef double2 vecType;
 #endif
 #elif defined(__SSE3__)
 #include "hl_cpu_simd_sse.cuh"
-#elif defined(__ARM_NEON) || defined(__ARM_NEON__)
+#elif (defined(__ARM_NEON) || defined(__ARM_NEON__)) && !defined(__NVCC__)
 #include "hl_cpu_simd_neon.cuh"
 #else
 #include "hl_cpu_scalar.cuh"