Unverified Commit 020428f0 authored by Tim Moon's avatar Tim Moon Committed by GitHub
Browse files

[PyTorch] Fix bug with clearing op outputs during backward (#2008)



Fix merge conflict bug with clearing op outputs
Signed-off-by: default avatarTim Moon <tmoon@nvidia.com>
parent feda5b55
......@@ -199,7 +199,7 @@ class _OperationFuserAutogradFunction(torch.autograd.Function):
# Mark output tensors as not deletable in backward
for tensor in [x] + extra_outputs_flat:
tensor.do_not_clear = True
tensor._do_not_clear = True
x.requires_grad_(fuser.first_op_requiring_backward < fuser._num_basic_ops)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment