[chore] update torch version to 2.9 (#12969)
Co-authored-by: Baizhou Zhang <sobereddiezhang@gmail.com>
This commit is contained in:
@@ -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),
|
||||
|
||||
Reference in New Issue
Block a user