fix bug (NVIDIA#3570)

byshiue · web-flow · commit 3aa37e6b72bd · 2025-04-15T16:50:22.000+08:00
Signed-off-by: bhsueh &lt;11360707+byshiue@users.noreply.github.com&gt;
diff --git a/tensorrt_llm/llmapi/trtllm-llmapi-launch b/tensorrt_llm/llmapi/trtllm-llmapi-launch
@@ -30,7 +30,7 @@ function export_free_tcp_addr {
     log_stderr "TLLM_SPAWN_PROXY_PROCESS_IPC_ADDR: $TLLM_SPAWN_PROXY_PROCESS_IPC_ADDR"
 }
 
-export tllm_mpi_size=$(mpi_size)
+export tllm_mpi_size=$(mpi_world_size)
 log_stderr "tllm_mpi_size: $tllm_mpi_size"
 
 export_free_tcp_addr

Original file line number	Diff line number	Diff line change
`@@ -30,7 +30,7 @@ function export_free_tcp_addr {`
`30`	`30`	`log_stderr "TLLM_SPAWN_PROXY_PROCESS_IPC_ADDR: $TLLM_SPAWN_PROXY_PROCESS_IPC_ADDR"`
`31`	`31`	`}`
`32`	`32`
`33`		`-export tllm_mpi_size=$(mpi_size)`
	`33`	`+export tllm_mpi_size=$(mpi_world_size)`
`34`	`34`	`log_stderr "tllm_mpi_size: $tllm_mpi_size"`
`35`	`35`
`36`	`36`	`export_free_tcp_addr`