[Fix] tensor pool prerequest bug
authorJihoon Lee <jhoon.it.lee@samsung.com>
Fri, 15 Oct 2021 09:23:51 +0000 (18:23 +0900)
committerJijoong Moon <jijoong.moon@samsung.com>
Tue, 19 Oct 2021 11:34:48 +0000 (20:34 +0900)
This patch fix tensor pool prereuqestedrequest bug by that shared_name
was wrong

**Self evaluation:**
1. Build test: [X]Passed [ ]Failed [ ]Skipped
2. Run test: [X]Passed [ ]Failed [ ]Skipped

Signed-off-by: Jihoon Lee <jhoon.it.lee@samsung.com>
nntrainer/tensor/manager.cpp

index ce45a6a22cc02b0c94b6bc4aa535313fac5d1c16..024dd83aefc78f008f98d4e204f3db07047295b3 100644 (file)
@@ -362,7 +362,7 @@ Manager::requestTensors(const GraphNode &node,
         grad = tensor_pool.requestPrerequestedTensor(
           dim, grad_exec_order, tspan,
           name + Var_Grad::grad_suffix, /// name
-          shared_name,
+          shared_name + Var_Grad::grad_suffix,
           Tensor::Initializer::ZEROS /// tensor initializer
         );
       }