Skip to content

Commit

Permalink
[FSU] Bugfix on FSU Inference
Browse files Browse the repository at this point in the history
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 <donghak.park@samsung.com>
Co-authored-by: Jijoong Moon <jijoong.moon@samsung.com>
  • Loading branch information
DonghakPark and jijoongmoon committed Jan 2, 2025
1 parent deec076 commit caef1fc
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions nntrainer/tensor/manager.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -381,9 +381,9 @@ std::vector<Weight *> 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;
}
}

Expand Down

0 comments on commit caef1fc

Please sign in to comment.