Skip to content

Commit c1c46f0

Browse files
QiJunedominicshanshan
authored andcommitted
[https://nvbugs/5531963][fix] cherry pick NVIDIA#7725 (NVIDIA#7907)
Signed-off-by: junq <[email protected]> Signed-off-by: Wangshanshan <[email protected]>
1 parent 3292f8b commit c1c46f0

File tree

1 file changed

+0
-1
lines changed

1 file changed

+0
-1
lines changed

tensorrt_llm/bench/dataclasses/configuration.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -100,7 +100,6 @@ def get_llm_args(self) -> Dict:
100100
if not batch_sizes_set and not max_batch_size_set:
101101
cuda_graph_config[
102102
"max_batch_size"] = self.settings_config.max_batch_size
103-
104103
updated_llm_args["cuda_graph_config"] = cuda_graph_config
105104

106105
return updated_llm_args

0 commit comments

Comments
 (0)