From 151f782ce1693830b2b6e865c862cb8059bc07a2 Mon Sep 17 00:00:00 2001
From: thibault allenet <thibault.allenet@cea.fr>
Date: Tue, 29 Aug 2023 14:40:40 +0000
Subject: [PATCH] Add an offset argument in the tensor copy CPP implementation
---
include/aidge/data/TensorImpl.hpp | 4 ++--
unit_tests/Test_TensorImpl.cpp | 1 +
2 files changed, 3 insertions(+), 2 deletions(-)
diff --git a/include/aidge/data/TensorImpl.hpp b/include/aidge/data/TensorImpl.hpp
index 002bad3c..c3cc1954 100644
--- a/include/aidge/data/TensorImpl.hpp
+++ b/include/aidge/data/TensorImpl.hpp
@@ -37,9 +37,9 @@ class TensorImpl_cpu : public TensorImpl {
std::size_t scalarSize() const override { return sizeof(T); }
- void copy(const void *src, NbElts_t length) override {
+ void copy(const void *src, NbElts_t length, std::size_t offset = 0) override {
std::copy(static_cast<const T *>(src), static_cast<const T *>(src) + length,
- static_cast<T *>(rawPtr()));
+ static_cast<T *>(rawPtr())+offset);
}
void *rawPtr() override {
diff --git a/unit_tests/Test_TensorImpl.cpp b/unit_tests/Test_TensorImpl.cpp
index ca9e7df5..d5aa94d0 100644
--- a/unit_tests/Test_TensorImpl.cpp
+++ b/unit_tests/Test_TensorImpl.cpp
@@ -57,3 +57,4 @@ TEST_CASE("Tensor creation") {
}
}
}
+
--
GitLab