Snap for 6718073 from 302959347fe3fafd4c61d0e068a76d941155ac7b to pie-vts-release
Change-Id: I7712dc799d1f01761600df071e60e8a60aa5f386
diff --git a/src/gpu/vk/GrVkMemory.cpp b/src/gpu/vk/GrVkMemory.cpp
index e391d02..affa5f6 100644
--- a/src/gpu/vk/GrVkMemory.cpp
+++ b/src/gpu/vk/GrVkMemory.cpp
@@ -191,7 +191,8 @@
VK_MEMORY_PROPERTY_DEVICE_LOCAL_BIT,
&typeIndex,
&heapIndex));
- if (memReqs.size <= kMaxSmallImageSize) {
+ // select the heap based on the device alignment
+ if (align_size(memReqs.size, memReqs.alignment) <= kMaxSmallImageSize) {
heap = gpu->getHeap(GrVkGpu::kSmallOptimalImage_Heap);
} else {
heap = gpu->getHeap(GrVkGpu::kOptimalImage_Heap);