Skip to content

Commit 16ab5f1

Browse files
committed
ggml: use __CUDACC__ to recognise nvcc compiler
1 parent 8a25bd4 commit 16ab5f1

File tree

1 file changed

+3
-4
lines changed

1 file changed

+3
-4
lines changed

ggml.h

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -255,11 +255,10 @@
255255
extern "C" {
256256
#endif
257257

258-
#if defined(__ARM_NEON) && !defined(GGML_CUDA_F16)
259-
// we use the built-in 16-bit float type
260-
typedef __fp16 ggml_fp16_t;
261-
#elif defined(GGML_CUDA_F16)
258+
#if defined(__ARM_NEON) && defined(__CUDACC__)
262259
typedef half ggml_fp16_t;
260+
#elif defined(__ARM_NEON)
261+
typedef __fp16 ggml_fp16_t;
263262
#else
264263
typedef uint16_t ggml_fp16_t;
265264
#endif

0 commit comments

Comments
 (0)