Skip to content

Commit 9a240e6

Browse files
author
Frankie Robertson
committed
Avoid using __fp16 on ARM with old nvcc
1 parent 3420909 commit 9a240e6

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

ggml/src/ggml-impl.h

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -310,14 +310,14 @@ void ggml_aligned_free(void * ptr, size_t size);
310310
// FP16 to FP32 conversion
311311

312312
#if defined(__ARM_NEON)
313-
#ifdef _MSC_VER
313+
#ifdef _MSC_VER || (defined(__CUDACC__) && __CUDACC_VER_MAJOR__ <= 11)
314314
typedef uint16_t ggml_fp16_internal_t;
315315
#else
316316
typedef __fp16 ggml_fp16_internal_t;
317317
#endif
318318
#endif
319319

320-
#if defined(__ARM_NEON) && !defined(_MSC_VER)
320+
#if defined(__ARM_NEON) && !defined(_MSC_VER) && !(defined(__CUDACC__) && __CUDACC_VER_MAJOR__ <= 11)
321321
#define GGML_COMPUTE_FP16_TO_FP32(x) ggml_compute_fp16_to_fp32(x)
322322
#define GGML_COMPUTE_FP32_TO_FP16(x) ggml_compute_fp32_to_fp16(x)
323323

0 commit comments

Comments
 (0)