fix: avoid tilelang cuda runtime pollution (#30870)

This commit is contained in:
Mick
2026-07-14 22:30:27 +08:00
committed by GitHub
parent f853293440
commit 04af94d150
3 changed files with 98 additions and 27 deletions
@@ -0,0 +1,38 @@
import io
import pytest
from sglang.test.ci.ci_register import register_cuda_ci
register_cuda_ci(est_time=2, stage="base-b", runner_config="1-gpu-small")
from sglang.srt.distributed.device_communicators import cuda_wrapper
def test_find_loaded_library_prefers_real_cudart_over_tilelang_stub(monkeypatch):
maps = """\
7f000000-7f010000 r-xp 00000000 00:00 0 /site-packages/tilelang/lib/libcudart_stub.so
7f020000-7f030000 r-xp 00000000 00:00 0 /cuda/lib64/libcudart.so.13
"""
monkeypatch.setattr("builtins.open", lambda *args, **kwargs: io.StringIO(maps))
assert (
cuda_wrapper.find_loaded_library("libcudart") == "/cuda/lib64/libcudart.so.13"
)
def test_find_loaded_library_strips_deleted_suffix(monkeypatch):
maps = """\
7f020000-7f030000 r-xp 00000000 00:00 0 /cuda/lib64/libcudart.so.13 (deleted)
"""
monkeypatch.setattr("builtins.open", lambda *args, **kwargs: io.StringIO(maps))
assert (
cuda_wrapper.find_loaded_library("libcudart") == "/cuda/lib64/libcudart.so.13"
)
if __name__ == "__main__":
raise SystemExit(pytest.main([__file__, "-v"]))