From 6ec0de5cb05d4e5e2a4d67d8a3e6a358b731e8e0 Mon Sep 17 00:00:00 2001
From: NAUD Maxence <maxence.naud@cea.fr>
Date: Fri, 29 Mar 2024 21:36:53 +0000
Subject: [PATCH] Fix
---
CMakeLists.txt | 2 +-
src/operator/GlobalAveragePooling.cpp | 7 ++++++-
2 files changed, 7 insertions(+), 2 deletions(-)
diff --git a/CMakeLists.txt b/CMakeLists.txt
index e157c6122..df8d6d4df 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -19,7 +19,7 @@ option(PYBIND "python binding" ON)
option(WERROR "Warning as error" OFF)
option(TEST "Enable tests" ON)
option(COVERAGE "Enable coverage" OFF)
-option(ENABLE_ASAN "Enable ASan (adress sanitizer) for runtime analysis of memory use (over/underflow, memory leak, ...)" OFF)
+option(ENABLE_ASAN "Enable ASan (AddressSanitizer) for runtime analysis of memory use (over/underflow, memory leak, ...)" OFF)
##############################################
# Import utils CMakeLists
diff --git a/src/operator/GlobalAveragePooling.cpp b/src/operator/GlobalAveragePooling.cpp
index 5781f0144..618ccc06f 100644
--- a/src/operator/GlobalAveragePooling.cpp
+++ b/src/operator/GlobalAveragePooling.cpp
@@ -9,10 +9,15 @@
*
********************************************************************************/
+#include <memory>
+#include <stdexcept> // std::runtime_error
#include <string>
#include <vector>
+#include "aidge/data/Tensor.hpp"
#include "aidge/operator/GlobalAveragePooling.hpp"
+#include "aidge/utils/ErrorHandling.hpp"
+#include "aidge/utils/Types.h"
const std::string Aidge::GlobalAveragePooling_Op::Type = "GlobalAveragePooling";
@@ -41,7 +46,7 @@ void Aidge::GlobalAveragePooling_Op::computeOutputDims() {
}
}
-void Aidge::GlobalAveragePooling_Op::setBackend(const std::string &name, DeviceIdx_t device) {
+void Aidge::GlobalAveragePooling_Op::setBackend(const std::string &name, Aidge::DeviceIdx_t device) {
SET_IMPL_MACRO(GlobalAveragePooling_Op, *this, name);
mOutputs[0]->setBackend(name, device);
}
\ No newline at end of file
--
GitLab