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 2792a04 commit ac9f2d4Copy full SHA for ac9f2d4
tests/py/test_api_multi_gpu.py
@@ -55,7 +55,7 @@ def test_suite():
55
56
if not torch.cuda.device_count() > 1:
57
raise ValueError("This test case is applicable for multi-gpu configurations only")
58
-
+
59
# Setting it up here so that all CUDA allocations are done on correct device
60
trtorch.set_device(gpu_id)
61
suite = test_suite()
0 commit comments