Skip to content

Commit 63886d1

Browse files
committed
renamed env var to VLLM_SKIP_NVLINK_CHECK
Signed-off-by: avtc <[email protected]>
1 parent bc16fcd commit 63886d1

File tree

1 file changed

+2
-5
lines changed

1 file changed

+2
-5
lines changed

vllm/platforms/cuda.py

Lines changed: 2 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -566,11 +566,8 @@ def is_fully_connected(cls, physical_device_ids: list[int]) -> bool:
566566
query if the set of gpus are fully connected by nvlink (1 hop)
567567
"""
568568

569-
# Check if is_fully_connected should be forced via environment variable
570-
if os.getenv("VLLM_FORCE_IS_FULLY_CONNECTED", "0") == "1":
571-
logger.warning(
572-
"Forcing is_fully_connected to True via VLLM_FORCE_IS_FULLY_CONNECTED environment variable."
573-
)
569+
if os.getenv("VLLM_SKIP_NVLINK_CHECK", "0") == "1":
570+
logger.info("Skipping NVLink check.")
574571
return True
575572

576573
handles = [

0 commit comments

Comments
 (0)