Skip to content

GitLab

Unverified Commit 5c8d34a4 authored by Louie Tsai's avatar Louie Tsai Committed by GitHub
Browse files

Support no privileged mode on CPU for docker and kubernetes deployments (#19241) 


Signed-off-by: default avatarTsai, Louie <louie.tsai@intel.com>
parent 3c8694ea
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 3 deletions
csrc/cpu/utils.cpp
View file @ 5c8d34a4
...@@ -54,8 +54,7 @@ std::string init_cpu_threads_env(const std::string& cpu_ids) { ...@@ -54,8 +54,7 @@ std::string init_cpu_threads_env(const std::string& cpu_ids) {
*(src_mask->maskp) = *(src_mask->maskp) ^ *(mask->maskp); *(src_mask->maskp) = *(src_mask->maskp) ^ *(mask->maskp);
int page_num = numa_migrate_pages(pid, src_mask, mask); int page_num = numa_migrate_pages(pid, src_mask, mask);
if (page_num == -1) { if (page_num == -1) {
TORCH_CHECK(false, TORCH_WARN("numa_migrate_pages failed. errno: " + std::to_string(errno));
"numa_migrate_pages failed. errno: " + std::to_string(errno));
} }
// restrict memory allocation node. // restrict memory allocation node.
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment