Merge pull request #114 from PAICookers/dev

Release: v1.1.1
PAICookers · Jul 9, 2024 · ff09f3f · ff09f3f
2 parents a4844cc + 064beb0
commit ff09f3f
Show file tree

Hide file tree

Showing 4 changed files with 10 additions and 62 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -67,3 +67,7 @@
   1. 子网络现在直接在主网络内部 `self.subnet=...` 例化即可
   2. 编译选项现在直接通过 `paibox.Mapper.compile(...)` 传入，默认配置不变
   3. 在 `paibox.Mapper.export()` 中使用 `split_by_chip` 指定配置帧文件是否以芯片分割，默认不分割。原 `split_by_coord` 弃用。
+
+## v1.1.1
+
+- 修复对权重RAM错误的配置
diff --git a/TODO.md b/TODO.md
diff --git a/paibox/backend/placement.py b/paibox/backend/placement.py
@@ -438,12 +438,16 @@ def _fold_raw_weights(self, raw_weights: list[WeightType]) -> WeightType:
         # LCN_EX > LCN_1X
         for raw_weight in raw_weights:
             w_folded_of_axon_segs.clear()
+            _n_axon_last = 0
 
             for s in self.source:
                 axon_seg = self.parent.axon_segments[s]
 
                 # Retrive the weight of the axon segment
-                w_of_axon_seg = raw_weight[: axon_seg.n_axon, :]
+                w_of_axon_seg = raw_weight[
+                    _n_axon_last : _n_axon_last + axon_seg.n_axon, :
+                ]
+                _n_axon_last += axon_seg.n_axon
 
                 # Fold the weight of axon segment
                 w_folded_of_axon_seg = self._nfold_weight(

diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "paibox"
-version = "1.1.0"
+version = "1.1.1"
 description = "Toolchain of PAICORE 2.0"
 authors = ["Ziru Pan <[email protected]>"]
 maintainers = [