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

…d_param/load_model first (Tencent#5410)
senli123 · May 7, 2024 · c5534ff · c5534ff
1 parent 97a5cc7
commit c5534ff
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