Skip to content

Commit 2357099

Browse files
committed
fix: Update test for new Exception syntax
Signed-off-by: Naren Dasan <[email protected]> Signed-off-by: Naren Dasan <[email protected]>
1 parent 17340fb commit 2357099

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

tests/core/conversion/evaluators/test_aten_evaluators.cpp

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -650,14 +650,15 @@ TEST(Evaluators, AtenFormatRaiseExceptionEvaluatesCorrectly) {
650650
%2 : str = prim::Constant[value="{} is not equal to {}"]()
651651
%3 : str = prim::Constant[value="res5_2"]()
652652
%5713 : Tensor = prim::Uninitialized()
653+
%32 : None = prim::Constant()
653654
%4 : str = aten::format(%2, %1, %3)
654655
%5 : bool = aten::eq(%1, %3)
655656
%y : Tensor = prim::If(%5)
656657
block0():
657658
%194 : Tensor = aten::add(%x_1, %x_2, %0)
658659
-> (%194)
659660
block1():
660-
prim::RaiseException(%4)
661+
prim::RaiseException(%4, %32)
661662
-> (%5713)
662663
return (%y))IR";
663664
auto g = std::make_shared<torch::jit::Graph>();

0 commit comments

Comments
 (0)