Revert "chore: update torch v2.5.1" (#2063)

2024-11-17 15:29:38 -08:00
parent 3b878863f7
commit c1f401fc58
10 changed files with 37 additions and 174 deletions
--- a/test/srt/test_torch_compile_moe.py
+++ b/test/srt/test_torch_compile_moe.py
@@ -66,7 +66,7 @@ class TestTorchCompile(unittest.TestCase):
        print(f"{res=}")
        throughput = max_tokens / (tok - tic)
        print(f"Throughput: {throughput} tokens/s")
-        self.assertGreaterEqual(throughput, 289)
+        self.assertGreaterEqual(throughput, 290)


 if __name__ == "__main__":