kanshan/TensorRT-LLMs
1
0
Fork 0
mirror of https://github.com/NVIDIA/TensorRT-LLM.git synced 2026-01-13 22:18:36 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

[None][chore] Modify python ipc_util to align with C++ path (#9894)

Browse Source 
Signed-off-by: yufeiwu <230315618+yufeiwu-nv@users.noreply.github.com>
Co-authored-by: ruodil <200874449+ruodil@users.noreply.github.com>
This commit is contained in:
yufeiwu-nv 2025-12-12 15:55:22 +08:00 committed by GitHub
parent 711016c799
commit fd3d3a553d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorrt_llm/_ipc_utils.py
View File

@ -47,7 +47,9 @@ def can_access_peer(mapping: Mapping) -> bool:
# Early exit if devices are on different nodes
if mapping.get_node_rank(rank) != mapping.node_rank:
logger.info(f"Detect inter-node TP between rank {mapping.rank} and rank {rank}")
logger.info(
f"Detect inter-node TP between rank {mapping.rank} and rank {rank}, fail to access peer GPU memory"
)
return False
# Skip if same device