update global thread test

microsoft · Jan 23, 2024 · 298347c · 298347c
1 parent 93373c9
commit 298347c
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/onnxruntime/test/global_thread_pools/test_inference.cc b/onnxruntime/test/global_thread_pools/test_inference.cc
@@ -15,7 +15,6 @@
 #include "test_allocator.h"
 #include "../shared_lib/test_fixture.h"
 #include <stdlib.h>
-#include "test/common/cuda_op_test_utils.h"
 
 struct Input {
   const char* name = nullptr;
@@ -56,11 +55,11 @@ static void RunSession(OrtAllocator& allocator, Ort::Session& session_object,
   // size_t total_len = type_info.GetElementCount();
   ASSERT_EQ(values_y.size(), static_cast<size_t>(5));
 
-  auto tolerance = 1e-6f;
+// test inference is using onnxruntime_shared_lib_test_LIBS, so HasCudaEnvironment(800) isn't available
 #ifdef USE_CUDA
-  if (HasCudaEnvironment(800)) {
-    tolerance = 1e-5f;
-  }
+  const float tolerance = 1e-5f;
+#else
+  const float tolerance = 1e-6f;
 #endif
   OutT* f = output_tensor->GetTensorMutableData<OutT>();
   for (size_t i = 0; i != static_cast<size_t>(5); ++i) {