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 df2ed80 commit db0ffb4Copy full SHA for db0ffb4
test/utils.py
@@ -5,7 +5,7 @@
5
dtypes.remove(torch.half)
6
7
devices = [torch.device('cpu')]
8
-if torch.cuda.is_available():
+if torch.cuda.is_available(): # pragma: no cover
9
devices += [torch.device('cuda:{}'.format(torch.cuda.current_device()))]
10
11
0 commit comments