[Vulkan] Workaround for zero size allocation (apache#7691)

neo-ai · May 11, 2021 · dd5dbd9 · dd5dbd9
1 parent 9dd88ef
commit dd5dbd9
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/runtime/vulkan/vulkan.cc b/src/runtime/vulkan/vulkan.cc
@@ -120,6 +120,10 @@ class VulkanDeviceAPI final : public DeviceAPI {
   std::vector<uint32_t> GetComputeQueueFamilies(VkPhysicalDevice phy_dev);
   void* AllocDataSpace(TVMContext ctx, size_t nbytes, size_t alignment,
                        DLDataType type_hint) final {
+    if (nbytes == 0) {
+      // Vulkan seems to have issues if we return nullptr on zero size alloc
+      nbytes = 1;
+    }
     const auto& vctx = context(ctx.device_id);
     VkBufferCreateInfo info;
     info.sType = VK_STRUCTURE_TYPE_BUFFER_CREATE_INFO;