Do not include arm_neon.h when compiling CUDA code (ggerganov#1028)

ypapadop-amd · Dec 2, 2024 · d54ace5 · d54ace5
1 parent 437f901
commit d54ace5
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(__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/