赛题八：paddleviz能力扩展- 暴露给python端反向节点命名优化 (PaddlePaddle#57626)

Frida-a · Oct 14, 2023 · 8694cfb · 8694cfb
1 parent b5cd5ef
commit 8694cfb
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 4 deletions.
diff --git a/paddle/fluid/eager/grad_node_info.cc b/paddle/fluid/eager/grad_node_info.cc
@@ -604,7 +604,7 @@ std::vector<std::shared_ptr<GradNodeBase>> GradNodeBase::NextFunctions() {
   return next_nodes;
 }
 
-uintptr_t GradNodeBase::GetThisPtr() const {
+uintptr_t GradNodeBase::GetPtr() const {
   return reinterpret_cast<uintptr_t>(this);
 }
 

diff --git a/paddle/fluid/eager/grad_node_info.h b/paddle/fluid/eager/grad_node_info.h
@@ -269,7 +269,7 @@ class GradNodeBase {
 
   std::vector<std::shared_ptr<egr::GradNodeBase>> NextFunctions();
 
-  uintptr_t GetThisPtr() const;
+  uintptr_t GetPtr() const;
 
   /**
    * Apply GradientHook

diff --git a/paddle/fluid/pybind/pybind.cc b/paddle/fluid/pybind/pybind.cc
@@ -868,8 +868,7 @@ PYBIND11_MODULE(libpaddle, m) {
           [](const std::shared_ptr<egr::GradNodeBase> &self) {
             return self->NextFunctions();
           })
-
-      .def("node_this_ptr", &egr::GradNodeBase::GetThisPtr)
+      .def("node_ptr", &egr::GradNodeBase::GetPtr)
       .def("input_meta",
            [](const std::shared_ptr<egr::GradNodeBase> &self) {
              return self->InputMeta();