diff --git a/ggml/src/ggml-cuda/fattn-common.cuh b/ggml/src/ggml-cuda/fattn-common.cuh
index 8dc82a9d3b8..57a9781248d 100644
--- a/ggml/src/ggml-cuda/fattn-common.cuh
+++ b/ggml/src/ggml-cuda/fattn-common.cuh
@@ -4,6 +4,7 @@
 #include "convert.cuh"
 #include "vecdotq.cuh"
 
+#include <cmath>
 #include <cstdint>
 
 #define FATTN_KQ_STRIDE       256
@@ -595,7 +596,7 @@ static __global__ void flash_attn_mask_to_KV_max(
 #pragma unroll
         for (int j = 0; j < ncols1; ++j) {
             const float2 tmp = __half22float2(mask[j*s31 + KV_max_sj/2 + tid]);
-            all_inf = all_inf && int(isinf(tmp.x)) && int(isinf(tmp.y));
+            all_inf = all_inf && int(std::isinf(tmp.x)) && int(std::isinf(tmp.y));
         }
 
         all_inf = warp_reduce_all(all_inf);