Test cuda distributed version.
(self)
| 383 | self._test(n_gpus=1) |
| 384 | |
| 385 | def test3(self): |
| 386 | """Test cuda distributed version.""" |
| 387 | if torch.cuda.is_available() and torch.cuda.device_count() >= 2: |
| 388 | self.start_process(n_gpus=min(2, torch.cuda.device_count())) |
| 389 | |
| 390 | |
| 391 | if __name__ == '__main__': |
nothing calls this directly
no test coverage detected