apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2026-04-23 00:17:25 +08:00
Code Issues Actions 19 Packages Projects Releases Wiki Activity

fix ci test case (#4635)

Browse Source
This commit is contained in:
RAM
2025-10-29 13:26:38 +08:00
committed by GitHub
parent 14e7d88ea4
commit a0d5426ab6
3 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/graph_optimization/test_cuda_graph_dynamic_subgraph.py
+1 -1
View File
@@ -161,7 +161,7 @@ class TestCUDAGrpahSubgraph(unittest.TestCase):
model_config = Mock()
model_config.max_model_len = 512
# Initialize cuda graph capture list
graph_opt_config._set_cudagraph_sizes(max_num_seqs=scheduler_config.max_num_seqs)
graph_opt_config._set_cudagraph_sizes(max_capture_size=scheduler_config.max_num_seqs)
graph_opt_config.init_with_cudagrpah_size(max_capture_size=scheduler_config.max_num_seqs)
fd_config = FDConfig(
graph_opt_config=graph_opt_config,
tests/graph_optimization/test_cuda_graph_spec_decode.py
+1 -1
View File
@@ -108,7 +108,7 @@ class TestCUDAGrpahSpecDecode(unittest.TestCase):
model_config = Mock()
model_config.max_model_len = 512
# Initialize cuda graph capture list
graph_opt_config._set_cudagraph_sizes(max_num_seqs=scheduler_config.max_num_seqs)
graph_opt_config._set_cudagraph_sizes(max_capture_size=scheduler_config.max_num_seqs)
graph_opt_config.init_with_cudagrpah_size(max_capture_size=scheduler_config.max_num_seqs)
fd_config = FDConfig(
graph_opt_config=graph_opt_config,
tests/graph_optimization/test_static_graph_cuda_graph_split.py
+1 -1
View File
@@ -91,7 +91,7 @@ class TestStaticGraphCUDAGraphSplit(unittest.TestCase):
# Set FastDeploy config
graph_opt_config = GraphOptimizationConfig({"use_cudagraph": True, "graph_opt_level": 1})
scheduler_config = SchedulerConfig({"max_num_seqs": 1})
graph_opt_config._set_cudagraph_sizes(max_num_seqs=scheduler_config.max_num_seqs)
graph_opt_config._set_cudagraph_sizes(max_capture_size=scheduler_config.max_num_seqs)
graph_opt_config.init_with_cudagrpah_size(max_capture_size=scheduler_config.max_num_seqs)
cache_config = CacheConfig({})
parallel_config = ParallelConfig(args={})