1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
|
# Copyright (c) 2016 Cisco and/or its affiliates.
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at:
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
"""CPU utilities library"""
from resources.libraries.python.ssh import SSH
__all__ = ["CpuUtils"]
class CpuUtils(object):
"""CPU utilities"""
@staticmethod
def __str2int(string):
"""Conversion from string to integer, 0 in case of empty string.
:param string: Input string.
:type string: str
:return: Integer converted from string, 0 in case of ValueError.
:rtype: int
"""
try:
return int(string)
except ValueError:
return 0
@staticmethod
def get_cpu_layout_from_all_nodes(nodes):
"""Retrieve cpu layout from all nodes, assuming all nodes
are Linux nodes.
:param nodes: DICT__nodes from Topology.DICT__nodes.
:type nodes: dict
"""
ssh = SSH()
for node in nodes.values():
ssh.connect(node)
cmd = "lscpu -p"
ret, stdout, stderr = ssh.exec_command(cmd)
# parsing of "lscpu -p" output:
# # CPU,Core,Socket,Node,,L1d,L1i,L2,L3
# 0,0,0,0,,0,0,0,0
# 1,1,0,0,,1,1,1,0
if ret != 0:
raise RuntimeError(
"Failed to execute ssh command, ret: {} err: {}".format(
ret, stderr))
node['cpuinfo'] = list()
for line in stdout.split("\n"):
if len(line) > 0 and line[0] != "#":
node['cpuinfo'].append([CpuUtils.__str2int(x) for x in
line.split(",")])
@staticmethod
def cpu_node_count(node):
"""Return count of numa nodes.
:param node: Targeted node.
:type node: dict
:return: Count of numa nodes.
:rtype: int
"""
cpuinfo = node.get("cpuinfo")
if cpuinfo is not None:
return node["cpuinfo"][-1][3] + 1
else:
raise RuntimeError("Node cpuinfo not available.")
@staticmethod
def cpu_list_per_node(node, cpu_node):
"""Return node related list of CPU numbers.
:param node: Node dictionary with cpuinfo.
:param cpu_node: Numa node number.
:type node: int
:type cpu_node: int
:return: List of cpu numbers related to numa from argument.
:rtype: list of int
"""
cpu_node = int(cpu_node)
cpuinfo = node.get("cpuinfo")
cpulist = []
if cpuinfo is not None:
for cpu in cpuinfo:
if cpu[3] == cpu_node:
cpulist.append(cpu[0])
else:
raise RuntimeError("Node cpuinfo not available.")
return cpulist
|
