feat:add bind numa

2023-09-19 18:34:30 +08:00 · 2023-09-19 18:34:30 +08:00 · ad73355215
parent d9fd76f8a7
commit ad73355215
1 changed files with 2 additions and 0 deletions
--- a/internlm/initialize/init.py
+++ b/internlm/initialize/init.py
@ -4,6 +4,7 @@ from .launch import (
    initialize_distributed_env,
    launch_from_slurm,
    launch_from_torch,
+    try_bind_numa,
 )

 __all__ = [
@ -12,4 +13,5 @@ __all__ = [
    "launch_from_slurm",
    "launch_from_torch",
    "initialize_distributed_env",
+    "try_bind_numa",
 ]