From a8147b884a982a3d364669c2c96cd53d84c26953 Mon Sep 17 00:00:00 2001
From: Olivier BICHLER <olivier.bichler@cea.fr>
Date: Tue, 16 Apr 2024 11:25:03 +0200
Subject: [PATCH] Make FC inherit MatMul and Add nodes name
---
src/recipes/FuseMulAdd.cpp | 8 ++++++--
1 file changed, 6 insertions(+), 2 deletions(-)
diff --git a/src/recipes/FuseMulAdd.cpp b/src/recipes/FuseMulAdd.cpp
index 6582038e9..9a89f8af5 100644
--- a/src/recipes/FuseMulAdd.cpp
+++ b/src/recipes/FuseMulAdd.cpp
@@ -85,8 +85,12 @@ void Aidge::fuseMulAdd(std::shared_ptr<Aidge::Node> matmulNode, std::shared_ptr<
AIDGE_ASSERT(outSize, "Couldnt get output number of channels for FC operator.");
// Instanciate FC
- //std::shared_ptr<Node> fc = FC(dim[0], false, "Fc");
- std::shared_ptr<Node> fc = std::make_shared<Node>(std::make_shared<FC_Op>(outSize, bias ? false : true));
+ std::string fcName = matmulNode->name();
+ if (!addNode->name().empty()) {
+ fcName += "_" + addNode->name();
+ }
+
+ std::shared_ptr<Node> fc = std::make_shared<Node>(std::make_shared<FC_Op>(outSize, bias ? false : true), fcName);
// Step 2 : Branch existing producers & create the others
// link weights & bias
--
GitLab