Skip to content

Commit 93c2384

Browse files
author
Anurag Dixit
committed
Merge branch 'anuragd/gelu_changes' into 'release/ngc/21.11'
(//tests): Fix gelu changes See merge request adlsa/TRTorch!6
2 parents 78636eb + 5c27343 commit 93c2384

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

tests/core/conversion/converters/test_activation.cpp

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -203,7 +203,8 @@ TEST(Converters, ATenEluConvertsCorrectly) {
203203
TEST(Converters, ATenGELUConvertsCorrectly) {
204204
const auto graph = R"IR(
205205
graph(%0 : Tensor):
206-
%3 : Tensor = aten::gelu(%0)
206+
%1 : bool = prim::Constant[value=0]()
207+
%3 : Tensor = aten::gelu(%0, %1)
207208
return (%3))IR";
208209

209210
auto g = std::make_shared<torch::jit::Graph>();

0 commit comments

Comments
 (0)