Skip to content

Commit

Permalink
Revert "dump xla output"
Browse files Browse the repository at this point in the history
This reverts commit 96e152a.
  • Loading branch information
samos123 committed Jan 22, 2025
1 parent 96e152a commit 4a6ada5
Showing 1 changed file with 0 additions and 1 deletion.
1 change: 0 additions & 1 deletion axlearn/cloud/gcp/job.py
Original file line number Diff line number Diff line change
Expand Up @@ -1080,7 +1080,6 @@ def _build_main_container(self) -> Nested[Any]:
"--xla_gpu_enable_reduce_scatter_combine_by_dim=false",
"--xla_disable_hlo_passes=rematerialization",
"--xla_gpu_enable_while_loop_double_buffering=true",
"--xla_dump_to=/output/xla",
]
env_vars["XLA_FLAGS"] = " ".join(default_xla_flags)

Expand Down

0 comments on commit 4a6ada5

Please sign in to comment.