pnnx: Fix save_ncnn with fp16=1

Signed-off-by: Molly Sophia <mollysophia379@gmail.com>
Tencent · Sep 1, 2023 · eb3061c · eb3061c
1 parent 14b000d
commit eb3061c
Showing 1 changed file with 9 additions and 10 deletions.
diff --git a/tools/pnnx/src/save_ncnn.cpp b/tools/pnnx/src/save_ncnn.cpp
@@ -262,6 +262,15 @@ int save_ncnn(const Graph& g, const std::string& parampath, const std::string& b
 
             const Attribute& attr = it.second;
 
+            if (fp16 && (((attr.type == 0 && attr.data == std::vector<char> {0, 0, 0, 0}) || attr.type == 1)))
+            {
+                // write fp16 flag
+                unsigned int fp16_flag = 0x01306B47;
+                fwrite((const char*)&fp16_flag, sizeof(fp16_flag), 1, binfp);
+
+                is_type_flag_fp32 = true;
+            }
+
             if (fp16 && is_type_flag_fp32)
             {
                 // fp32 -> fp16
@@ -287,16 +296,6 @@ int save_ncnn(const Graph& g, const std::string& parampath, const std::string& b
                 continue;
             }
 
-            if (fp16 && attr.type == 0 && attr.data == std::vector<char> {0, 0, 0, 0})
-            {
-                // write fp16 flag
-                unsigned int fp16_flag = 0x01306B47;
-                fwrite((const char*)&fp16_flag, sizeof(fp16_flag), 1, binfp);
-
-                is_type_flag_fp32 = true;
-                continue;
-            }
-
             fwrite(attr.data.data(), attr.data.size(), 1, binfp);
         }