Skip to content

Commit c16bc3a

Browse files
committed
fix: Remove eliminate exceptions lowering pass
- Torch-TensorRT has evaluator support for exception elimination and this particular lowering pass causes certain models to halt compilation indefinitely - Remove this lowering pass temporarily while a more robust and safer long-term solution is in development
1 parent b3f433a commit c16bc3a

File tree

1 file changed

+0
-2
lines changed

1 file changed

+0
-2
lines changed

core/lowering/lowering.cpp

-2
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,6 @@
99
#include "torch/csrc/jit/passes/lower_graph.h"
1010
#include "torch/csrc/jit/passes/lower_tuples.h"
1111
#include "torch/csrc/jit/passes/peephole.h"
12-
#include "torch/csrc/jit/passes/remove_exceptions.h"
1312
#include "torch/csrc/jit/passes/remove_mutation.h"
1413

1514
#include "core/lowering/lowering.h"
@@ -105,7 +104,6 @@ void LowerGraph(std::shared_ptr<torch::jit::Graph>& g, std::vector<torch::jit::I
105104
torch::jit::InlineFunctionalGraphs(g);
106105
torch::jit::PeepholeOptimize(g, false);
107106
torch::jit::FuseLinear(g);
108-
torch::jit::EliminateExceptions(g);
109107
if (!lower_info.disable_cse) {
110108
torch::jit::EliminateCommonSubexpression(g);
111109
}

0 commit comments

Comments
 (0)