From fa1ce74053d4a5f18162c55a47dbc31f668f98e4 Mon Sep 17 00:00:00 2001 From: Alexander Bengtsson Date: Tue, 8 Apr 2025 14:10:42 +0200 Subject: [PATCH] MLBEDSW-10663: Skip OpType::Rescale in Ethos-U55 SupportedDtypes - Skip DataType-checks for Rescale on Ethos-U55 as the constraints differ from regular pooling constraints. This aligns with compiler behaviour before MLBEDSW-10488 Change-Id: Id3ddc2d4de680cf1b4e4acc11582b1daa0927746 Signed-off-by: Alexander Bengtsson --- ethosu/regor/architecture/ethosu55/ethos_u55_constraints.cpp | 5 +++-- ethosu/regor/architecture/ethosu85/ethos_u85_constraints.cpp | 3 ++- 2 files changed, 5 insertions(+), 3 deletions(-) diff --git a/ethosu/regor/architecture/ethosu55/ethos_u55_constraints.cpp b/ethosu/regor/architecture/ethosu55/ethos_u55_constraints.cpp index 37933d51..49742cee 100644 --- a/ethosu/regor/architecture/ethosu55/ethos_u55_constraints.cpp +++ b/ethosu/regor/architecture/ethosu55/ethos_u55_constraints.cpp @@ -180,9 +180,10 @@ bool EthosU55Constraints::SupportedDtypes(OpType opType, DataType ifmType, DataT return false; } - if ( _arch->UseAvgPoolNop(opType) ) + if ( _arch->UseAvgPoolNop(opType) || opType == OpType::Rescale ) { - // The rules for UseAvgPoolNop are not the same as for a Pooling operation, so skip checks for now + // TODO MLBEDSW-10667: The rules for UseAvgPoolNop are not the same as for a Pooling operation, so skip checks + // for now return true; } diff --git a/ethosu/regor/architecture/ethosu85/ethos_u85_constraints.cpp b/ethosu/regor/architecture/ethosu85/ethos_u85_constraints.cpp index a7bc9315..f43a8407 100644 --- a/ethosu/regor/architecture/ethosu85/ethos_u85_constraints.cpp +++ b/ethosu/regor/architecture/ethosu85/ethos_u85_constraints.cpp @@ -219,7 +219,8 @@ bool EthosU85Constraints::SupportedDtypes(OpType opType, DataType ifmType, DataT if ( _arch->UseAvgPoolNop(opType) ) { - // The rules for UseAvgPoolNop are not the same as for a Pooling operation, skip checks for now + // TODO MLBEDSW-10667: The rules for UseAvgPoolNop are not the same as for a Pooling operation, skip checks for + // now return true; } -- GitLab