[Bugfix][Vulkan] Call VulkanDeviceAPI destructor on program exit (#7997)

Most of the TVM Global() functions allocate with "new" and do not deallocate, as the OS can clean up any leftover buffers at the end. In this case, we need the VulkanDeviceAPI destructor to call vkDestroyInstance, to prevent a segfault on exit when using some nvidia drivers. Co-authored-by: Eric Lunderberg <[email protected]>
apache · May 11, 2021 · 58c3413 · 58c3413
1 parent 7b49744
commit 58c3413
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/src/runtime/vulkan/vulkan.cc b/src/runtime/vulkan/vulkan.cc
@@ -412,8 +412,13 @@ class VulkanDeviceAPI final : public DeviceAPI {
   }
 
   static VulkanDeviceAPI* Global() {
-    static VulkanDeviceAPI* inst = new VulkanDeviceAPI();
-    return inst;
+    // Most of the TVM Global() functions allocate with "new" and do
+    // not deallocate, as the OS can clean up any leftover buffers at
+    // the end.  In this case, we need the VulkanDeviceAPI destructor
+    // to call vkDestroyInstance, to prevent a segfault on exit when
+    // using some nvidia drivers.
+    static VulkanDeviceAPI inst;
+    return &inst;
   }
 
   const VulkanContext& context(size_t device_id) const {