Skip to content

Commit 625b711

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

File tree

1 file changed

+4
-5
lines changed

1 file changed

+4
-5
lines changed

ggml.h

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -255,13 +255,12 @@
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
264-
typedef uint16_t ggml_fp16_t;
263+
typedef uint64_t ggml_fp16_t;
265264
#endif
266265

267266
// convert FP16 <-> FP32

0 commit comments

Comments
 (0)