From 3f6ee54d108e7b51c39931016b955b1ee2ec5bdd Mon Sep 17 00:00:00 2001
From: hrouis <houssemeddine.rouis92@gmail.com>
Date: Fri, 29 Mar 2024 10:26:43 +0100
Subject: [PATCH] fix OperatorImpl call
---
include/aidge/backend/cuda/operator/AvgPoolingImpl.hpp | 2 +-
include/aidge/backend/cuda/operator/ConvImpl.hpp | 2 +-
include/aidge/backend/cuda/operator/FCImpl.hpp | 2 +-
include/aidge/backend/cuda/operator/MaxPoolingImpl.hpp | 2 +-
include/aidge/backend/cuda/operator/ReLUImpl.hpp | 2 +-
5 files changed, 5 insertions(+), 5 deletions(-)
diff --git a/include/aidge/backend/cuda/operator/AvgPoolingImpl.hpp b/include/aidge/backend/cuda/operator/AvgPoolingImpl.hpp
index bec3728..43a6bd5 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 1f692b0..8c59192 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 023757b..3d8a134 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 9216eca..4d08d7f 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 27d0a61..6570662 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);
--
GitLab