diff --git a/onnxruntime/test/shared_lib/test_inference.cc b/onnxruntime/test/shared_lib/test_inference.cc index f53a93225b..c352077e34 100644 --- a/onnxruntime/test/shared_lib/test_inference.cc +++ b/onnxruntime/test/shared_lib/test_inference.cc @@ -81,7 +81,11 @@ void RunSession(OrtAllocator* allocator, Ort::Session& session_object, OutT* f = output_tensor->GetTensorMutableData(); for (size_t i = 0; i != total_len; ++i) { - ASSERT_EQ(values_y[i], f[i]); + if constexpr (std::is_same::value || std::is_same::value) { + ASSERT_NEAR(values_y[i], f[i], 1e-3); + } else { + ASSERT_EQ(values_y[i], f[i]); + } } }