def gpu_resources(cls, jobs_resources):
jobs_resources = to_list(jobs_resources)
click.clear()
data = [[
'job_name', 'name', 'GPU Usage', 'GPU Mem Usage / Total',
'GPU Temperature', 'Power Draw / Limit'
]]
non_gpu_jobs = 0
for job_resources in jobs_resources:
job_resources = ContainerResourcesConfig.from_dict(job_resources)
line = []
if not job_resources.gpu_resources:
non_gpu_jobs += 1
continue
for gpu_resources in job_resources.gpu_resources:
line += [
job_resources.job_name,
gpu_resources.name,
to_percentage(gpu_resources.utilization_gpu / 100),
'{} / {}'.format(
to_unit_memory(gpu_resources.memory_used),
to_unit_memory(gpu_resources.memory_total)),
gpu_resources.temperature_gpu,
'{} / {}'.format(gpu_resources.power_draw,
gpu_resources.power_limit),
]
data.append(line)
if non_gpu_jobs == len(jobs_resources):
Printer.print_error(
'No GPU job was found, please run `resources` command without `-g | --gpu` option.'
)
exit(1)
click.echo(tabulate(data, headers="firstrow"))
sys.stdout.flush()
def humanize_attrs(cls, obj):
humanized_attrs = {}
for attr in cls.DATETIME_ATTRIBUTES:
humanized_attrs[attr] = humanize_timesince(getattr(obj, attr))
for attr in cls.PERCENT_ATTRIBUTES:
humanized_attrs[attr] = to_percentage(getattr(obj, attr),
cls.ROUNDING)
for attr in cls.MEM_SIZE_ATTRIBUTES:
humanized_attrs[attr] = to_unit_memory(getattr(obj, attr))
return humanized_attrs
def test_works_as_expected_for_valid_values(self):
test_data = [
(0, '0%'),
(0.25, '25%'),
(-0.25, '-25%'),
(12, '1200%'),
(0.123, '12.3%'),
(0.12345, '12.35%'),
(0.12001, '12%'),
(0.12101, '12.1%'),
('0', '0%'),
('0.25', '25%'),
]
for value, expected in test_data:
result = to_percentage(value)
self.assertEqual(result, expected)
def resources(cls, jobs_resources):
jobs_resources = to_list(jobs_resources)
click.clear()
data = [['Job', 'Mem Usage / Total', 'CPU% - CPUs']]
for job_resources in jobs_resources:
job_resources = ContainerResourcesConfig.from_dict(job_resources)
line = [
job_resources.job_name,
'{} / {}'.format(to_unit_memory(job_resources.memory_used),
to_unit_memory(job_resources.memory_limit)),
'{} - {}'.format(
to_percentage(job_resources.cpu_percentage / 100),
job_resources.n_cpus)
]
data.append(line)
click.echo(tabulate(data, headers="firstrow"))
sys.stdout.flush()
def test_raises_value_error_for_invalid_types(self):
with self.assertRaises(ValueError):
to_percentage('foo')