usertools/cpu_layout.py

   1 #!/usr/bin/env python
   2 # SPDX-License-Identifier: BSD-3-Clause
   3 # Copyright(c) 2010-2014 Intel Corporation
   4 # Copyright(c) 2017 Cavium, Inc. All rights reserved.
   5
   6 from __future__ import print_function
   7 import sys
   8 try:
   9     xrange # Python 2
  10 except NameError:
  11     xrange = range # Python 3
  12
  13 sockets = []
  14 cores = []
  15 core_map = {}
  16 base_path = "/sys/devices/system/cpu"
  17 fd = open("{}/kernel_max".format(base_path))
  18 max_cpus = int(fd.read())
  19 fd.close()
  20 for cpu in xrange(max_cpus + 1):
  21     try:
  22         fd = open("{}/cpu{}/topology/core_id".format(base_path, cpu))
  23     except IOError:
  24         continue
  25     except:
  26         break
  27     core = int(fd.read())
  28     fd.close()
  29     fd = open("{}/cpu{}/topology/physical_package_id".format(base_path, cpu))
  30     socket = int(fd.read())
  31     fd.close()
  32     if core not in cores:
  33         cores.append(core)
  34     if socket not in sockets:
  35         sockets.append(socket)
  36     key = (socket, core)
  37     if key not in core_map:
  38         core_map[key] = []
  39     core_map[key].append(cpu)
  40
  41 print(format("=" * (47 + len(base_path))))
  42 print("Core and Socket Information (as reported by '{}')".format(base_path))
  43 print("{}\n".format("=" * (47 + len(base_path))))
  44 print("cores = ", cores)
  45 print("sockets = ", sockets)
  46 print("")
  47
  48 max_processor_len = len(str(len(cores) * len(sockets) * 2 - 1))
  49 max_thread_count = len(list(core_map.values())[0])
  50 max_core_map_len = (max_processor_len * max_thread_count)  \
  51                       + len(", ") * (max_thread_count - 1) \
  52                       + len('[]') + len('Socket ')
  53 max_core_id_len = len(str(max(cores)))
  54
  55 output = " ".ljust(max_core_id_len + len('Core '))
  56 for s in sockets:
  57     output += " Socket %s" % str(s).ljust(max_core_map_len - len('Socket '))
  58 print(output)
  59
  60 output = " ".ljust(max_core_id_len + len('Core '))
  61 for s in sockets:
  62     output += " --------".ljust(max_core_map_len)
  63     output += " "
  64 print(output)
  65
  66 for c in cores:
  67     output = "Core %s" % str(c).ljust(max_core_id_len)
  68     for s in sockets:
  69         if (s,c) in core_map:
  70             output += " " + str(core_map[(s, c)]).ljust(max_core_map_len)
  71         else:
  72             output += " " * (max_core_map_len + 1)
  73     print(output)