From caef1fc781658a9dbd54dc7eb22a2fcd9ddf5e96 Mon Sep 17 00:00:00 2001 From: Donghak PARK Date: Tue, 31 Dec 2024 10:34:39 +0900 Subject: [PATCH] [FSU] Bugfix on FSU Inference at case of inference there are some bug - when set lifespan to tensor swap & non-swap value is exchanged - set swap case - forward_func_lifespan - set non-swap case - forward_infer_lifespan **Self evaluation:** 1. Build test: [X]Passed [ ]Failed [ ]Skipped 2. Run test: [X]Passed [ ]Failed [ ]Skipped Signed-off-by: Donghak PARK Co-authored-by: Jijoong Moon --- nntrainer/tensor/manager.cpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/nntrainer/tensor/manager.cpp b/nntrainer/tensor/manager.cpp index 8500db83d3..1e0b1ddce2 100644 --- a/nntrainer/tensor/manager.cpp +++ b/nntrainer/tensor/manager.cpp @@ -381,9 +381,9 @@ std::vector Manager::requestWeights( var_ls = TensorLifespan::MAX_LIFESPAN; } else { if (enable_swap) { - var_ls = TensorLifespan::FORWARD_INFER_LIFESPAN; - } else { var_ls = TensorLifespan::FORWARD_FUNC_LIFESPAN; + } else { + var_ls = TensorLifespan::FORWARD_INFER_LIFESPAN; } }