fix the insert_guard for norm decomposation

wanchaol · wanchaol · commit 67ac45eb60f2 · 2019-04-24T16:27:05.000-07:00
diff --git a/torch/csrc/jit/passes/graph_fuser.cpp b/torch/csrc/jit/passes/graph_fuser.cpp
@@ -292,8 +292,6 @@ struct GraphFuser {
         source,
         method_name);
 
-    AT_ASSERT(isFusableNorm(normalization_op));
-    WithInsertPoint insert_guard{normalization_op};
     Value* new_output =
         SubgraphUtils::inlineGraph(nm_graph, inputs, normalization_op).at(0);
     return new_output;
@@ -327,6 +325,8 @@ struct GraphFuser {
 
             return (input - mean) * invstd
       )SCRIPT";
+    AT_ASSERT(isFusableNorm(normalization_op));
+    WithInsertPoint insert_guard{normalization_op};
     Value* input = normalization_op->namedInput(attr::input);
     if (normalization_op->kind() == aten::batch_norm) {
       Value* input_dim = graph_->insert(aten::dim, {input});
