Update on "JIT pass for add relu fusion."

kimishpatel · kimishpatel · commit b2628b461bb4 · 2020-07-08T16:53:00.000-07:00
Summary: Building on top of previous PR that adds fused add_relu op, this PR adds a JIT pass to transform input graph to find all fusable instancs of add + relu and fuses them. Test Plan: python test/test_jit.py TestJit.test_add_relu_fusion Reviewers: Subscribers: Tasks: Tags: Differential Revision: [D21822396](https://our.internmc.facebook.com/intern/diff/D21822396) [ghstack-poisoned]
diff --git a/torch/csrc/jit/passes/fuse_relu.cpp b/torch/csrc/jit/passes/fuse_relu.cpp
@@ -1,4 +1,3 @@
-
 #include <torch/csrc/jit/passes/fuse_relu.h>
 #include <torch/csrc/jit/ir/ir.h>
 #include <torch/csrc/jit/ir/subgraph_matcher.h>

Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,3 @@`
`1`		`-`
`2`	`1`	`#include <torch/csrc/jit/passes/fuse_relu.h>`
`3`	`2`	`#include <torch/csrc/jit/ir/ir.h>`
`4`	`3`	`#include <torch/csrc/jit/ir/subgraph_matcher.h>`