Vulkan: Support dumping VMA stats string

src/libANGLE/renderer/vulkan/RendererVk.cpp

@@ -64,6 +64,8 @@ constexpr uint32_t kPipelineCacheVkUpdatePeriod = 60;
 // version of Vulkan.
 constexpr uint32_t kPreferredVulkanAPIVersion = VK_API_VERSION_1_1;
 
+constexpr bool kOutputVmaStatsString = false;
+
 angle::vk::ICD ChooseICDFromAttribs(const egl::AttributeMap &attribs)
 {
 #if !defined(ANGLE_PLATFORM_ANDROID)
@@ -2182,6 +2184,17 @@ angle::Result RendererVk::queueSubmit(vk::Context *context,
                                       const vk::Fence *fence,
                                       Serial *serialOut)
 {
+    if (kOutputVmaStatsString)
+    {
+        // Output the VMA stats string
+        // This JSON string can be passed to VmaDumpVis.py to generate a visualization of the
+        // allocations the VMA has performed.
+        char *statsString;
+        mAllocator.buildStatsString(&statsString, true);
+        INFO() << std::endl << statsString << std::endl;
+        mAllocator.freeStatsString(statsString);
+    }
+
     if (getFeatures().enableCommandProcessingThread.enabled)
     {
         // For initial threading phase 1 code make sure any outstanding command processing

src/libANGLE/renderer/vulkan/vk_mem_alloc_wrapper.cpp

@@ -148,4 +148,14 @@ void InvalidateAllocation(VmaAllocator allocator,
 {
     vmaInvalidateAllocation(allocator, allocation, offset, size);
 }
+
+void BuildStatsString(VmaAllocator allocator, char **statsString, VkBool32 detailedMap)
+{
+    vmaBuildStatsString(allocator, statsString, detailedMap);
+}
+
+void FreeStatsString(VmaAllocator allocator, char *statsString)
+{
+    vmaFreeStatsString(allocator, statsString);
+}
 }  // namespace vma

src/libANGLE/renderer/vulkan/vk_mem_alloc_wrapper.h

@@ -62,6 +62,9 @@ void InvalidateAllocation(VmaAllocator allocator,
                           VkDeviceSize offset,
                           VkDeviceSize size);
 
+void BuildStatsString(VmaAllocator allocator, char **statsString, VkBool32 detailedMap);
+void FreeStatsString(VmaAllocator allocator, char *statsString);
+
 }  // namespace vma
 
 #endif  // LIBANGLE_RENDERER_VULKAN_VK_MEM_ALLOC_WRAPPER_H_

src/libANGLE/renderer/vulkan/vk_wrapper.h

@@ -490,6 +490,9 @@ class Allocator : public WrappedObject<Allocator, VmaAllocator>
                                               VkMemoryPropertyFlags preferredFlags,
                                               bool persistentlyMappedBuffers,
                                               uint32_t *memoryTypeIndexOut) const;
+
+    void buildStatsString(char **statsString, VkBool32 detailedMap);
+    void freeStatsString(char *statsString);
 };
 
 class Allocation final : public WrappedObject<Allocation, VmaAllocation>
@@ -1457,6 +1460,18 @@ Allocator::findMemoryTypeIndexForBufferInfo(const VkBufferCreateInfo &bufferCrea
                                                  memoryTypeIndexOut);
 }
 
+ANGLE_INLINE void Allocator::buildStatsString(char **statsString, VkBool32 detailedMap)
+{
+    ASSERT(valid());
+    vma::BuildStatsString(mHandle, statsString, detailedMap);
+}
+
+ANGLE_INLINE void Allocator::freeStatsString(char *statsString)
+{
+    ASSERT(valid());
+    vma::FreeStatsString(mHandle, statsString);
+}
+
 // Allocation implementation.
 ANGLE_INLINE void Allocation::destroy(const Allocator &allocator)
 {