diff --git a/tests/gtests/internals/test_sdpa.cpp b/tests/gtests/internals/test_sdpa.cpp index 8a0b3ed1f68..409bbdb498a 100644 --- a/tests/gtests/internals/test_sdpa.cpp +++ b/tests/gtests/internals/test_sdpa.cpp @@ -1669,7 +1669,13 @@ void check_memory(memory &gold, memory &test) { int mismatches = 0; int total = 0; - float fthreshold = 0.000978; + float fthreshold = 0.f; + if (std::is_same::value) { + fthreshold = 0.000978f; + } else { + fthreshold = 0.0079f; + } + float max_diff = std::numeric_limits::min(); std::map> hist; for_(int l = 0; l < dims[0]; l++)