diff --git a/CMakeLists.txt b/CMakeLists.txt index e157c61221f77995bd72125ce848aa0024012a50..df8d6d4dff7be783aef58a5beca33d3a922caa1f 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 5781f014483164142187f07edb402cbde086dc43..618ccc06f40da4b1f1c491487fd978da768652e4 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