def get_gpu_resources(): if not polyaxon_gpustat.has_gpu_nvidia: return try: return polyaxon_gpustat.query() except: polyaxon_gpustat.has_gpu_nvidia = False
def test_query_mocked_nonexistent_pid(self, N, Process): """ Test a case where nvidia query returns non-existent pids (see #16, #18) """ _configure_mock(N, Process, scenario_nonexistent_pid=True) gpustats = polyaxon_gpustat.query() self.assertEqual(gpustats, MOCK_EXPECTED_OUTPUT_FULL)
def test_query_mocked(self, N, Process): """ A basic functionality test, in a case where everything is just normal. """ _configure_mock(N, Process) gpustats = polyaxon_gpustat.query() self.assertEqual(gpustats[0], MOCK_EXPECTED_OUTPUT_FULL[0]) self.assertEqual(gpustats[1], MOCK_EXPECTED_OUTPUT_FULL[1]) self.assertEqual(gpustats[2].pop('processes'), None) mock_stats = copy.copy(MOCK_EXPECTED_OUTPUT_FULL[2]) mock_stats.pop('processes') self.assertEqual(gpustats[2], mock_stats)
def get_gpu_resources(): try: return polyaxon_gpustat.query() except: # noqa return []
def get_gpu_resources(): try: return polyaxon_gpustat.query() except: # noqa return []