Skip to content

Commit d7fd063

Browse files
committed
updated param
Signed-off-by: Malay Nagda <[email protected]>
1 parent c07da88 commit d7fd063

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

scripts/llm/performance/llama3_8b.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -158,7 +158,7 @@ def llama3_8b_performance_recipe(
158158
# following line ensures file is at- `<log_dir>/lightning_logs/tb_logs/default/<tfevents_file>`
159159
recipe.log.log_dir = "/nemo_run/lightning_logs"
160160

161-
plugins = [PerfEnvPlugin(enable_vboost=True, nccl_pp_comm_chunksize=2097152)]
161+
plugins = [PerfEnvPlugin(enable_vboost=True)]
162162
if args.enable_profiling:
163163
plugins.append(NsysPlugin(start_step=5, end_step=6))
164164

0 commit comments

Comments
 (0)