/
constants.py
67 lines (59 loc) · 1.85 KB
/
constants.py
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
"""
Constants for ReFrame tests
"""
AMD = 'AMD'
CI = 'CI'
CPU = 'CPU'
CPU_SOCKET = 'CPU_SOCKET'
GPU = 'GPU'
GPU_VENDOR = 'GPU_VENDOR'
INTEL = 'INTEL'
NODE = 'NODE'
NVIDIA = 'NVIDIA'
ALWAYS_REQUEST_GPUS = 'ALWAYS_REQUEST_GPUS'
DEVICE_TYPES = {
CPU: 'cpu',
GPU: 'gpu',
}
COMPUTE_UNIT = {
CPU: 'cpu',
CPU_SOCKET: 'cpu_socket',
GPU: 'gpu',
NODE: 'node',
}
TAGS = {
CI: 'CI',
}
FEATURES = {
CPU: 'cpu',
GPU: 'gpu',
ALWAYS_REQUEST_GPUS: 'always_request_gpus',
}
GPU_VENDORS = {
AMD: 'amd',
INTEL: 'intel',
NVIDIA: 'nvidia',
}
SCALES = {
# required keys:
# - num_nodes
# - either node_part or (num_cpus_per_node and num_gpus_per_node)
# num_cpus_per_node and num_gpus_per_node are upper limits:
# the actual count depends on the specific configuration of cores, gpus, and sockets within the node,
# as well as the specific test being carried out.
'1_core': {'num_nodes': 1, 'num_cpus_per_node': 1, 'num_gpus_per_node': 1},
'2_cores': {'num_nodes': 1, 'num_cpus_per_node': 2, 'num_gpus_per_node': 1},
'4_cores': {'num_nodes': 1, 'num_cpus_per_node': 4, 'num_gpus_per_node': 1},
'1_cpn_2_nodes': {'num_nodes': 2, 'num_cpus_per_node': 1, 'num_gpus_per_node': 1},
'1_cpn_4_nodes': {'num_nodes': 4, 'num_cpus_per_node': 1, 'num_gpus_per_node': 1},
'1_8_node': {'num_nodes': 1, 'node_part': 8}, # 1/8 node
'1_4_node': {'num_nodes': 1, 'node_part': 4}, # 1/4 node
'1_2_node': {'num_nodes': 1, 'node_part': 2}, # 1/2 node
'1_node': {'num_nodes': 1, 'node_part': 1},
'2_nodes': {'num_nodes': 2, 'node_part': 1},
'4_nodes': {'num_nodes': 4, 'node_part': 1},
'8_nodes': {'num_nodes': 8, 'node_part': 1},
'16_nodes': {'num_nodes': 16, 'node_part': 1},
}
# When tests are filtered by the hooks, the valid_systems is set to this system name:
INVALID_SYSTEM = "INVALID_SYSTEM"