From 970f209825237a8e9ceafc2d3336f7324a0aa0e2 Mon Sep 17 00:00:00 2001 From: Donghyeon Jeong Date: Tue, 22 Aug 2023 10:59:59 +0900 Subject: [PATCH] [TensorPool] Check tensor type in view This PR enables the TensorPool view to filter call from different tensor type **Self evaluation:** 1. Build test: [X]Passed [ ]Failed [ ]Skipped 2. Run test: [X]Passed [ ]Failed [ ]Skipped Signed-off-by: Donghyeon Jeong --- nntrainer/tensor/tensor_pool.cpp | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/nntrainer/tensor/tensor_pool.cpp b/nntrainer/tensor/tensor_pool.cpp index ba1021c..de45099 100644 --- a/nntrainer/tensor/tensor_pool.cpp +++ b/nntrainer/tensor/tensor_pool.cpp @@ -62,6 +62,13 @@ Tensor *TensorPool::view(const std::string &name, const std::string &reference, const std::vector &exec_order, TensorLifespan lifespan, const size_t offset) { auto &spec = getSourceSpec(reference); + + NNTR_THROW_IF(spec.tensor->getDataType() != dim.getDataType() || + spec.tensor->getFormat() != dim.getFormat(), + std::invalid_argument) + << "view tensor type != source tensor type, view tensor type: " << dim + << " source tensor: " << spec.tensor->getDim(); + unsigned adjusted_offset = std::visit( [](const auto &s) { using T = std::decay_t; -- 2.7.4