From 41b356b3f46b890c41946fa07a7e1461015a1e88 Mon Sep 17 00:00:00 2001
From: Olivier BICHLER <olivier.bichler@cea.fr>
Date: Tue, 26 Nov 2024 12:16:02 +0100
Subject: [PATCH] Fixed unit test following eclipse/aidge/aidge_core#195 fix
---
unit_tests/operator/Test_GlobalAveragePoolingImpl.cpp | 8 ++++----
1 file changed, 4 insertions(+), 4 deletions(-)
diff --git a/unit_tests/operator/Test_GlobalAveragePoolingImpl.cpp b/unit_tests/operator/Test_GlobalAveragePoolingImpl.cpp
index d5f2065b..51b4366d 100644
--- a/unit_tests/operator/Test_GlobalAveragePoolingImpl.cpp
+++ b/unit_tests/operator/Test_GlobalAveragePoolingImpl.cpp
@@ -151,7 +151,7 @@ TEST_CASE("[cpu/operator] GlobalAveragePooling",
T0->getImpl()->setRawPtr(array0, in_nb_elems);
// results
- Tres->resize(dims_out);
+ Tres->resize(dims_in);
Tres->getImpl()->setRawPtr(result, out_nb_elems);
op->forwardDims();
@@ -222,7 +222,7 @@ TEST_CASE("[cpu/operator] GlobalAveragePooling",
T0->getImpl()->setRawPtr(array0, in_nb_elems);
// results
- Tres->resize(dims_out);
+ Tres->resize(dims_in);
Tres->getImpl()->setRawPtr(result, out_nb_elems);
op->forwardDims();
@@ -348,7 +348,7 @@ TEST_CASE("[cpu/operator] GlobalAveragePooling",
T0->getImpl()->setRawPtr(input, in_nb_elems);
// results
- Tres->resize(out_dims);
+ Tres->resize(in_dims);
Tres->getImpl()->setRawPtr(result, out_nb_elems);
op->forwardDims();
start = std::chrono::system_clock::now();
@@ -535,7 +535,7 @@ TEST_CASE("[cpu/operator] GlobalAveragePooling",
T0->getImpl()->setRawPtr(input, in_nb_elems);
// results
- Tres->resize(out_dims);
+ Tres->resize(in_dims);
Tres->getImpl()->setRawPtr(result, out_nb_elems);
op->forwardDims();
start = std::chrono::system_clock::now();
--
GitLab