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 961f24e commit 8d40b21Copy full SHA for 8d40b21
thunder/tests/distributed/test_dtensor.py
@@ -4,7 +4,7 @@
4
import pytest
5
import torch
6
7
-if not torch.distributed.is_available():
+if not torch.distributed.is_available() and not (torch.cuda.is_available() and torch.distributed.is_nccl_available()):
8
pytest.skip(allow_module_level=True)
9
10
import thunder
0 commit comments