github
/
InternLM
mirror of https://github.com/InternLM/InternLM
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

feat:add bind numa

pull/320/head
li126com 2023-09-19 18:34:30 +08:00
parent d9fd76f8a7
commit ad73355215
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
internlm/initialize/__init__.py
View File

@ -4,6 +4,7 @@ from .launch import (
initialize_distributed_env, initialize_distributed_env,
launch_from_slurm, launch_from_slurm,
launch_from_torch, launch_from_torch,
try_bind_numa,
) )
__all__ = [ __all__ = [
@ -12,4 +13,5 @@ __all__ = [
"launch_from_slurm", "launch_from_slurm",
"launch_from_torch", "launch_from_torch",
"initialize_distributed_env", "initialize_distributed_env",
"try_bind_numa",
] ]