[Fix] Update multimodal CUDA VMM helper import (#34358)

2026-08-10 22:41:21 -07:00
parent b3c02cbce7
commit 13aeb91b6e
1 changed files with 1 additions and 1 deletions
@@ -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