From dd2529ecb6a4af3d92f126b0bbbc0b6b6e02ef5a Mon Sep 17 00:00:00 2001
From: Olivier BICHLER <olivier.bichler@cea.fr>
Date: Fri, 8 Dec 2023 22:40:51 +0100
Subject: [PATCH] Minor adaptations
---
include/aidge/backend/cuda/data/TensorImpl.hpp | 6 ++++--
1 file changed, 4 insertions(+), 2 deletions(-)
diff --git a/include/aidge/backend/cuda/data/TensorImpl.hpp b/include/aidge/backend/cuda/data/TensorImpl.hpp
index 4f66a93..6b309c5 100644
--- a/include/aidge/backend/cuda/data/TensorImpl.hpp
+++ b/include/aidge/backend/cuda/data/TensorImpl.hpp
@@ -81,6 +81,10 @@ public:
}
void copyCast(const void *src, NbElts_t length, const DataType srcDt) override {
+ if (length == 0) {
+ return;
+ }
+
if (srcDt == DataType::Float64) {
thrust_copy(static_cast<const double*>(src),
static_cast<T*>(rawPtr()),
@@ -208,8 +212,6 @@ public:
private:
void lazyInit() {
- AIDGE_INTERNAL_ASSERT(mTensor.dataType() == NativeType<T>::type);
-
if (mData.size() < mTensor.size()) {
// Need more data, a re-allocation will occur
AIDGE_ASSERT(mData.empty() || mDataOwner != nullptr, "trying to enlarge non-owned data");
--
GitLab