From 612d5ffdab4bc48e7eae327bc5903920be528db2 Mon Sep 17 00:00:00 2001
From: Angela Yi <yiangela7@gmail.com>
Date: Mon, 22 Dec 2025 17:56:47 -0800
Subject: [PATCH] [ci] Fix Pytorch compilation test oom in 2.10 (#31194)

Signed-off-by: angelayi <yiangela7@gmail.com>
---
 tests/compile/test_dynamic_shapes_compilation.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/tests/compile/test_dynamic_shapes_compilation.py b/tests/compile/test_dynamic_shapes_compilation.py
index 9ccb363b088f5..1fda21dea6361 100644
--- a/tests/compile/test_dynamic_shapes_compilation.py
+++ b/tests/compile/test_dynamic_shapes_compilation.py
@@ -77,6 +77,7 @@ def test_dynamic_shapes_compilation(
                 "evaluate_guards": evaluate_guards,
             },
         },
+        max_model_len=1024,
     )
 
     output = model.generate(prompt)