File tree Expand file tree Collapse file tree 1 file changed +2
-6
lines changed
tests/core/conversion/converters Expand file tree Collapse file tree 1 file changed +2
-6
lines changed Original file line number Diff line number Diff line change @@ -200,13 +200,10 @@ 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
204-
205203TEST (Converters, ATenGELUConvertsCorrectly) {
206204 const auto graph = R"IR(
207205 graph(%0 : Tensor):
208- %1 : bool = prim::Constant[value=0]()
209- %3 : Tensor = aten::gelu(%0, %1)
206+ %3 : Tensor = aten::gelu(%0)
210207 return (%3))IR" ;
211208
212209 auto g = std::make_shared<torch::jit::Graph>();
@@ -226,6 +223,5 @@ TEST(Converters, ATenGELUConvertsCorrectly) {
226223 // c10::cuda::compat::normcdf to compute Phi(x). So there's a difference here and therefore the threshold is slightly
227224 // higher than other ops. One in ten runs will give you an out of normal threshold result
228225
229- ASSERT_TRUE (torch_tensorrt::tests::util::almostEqual (jit_results[0 ], trt_results[0 ], 5e-3 ));
226+ ASSERT_TRUE (torch_tensorrt::tests::util::almostEqual (jit_results[0 ], trt_results[0 ], 5e-2 ));
230227}
231- #endif
You can’t perform that action at this time.
0 commit comments