Skip to content
Snippets Groups Projects
Commit a898a52d authored by Houssem ROUIS's avatar Houssem ROUIS
Browse files

fix RM test

parent a28eb541
Branches
Tags
2 merge requests!93Release v0.3.0,!75Learning backend cuda
...@@ -207,12 +207,12 @@ TEST_CASE("[cpu/operator] ReduceMean(forward)", "[ReduceMean][CPU]") { ...@@ -207,12 +207,12 @@ TEST_CASE("[cpu/operator] ReduceMean(forward)", "[ReduceMean][CPU]") {
} }
}); });
std::shared_ptr<Node> myReduceSum = ReduceSum({}, 0, 1); std::shared_ptr<Node> myReduceMean = ReduceMean({}, 0, 1);
auto op = std::static_pointer_cast<OperatorTensor>(myReduceSum -> getOperator()); auto op = std::static_pointer_cast<OperatorTensor>(myReduceMean -> getOperator());
op->associateInput(0,myInput); op->associateInput(0,myInput);
op->setDataType(DataType::Float32); op->setDataType(DataType::Float32);
op->setBackend("cpu"); op->setBackend("cpu");
myReduceSum->forward(); myReduceMean->forward();
op->getOutput(0)->print(); op->getOutput(0)->print();
REQUIRE(*(op->getOutput(0)) == *myInput); REQUIRE(*(op->getOutput(0)) == *myInput);
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment