Fix batch invariant ops (#11368)

2025-10-10 20:49:08 -07:00
parent 2674c1d280
commit eae9a9fb9d
3 changed files with 168 additions and 6 deletions
--- a/test/srt/run_suite.py
+++ b/test/srt/run_suite.py
@@ -33,6 +33,7 @@ suites = {
        TestFile("models/test_generation_models.py", 103),
        TestFile("models/test_nvidia_nemotron_nano_v2.py", 180),
        TestFile("models/test_qwen_models.py", 82),
+        TestFile("batch_invariant/test_batch_invariant_ops.py", 10),
        TestFile("models/test_reward_models.py", 132),
        TestFile("models/test_transformers_models.py", 320),
        TestFile("models/test_vlm_models.py", 741),