From 13aeb91b6e030d238fa8b36194db5db2be2174ed Mon Sep 17 00:00:00 2001
From: Xinyuan Tong <115166877+JustinTong0323@users.noreply.github.com>
Date: Tue, 11 Aug 2026 13:41:21 +0800
Subject: [PATCH] [Fix] Update multimodal CUDA VMM helper import (#34358)

---
 python/sglang/srt/multimodal/transport/memory_pool.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/python/sglang/srt/multimodal/transport/memory_pool.py b/python/sglang/srt/multimodal/transport/memory_pool.py
index c9df875b9..aa23a40e5 100644
--- a/python/sglang/srt/multimodal/transport/memory_pool.py
+++ b/python/sglang/srt/multimodal/transport/memory_pool.py
@@ -21,7 +21,7 @@ def align_up(value: int, alignment: int) -> int:
 def _driver_modules():
     from cuda.bindings import driver as cuda
 
-    from sglang.srt.distributed.device_communicators.vmm_utils import check_drv
+    from sglang.srt.cuda_vmm_utils import check_drv
 
     return cuda, check_drv