From f584416aa29164e1ad10bdfbfe455cd2360da05c Mon Sep 17 00:00:00 2001
From: Megvii Engine Team <megengine@megvii.com>
Date: Fri, 14 May 2021 14:22:08 +0800
Subject: [PATCH] fix(dnn/bn): revise the conditions for inplace flag

GitOrigin-RevId: 59a104bf6a0dd4989c8a24f2314b7232796be818
---
 src/opr/impl/dnn/batch_norm.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/opr/impl/dnn/batch_norm.cpp b/src/opr/impl/dnn/batch_norm.cpp
index 4cd502d2..98975530 100644
--- a/src/opr/impl/dnn/batch_norm.cpp
+++ b/src/opr/impl/dnn/batch_norm.cpp
@@ -137,7 +137,7 @@ SymbolVarArray BatchNormForward::make(SymbolVar x,
 cg::OperatorNodeBase::NodeProp*
 BatchNormForward::do_make_node_prop() const {
     auto ret = Super::do_make_node_prop();
-    if (need_stats()) {
+    if (need_stats() && m_force_inplace) {
         ret->add_flag(NodeProp::Flag::FORCE_UPDATE_INPUT_VAR);
     }
     return ret;
-- 
GitLab