Polish code

test=develop
6 years ago · 212242c4e4
parent 1b10a7843c
commit 212242c4e4
2 changed files with 189 additions and 192 deletions
--- a/paddle/fluid/imperative/layer.h
+++ b/paddle/fluid/imperative/layer.h
@ -243,8 +243,10 @@ class PYBIND11_HIDDEN OpBase {
    }

    // remove op desc from block desc
-    if (block_) {
-      block_->RemoveOpInternal(op_desc_);
+    if (op_desc_) {
+      if (block_) {
+        block_->RemoveOpInternal(op_desc_);
+      }
    }

    // release resource
--- a/python/paddle/fluid/tests/unittests/test_imperative_basic.py
+++ b/python/paddle/fluid/tests/unittests/test_imperative_basic.py