Skip to content

Commit 6d1a1fd

Browse files
author
Anurag Dixit
committed
fix: Guard test activation for CI testing
Signed-off-by: Anurag Dixit <[email protected]>
1 parent 2699f63 commit 6d1a1fd

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

Diff for: tests/core/conversion/converters/test_activation.cpp

+2
Original file line numberDiff line numberDiff line change
@@ -200,6 +200,7 @@ TEST(Converters, ATenEluConvertsCorrectly) {
200200
ASSERT_TRUE(torch_tensorrt::tests::util::almostEqual(jit_results[0], trt_results[0], 2e-6));
201201
}
202202

203+
#ifndef DISABLE_TEST_IN_CI
203204
TEST(Converters, ATenGELUConvertsCorrectly) {
204205
const auto graph = R"IR(
205206
graph(%0 : Tensor):
@@ -225,3 +226,4 @@ TEST(Converters, ATenGELUConvertsCorrectly) {
225226

226227
ASSERT_TRUE(torch_tensorrt::tests::util::almostEqual(jit_results[0], trt_results[0], 5e-2));
227228
}
229+
#endif

0 commit comments

Comments
 (0)