Do not include arm_neon.h when running under nvcc

ggerganov · Nov 26, 2024 · 88b306f · 88b306f
1 parent 6fcbd60
commit 88b306f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/ggml-impl.h b/src/ggml-impl.h
@@ -14,7 +14,7 @@
 #include <arm_sve.h>
 #endif // __ARM_FEATURE_SVE
 
-#if defined(__ARM_NEON)
+#if defined(__ARM_NEON) && !(defined(__NVCC__) && defined(__CUDACC__))
 // if YCM cannot find <arm_neon.h>, make a symbolic link to it, for example:
 //
 //   $ ln -sfn /Library/Developer/CommandLineTools/usr/lib/clang/13.1.6/include/arm_neon.h ./src/