diff --git a/tests/core/conversion/evaluators/test_aten_evaluators.cpp b/tests/core/conversion/evaluators/test_aten_evaluators.cpp index b7aae61bb0..29bdc89152 100644 --- a/tests/core/conversion/evaluators/test_aten_evaluators.cpp +++ b/tests/core/conversion/evaluators/test_aten_evaluators.cpp @@ -650,6 +650,7 @@ TEST(Evaluators, AtenFormatRaiseExceptionEvaluatesCorrectly) { %2 : str = prim::Constant[value="{} is not equal to {}"]() %3 : str = prim::Constant[value="res5_2"]() %5713 : Tensor = prim::Uninitialized() + %32 : None = prim::Constant() %4 : str = aten::format(%2, %1, %3) %5 : bool = aten::eq(%1, %3) %y : Tensor = prim::If(%5) @@ -657,7 +658,7 @@ TEST(Evaluators, AtenFormatRaiseExceptionEvaluatesCorrectly) { %194 : Tensor = aten::add(%x_1, %x_2, %0) -> (%194) block1(): - prim::RaiseException(%4) + prim::RaiseException(%4, %32) -> (%5713) return (%y))IR"; auto g = std::make_shared();