From 03241e36819a8aa49ace5712b033b4d809b2a85b Mon Sep 17 00:00:00 2001
From: Antoni Olivier <olivier.antoni@cea.fr>
Date: Tue, 25 Jun 2024 12:40:42 +0200
Subject: [PATCH] Fix Adam optimizer unit test
---
unit_tests/optimizer/Test_Adam.cpp | 1 +
1 file changed, 1 insertion(+)
diff --git a/unit_tests/optimizer/Test_Adam.cpp b/unit_tests/optimizer/Test_Adam.cpp
index ef5482d..e3957a7 100644
--- a/unit_tests/optimizer/Test_Adam.cpp
+++ b/unit_tests/optimizer/Test_Adam.cpp
@@ -23,6 +23,7 @@
#include "aidge/learning/learningRate/LRSchedulerList.hpp"
#include "aidge/learning/optimizer/Optimizer.hpp"
#include "aidge/learning/optimizer/Adam.hpp"
+#include "aidge/backend/cpu/operator/SqrtImpl.hpp"
#include "aidge/utils/TensorUtils.hpp"
namespace Aidge {
--
GitLab