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

louie-tsai · web-flow · commit 5c8d34a42cff · 2025-06-11T04:11:47.000-07:00
Signed-off-by: Tsai, Louie &lt;louie.tsai@intel.com&gt;
diff --git a/csrc/cpu/utils.cpp b/csrc/cpu/utils.cpp
@@ -54,8 +54,7 @@ std::string init_cpu_threads_env(const std::string& cpu_ids) {
     *(src_mask->maskp) = *(src_mask->maskp) ^ *(mask->maskp);
     int page_num = numa_migrate_pages(pid, src_mask, mask);
     if (page_num == -1) {
-      TORCH_CHECK(false,
-                  "numa_migrate_pages failed. errno: " + std::to_string(errno));
+      TORCH_WARN("numa_migrate_pages failed. errno: " + std::to_string(errno));
     }
 
     // restrict memory allocation node.
@@ -105,4 +104,4 @@ std::string init_cpu_threads_env(const std::string& cpu_ids) {
 
   return ss.str();
 }
-#endif
+#endif

Original file line number	Diff line number	Diff line change
`@@ -54,8 +54,7 @@ std::string init_cpu_threads_env(const std::string& cpu_ids) {`
`54`	`54`	`(src_mask->maskp) = (src_mask->maskp) ^ *(mask->maskp);`
`55`	`55`	`int page_num = numa_migrate_pages(pid, src_mask, mask);`
`56`	`56`	`if (page_num == -1) {`
`57`		`- TORCH_CHECK(false,`
`58`		`- "numa_migrate_pages failed. errno: " + std::to_string(errno));`
	`57`	`+ TORCH_WARN("numa_migrate_pages failed. errno: " + std::to_string(errno));`
`59`	`58`	`}`
`60`	`59`
`61`	`60`	`// restrict memory allocation node.`
`@@ -105,4 +104,4 @@ std::string init_cpu_threads_env(const std::string& cpu_ids) {`
`105`	`104`
`106`	`105`	`return ss.str();`
`107`	`106`	`}`
`108`		`-#endif`
	`107`	`+#endif`