chore: Increase threshold for topk

Signed-off-by: Dheeraj Peri <[email protected]>

Author: peri044

tests/core/conversion/converters/test_topk.cpp

@@ -26,7 +26,7 @@ TEST(Converters, ATenTopKConvertsCorrectly) {
   auto trt_results = torch_tensorrt::tests::util::RunGraphEngine(g, params, {in});
 
   ASSERT_TRUE(
-      torch_tensorrt::tests::util::almostEqual(jit_results[0], trt_results[0].reshape_as(jit_results[0]), 2e-6));
+      torch_tensorrt::tests::util::almostEqual(jit_results[0], trt_results[0].reshape_as(jit_results[0]), 8e-5));
   ASSERT_TRUE(
-      torch_tensorrt::tests::util::almostEqual(jit_results[1], trt_results[1].reshape_as(jit_results[1]), 2e-6));
+      torch_tensorrt::tests::util::almostEqual(jit_results[1], trt_results[1].reshape_as(jit_results[1]), 8e-5));
 }

tests/util/util.cpp

@@ -11,9 +11,9 @@ bool cosineSimEqual(const at::Tensor& computed_tensor, const at::Tensor& gt_tens
       computed_tensor.flatten(), gt_tensor.flatten(), torch::nn::functional::CosineSimilarityFuncOptions().dim(0));
   std::ostringstream ss;
   ss << computed_tensor << std::endl << gt_tensor << std::endl;
-  LOG_GRAPH(ss.str());
-  LOG_GRAPH(std::string("Cosine Similarity score: ") + std::to_string(cosine_sim.item<float>()));
-  LOG_GRAPH(std::string("Acceptable Threshold: ") + std::to_string(threshold));
+  LOG_DEBUG(ss.str());
+  LOG_DEBUG(std::string("Cosine Similarity score: ") + std::to_string(cosine_sim.item<float>()));
+  LOG_DEBUG(std::string("Acceptable Threshold: ") + std::to_string(threshold));
 
   return cosine_sim.item<float>() >= threshold;
 }
@@ -31,14 +31,14 @@ bool almostEqual(const at::Tensor& computed_tensor, const at::Tensor& gt_tensor,
   auto result = diff.abs().max().item<float>();
   auto threshold = atol + (rtol * gt_tensor.abs().max().item<float>());
 
-  LOG_GRAPH(std::string("Max Difference: ") + std::to_string(result));
-  LOG_GRAPH(std::string("Acceptable Threshold: ") + std::to_string(threshold));
+  LOG_DEBUG(std::string("Max Difference: ") + std::to_string(result));
+  LOG_DEBUG(std::string("Acceptable Threshold: ") + std::to_string(threshold));
 
   return result <= threshold;
 }
 
 bool exactlyEqual(const at::Tensor& computed_tensor, const at::Tensor& gt_tensor) {
-  LOG_GRAPH(computed_tensor << std::endl << gt_tensor << std::endl);
+  LOG_DEBUG(computed_tensor << std::endl << gt_tensor << std::endl);
   std::cout << "Max Difference: " << (computed_tensor - gt_tensor).abs().max().item<float>() << std::endl;
 
   return (computed_tensor - gt_tensor).abs().max().item<float>() == 0.f;