fix create_pipeline crash in vulkan-enabled layer without calling loa…

…d_param/load_model first (#5410)
Tencent · Apr 8, 2024 · 056509a · 056509a
1 parent 3b048d1
commit 056509a
Showing 1 changed file with 10 additions and 3 deletions.
diff --git a/src/layer.cpp b/src/layer.cpp
@@ -362,9 +362,16 @@ class Layer_final : public Layer
 #if NCNN_VULKAN
         if (layer_vulkan)
         {
-            int ret = layer_vulkan->create_pipeline(opt);
-            get_layer_properties();
-            return ret;
+            if (vkdev)
+            {
+                int ret = layer_vulkan->create_pipeline(opt);
+                get_layer_properties();
+                return ret;
+            }
+
+            // fallback to cpu layer
+            delete layer_vulkan;
+            layer_vulkan = 0;
         }
 #endif // NCNN_VULKAN