Add header for FLT_MIN/FLT_MAX

matthewdouglas · matthewdouglas · commit 12297f3b6eff · 2025-12-10T14:33:18.000-05:00
diff --git a/csrc/cuda/blockwise_quantization.cu b/csrc/cuda/blockwise_quantization.cu
@@ -2,6 +2,7 @@
 #include "ops.cuh" // For CUDA_CHECK_RETURN, some typedefs
 #include <cub/cub.cuh>
 #include <cuda_fp16.h>
+#include <float.h> // For FLT_MIN/FLT_MAX
 
 // from kernels.cu
 // TODO move somewhere like common.cuh or cub_utils.cuh etc
diff --git a/csrc/cuda/int8.cu b/csrc/cuda/int8.cu
@@ -3,6 +3,7 @@
 #include <cub/cub.cuh>
 #include <cublasLt.h>
 #include <cuda_fp16.h>
+#include <float.h> // For FLT_MIN/FLT_MAX
 
 #define MM_DEQUANT_CONST 6.200012e-05f // 1.0f/(127.0f*127.0f)
 
diff --git a/csrc/cuda/optimizers.cu b/csrc/cuda/optimizers.cu
@@ -1,5 +1,6 @@
 #include "ops.cuh" // For Optimizer_t, CUDA_CHECK_RETURN
 #include <cub/cub.cuh>
+#include <float.h> // For FLT_MIN/FLT_MAX
 
 // constants for 32bit optimizers
 #define TH 1024
