fix bug in weight reorder

conv with stride > 15 case weight reorder bugfix Change-Id: Ief568e5253828bfdb9f4db198677878af7897d72
sophgo · Feb 25, 2025 · 6d4ae8b · 6d4ae8b
1 parent 5167d96
commit 6d4ae8b
Show file tree

Hide file tree

Showing 6 changed files with 2 additions and 2 deletions.
diff --git a/lib/Dialect/Tpu/Transforms/WeightReorder/BM1684X/Conv2D.cpp b/lib/Dialect/Tpu/Transforms/WeightReorder/BM1684X/Conv2D.cpp
@@ -987,7 +987,7 @@ LogicalResult WeightReorder<tpu::Conv2DOp, Float32Type>::matchAndRewriteImpl(
                                       : align_up(oc_per_groups, npu_num)) *
             gic * cell_num_h * max_cell_h * cell_num_w * max_cell_w;
         size_t weight_size = groups * weight_size_per_group;
-        auto data_f32 = std::make_shared<std::vector<float>>(weight_size);
+        data_f32->resize(weight_size);
         int ocloops = ceiling_func(oc_per_groups, npu_num);
         for (int group_idx = 0; group_idx < groups; group_idx++) {
           for (int oc = 0; oc < oc_per_groups; oc++) {

diff --git a/third_party/nntoolchain/README.md b/third_party/nntoolchain/README.md
@@ -28,7 +28,7 @@ cp build/firmware_core/libfirmware_core.so /workspace/tpu-mlir/third_party/nntoo
 cp build/firmware_core/libfirmware_core.a /workspace/tpu-mlir/third_party/nntoolchain/lib/libbm1684x_kernel_module.a
 
 #bm1688 sha256: 46ba665895c7fd919d01a81262220c23691a6f02
-
+# clip build bugfix: + 817fe9ddd26cf59c0b3ece40c92f087fab1a4607
 cd TPU1686
 source  scripts/envsetup.sh bm1686
 debug: rebuild_backend_lib_cmodel

diff --git a/third_party/nntoolchain/lib/libbackend_1688.so b/third_party/nntoolchain/lib/libbackend_1688.so
diff --git a/third_party/nntoolchain/lib/libbmtpulv60_kernel_module.a b/third_party/nntoolchain/lib/libbmtpulv60_kernel_module.a
diff --git a/third_party/nntoolchain/lib/libbmtpulv60_kernel_module.so b/third_party/nntoolchain/lib/libbmtpulv60_kernel_module.so
diff --git a/third_party/nntoolchain/lib/libcmodel_1688.so b/third_party/nntoolchain/lib/libcmodel_1688.so