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);