Reduce required precision in test

mfeliz-cruise · mfeliz-cruise · commit 0b0666ac45a9 · 2023-01-09T13:33:57.000-08:00
diff --git a/tests/core/conversion/converters/test_batch_norm.cpp b/tests/core/conversion/converters/test_batch_norm.cpp
@@ -165,5 +165,5 @@ TEST(Converters, ATenBatchNormHalfConvertsCorrectly) {
   params = torch_tensorrt::core::ir::get_static_params(g->inputs(), {trt_mean, trt_var});
   auto trt_results = torch_tensorrt::tests::util::RunGraphEngine(g, params, {trt_in}, {nvinfer1::DataType::kHALF});
 
-  ASSERT_TRUE(torch_tensorrt::tests::util::almostEqual(jit_results[0], trt_results[0], 2e-6));
+  ASSERT_TRUE(torch_tensorrt::tests::util::almostEqual(jit_results[0], trt_results[0], 2e-2));
 }

Original file line number	Diff line number	Diff line change
`@@ -165,5 +165,5 @@ TEST(Converters, ATenBatchNormHalfConvertsCorrectly) {`
`165`	`165`	`params = torch_tensorrt::core::ir::get_static_params(g->inputs(), {trt_mean, trt_var});`
`166`	`166`	`auto trt_results = torch_tensorrt::tests::util::RunGraphEngine(g, params, {trt_in}, {nvinfer1::DataType::kHALF});`
`167`	`167`
`168`		`- ASSERT_TRUE(torch_tensorrt::tests::util::almostEqual(jit_results[0], trt_results[0], 2e-6));`
	`168`	`+ ASSERT_TRUE(torch_tensorrt::tests::util::almostEqual(jit_results[0], trt_results[0], 2e-2));`
`169`	`169`	`}`