From ad73355215f81ada6a8eacd51b01682bbb2b11e7 Mon Sep 17 00:00:00 2001 From: li126com Date: Tue, 19 Sep 2023 18:34:30 +0800 Subject: [PATCH] feat:add bind numa --- internlm/initialize/__init__.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/internlm/initialize/__init__.py b/internlm/initialize/__init__.py index ae94e0a..14fe06b 100644 --- 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", ]