Skip to content

Commit 40bc4e3

Browse files
committed
fix: aten::gelu call was wrong in test
Signed-off-by: Naren Dasan <[email protected]> Signed-off-by: Naren Dasan <[email protected]>
1 parent cc7d0b7 commit 40bc4e3

File tree

1 file changed

+2
-6
lines changed

1 file changed

+2
-6
lines changed

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

+2-6
Original file line numberDiff line numberDiff 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-
205203
TEST(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

0 commit comments

Comments
 (0)