[Eager] just need to run needed nodes (#44910)

* [Eager] just need to run needed nodes

* [Eager] just need to run needed nodes

paddle/fluid/eager/backward.cc

@@ -360,6 +360,16 @@ std::vector<paddle::experimental::Tensor> RunBackward(
                 "Node's in-degree cannot be negative.",
                 next_node->name()));
 
+        if (is_general_grad) {
+          if (node_in_degree_map[next_node] == 0 &&
+              GeneralGrad::Instance().IsNeededNodes(next_node)) {
+            if (dynamic_cast<egr::GradNodeAccumulation*>(next_node)) {
+              queue.push_front(std::move(next_node));
+            } else {
+              queue.push_back(std::move(next_node));
+            }
+          }
+        } else {
           if (node_in_degree_map[next_node] == 0) {
             if (dynamic_cast<egr::GradNodeAccumulation*>(next_node)) {
               queue.push_front(std::move(next_node));
@@ -370,6 +380,7 @@ std::vector<paddle::experimental::Tensor> RunBackward(
         }
       }
     }
+  }
 
   VLOG(6) << "Run Backward Final hook size: "
           << egr::Controller::Instance().FinalBackwardHooks().size();