From 2b8c87250656df91dc5d26404a9a4742b4eb8e5a Mon Sep 17 00:00:00 2001 From: Johan Gunnarsson Date: Thu, 10 Apr 2025 17:58:33 +0200 Subject: [PATCH] MLBEDSW-10679: Restore allocation of CPU-only tensors Restore allocation of CPU-only tensors when the --separate-io-regions flag is not enabled. This matches the old behavior. This partially reverts d4c6dbd7c. Signed-off-by: Johan Gunnarsson Change-Id: I4d36ce2197f4386d47cdd96a1859f3be44f63db6 --- ethosu/regor/compiler/scheduler.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ethosu/regor/compiler/scheduler.cpp b/ethosu/regor/compiler/scheduler.cpp index 112b9486..3a8722d3 100644 --- a/ethosu/regor/compiler/scheduler.cpp +++ b/ethosu/regor/compiler/scheduler.cpp @@ -334,7 +334,7 @@ int Scheduler::UpdateSchedulerTensor(TensorUsage usage, SchedulerConnection *con conn->requireFullTensor = conn->requireFullTensor || cpuTensor; tensor->needsLinearFormat = tensor->needsLinearFormat || cpuTensor || CheckLinearFormatForConcatSplit(tensor); - if ( cpuTensor && !tensor->hasNPUWriters && !tensor->hasNPUReaders ) + if ( _options.separateIORegions && cpuTensor && !tensor->hasNPUWriters && !tensor->hasNPUReaders ) { tensor->memArea = _arch->CPUMemory(); } -- GitLab