[Graph Optimization] Add dy_runnable and introduce cudagraph_switch_threshold for cudagraph mode switching (#4578)

* add new branch for sot

* reorder

* fix batch bug
This commit is contained in:
Ryan
2025-10-24 18:36:52 +08:00
committed by GitHub
parent e02a812880
commit f42ed6d5f2
2 changed files with 8 additions and 4 deletions
@@ -83,7 +83,7 @@ class TestGraphOptBackend(unittest.TestCase):
self.input_shape = (4, 8)
self.dtype = "int32"
self.model_config = {}
self.max_num_seqs = 1
self.max_num_seqs = 4
# Create baseline configuration (dynamic graph, no cudagraph)
baseline_graph_opt_config = GraphOptimizationConfig(args={})