From ad4b1edbec2551163cbc2b39b2f8a745cf091998 Mon Sep 17 00:00:00 2001
From: cmoineau <cyril.moineau@cea.fr>
Date: Fri, 3 May 2024 09:19:45 +0000
Subject: [PATCH] Add clearer error message for missing input gradient for FC.
---
src/operator/FCImpl.cpp | 5 ++++-
1 file changed, 4 insertions(+), 1 deletion(-)
diff --git a/src/operator/FCImpl.cpp b/src/operator/FCImpl.cpp
index eecff38a..d9edf3a9 100644
--- a/src/operator/FCImpl.cpp
+++ b/src/operator/FCImpl.cpp
@@ -72,7 +72,10 @@ void Aidge::FCImpl_cpu::backward()
{
const FC_Op& op_ = dynamic_cast<const FC_Op&>(mOp);
const auto& fc_grad = op_.getOutput(0)->grad();
- assert(fc_grad && "missing ouput #0 gradient");
+ AIDGE_ASSERT(fc_grad, "missing ouput #0 gradient");
+ AIDGE_ASSERT(op_.getInput(0)->grad(), "missing input #0 gradient");
+ AIDGE_ASSERT(op_.getInput(1)->grad(), "missing input #1 gradient");
+ AIDGE_ASSERT(op_.getInput(2)->grad(), "missing input #2 gradient");
// Find the correct kernel type
const Registrar<FCImplBackward_cpu>::registrar_key registrarKey = {
--
GitLab