Skip to content

[DoubleGrad PR #3] Supported higher-order GradNode generation #41051

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.

Sign up for GitHub

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

Merged
merged 14 commits into from
Mar 30, 2022
Merged

[DoubleGrad PR #3] Supported higher-order GradNode generation #41051

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
14 commits
Select commit Hold shift + click to select a range
5e95fd1
[Refactor] refactored eager_gen.py PR #2
jim19930609 Mar 24, 2022
214ad36
[DoubleGrad PR #1] Decoupled code generation logics for Dygraph Forwa…
jim19930609 Mar 25, 2022
14ebc19
Fixed minor issue
jim19930609 Mar 25, 2022
62179ba
Merged develop
jim19930609 Mar 25, 2022
2a83dd9
Merged develop from [#40935] Bug fix for intermediate support in Yaml
jim19930609 Mar 28, 2022
9e7ecf6
Adjusted logics of GenerateNodeCreationCodes and GenerateForwardDefin…
jim19930609 Mar 28, 2022
b945424
Fixed issues
jim19930609 Mar 28, 2022
033913c
Merged develop
jim19930609 Mar 28, 2022
84ea03e
Supported higher-order grad node generation
jim19930609 Mar 28, 2022
f29de18
Merge branch 'refactored_node_creation_codes' of https://github.com/j…
jim19930609 Mar 28, 2022
6117fd1
[DoubleGrad PR #4] Supported higher-order GradNode generation
jim19930609 Mar 29, 2022
8785075
Merged develop
jim19930609 Mar 29, 2022
f647fb4
Merged develop
jim19930609 Mar 30, 2022
e26b572
Fixed yaml typo
jim19930609 Mar 30, 2022
File filter

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
10 changes: 10 additions & 0 deletions paddle/fluid/eager/auto_code_generator/final_state_generator/codegen_utils.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -89,6 +89,10 @@ def FindForwardName(string):
return string[:-5]


def IsGradName(string):
return string.endswith("_grad")


def IsPlainTensorType(string):
plain_tensor_types = ['Tensor&', 'Tensor', 'const Tensor&', 'const Tensor']
if string in plain_tensor_types:
Expand Down Expand Up @@ -166,6 +170,12 @@ def GetForwardFunctionName(string):
return f"{string}_final_state_dygraph_function"


def TransformGradVarNameForDoubleGradGeneration(string):
if IsGradName(string):
string = "grad_" + string[:-5]
return string


######################
### Yaml Parsers ###
######################
Expand Down
Loading