diff --git a/unit_tests/operator/Test_ReduceSumImpl.cpp b/unit_tests/operator/Test_ReduceSumImpl.cpp index 49569d1f65ff6c51f9681632b16375605ab326e7..0aa543da4f9d55dfa672790a5d99467cd794001a 100644 --- a/unit_tests/operator/Test_ReduceSumImpl.cpp +++ b/unit_tests/operator/Test_ReduceSumImpl.cpp @@ -9,17 +9,22 @@ * ********************************************************************************/ -#include <catch2/catch_test_macros.hpp> +#include <cstddef> // std::size_t +#include <cstdint> // std::uint16_t, std::int32_t #include <memory> -#include <numeric> // std::accumulate #include <random> // std::random_device, std::mt19937, std::uniform_real_distribution +#include <vector> + +#include <catch2/catch_test_macros.hpp> +#include "aidge/backend/cpu/data/TensorImpl.hpp" +#include "aidge/data/Data.hpp" // DataType #include "aidge/data/Tensor.hpp" +#include "aidge/graph/Node.hpp" +#include "aidge/operator/OperatorTensor.hpp" #include "aidge/operator/ReduceSum.hpp" -#include "aidge/operator/Conv.hpp" - -#include "aidge/backend/cpu.hpp" #include "aidge/utils/TensorUtils.hpp" +#include "aidge/utils/Types.h" using namespace Aidge; @@ -112,7 +117,7 @@ TEST_CASE("[cpu/operator] ReduceSum(forward)", "[ReduceSum][CPU]") { std::shared_ptr<Tensor> myInput = std::make_shared<Tensor>(dims); myInput->setBackend("cpu"); myInput->setDataType(DataType::Float32); - std::shared_ptr<Node> myReduceSum = ReduceSum(std::vector<int32_t>{}, false, true); + std::shared_ptr<Node> myReduceSum = ReduceSum(std::vector<std::int32_t>{}, false, true); auto op = std::static_pointer_cast<OperatorTensor>(myReduceSum -> getOperator()); op->associateInput(0,myInput); op->setDataType(DataType::Float32);