From 3010d9b5c90628e07c7d0f0c33e7355b8bc3e19d Mon Sep 17 00:00:00 2001 From: Patrik Gustavsson Date: Thu, 1 Oct 2020 08:22:10 +0200 Subject: MLBEDSW-3060 Adjust check if weights fit in sram When deciding if weights fit sram: A compression of the weights has been added when a weight compression test limit makes it impossible to fit weights in a double buffer in sram. The worst compression ratio from compression, is used to decide if weights can be fit in sram. Signed-off-by: Patrik Gustavsson Change-Id: I9458769866b3f9fc15659185aae09658ed10fb38 --- ethosu/vela/mark_tensors.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'ethosu/vela/mark_tensors.py') diff --git a/ethosu/vela/mark_tensors.py b/ethosu/vela/mark_tensors.py index a971ef23..c4496cdc 100644 --- a/ethosu/vela/mark_tensors.py +++ b/ethosu/vela/mark_tensors.py @@ -266,7 +266,7 @@ def mark_tensor_purpose(nng, arch, verbose_tensor_purpose=False): ) # special case constants, as they must be in permanent storage tens.mem_type = MemType.Permanent_NPU - def rewrite_mark_tensor_purpose(op, arch): + def rewrite_mark_tensor_purpose(op, arch, nng): # find disconnected outputs and mark as parameters for tens in op.outputs: if not tens.consumers(): @@ -308,7 +308,7 @@ def mark_tensor_purpose(nng, arch, verbose_tensor_purpose=False): return op for sg in nng.subgraphs: - sg = rewrite_graph.rewrite_graph_pre_order(sg, arch, [], [rewrite_mark_tensor_purpose]) + sg = rewrite_graph.rewrite_graph_pre_order(nng, sg, arch, [], [rewrite_mark_tensor_purpose]) for tens in sg.output_tensors: mark_tensor_helper(tens, TensorPurpose.FeatureMap) -- cgit v1.2.1