From 0014b849bfb66a853990dad780693921c764c51a Mon Sep 17 00:00:00 2001 From: Naren Dasan Date: Thu, 4 Jun 2020 17:05:32 -0700 Subject: [PATCH] feat(//core/lowering): Adds peephole optimization pass Signed-off-by: Naren Dasan Signed-off-by: Naren Dasan --- core/lowering/lowering.cpp | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/core/lowering/lowering.cpp b/core/lowering/lowering.cpp index 7ce259896b..3424a2ea97 100644 --- a/core/lowering/lowering.cpp +++ b/core/lowering/lowering.cpp @@ -2,11 +2,12 @@ #include "torch/csrc/jit/passes/dead_code_elimination.h" #include "torch/csrc/jit/passes/fuse_linear.h" #include "torch/csrc/jit/passes/freeze_module.h" +#include "torch/csrc/jit/passes/guard_elimination.h" #include "torch/csrc/jit/passes/loop_unrolling.h" #include "torch/csrc/jit/passes/lower_graph.h" #include "torch/csrc/jit/passes/lower_tuples.h" +#include "torch/csrc/jit/passes/peephole.h" #include "torch/csrc/jit/passes/quantization.h" -#include "torch/csrc/jit/passes/guard_elimination.h" #include "core/util/prelude.h" #include "core/lowering/lowering.h" @@ -24,6 +25,7 @@ void LowerBlock(torch::jit::Block* b) { void LowerGraph(std::shared_ptr& g) { torch::jit::EliminateRedundantGuards(g); + torch::jit::PeepholeOptimize(g, false); passes::EliminateExceptionOrPassPattern(g); torch::jit::FuseLinear(g); torch::jit::LowerAllTuples(g);