disable arm neon when using musa

ggerganov · Dec 11, 2024 · 0eb719d · 0eb719d
1 parent ead1612
commit 0eb719d
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 0 deletions.
diff --git a/ggml/include/ggml-cuda.h b/ggml/include/ggml-cuda.h
@@ -1,5 +1,9 @@
 #pragma once
 
+#ifdef GGML_USE_MUSA
+#undef __ARM_NEON
+#endif
+
 #include "ggml.h"
 #include "ggml-backend.h"
 

diff --git a/ggml/src/ggml-cuda/common.cuh b/ggml/src/ggml-cuda/common.cuh
@@ -1,5 +1,9 @@
 #pragma once
 
+#ifdef GGML_USE_MUSA
+#undef __ARM_NEON
+#endif
+
 #include "ggml.h"
 #include "ggml-cuda.h"
 

diff --git a/ggml/src/ggml-cuda/opt-step-adamw.cu b/ggml/src/ggml-cuda/opt-step-adamw.cu
@@ -1,3 +1,7 @@
+#ifdef GGML_USE_MUSA
+#undef __ARM_NEON
+#endif
+
 #include "ggml-impl.h"
 #include "opt-step-adamw.cuh"