From a0dd82ccbaaddaa519c44a3293207bead28a9450 Mon Sep 17 00:00:00 2001 From: Antoni Olivier <olivier.antoni@cea.fr> Date: Tue, 25 Jun 2024 14:53:58 +0200 Subject: [PATCH] Fix Adam optimizer unit test --- unit_tests/optimizer/Test_Adam.cpp | 3 +++ 1 file changed, 3 insertions(+) diff --git a/unit_tests/optimizer/Test_Adam.cpp b/unit_tests/optimizer/Test_Adam.cpp index e3957a7..cf132a2 100644 --- a/unit_tests/optimizer/Test_Adam.cpp +++ b/unit_tests/optimizer/Test_Adam.cpp @@ -23,6 +23,9 @@ #include "aidge/learning/learningRate/LRSchedulerList.hpp" #include "aidge/learning/optimizer/Optimizer.hpp" #include "aidge/learning/optimizer/Adam.hpp" +#include "aidge/backend/cpu/operator/AddImpl.hpp" +#include "aidge/backend/cpu/operator/MulImpl.hpp" +#include "aidge/backend/cpu/operator/SubImpl.hpp" #include "aidge/backend/cpu/operator/SqrtImpl.hpp" #include "aidge/utils/TensorUtils.hpp" -- GitLab