We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent e7cd77a commit 036b114Copy full SHA for 036b114
.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