From 0f65277f914fc9cd32613db4b4679655a24f58c8 Mon Sep 17 00:00:00 2001
From: Stephan Seitz <stephan.seitz@fau.de>
Date: Fri, 28 Feb 2020 17:56:53 +0100
Subject: [PATCH] Add TODO
---
src/pystencils_autodiff/backends/_torch_native.py | 2 ++
1 file changed, 2 insertions(+)
diff --git a/src/pystencils_autodiff/backends/_torch_native.py b/src/pystencils_autodiff/backends/_torch_native.py
index fa5c299..bba20bc 100644
--- a/src/pystencils_autodiff/backends/_torch_native.py
+++ b/src/pystencils_autodiff/backends/_torch_native.py
@@ -73,6 +73,8 @@ def create_autograd_function(autodiff_obj, use_cuda):
output_tensors = OrderedDict({f.name:
field_to_tensor_dict.get(f, kwargs[f.name])
for f in autodiff_obj.forward_output_fields})
+
+ # TODO: only if any of input tensors requires grad
for o in output_tensors.values():
if isinstance(o, torch.Tensor):
o.requires_grad = True
--
GitLab