diff --git a/unit_tests/optimizer/Test_Adam.cpp b/unit_tests/optimizer/Test_Adam.cpp index ef5482d448f49e3c44f7fa97fb8ae67f40e9cb44..e3957a7fe5b571115fc067240b72029dea1c6bae 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 {