diff options
Diffstat (limited to 'tools')
| -rw-r--r-- | tools/autograd/gen_variable_type.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tools/autograd/gen_variable_type.py b/tools/autograd/gen_variable_type.py index c672c447ce..a47c46d422 100644 --- a/tools/autograd/gen_variable_type.py +++ b/tools/autograd/gen_variable_type.py @@ -93,7 +93,7 @@ if (compute_requires_grad( ${args_with_derivatives} )) { """) ASSIGN_GRAD_FN = CodeTemplate("""\ -grad_fn = std::make_shared<${op}>(${op_ctor}); +grad_fn = std::shared_ptr<${op}>(new ${op}(${op_ctor}), deleteFunction); grad_fn->set_next_edges(collect_next_edges( ${args_with_derivatives} )); """) |