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 38c8b3a commit a1f7031Copy full SHA for a1f7031
test/utils.py
@@ -8,7 +8,7 @@
8
grad_dtypes = [torch.float, torch.double]
9
10
devices = [torch.device('cpu')]
11
-if torch.cuda.is_available(): # pragma: no cover
+if torch.cuda.is_available():
12
devices += [torch.device('cuda:{}'.format(torch.cuda.current_device()))]
13
14
0 commit comments