Fix torch.compile for MoE (#2033)

2024-11-14 01:30:24 -08:00
parent b275ce0043
commit c3eac1b010
10 changed files with 89 additions and 12 deletions
--- a/test/srt/test_double_sparsity.py
+++ b/test/srt/test_double_sparsity.py
@@ -55,7 +55,7 @@ class TestDoubleSparsity(unittest.TestCase):
        )

        metrics = run_eval(args)
-        assert metrics["score"] >= 0.65
+        self.assertGreaterEqual(metrics["score"], 0.65)


 if __name__ == "__main__":