feat(//core/lowering): Add tuple lowering pass to remove tuples if

possible Signed-off-by: Naren Dasan <[email protected]> Signed-off-by: Naren Dasan <[email protected]>
pytorch · May 22, 2020 · ce6cf75 · ce6cf75
1 parent d46676d
commit ce6cf75
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/core/lowering/lowering.cpp b/core/lowering/lowering.cpp
@@ -2,6 +2,7 @@
 #include "torch/csrc/jit/passes/fuse_linear.h"
 #include "torch/csrc/jit/passes/freeze_module.h"
 #include "torch/csrc/jit/passes/lower_graph.h"
+#include "torch/csrc/jit/passes/lower_tuples.h"
 #include "torch/csrc/jit/passes/quantization.h"
 #include "torch/csrc/jit/passes/guard_elimination.h"
 
@@ -23,6 +24,7 @@ void LowerGraph(std::shared_ptr<torch::jit::Graph>& g) {
     torch::jit::EliminateRedundantGuards(g);
     passes::EliminateExceptionOrPassPattern(g);
     torch::jit::FuseLinear(g);
+    torch::jit::LowerAllTuples(g);
     passes::RemoveDropout(g);
     passes::FuseFlattenLinear(g);
     passes::Conv2DToConvolution(g);