From 735ce2254a839d1f0e76f1211913df640b48f4b6 Mon Sep 17 00:00:00 2001 From: Alexander Bengtsson Date: Mon, 13 Jan 2025 16:07:22 +0100 Subject: [PATCH] MLBEDSW-10253: Normalise scales when ofm-fusing - Unit scales were not normalized when performing OFM-fusing This triggered asserts in RCS-gen for operations that can only handle unit scaling Change-Id: I8f06f070aa4bb308afa5223f15d2e1bb88465210 Signed-off-by: Alexander Bengtsson --- ethosu/regor/compiler/graphir_optimiser.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ethosu/regor/compiler/graphir_optimiser.cpp b/ethosu/regor/compiler/graphir_optimiser.cpp index 2b44bd02..590bbc8a 100644 --- a/ethosu/regor/compiler/graphir_optimiser.cpp +++ b/ethosu/regor/compiler/graphir_optimiser.cpp @@ -959,7 +959,7 @@ Operation *GraphIrOptimiser::FuseRescale(Graph *const graph, Operation *const op { // Propagate rescaling to output of previous op producer->CopyOutput(TensorUsage::OFM, *ofmConn); - producer->Output(TensorUsage::OFM)->Set(ofmConn->rounding); + producer->Output(TensorUsage::OFM)->Set(ofmConn->rounding).Set(ofmQuant); returnOp = producer.get(); } } -- GitLab