Skip to content
Snippets Groups Projects
Commit 6ec0de5c authored by Maxence Naud's avatar Maxence Naud
Browse files

Fix

parent 62c5328d
No related branches found
No related tags found
2 merge requests!105version 0.2.0,!91Feat/operator global average pooling
Pipeline #42563 passed
......@@ -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
......
......@@ -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
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment