[chore] update torch version to 2.9 (#12969)

Co-authored-by: Baizhou Zhang <sobereddiezhang@gmail.com>
This commit is contained in:
Fan Yin
2025-11-25 14:47:34 -08:00
committed by GitHub
co-authored by Baizhou Zhang
parent 64a11303ce
commit 36b1bcd242
13 changed files with 44 additions and 87 deletions
+2
View File
@@ -485,6 +485,7 @@ def run_lora_multiple_batch_on_model_cases(
attention_backend: str = "torch_native",
disable_cuda_graph: bool = True,
enable_deterministic_inference: bool = False,
disable_radix_cache: bool = True,
):
for model_case in model_cases:
for torch_dtype in TORCH_DTYPES:
@@ -523,6 +524,7 @@ def run_lora_multiple_batch_on_model_cases(
attention_backend=attention_backend,
enable_deterministic_inference=enable_deterministic_inference,
disable_cuda_graph=disable_cuda_graph,
disable_radix_cache=disable_radix_cache,
**spec_args,
)
+2 -2
View File
@@ -17,7 +17,6 @@ suites = {
TestFile("lora/test_lora_eviction.py", 240),
TestFile("lora/test_lora_update.py", 600),
TestFile("lora/test_lora_backend.py", 99),
TestFile("lora/test_lora_spec_decoding.py", 150),
TestFile("lora/test_multi_lora_backend.py", 60),
TestFile("models/test_compressed_tensors_models.py", 42),
TestFile("models/test_cross_encoder_models.py", 100),
@@ -186,7 +185,8 @@ suites = {
],
"per-commit-4-gpu-deepep": [
TestFile("ep/test_deepep_small.py", 531),
TestFile("ep/test_mooncake_ep_small.py", 450),
# TODO: Add it back after mooncake supports torch 2.9
# TestFile("ep/test_mooncake_ep_small.py", 450),
],
"per-commit-8-gpu-h200-deepep": [
TestFile("ep/test_deepep_large.py", 338),