Merge pull request #724 from NVIDIA/anuragd/ci_flag_guard_test_activation

andi4191 · web-flow · commit a234edd328b4 · 2021-11-24T11:43:45.000-08:00
fix: Guard test activation for CI testing
diff --git a/tests/core/conversion/converters/test_activation.cpp b/tests/core/conversion/converters/test_activation.cpp
@@ -201,6 +201,7 @@ TEST(Converters, ATenEluConvertsCorrectly) {
   ASSERT_TRUE(torch_tensorrt::tests::util::almostEqual(jit_results[0], trt_results[0], 2e-6));
 }
 
+#ifndef DISABLE_TEST_IN_CI
 TEST(Converters, ATenGELUConvertsCorrectly) {
   const auto graph = R"IR(
       graph(%0 : Tensor):
@@ -230,3 +231,4 @@ TEST(Converters, ATenGELUConvertsCorrectly) {
 
   ASSERT_TRUE(torch_tensorrt::tests::util::almostEqual(jit_results[0], trt_results[0], 5e-2));
 }
+#endif

Original file line number	Diff line number	Diff line change
`@@ -201,6 +201,7 @@ TEST(Converters, ATenEluConvertsCorrectly) {`
`201`	`201`	`ASSERT_TRUE(torch_tensorrt::tests::util::almostEqual(jit_results[0], trt_results[0], 2e-6));`
`202`	`202`	`}`
`203`	`203`
	`204`	`+#ifndef DISABLE_TEST_IN_CI`
`204`	`205`	`TEST(Converters, ATenGELUConvertsCorrectly) {`
`205`	`206`	`const auto graph = R"IR(`
`206`	`207`	`graph(%0 : Tensor):`
`@@ -230,3 +231,4 @@ TEST(Converters, ATenGELUConvertsCorrectly) {`
`230`	`231`
`231`	`232`	`ASSERT_TRUE(torch_tensorrt::tests::util::almostEqual(jit_results[0], trt_results[0], 5e-2));`
`232`	`233`	`}`
	`234`	`+#endif`