From b5210d987b1d4acd4a149667db3800dc441c4ecd Mon Sep 17 00:00:00 2001
From: Olivier BICHLER <olivier.bichler@cea.fr>
Date: Thu, 16 Jan 2025 09:09:48 +0100
Subject: [PATCH] Fixed identity to ensure identity, not assume it
---
src/operator/Identity.cpp | 3 +++
1 file changed, 3 insertions(+)
diff --git a/src/operator/Identity.cpp b/src/operator/Identity.cpp
index f0b8720bc..d01d57678 100644
--- a/src/operator/Identity.cpp
+++ b/src/operator/Identity.cpp
@@ -15,6 +15,9 @@
void Aidge::Identity_OpImpl::forward() {
const Identity_Op& op = dynamic_cast<const Identity_Op&>(mOp);
+ op.getOutput(0)->setBackend(op.getInput(0)->backend(), op.getInput(0)->device());
+ op.getOutput(0)->setDataType(op.getInput(0)->dataType());
+ op.getOutput(0)->setDataFormat(op.getInput(0)->dataFormat());
op.getOutput(0)->getImpl()->copy(op.getInput(0)->getImpl()->rawPtr(), op.getInput(0)->size());
}
--
GitLab