"server/vscode:/vscode.git/clone" did not exist on "3f3b7ffd67364c5fb3a362bb1fb475e1b0f0ce5f"
Unverified Commit 3c5538f7 authored by Lianmin Zheng's avatar Lianmin Zheng Committed by GitHub
Browse files

Update CI threshold (#2186)

parent 10189d08
...@@ -65,7 +65,9 @@ def patch_model( ...@@ -65,7 +65,9 @@ def patch_model(
_to_torch(model) _to_torch(model)
monkey_patch_vllm_all_gather() monkey_patch_vllm_all_gather()
backup_ca_comm = tp_group.ca_comm backup_ca_comm = tp_group.ca_comm
# Use custom-allreduce here # Use custom-allreduce here.
# We found the custom allreduce is much faster than the built-in allreduce in torch,
# even with ENABLE_INTRA_NODE_COMM=1.
# tp_group.ca_comm = None # tp_group.ca_comm = None
yield torch.compile( yield torch.compile(
torch.no_grad()(model.forward), mode="max-autotune-no-cudagraphs" torch.no_grad()(model.forward), mode="max-autotune-no-cudagraphs"
......
...@@ -36,7 +36,9 @@ class TestBenchServing(unittest.TestCase): ...@@ -36,7 +36,9 @@ class TestBenchServing(unittest.TestCase):
) )
if is_in_ci(): if is_in_ci():
self.assertGreater(res["output_throughput"], 950) # There is a regression with torch 2.5
# This number was 950 for torch 2.4
self.assertGreater(res["output_throughput"], 800)
def test_offline_throughput_without_radix_cache(self): def test_offline_throughput_without_radix_cache(self):
res = run_bench_serving( res = run_bench_serving(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment