Skip to content

Commit 36e72c6

Browse files
chore(CI): prune compute_device
Integrated it into akida 2.18. Signed-off-by: jmejia <jmejia@brainchipinc.com>
1 parent 09540f6 commit 36e72c6

2 files changed

Lines changed: 1 addition & 261 deletions

File tree

CI/check_model.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -4,8 +4,6 @@
44
from cnn2snn import convert
55
from quantizeml import load_model
66

7-
from compute_device import compute_min_device
8-
97

108
def process_model(file_path):
119
try:
@@ -20,7 +18,7 @@ def process_model(file_path):
2018
raise RuntimeError(f"❌ Error converting {file_path}: {e}") from e
2119

2220
try:
23-
device = compute_min_device(model_ak, enable_hwpr=True)
21+
device = akida.compute_min_device(model_ak, enable_hwpr=True)
2422
result = len(device.mesh.nps) // 4
2523
print(f"✅ {file_path}: needs {result} Akida nodes")
2624
except Exception as e:
@@ -33,4 +31,3 @@ def process_model(file_path):
3331
args = parser.parse_args()
3432

3533
process_model(args.model)
36-

CI/compute_device.py

Lines changed: 0 additions & 257 deletions
This file was deleted.

0 commit comments

Comments
 (0)