[XLA/GPU] Address review comments.

This commit is contained in:
Trent Lo 2020-10-12 23:10:26 -07:00
parent feadb9ae1f
commit 55a1d7cbbd
2 changed files with 3 additions and 2 deletions

View File

@ -580,7 +580,8 @@ static void AllocateFlags() {
"Generates multiple heaps (i.e., temp buffers) with a size "
"constraint on each heap to avoid Out-of-Memory due to memory "
"fragmentation. The constraint is soft, so it works with tensors "
"larger than the given constraint size."));
"larger than the given constraint size. -1 corresponds to no "
"constraints."));
ParseFlagsFromEnvAndDieIfUnknown("XLA_FLAGS", *flag_objects);
}

View File

@ -608,7 +608,7 @@ void BufferAssignment::AddAssignment(BufferAllocation* allocation,
void BufferAssignment::CombineTempAllocations() {
VLOG(1) << "CombineTempAllocations()";
// Stores the combined allocations.
std::vector<BufferAllocation> combined_allocations;
std::deque<BufferAllocation> combined_allocations;
// Holds the pointer to a combined allocation of each color, if any.
flat_hash_map<BufferValue::Color, BufferAllocation*> combined_allocation_map;