diff --git a/tests/core/conversion/converters/test_activation.cpp b/tests/core/conversion/converters/test_activation.cpp index 78ffe0017f..885b7caa91 100644 --- a/tests/core/conversion/converters/test_activation.cpp +++ b/tests/core/conversion/converters/test_activation.cpp @@ -200,6 +200,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): @@ -225,3 +226,4 @@ TEST(Converters, ATenGELUConvertsCorrectly) { ASSERT_TRUE(torch_tensorrt::tests::util::almostEqual(jit_results[0], trt_results[0], 5e-2)); } +#endif