From 8600397a8bb25b59d8460f494274d101dc015e29 Mon Sep 17 00:00:00 2001 From: Pablo Gonzalez Date: Thu, 14 Dec 2023 16:17:43 -0500 Subject: [PATCH] Add node count to summary --- mlperf_logging/result_summarizer/result_summarizer.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/mlperf_logging/result_summarizer/result_summarizer.py b/mlperf_logging/result_summarizer/result_summarizer.py index 318f906..299801c 100644 --- a/mlperf_logging/result_summarizer/result_summarizer.py +++ b/mlperf_logging/result_summarizer/result_summarizer.py @@ -273,6 +273,7 @@ def _get_column_schema(usage, ruleset, weak_scaling=False): 'availability': str, 'submitter': str, 'system': str, + 'number_of_nodes': str, 'host_processor_model_name': str, 'host_processors_count': int, 'accelerator_model_name': str, @@ -624,6 +625,7 @@ def _check_and_update_system_specs(desc_keys, column_name, query=None): query=_pretty_system_name) _check_and_update_system_specs('host_processor_model_name', 'host_processor_model_name') + _check_and_update_system_specs('number_of_nodes', 'number_of_nodes') _check_and_update_system_specs( [ 'host_processor_core_count', 'host_processors_per_node',