onnx2pnnx do not fold single constant for gemm weight (#5634)

Tencent · Aug 17, 2024 · 4de5369 · 4de5369
1 parent 789d868
commit 4de5369
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tools/pnnx/src/pass_onnx.cpp b/tools/pnnx/src/pass_onnx.cpp
@@ -820,6 +820,8 @@ void pass_onnx(const onnx::ModelProto& model, Graph& pnnx_graph)
                         is_attr_weight = true;
                     if (sim_op_type == "Gather" && j == 0)
                         is_attr_weight = true;
+                    if (sim_op_type == "Gemm" && (j == 1 || j == 2))
+                        is_attr_weight = true;
                     if (sim_op_type == "GroupNormalization" && (j == 1 || j == 2))
                         is_attr_weight = true;
                     if (sim_op_type == "GRU" && (j == 1 || j == 2 || j == 3 || j == 5))