[Eager]fix_stop_gradient (#45154)

* fix_stop_gradient

paddle/fluid/eager/autograd_meta.h

@@ -123,12 +123,6 @@ class AutogradMeta : public AbstractAutogradMeta {
     stop_gradient_ = static_cast<int>(stop_gradient);
   }
 
-  void WeakSetStopGradient(bool stop_gradient) {
-    if (stop_gradient_ == -1) {
-      stop_gradient_ = static_cast<int>(stop_gradient);
-    }
-  }
-
   bool Persistable() const { return persistable_; }
 
   void SetPersistable(bool persistable) { persistable_ = persistable; }

paddle/fluid/eager/tests/task_tests/eager_utils_test.cc

@@ -163,7 +163,7 @@ TEST(EagerUtils, PassStopGradient) {
                                     auto_grad1.get(),
                                     auto_grad2.get(),
                                     auto_grad3.get());
-  CHECK(auto_grad0->StopGradient() == false);
+  CHECK(auto_grad0->StopGradient() == true);
   CHECK(auto_grad1->StopGradient() == true);
   CHECK(auto_grad2->StopGradient() == true);
   CHECK(auto_grad3->StopGradient() == true);

paddle/fluid/eager/utils.h

@@ -77,7 +77,7 @@ class PassStopGradientIter : public IterHelper<AutogradMeta*> {
       VLOG(2) << "Tensor is NULL";
       return;
     }
-    element->WeakSetStopGradient(stop_gradient_);
+    element->SetStopGradient(stop_gradient_);
   }
 
   bool stop_gradient_ = true;

paddle/fluid/pybind/eager_py_layer.cc

@@ -315,7 +315,7 @@ PyObject* pylayer_method_apply(PyObject* cls,
             non_differentiable.end()) {
           outputs_autograd_meta[i][j]->SetStopGradient(true);
         } else {
-          outputs_autograd_meta[i][j]->WeakSetStopGradient(false);
+          outputs_autograd_meta[i][j]->SetStopGradient(false);
         }
       }
     }