Просмотр исходного кода

Do not include arm_neon.h when compiling CUDA code (ggml/1028)

Frankie Robertson 1 год назад
Родитель
Сommit
9150f8fef9
1 измененных файлов с 1 добавлено и 1 удалено
  1. 1 1
      ggml/src/ggml-impl.h

+ 1 - 1
ggml/src/ggml-impl.h

@@ -14,7 +14,7 @@
 #include <arm_sve.h>
 #include <arm_sve.h>
 #endif // __ARM_FEATURE_SVE
 #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:
 // 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/
 //   $ ln -sfn /Library/Developer/CommandLineTools/usr/lib/clang/13.1.6/include/arm_neon.h ./src/