File tree Expand file tree Collapse file tree 3 files changed +1
-18
lines changed Expand file tree Collapse file tree 3 files changed +1
-18
lines changed Original file line number Diff line number Diff line change 1- cc_test(test_egr_ds_eager_utils SRCS eager_utils_test.cc DEPS ${eager_deps} )
1+ cc_test(test_egr_task_eager_utils SRCS eager_utils_test.cc DEPS ${eager_deps} )
Original file line number Diff line number Diff line change @@ -188,18 +188,4 @@ egr::EagerTensor EagerUtils::GetOutput(
188188 return EagerTensor ((*(out.get ())));
189189}
190190
191- AutogradMeta* EagerUtils::unsafe_autograd_meta (const egr::EagerTensor& target) {
192- auto * p_autograd_meta = target.get_autograd_meta ();
193- PADDLE_ENFORCE (p_autograd_meta,
194- paddle::platform::errors::Fatal (
195- " Null autograd_meta gotten from unsafe_autograd_meta(), "
196- " if you are using unsafe_autograd_meta, please make sure "
197- " your tensor's autograd_meta is set" ));
198- return static_cast <AutogradMeta*>(p_autograd_meta);
199- }
200-
201- std::pair<size_t , size_t > EagerUtils::OutRankInfo (
202- const egr::EagerTensor& target) {
203- return unsafe_autograd_meta (target)->OutRankInfo ();
204- }
205191} // namespace egr
Original file line number Diff line number Diff line change @@ -130,9 +130,6 @@ class EagerUtils {
130130 iter.SetStopGradient (stop_gradient);
131131 iter.apply (std::forward<Args>(args)...);
132132 }
133- static std::pair<size_t , size_t > OutRankInfo (const egr::EagerTensor& target);
134- // This method will return an AutogradMeta pointer unsafely.
135- static AutogradMeta* unsafe_autograd_meta (const egr::EagerTensor& target);
136133
137134 // Intermidate needed remove this once we don't need legacy
138135 static std::vector<std::shared_ptr<egr::EagerTensor>> SyncToVars (
You can’t perform that action at this time.
0 commit comments