Update

[ghstack-poisoned]
pytorch · Oct 11, 2024 · fddc70f · fddc70f
1 parent 610e908
commit fddc70f
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tensordict/nn/cudagraphs.py b/tensordict/nn/cudagraphs.py
@@ -298,7 +298,8 @@ def _call(*args: torch.Tensor, **kwargs: torch.Tensor):
                         tree_leaves((args, kwargs)), self._flat_tree
                     ):
                         self._maybe_copy_onto_(arg_src, arg_dest, srcs, dests)
-                    torch._foreach_copy_(dests, srcs)
+                    if dests:
+                        torch._foreach_copy_(dests, srcs)
                     torch.cuda.synchronize()
                     self.graph.replay()
                     if self._return_unchanged == "clone":