From d87663c38e99aa3d901fb0442e3590193f3eee34 Mon Sep 17 00:00:00 2001
From: NAUD Maxence <maxence.naud@cea.fr>
Date: Wed, 7 Feb 2024 13:51:13 +0000
Subject: [PATCH] Add tag to ReduceMean test
---
unit_tests/operator/Test_ReduceMeanImpl.cpp | 6 +++---
1 file changed, 3 insertions(+), 3 deletions(-)
diff --git a/unit_tests/operator/Test_ReduceMeanImpl.cpp b/unit_tests/operator/Test_ReduceMeanImpl.cpp
index 6a5166c7..2cde38d6 100644
--- a/unit_tests/operator/Test_ReduceMeanImpl.cpp
+++ b/unit_tests/operator/Test_ReduceMeanImpl.cpp
@@ -20,7 +20,7 @@
using namespace Aidge;
-TEST_CASE("[cpu/operator] ReduceMean(forward)") {
+TEST_CASE("[cpu/operator] ReduceMean(forward)", "[ReduceMean][CPU]") {
SECTION("KeepDims") {
std::shared_ptr<Tensor> myInput = std::make_shared<Tensor>(Array3D<float,3,2,2> {
{
@@ -76,7 +76,7 @@ TEST_CASE("[cpu/operator] ReduceMean(forward)") {
}
});
std::shared_ptr<Tensor> myOutput = std::make_shared<Tensor>(Array2D<float,3,2> {
- {
+ {
{ 12.5, 1.5 },
{ 35.0, 1.5 },
{ 57.5, 1.5 }
@@ -124,7 +124,7 @@ TEST_CASE("[cpu/operator] ReduceMean(forward)") {
op->computeOutputDims();
myReduceMean->forward();
op->getOutput(0)->print();
-
+
REQUIRE(*(op->getOutput(0)) == *myOutput);
}
}
\ No newline at end of file
--
GitLab