Skip to content

Commit 8b1bfed

Browse files
cheshirecopybara-github
authored andcommitted
[PJRT] [XLA:GPU] Bump down % of memory available to XLA to 0.8
We observe crashes from not enough memory available to NCCL otherwise. 0.85 does not seem to do the trick either. PiperOrigin-RevId: 508387705
1 parent 0f5c989 commit 8b1bfed

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

xla/pjrt/gpu/gpu_helpers.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -50,7 +50,7 @@ struct GpuAllocatorConfig {
5050

5151
// Only used if kind == kBFC. The maximum fraction of available memory to
5252
// allocate.
53-
double memory_fraction = 0.9;
53+
double memory_fraction = 0.8;
5454

5555
// Only used if kind == kBFC. If true, the allocator will immediately allocate
5656
// the maximum amount allowed by `memory_fraction`. This reduces

0 commit comments

Comments
 (0)