Merge branch 'main' into financebench

2025-08-20 20:40:27 -07:00
parent 5c163737c4 0174ba5571
commit 772894012e
8 changed files with 3857 additions and 3816 deletions
--- a/benchmarks/simple_mac_tpt_test.py
+++ b/benchmarks/simple_mac_tpt_test.py
@@ -183,6 +183,9 @@ class Benchmark:
        start_time = time.time()
        with torch.no_grad():
            self.model(input_ids=input_ids, attention_mask=attention_mask)
+        # mps sync
+        if torch.backends.mps.is_available():
+            torch.mps.synchronize()
        end_time = time.time()

        return end_time - start_time