diff --git a/ethosu/regor/architecture/ethosu85/ethos_u85.cpp b/ethosu/regor/architecture/ethosu85/ethos_u85.cpp index 1f3907b779ff934089c7b190ab8e0f06409d2a17..04a3e0b200903e30a38d792587a8aec05d8c9cf6 100644 --- a/ethosu/regor/architecture/ethosu85/ethos_u85.cpp +++ b/ethosu/regor/architecture/ethosu85/ethos_u85.cpp @@ -1779,12 +1779,12 @@ bool EthosU85OpGroup::CanRunOnNPU(const ArchitectureOpGroupQuery &op) return true; } - ArchOperatorQuery query; - query.transposeMask = op.ofm.transpose; - query.reverseMask = op.ofm.reverse; - if ( !_arch->_constraints->OperatorQuery(OpType::MemoryCopy, &query, nullptr).Any(QueryResult::Native) ) + if ( op.type == OpType::Transpose || op.type == OpType::Reverse ) { - return false; + ArchOperatorQuery query; + query.transposeMask = op.ofm.transpose; + query.reverseMask = op.ofm.reverse; + return _arch->_constraints->OperatorQuery(OpType::MemoryCopy, &query, nullptr).Any(QueryResult::Native); } auto map = s_opDataTypeSupport.find(npuOp);