diff --git a/include/aidge/backend/cuda.hpp b/include/aidge/backend/cuda.hpp index 02cfdb1502bfca0669fc503d45cb8d87cb91ad63..19773878c28c0fa09379af2033e058695f4175f3 100644 --- a/include/aidge/backend/cuda.hpp +++ b/include/aidge/backend/cuda.hpp @@ -20,6 +20,7 @@ #include "aidge/backend/cuda/operator/BatchNormImpl.hpp" #include "aidge/backend/cuda/operator/ConvImpl.hpp" #include "aidge/backend/cuda/operator/DivImpl.hpp" +#include "aidge/backend/cuda/operator/ErfImpl.hpp" #include "aidge/backend/cuda/operator/FCImpl.hpp" #include "aidge/backend/cuda/operator/GlobalAveragePoolingImpl.hpp" #include "aidge/backend/cuda/operator/LnImpl.hpp" @@ -35,6 +36,7 @@ #include "aidge/backend/cuda/operator/ShiftGELUImpl.hpp" #include "aidge/backend/cuda/operator/ReshapeImpl.hpp" #include "aidge/backend/cuda/operator/SigmoidImpl.hpp" +#include "aidge/backend/cuda/operator/SoftmaxImpl.hpp" #include "aidge/backend/cuda/operator/SubImpl.hpp" #include "aidge/backend/cuda/operator/TanhImpl.hpp"