[AMD] Disable CUDA IPC multimodal transport on ROCm in MMMU VLM tests (#31342)

This commit is contained in:
YC Yen-Ching Tseng
2026-07-15 18:42:30 -07:00
committed by GitHub
parent d9003dd452
commit 14095ef78a
+6 -3
View File
@@ -7,7 +7,7 @@ import tempfile
from pathlib import Path
from types import SimpleNamespace
from sglang.srt.utils import kill_process_tree
from sglang.srt.utils import is_cuda, kill_process_tree
from sglang.srt.utils.common import temp_set_env
from sglang.test.test_utils import (
DEFAULT_TIMEOUT_FOR_SERVER_LAUNCH,
@@ -349,8 +349,11 @@ class MMMUMultiModelTestBase(CustomTestCase):
process_env = os.environ.copy()
if custom_env:
process_env.update(custom_env)
# if test vlm with cuda_ipc feature, open this env_var
process_env["SGLANG_USE_CUDA_IPC_TRANSPORT"] = "1"
# CUDA IPC multimodal-feature transport is NVIDIA-only; enabling it
# on ROCm/HIP makes ServerArgs raise "requires NVIDIA CUDA" and the
# server exits. Only opt in on CUDA so AMD exercises CPU transport.
if is_cuda():
process_env["SGLANG_USE_CUDA_IPC_TRANSPORT"] = "1"
# Prepare stdout/stderr redirection if needed
stdout_file = None