Skip to content

Commit 5c27343

Browse files
author
Anurag Dixit
committed
(//tests): Fix gelu changes
Signed-off-by: Anurag Dixit <[email protected]>
1 parent 65ffaef commit 5c27343

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)