Add node count to summary

mlcommons · Jan 23, 2024 · 72f0403 · 72f0403
1 parent 9d75268
commit 72f0403
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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',