Stronger guard against linear tiling

Differential Revision: D66776766 Pull Request resolved: pytorch#7181
kirklandsign · Dec 5, 2024 · 6b41c3a · 6b41c3a
1 parent daf9aee
commit 6b41c3a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/backends/vulkan/runtime/vk_api/Adapter.cpp b/backends/vulkan/runtime/vk_api/Adapter.cpp
@@ -182,9 +182,9 @@ Adapter::Adapter(
   VkImage image = VK_NULL_HANDLE;
   VkResult res =
       vkCreateImage(device_.handle, &image_create_info, nullptr, &image);
-  if (res == VK_ERROR_FEATURE_NOT_PRESENT) {
+  if (res != VK_SUCCESS) {
     linear_tiling_3d_enabled_ = false;
-  } else if (res == VK_SUCCESS) {
+  } else {
     vkDestroyImage(device_.handle, image, nullptr);
   }
   return;