[ci][cu13] Bump torch_memory_saver to 0.0.9.post1; restore manual tests (#23182)
Co-authored-by: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
This commit is contained in:
co-authored by
Claude Opus 4.7
parent
674a80d5d8
commit
cd27baaffd
@@ -63,7 +63,7 @@ dependencies = [
|
|||||||
"soundfile==0.13.1",
|
"soundfile==0.13.1",
|
||||||
"tiktoken",
|
"tiktoken",
|
||||||
"timm==1.0.16",
|
"timm==1.0.16",
|
||||||
"torch_memory_saver==0.0.9",
|
"torch_memory_saver>=0.0.9.post1",
|
||||||
"torch==2.9.1",
|
"torch==2.9.1",
|
||||||
"torchao==0.17.0",
|
"torchao==0.17.0",
|
||||||
"torchaudio==2.9.1",
|
"torchaudio==2.9.1",
|
||||||
|
|||||||
-5
@@ -14,11 +14,6 @@ from sglang.test.test_utils import DEFAULT_SMALL_MODEL_NAME_FOR_TEST
|
|||||||
|
|
||||||
register_cuda_ci(est_time=32, suite="stage-b-test-1-gpu-large")
|
register_cuda_ci(est_time=32, suite="stage-b-test-1-gpu-large")
|
||||||
|
|
||||||
"""
|
|
||||||
# TODO: torch_memory_saver wheel is built against libcudart.so.12, fails to LD_PRELOAD in Cu13 venv. Ref: https://github.com/sgl-project/sglang/actions/runs/24604424372/job/71968573867
|
|
||||||
# Should move back to registered test after it's fixed
|
|
||||||
"""
|
|
||||||
|
|
||||||
|
|
||||||
class AsyncEngine(Engine):
|
class AsyncEngine(Engine):
|
||||||
def __init__(self, **kwargs):
|
def __init__(self, **kwargs):
|
||||||
-5
@@ -1,8 +1,3 @@
|
|||||||
"""
|
|
||||||
# TODO: Fails on cu13 venv migration. Ref: https://github.com/sgl-project/sglang/actions/runs/24616960626/job/71980705675?pr=23119
|
|
||||||
# Should move back to registered test after it's fixed
|
|
||||||
"""
|
|
||||||
|
|
||||||
import gc
|
import gc
|
||||||
import multiprocessing
|
import multiprocessing
|
||||||
import os
|
import os
|
||||||
Reference in New Issue
Block a user