Searched refs:_test_memory_stats_generator (Results 1 – 1 of 1) sorted by relevance
130 def _test_memory_stats_generator(self, device=None, N=35): member in TestCudaMultiGPU288 for _ in self._test_memory_stats_generator(self):305 gen0 = self._test_memory_stats_generator(self, device="cuda:0", N=35)306 gen1 = self._test_memory_stats_generator(316 gen0 = self._test_memory_stats_generator(self, device=0, N=35)317 gen1 = self._test_memory_stats_generator(