From 16ab5f1b184f90552cc944e82b2f25d68d2ce367 Mon Sep 17 00:00:00 2001
From: Kylin <y13308074856@gmail.com>
Date: Sun, 20 Aug 2023 01:57:24 +0800
Subject: [PATCH] ggml: use __CUDACC__ to recognise nvcc compiler

---
 ggml.h | 7 +++----
 1 file changed, 3 insertions(+), 4 deletions(-)

diff --git a/ggml.h b/ggml.h
index 413bc1de5..4fa78a427 100644
--- a/ggml.h
+++ b/ggml.h
@@ -255,11 +255,10 @@
 extern "C" {
 #endif
 
-#if defined(__ARM_NEON) && !defined(GGML_CUDA_F16)
-    // we use the built-in 16-bit float type
-    typedef __fp16 ggml_fp16_t;
-#elif defined(GGML_CUDA_F16)
+#if defined(__ARM_NEON) && defined(__CUDACC__)
     typedef half ggml_fp16_t;
+#elif defined(__ARM_NEON)
+    typedef __fp16 ggml_fp16_t;
 #else
     typedef uint16_t ggml_fp16_t;
 #endif