-
Notifications
You must be signed in to change notification settings - Fork 5.6k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Added tensor utils to Eager Dygraph #37478
Added tensor utils to Eager Dygraph #37478
Conversation
Thanks for your contribution! |
… dygraph_second_pr_tensor_utils
… dygraph_second_pr_tensor_utils
bf583ee
to
0310154
Compare
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM
|
||
namespace egr { | ||
|
||
bool IsLeafTensor(const egr::EagerTensor& target) { |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Change this to namespace egr_utils_api{}
in later PR
@@ -42,6 +42,7 @@ TEST(AccumulationNode, EagerTensor) { | |||
std::make_shared<paddle::experimental::DefaultAllocator>( | |||
paddle::platform::CPUPlace()), | |||
meta); | |||
|
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
remove additional line here
@@ -172,4 +172,5 @@ inline void InitEnv(paddle::platform::Place place) { | |||
// Init Tracer Place | |||
egr::Controller::Instance().SetExpectedPlace(place); | |||
} | |||
|
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
remove additional line
* Added GradTensorHolder to Eager Dygraph * Added accumulation codes to Eager Dygraph * Added tensor utils to Eager Dygraph * Resolve compilation issues * Fixed issues
PR types
New features
PR changes
Others
Describe
Supported tensor utils