Fix extra calls to get_numa_node_if_available to clean up logs (#21781)

This commit is contained in:
Trevor Morris
2026-04-06 16:18:40 -07:00
committed by GitHub
parent 56266de624
commit 5cc246e095
2 changed files with 15 additions and 11 deletions
+2 -1
View File
@@ -3578,8 +3578,9 @@ def run_scheduler_process(
set_gpu_proc_affinity( set_gpu_proc_affinity(
server_args.pp_size, server_args.tp_size, server_args.nnodes, gpu_id server_args.pp_size, server_args.tp_size, server_args.nnodes, gpu_id
) )
if not envs.SGLANG_NUMA_BIND_V2.get():
numa_node = get_numa_node_if_available(server_args, gpu_id) numa_node = get_numa_node_if_available(server_args, gpu_id)
if numa_node is not None and not envs.SGLANG_NUMA_BIND_V2.get(): if numa_node is not None:
numa_bind_to_node(numa_node) numa_bind_to_node(numa_node)
# Set up tracing # Set up tracing
+6 -3
View File
@@ -24,13 +24,16 @@ logger = logging.getLogger(__name__)
@contextmanager @contextmanager
def configure_subprocess(server_args: ServerArgs, gpu_id: int): def configure_subprocess(server_args: ServerArgs, gpu_id: int):
if envs.SGLANG_NUMA_BIND_V2.get():
numa_node = get_numa_node_if_available(server_args, gpu_id) numa_node = get_numa_node_if_available(server_args, gpu_id)
if numa_node is not None and envs.SGLANG_NUMA_BIND_V2.get(): if numa_node is not None:
numactl_args = f"--cpunodebind={numa_node} --membind={numa_node}" numactl_args = f"--cpunodebind={numa_node} --membind={numa_node}"
executable, debug_str = _create_numactl_executable(numactl_args=numactl_args) executable, debug_str = _create_numactl_executable(
numactl_args=numactl_args
)
with _mp_set_executable(executable=executable, debug_str=debug_str): with _mp_set_executable(executable=executable, debug_str=debug_str):
yield yield
else: return
yield yield