Fix retract for page size > 1 (#4914)

2025-03-30 02:57:15 -07:00
parent b26bc86b36
commit 4ede6770cd
10 changed files with 68 additions and 120 deletions
--- a/test/srt/test_dp_attention.py
+++ b/test/srt/test_dp_attention.py
@@ -60,3 +60,7 @@ class TestDPAttentionDP2TP2(CustomTestCase):
        metrics = run_eval(args)
        print(f"{metrics=}")
        self.assertGreater(metrics["score"], 0.8)
+
+
+if __name__ == "__main__":
+    unittest.main()