[vulkan] Add new VMA vulkan functions. (#4893)

* Add new VMA vulkan functions. * fix
taichi-dev · Apr 30, 2022 · e583b2d · e583b2d
1 parent 6d223da
commit e583b2d
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/taichi/backends/vulkan/vulkan_device.cpp b/taichi/backends/vulkan/vulkan_device.cpp
@@ -2007,6 +2007,10 @@ void VulkanDevice::create_vma_allocator() {
   vk_vma_functions.vkGetPhysicalDeviceMemoryProperties2KHR =
       PFN_vkGetPhysicalDeviceMemoryProperties2KHR(vkGetInstanceProcAddr(
           volkGetLoadedInstance(), "vkGetPhysicalDeviceMemoryProperties2KHR"));
+  vk_vma_functions.vkGetDeviceBufferMemoryRequirements =
+      table.vkGetDeviceBufferMemoryRequirements;
+  vk_vma_functions.vkGetDeviceImageMemoryRequirements =
+      table.vkGetDeviceImageMemoryRequirements;
 
   allocatorInfo.pVulkanFunctions = &vk_vma_functions;