diff --git a/confluent_client/bin/nodeinventory b/confluent_client/bin/nodeinventory index 1fb3d151..6551c1b5 100755 --- a/confluent_client/bin/nodeinventory +++ b/confluent_client/bin/nodeinventory @@ -60,16 +60,21 @@ def print_mem_info(node, prefix, meminfo): elif 'DCPMM' in meminfo['memory_type']: memdescfmt = '{0}GB {1} ' meminfo['module_type'] = 'DCPMM' + elif meminfo['memory_type'] == 'HBM': + memdescfmt = '{0}GB HBM ' else: print('{0}: {1}: Unrecognized Memory'.format(node, prefix)) return if meminfo.get('ecc', False): memdescfmt += 'ECC ' - capacity = meminfo['capacity_mb'] / 1024 modtype = meminfo.get('module_type', None) if modtype: memdescfmt += modtype - memdesc = memdescfmt.format(capacity, meminfo['speed']) + if meminfo.get('capacity_mb', None): + capacity = meminfo['capacity_mb'] // 1024 + memdesc = memdescfmt.format(capacity, meminfo['speed']) + else: + memdesc = 'Unspecified Module' print('{0}: {1} description: {2}'.format(node, prefix, memdesc)) print('{0}: {1} manufacturer: {2}'.format( node, prefix, meminfo['manufacturer']))