From a898a52d810d296d49c14ec6f7a9d94c3d535bb8 Mon Sep 17 00:00:00 2001
From: hrouis <houssemeddine.rouis92@gmail.com>
Date: Tue, 13 Aug 2024 12:16:34 +0200
Subject: [PATCH] fix RM 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 58ed69e7..0aed1515 100644
--- a/unit_tests/operator/Test_ReduceMeanImpl.cpp
+++ b/unit_tests/operator/Test_ReduceMeanImpl.cpp
@@ -207,12 +207,12 @@ TEST_CASE("[cpu/operator] ReduceMean(forward)", "[ReduceMean][CPU]") {
}
});
- std::shared_ptr<Node> myReduceSum = ReduceSum({}, 0, 1);
- auto op = std::static_pointer_cast<OperatorTensor>(myReduceSum -> getOperator());
+ std::shared_ptr<Node> myReduceMean = ReduceMean({}, 0, 1);
+ auto op = std::static_pointer_cast<OperatorTensor>(myReduceMean -> getOperator());
op->associateInput(0,myInput);
op->setDataType(DataType::Float32);
op->setBackend("cpu");
- myReduceSum->forward();
+ myReduceMean->forward();
op->getOutput(0)->print();
REQUIRE(*(op->getOutput(0)) == *myInput);
--
GitLab