From 04bd3e91ccc43ff09cb98d792a544997e880dc18 Mon Sep 17 00:00:00 2001 From: Louis Verhaard Date: Thu, 19 Aug 2021 16:36:32 +0200 Subject: MLBEDSW-4978: Cascading of operators requiring full IFM/OFM Bug fix in cascade builder: tensors produced with operators requiring full OFM or consumed by operators requiring full IFM could be added as intermediate buffers to a cascade. Signed-off-by: Louis Verhaard Change-Id: Id84e9e1940bf85ab4cbc42a03e65f64da16a094c --- ethosu/vela/tensor.py | 1 + 1 file changed, 1 insertion(+) (limited to 'ethosu/vela/tensor.py') diff --git a/ethosu/vela/tensor.py b/ethosu/vela/tensor.py index 677757ca..8304a657 100644 --- a/ethosu/vela/tensor.py +++ b/ethosu/vela/tensor.py @@ -478,6 +478,7 @@ class Tensor: if shape_len > 4: return + assert not (self.needs_linear_format and fmt == TensorFormat.NHCWB16) self.storage_rounding_quantum = arch.storage_rounding_quantums[self.format] self.storage_rounding_quantum = tuple(self.storage_rounding_quantum[-shape_len:]) self.brick_size = arch.brick_sizes[self.format] -- cgit v1.2.1