diff --git a/common/NUMAHWPolicy.cpp b/common/NUMAHWPolicy.cpp index 42043e95..e3127452 100644 --- a/common/NUMAHWPolicy.cpp +++ b/common/NUMAHWPolicy.cpp @@ -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}); diff --git a/receiver/jfjoch_action_test.cpp b/receiver/jfjoch_action_test.cpp index 58a31434..3c792463 100644 --- a/receiver/jfjoch_action_test.cpp +++ b/receiver/jfjoch_action_test.cpp @@ -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);