[misc] Use --cuda-graph-max-bs-decode in tests, examples, and docs (#29591)

2026-06-28 18:38:28 -07:00
parent 3217410cf6
commit 909123ddb8
140 changed files with 304 additions and 290 deletions
@@ -35,7 +35,7 @@ class TestFlashMLAAttnBackend(unittest.TestCase):
        if torch.cuda.is_available() and torch.version.cuda:
            other_args.extend(
                [
-                    "--cuda-graph-max-bs",
+                    "--cuda-graph-max-bs-decode",
                    "2",
                    "--attention-backend",
                    "flashmla",