diff --git a/paddle/framework/op_desc.cc b/paddle/framework/op_desc.cc
index a217cdd20964452cebda101462823cc47e0e91fc..a5d515bbca729220ca6df5fa07d02f1b3f025109 100644
--- a/paddle/framework/op_desc.cc
+++ b/paddle/framework/op_desc.cc
@@ -103,6 +103,7 @@ void OpDescBind::SetAttr(const std::string &name, const Attribute &v) {
 void OpDescBind::SetBlockAttr(const std::string &name, BlockDescBind &block) {
   BlockDesc *desc = block.RawPtr();
   this->attrs_[name] = desc;
+  need_update_ = true;
 }
 
 void OpDescBind::SetAttrMap(