Disable neon when enable gpu.

4ed5f7db · Dong Li · 688305f8 · 4ed5f7db
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

paddle/cuda/include/hl_matrix_type.cuh paddle/cuda/include/hl_matrix_type.cuh +1 -1

未找到文件。
--- 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"