From fe6588d9851a671ba318e4d8aef1c5856a90bc8b Mon Sep 17 00:00:00 2001 From: William Isaksson Date: Thu, 10 Apr 2025 08:07:17 +0200 Subject: [PATCH] MLBEDSW-10682: Add _srcKey tracking to scheduler decompose Makes sure ops created by scheduler decompose get the right source key attached Change-Id: I3f6aa85e05d1166e3c004b5d2e0d8dccb68fefaf Signed-off-by: William Isaksson --- ethosu/regor/compiler/scheduler_packing.cpp | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/ethosu/regor/compiler/scheduler_packing.cpp b/ethosu/regor/compiler/scheduler_packing.cpp index 745e0252..7af4acf2 100644 --- a/ethosu/regor/compiler/scheduler_packing.cpp +++ b/ethosu/regor/compiler/scheduler_packing.cpp @@ -140,7 +140,16 @@ void SchedulerPacking::FilterOperations(const std::vector &executio if ( ShouldDecompose(_arch, schedOp.get()) ) { + auto srcKey = schedOp->_srcKey; auto schedOps = DecomposeSchedulerOperation(std::move(schedOp)); + // Track source keys + for ( auto &newOp : schedOps ) + { + if ( !newOp->_srcKey ) + { + newOp->_srcKey = srcKey; + } + } _schedList.insert( _schedList.end(), std::make_move_iterator(schedOps.begin()), std::make_move_iterator(schedOps.end())); } -- GitLab