We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
2 parents 3d1fc95 + 3632616 commit b2200e1Copy full SHA for b2200e1
.github/eks-workflow-files/maxtext-job.yaml
@@ -65,6 +65,7 @@ spec:
65
use_iota_embed=true \
66
logits_dot_in_fp32=false \
67
profiler=nsys \
68
+ upload_all_profiler_results=true \
69
skip_first_n_steps_for_profiler=3 \
70
profiler_steps=8" \
71
|& tee /opt/output/output.$NODE_RANK.log >"${console}"
0 commit comments