From 8af2ffc37996b4696ec3e06fabe2f79fb2f4ff77 Mon Sep 17 00:00:00 2001 From: Bo Wang Date: Sun, 11 Sep 2022 23:44:49 -0700 Subject: [PATCH 1/2] fix: add remove_exception pass from torch to fix uninitialized tensor bug Signed-off-by: Bo Wang --- core/lowering/passes/exception_elimination.cpp | 2 ++ 1 file changed, 2 insertions(+) diff --git a/core/lowering/passes/exception_elimination.cpp b/core/lowering/passes/exception_elimination.cpp index 63581af1ca..79ff2c2a06 100644 --- a/core/lowering/passes/exception_elimination.cpp +++ b/core/lowering/passes/exception_elimination.cpp @@ -2,6 +2,7 @@ #include "torch/csrc/jit/jit_log.h" #include "torch/csrc/jit/passes/constant_propagation.h" #include "torch/csrc/jit/passes/dead_code_elimination.h" +#include "torch/csrc/jit/passes/remove_exceptions.h" #include "torch/csrc/jit/passes/guard_elimination.h" #include "torch/csrc/jit/passes/peephole.h" #include "torch/csrc/jit/runtime/graph_executor.h" @@ -21,6 +22,7 @@ struct ExceptionOrPassPatternElimination { void run() { findExceptionOrPassNodes(graph_->block()); + torch::jit::EliminateExceptions(graph_); torch::jit::EliminateDeadCode(graph_); LOG_GRAPH("Post exeception or pass elimination: " << *graph_); } From 574402b88a820a0ac8039eb932aa07dd778d4242 Mon Sep 17 00:00:00 2001 From: Naren Dasan Date: Mon, 12 Sep 2022 09:28:06 -0700 Subject: [PATCH 2/2] chore: apply linting Signed-off-by: Naren Dasan Signed-off-by: Naren Dasan --- core/lowering/passes/exception_elimination.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/core/lowering/passes/exception_elimination.cpp b/core/lowering/passes/exception_elimination.cpp index 79ff2c2a06..02fb773653 100644 --- a/core/lowering/passes/exception_elimination.cpp +++ b/core/lowering/passes/exception_elimination.cpp @@ -2,9 +2,9 @@ #include "torch/csrc/jit/jit_log.h" #include "torch/csrc/jit/passes/constant_propagation.h" #include "torch/csrc/jit/passes/dead_code_elimination.h" -#include "torch/csrc/jit/passes/remove_exceptions.h" #include "torch/csrc/jit/passes/guard_elimination.h" #include "torch/csrc/jit/passes/peephole.h" +#include "torch/csrc/jit/passes/remove_exceptions.h" #include "torch/csrc/jit/runtime/graph_executor.h" #include "core/util/prelude.h"