[Bug] Fix memory access error in addValue

- Previously memory access to tensor data was incorrect
- Change to direct access to data with index instead of calculating the index

Signed-off-by: Donghyeon Jeong <dhyeon.jeong@samsung.com>

diff --git a/nntrainer/tensor/tensor.h b/nntrainer/tensor/tensor.h
index ad68f37..eb1279f 100644 (file)
--- a/nntrainer/tensor/tensor.h
+++ b/nntrainer/tensor/tensor.h
@@ -1315,9 +1315,11 @@ public:
                 unsigned int w, float value, float beta) noexcept {
     auto const &idx = getIndex(batch, c, h, w);
     if (dim.getDataType() == Tdatatype::FP32) {
-      *(float *)(getData(idx)) = value + *(float *)(getData(idx)) * beta;
+      getData<float>()[idx] *= beta;
+      getData<float>()[idx] += value;
     } else if (dim.getDataType() == Tdatatype::FP16) {
-      *(__fp16 *)(getData(idx)) = value + *(__fp16 *)(getData(idx)) * beta;
+      getData<__fp16>()[idx] *= beta;
+      getData<__fp16>()[idx] += value;
     }
   }