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