Skip to content

Commit

Permalink
minor change
Browse files Browse the repository at this point in the history
  • Loading branch information
keshavb96 committed Oct 18, 2024
1 parent 318d6e8 commit 6df8a71
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion docs/gpu_memory_allocation.rst
Original file line number Diff line number Diff line change
Expand Up @@ -63,6 +63,6 @@ Common causes of OOM failures

**Disabling rematerialization HLO pass**
Sometimes disabling the rematerialization HLO pass is favorable to avoid
poor remat choices by the compiler. The pass can be disabled by
poor remat choices by the compiler. The pass can be disabled by adding
:code:`jax.config.update('enable_remat_opt_pass', False)`. But this can
sometimes lead to OOM failures.

0 comments on commit 6df8a71

Please sign in to comment.