diff --git a/tests/test-flash-attention.cpp b/tests/test-flash-attention.cpp
index c99ad719d..fb5e2a8bc 100644
--- a/tests/test-flash-attention.cpp
+++ b/tests/test-flash-attention.cpp
@@ -16,6 +16,7 @@
 #include <map>
 #include <string>
 #include <vector>
+#include <cfloat>
 
 struct test_model {
     struct ggml_tensor * q;