Polish code

test=develop
6 years ago · fe406b98c9
parent f469bb6b36
commit fe406b98c9
2 changed files with 189 additions and 192 deletions
--- a/paddle/fluid/imperative/layer.h
+++ b/paddle/fluid/imperative/layer.h
@ -234,8 +234,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