From c411d443eebf9e81552006d04ef0f3d92af16a0e Mon Sep 17 00:00:00 2001 From: Iwan Kawrakow Date: Tue, 26 Aug 2025 08:43:59 +0300 Subject: [PATCH] Add CUDA fp8 header --- ggml/src/ggml-cuda/vendors/cuda.h | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/ggml/src/ggml-cuda/vendors/cuda.h b/ggml/src/ggml-cuda/vendors/cuda.h index 840809a1..30ddbb3c 100644 --- a/ggml/src/ggml-cuda/vendors/cuda.h +++ b/ggml/src/ggml-cuda/vendors/cuda.h @@ -6,6 +6,10 @@ #include #include +#if CUDART_VERSION >= 12050 +#include +#endif // CUDART_VERSION >= 12050 + #if CUDART_VERSION < 11020 #define CU_DEVICE_ATTRIBUTE_VIRTUAL_MEMORY_MANAGEMENT_SUPPORTED CU_DEVICE_ATTRIBUTE_VIRTUAL_ADDRESS_MANAGEMENT_SUPPORTED #define CUBLAS_TF32_TENSOR_OP_MATH CUBLAS_TENSOR_OP_MATH