fix ncnn tools crash with compound layer creation

Tencent · Apr 10, 2024 · b1d0b70 · b1d0b70
1 parent db035d6
commit b1d0b70
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 8 deletions.
diff --git a/tools/modelwriter.h b/tools/modelwriter.h
@@ -796,7 +796,7 @@ int ModelWriter::save(const char* parampath, const char* binpath)
             continue;
         }
 
-        ncnn::Layer* layer_default = ncnn::create_layer(layer->typeindex);
+        ncnn::Layer* layer_default = ncnn::create_layer_cpu(layer->typeindex);
 
         ncnn::ParamDict pd;
         layer_default->load_param(pd);

diff --git a/tools/ncnnoptimize.cpp b/tools/ncnnoptimize.cpp
@@ -1993,7 +1993,7 @@ int NetOptimize::fuse_binaryop_eltwise()
 
         fprintf(stderr, "fuse_binaryop_eltwise %s %s %s\n", binaryop0->name.c_str(), binaryop1->name.c_str(), binaryop->name.c_str());
 
-        ncnn::Eltwise* eltwise = (ncnn::Eltwise*)ncnn::create_layer("Eltwise");
+        ncnn::Eltwise* eltwise = (ncnn::Eltwise*)ncnn::create_layer_cpu("Eltwise");
 
         eltwise->type = "Eltwise";
         eltwise->name = binaryop->name;
@@ -2554,7 +2554,7 @@ int NetOptimize::replace_reduction_with_global_pooling()
 
         fprintf(stderr, "replace_reduction_with_global_pooling %s %s\n", reduction1->name.c_str(), reduction2->name.c_str());
 
-        ncnn::Pooling* pooling = (ncnn::Pooling*)ncnn::create_layer("Pooling");
+        ncnn::Pooling* pooling = (ncnn::Pooling*)ncnn::create_layer_cpu("Pooling");
 
         pooling->type = "Pooling";
         pooling->name = reduction2->name;
@@ -2593,7 +2593,7 @@ int NetOptimize::replace_prelu_with_leaky_relu()
 
         fprintf(stderr, "replace_prelu_with_leaky_relu %s\n", prelu->name.c_str());
 
-        ncnn::ReLU* relu = (ncnn::ReLU*)ncnn::create_layer("ReLU");
+        ncnn::ReLU* relu = (ncnn::ReLU*)ncnn::create_layer_cpu("ReLU");
 
         relu->type = "ReLU";
         relu->name = prelu->name;
@@ -2647,7 +2647,7 @@ int NetOptimize::replace_convolution_with_innerproduct_after_global_pooling()
 
         fprintf(stderr, "replace_convolution_with_innerproduct_after_global_pooling %s %s\n", pooling->name.c_str(), convolution->name.c_str());
 
-        ncnn::InnerProduct* innerproduct = (ncnn::InnerProduct*)ncnn::create_layer("InnerProduct");
+        ncnn::InnerProduct* innerproduct = (ncnn::InnerProduct*)ncnn::create_layer_cpu("InnerProduct");
 
         innerproduct->type = "InnerProduct";
         innerproduct->name = convolution->name;
@@ -2715,7 +2715,7 @@ int NetOptimize::replace_convolution_with_innerproduct_after_innerproduct()
 
             fprintf(stderr, "replace_convolution_with_innerproduct_after_innerproduct %s %s\n", innerproduct->name.c_str(), convolution->name.c_str());
 
-            ncnn::InnerProduct* innerproduct2 = (ncnn::InnerProduct*)ncnn::create_layer("InnerProduct");
+            ncnn::InnerProduct* innerproduct2 = (ncnn::InnerProduct*)ncnn::create_layer_cpu("InnerProduct");
 
             innerproduct2->type = "InnerProduct";
             innerproduct2->name = convolution->name;

diff --git a/tools/quantize/ncnn2table.cpp b/tools/quantize/ncnn2table.cpp
@@ -1112,7 +1112,7 @@ int QuantNet::quantize_EQ()
                 ncnn::Mat out;
                 ex.extract(conv_top_blobs[i], out);
 
-                ncnn::Layer* layer_int8 = ncnn::create_layer(layer->typeindex);
+                ncnn::Layer* layer_int8 = ncnn::create_layer_cpu(layer->typeindex);
 
                 ncnn::ParamDict pd;
                 get_layer_param(layer, pd);
@@ -1222,7 +1222,7 @@ int QuantNet::quantize_EQ()
                 ncnn::Mat out;
                 ex.extract(conv_top_blobs[i], out);
 
-                ncnn::Layer* layer_int8 = ncnn::create_layer(layer->typeindex);
+                ncnn::Layer* layer_int8 = ncnn::create_layer_cpu(layer->typeindex);
 
                 ncnn::ParamDict pd;
                 get_layer_param(layer, pd);