We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 4dd9e25 commit 933036dCopy full SHA for 933036d
src/lightning/fabric/strategies/ddp.py
@@ -245,6 +245,7 @@ def _create_stream_context(self, device_ids=None):
245
ctx = self.torch_lib.stream(stream)
246
else:
247
from contextlib import nullcontext
248
+
249
ctx = nullcontext()
250
return ctx
251
src/lightning/pytorch/strategies/ddp.py
@@ -434,6 +434,7 @@ def _create_stream_context(self, device_ids=None):
434
435
436
437
438
439
440
0 commit comments