From c3982bdecccd3d9046a3232cd9aa25b3560dda6a Mon Sep 17 00:00:00 2001 From: Max Bergfelt Date: Wed, 12 Mar 2025 16:53:20 +0100 Subject: [PATCH] MLBEDSW-10449: Failed to find block configuration Fixed a non-negative check of the OFM slice shape size in the HandleDilation method of scheduler_decompose.cpp which was causing a fail to find the block config. The bug caused problems for certain networks with a dilation size greater than the IFM shape. Change-Id: Ibbb9c47a6bf6b032087c30c926c0abbdf8f17b35 Signed-off-by: Max Bergfelt --- ethosu/regor/compiler/scheduler_decompose.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ethosu/regor/compiler/scheduler_decompose.cpp b/ethosu/regor/compiler/scheduler_decompose.cpp index 9dafb6ce..096bae9a 100644 --- a/ethosu/regor/compiler/scheduler_decompose.cpp +++ b/ethosu/regor/compiler/scheduler_decompose.cpp @@ -433,7 +433,7 @@ HandleDilation(Architecture *arch, std::unique_ptr op, const newOfmSlice.offset.WithHeight(newOfmSlice.offset.Height() + dy).WithWidth(newOfmSlice.offset.Width() + dx); newOfmSlice.shape = newOfmSlice.shape.WithHeight(newOfmSlice.shape.Height() - dy).WithWidth(newOfmSlice.shape.Width() - dx); - if ( newOfmSlice.shape.Elements() > 0 ) + if ( newOfmSlice.shape.Width() > 0 && newOfmSlice.shape.Height() > 0 ) { ofmStrides.y *= DY; ofmStrides.x *= DX; -- GitLab