diff --git a/include/aidge/backend/cuda/operator/AvgPoolingImpl.hpp b/include/aidge/backend/cuda/operator/AvgPoolingImpl.hpp index bec3728579a3299f11cf21d7140df8adaef13b98..43a6bd57c0c6431705abe73d3f3c175046d72dc9 100644 --- a/include/aidge/backend/cuda/operator/AvgPoolingImpl.hpp +++ b/include/aidge/backend/cuda/operator/AvgPoolingImpl.hpp @@ -36,7 +36,7 @@ private: std::shared_ptr<Tensor> mInputFallback; public: - AvgPoolingImpl_cuda(const AvgPooling_Op<DIM> &op) : OperatorImpl(op) {} + AvgPoolingImpl_cuda(const AvgPooling_Op<DIM> &op) : OperatorImpl(op, "cuda") {} static std::unique_ptr<AvgPoolingImpl_cuda> create(const AvgPooling_Op<2> &op) { return std::make_unique<AvgPoolingImpl_cuda>(op); diff --git a/include/aidge/backend/cuda/operator/ConvImpl.hpp b/include/aidge/backend/cuda/operator/ConvImpl.hpp index 1f692b09cf44e5d54d1bc9d5b998b90a800f719f..8c591927ce0e52daeff447726c114ce3ae4d0103 100644 --- a/include/aidge/backend/cuda/operator/ConvImpl.hpp +++ b/include/aidge/backend/cuda/operator/ConvImpl.hpp @@ -44,7 +44,7 @@ private: std::shared_ptr<Tensor> mInput2Fallback; public: - ConvImpl_cuda(const Conv_Op<DIM> &op) : OperatorImpl(op) {} + ConvImpl_cuda(const Conv_Op<DIM> &op) : OperatorImpl(op, "cuda") {} static std::unique_ptr<ConvImpl_cuda> create(const Conv_Op<DIM> &op) { return std::make_unique<ConvImpl_cuda>(op); diff --git a/include/aidge/backend/cuda/operator/FCImpl.hpp b/include/aidge/backend/cuda/operator/FCImpl.hpp index 023757b17eb80249f184a144cd7c38b7ca5758e6..3d8a1348d500fc533c7c9b601b09629995f97427 100644 --- a/include/aidge/backend/cuda/operator/FCImpl.hpp +++ b/include/aidge/backend/cuda/operator/FCImpl.hpp @@ -38,7 +38,7 @@ private: public: - FCImpl_cuda(const FC_Op &op) : OperatorImpl(op) {} + FCImpl_cuda(const FC_Op &op) : OperatorImpl(op, "cuda") {} static std::unique_ptr<FCImpl_cuda> create(const FC_Op &op) { return std::make_unique<FCImpl_cuda>(op); diff --git a/include/aidge/backend/cuda/operator/MaxPoolingImpl.hpp b/include/aidge/backend/cuda/operator/MaxPoolingImpl.hpp index 9216eca4b901b9d7b92dbf8efb4d0a419dac9876..4d08d7fab3c7cb2baa18838fd872e44a8eccc923 100644 --- a/include/aidge/backend/cuda/operator/MaxPoolingImpl.hpp +++ b/include/aidge/backend/cuda/operator/MaxPoolingImpl.hpp @@ -36,7 +36,7 @@ private: std::shared_ptr<Tensor> mInputFallback; public: - MaxPoolingImpl_cuda(const MaxPooling_Op<DIM> &op) : OperatorImpl(op) {} + MaxPoolingImpl_cuda(const MaxPooling_Op<DIM> &op) : OperatorImpl(op, "cuda") {} static std::unique_ptr<MaxPoolingImpl_cuda> create(const MaxPooling_Op<2> &op) { return std::make_unique<MaxPoolingImpl_cuda>(op); diff --git a/include/aidge/backend/cuda/operator/ReLUImpl.hpp b/include/aidge/backend/cuda/operator/ReLUImpl.hpp index 27d0a61824df78bcc1c001e6fdc4a28056d5459c..6570662fa5df27e54a9df6f357e918243a71330a 100644 --- a/include/aidge/backend/cuda/operator/ReLUImpl.hpp +++ b/include/aidge/backend/cuda/operator/ReLUImpl.hpp @@ -38,7 +38,7 @@ private: std::shared_ptr<Tensor> mInputFallback; public: - ReLUImpl_cuda(const ReLU_Op &op) : OperatorImpl(op) {} + ReLUImpl_cuda(const ReLU_Op &op) : OperatorImpl(op, "cuda") {} static std::unique_ptr<ReLUImpl_cuda> create(const ReLU_Op &op) { return std::make_unique<ReLUImpl_cuda>(op);