Skip to content

Commit 3632616

Browse files
authored
Set upload_all_profiler_results in EKS MaxText workload (#1948)
Fixes [failing EKS MaxText workload ](https://github.com/NVIDIA/JAX-Toolbox/actions/runs/21735284056/job/62742958259#step:9:2992)due to missing profile c.f. [passing](https://github.com/NVIDIA/JAX-Toolbox/actions/runs/21751322430/job/62749748788#step:9:1797)
1 parent e7cd77a commit 3632616

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

.github/eks-workflow-files/maxtext-job.yaml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -65,6 +65,7 @@ spec:
6565
use_iota_embed=true \
6666
logits_dot_in_fp32=false \
6767
profiler=nsys \
68+
upload_all_profiler_results=true \
6869
skip_first_n_steps_for_profiler=3 \
6970
profiler_steps=8" \
7071
|& tee /opt/output/output.$NODE_RANK.log >"${console}"

0 commit comments

Comments
 (0)