NUMAHWPolicy: Add more policies for 2-node system
This commit is contained in:
@@ -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});
|
||||
|
||||
@@ -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);
|
||||
|
||||
Reference in New Issue
Block a user