mirror of
https://github.com/NVIDIA/TensorRT-LLM.git
synced 2026-01-14 06:27:45 +08:00
[https://nvbugs/5449032][fix] Add more llm-args to llm_mgmn_trtllm_bench.sh (#7144)
Signed-off-by: Balaram Buddharaju <169953907+brb-nv@users.noreply.github.com>
This commit is contained in:
parent
1e5a6be55d
commit
79f1e6c867
@ -76,6 +76,7 @@ srun -l \
|
||||
|
||||
# This is optional
|
||||
cat > /tmp/pytorch_extra_args.txt << EOF
|
||||
cuda_graph_config: null
|
||||
print_iter_log: true
|
||||
enable_attention_dp: false
|
||||
EOF
|
||||
|
||||
Loading…
Reference in New Issue
Block a user