Commit 898b3486 authored by Justin Lebar's avatar Justin Lebar Committed by TensorFlower Gardener
Browse files

Limit internal fragmentation in BFCAllocator to 128mb per allocation.

Previously, if you had a very large allocation, it would round up to the
next power of 2, and then, if this didn't fit in your GPU's available
memory, eat all remaining memory in the device.

Now we waste at most 128mb of memory in a large alloc.

PiperOrigin-RevId: 175209995
parent 9d5a6650
Loading
Loading
Loading
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment