NUMAHWPolicy: Add more policies for 2-node system

This commit is contained in:
2023-08-04 09:31:18 +02:00
parent 45336015c5
commit d996ca42c9
2 changed files with 11 additions and 1 deletions
+10
View File
@@ -16,6 +16,16 @@ NUMAHWPolicy::NUMAHWPolicy(const std::string &policy) : name(policy) {
} else if (policy == "n2g2") {
bindings.emplace_back(NUMABinding{.cpu_node = 0, .mem_node = 0, .gpu = 0});
bindings.emplace_back(NUMABinding{.cpu_node = 1, .mem_node = 1, .gpu = 1});
} else if (policy == "n2g4") {
bindings.emplace_back(NUMABinding{.cpu_node = 0, .mem_node = 0, .gpu = 0});
bindings.emplace_back(NUMABinding{.cpu_node = 1, .mem_node = 1, .gpu = 2});
bindings.emplace_back(NUMABinding{.cpu_node = 0, .mem_node = 0, .gpu = 1});
bindings.emplace_back(NUMABinding{.cpu_node = 1, .mem_node = 1, .gpu = 3});
} else if (policy == "n2g4_hbm") {
bindings.emplace_back(NUMABinding{.cpu_node = 0, .mem_node = 2, .gpu = 0});
bindings.emplace_back(NUMABinding{.cpu_node = 1, .mem_node = 3, .gpu = 2});
bindings.emplace_back(NUMABinding{.cpu_node = 0, .mem_node = 2, .gpu = 1});
bindings.emplace_back(NUMABinding{.cpu_node = 1, .mem_node = 3, .gpu = 3});
} else if (policy == "n8g4") {
for (int32_t i = 0; i < 8; i++)
bindings.emplace_back(NUMABinding{.cpu_node = i, .mem_node = i, .gpu = i/2});
+1 -1
View File
@@ -37,7 +37,7 @@ int main(int argc, char **argv) {
bool use_mock_device = false;
bool nonblocking_mode = true;
bool verbose = false;
std::string numa_policy_name = "";
std::string numa_policy_name;
if (argc == 1)
print_usage(logger);