From 5f4f881427b9efc8da83d87c7781000f53a3b0e8 Mon Sep 17 00:00:00 2001
From: Olivier BICHLER <olivier.bichler@cea.fr>
Date: Tue, 26 Nov 2024 14:17:14 +0100
Subject: [PATCH] FIxed dims in unit tests
---
.../Test_GlobalAveragePoolingImpl.cpp | 24 ++++++++++++-------
1 file changed, 16 insertions(+), 8 deletions(-)
diff --git a/unit_tests/operator/Test_GlobalAveragePoolingImpl.cpp b/unit_tests/operator/Test_GlobalAveragePoolingImpl.cpp
index 51b4366d..43af5448 100644
--- a/unit_tests/operator/Test_GlobalAveragePoolingImpl.cpp
+++ b/unit_tests/operator/Test_GlobalAveragePoolingImpl.cpp
@@ -124,7 +124,9 @@ TEST_CASE("[cpu/operator] GlobalAveragePooling",
dims_in[1]; // averaging per channel : 1 addition per element in
// the channel + 1 division this for every batch
// create out nb_elems
- std::vector<std::size_t> dims_out{dims_in[0], dims_in[1]};
+ std::vector<std::size_t> dims_out(dims_in.size(), 1);
+ dims_out[0] = dims_in[0];
+ dims_out[1] = dims_in[1];
const std::size_t out_nb_elems =
std::accumulate(dims_out.cbegin(), dims_out.cend(), std::size_t(1),
std::multiplies<std::size_t>());
@@ -151,7 +153,7 @@ TEST_CASE("[cpu/operator] GlobalAveragePooling",
T0->getImpl()->setRawPtr(array0, in_nb_elems);
// results
- Tres->resize(dims_in);
+ Tres->resize(dims_out);
Tres->getImpl()->setRawPtr(result, out_nb_elems);
op->forwardDims();
@@ -192,7 +194,9 @@ TEST_CASE("[cpu/operator] GlobalAveragePooling",
// the channel + 1 division this for every batch
// create out nb_elems
- std::vector<std::size_t> dims_out{dims_in[0], dims_in[1]};
+ std::vector<std::size_t> dims_out(dims_in.size(), 1);
+ dims_out[0] = dims_in[0];
+ dims_out[1] = dims_in[1];
const std::size_t out_nb_elems =
std::accumulate(dims_out.cbegin(), dims_out.cend(),
std::size_t(1), std::multiplies<std::size_t>());
@@ -222,7 +226,7 @@ TEST_CASE("[cpu/operator] GlobalAveragePooling",
T0->getImpl()->setRawPtr(array0, in_nb_elems);
// results
- Tres->resize(dims_in);
+ Tres->resize(dims_out);
Tres->getImpl()->setRawPtr(result, out_nb_elems);
op->forwardDims();
@@ -253,7 +257,9 @@ TEST_CASE("[cpu/operator] GlobalAveragePooling",
SECTION("2D_img") {
const std::vector<DimSize_t> in_dims{batch_size, channels, height,
width};
- const std::vector<DimSize_t> out_dims{batch_size, channels};
+ std::vector<std::size_t> out_dims(in_dims.size(), 1);
+ out_dims[0] = in_dims[0];
+ out_dims[1] = in_dims[1];
DimSize_t in_nb_elems = batch_size * channels * height * width;
DimSize_t out_nb_elems = batch_size * channels;
number_of_operation +=
@@ -348,7 +354,7 @@ TEST_CASE("[cpu/operator] GlobalAveragePooling",
T0->getImpl()->setRawPtr(input, in_nb_elems);
// results
- Tres->resize(in_dims);
+ Tres->resize(out_dims);
Tres->getImpl()->setRawPtr(result, out_nb_elems);
op->forwardDims();
start = std::chrono::system_clock::now();
@@ -368,7 +374,9 @@ TEST_CASE("[cpu/operator] GlobalAveragePooling",
SECTION("3D_img") {
const std::vector<DimSize_t> in_dims{batch_size, channels, height,
width, depth};
- const std::vector<DimSize_t> out_dims{batch_size, channels};
+ std::vector<std::size_t> out_dims(in_dims.size(), 1);
+ out_dims[0] = in_dims[0];
+ out_dims[1] = in_dims[1];
DimSize_t in_nb_elems =
batch_size * channels * height * width * depth;
number_of_operation +=
@@ -535,7 +543,7 @@ TEST_CASE("[cpu/operator] GlobalAveragePooling",
T0->getImpl()->setRawPtr(input, in_nb_elems);
// results
- Tres->resize(in_dims);
+ Tres->resize(out_dims);
Tres->getImpl()->setRawPtr(result, out_nb_elems);
op->forwardDims();
start = std::chrono::system_clock::now();
--
GitLab