Skip to content
Snippets Groups Projects
Commit c215a4ca authored by Maxence Naud's avatar Maxence Naud
Browse files

[Upd] Slice test

parent a679ca4a
No related branches found
No related tags found
2 merge requests!22Update operators implementation,!16Draft: Tiling
Pipeline #32951 failed
...@@ -33,8 +33,10 @@ TEST_CASE("[cpu/operator] Slice(forward)", "[Slice]") { ...@@ -33,8 +33,10 @@ TEST_CASE("[cpu/operator] Slice(forward)", "[Slice]") {
mySlice->getOperator()->associateInput(0,input0); mySlice->getOperator()->associateInput(0,input0);
mySlice->getOperator()->computeOutputDims(); mySlice->getOperator()->computeOutputDims();
mySlice->forward(); mySlice->forward();
mySlice->getOperator()->output(0).print(); // mySlice->getOperator()->output(0).print();
REQUIRE(mySlice->getOperator()->output(0) == *expectedOutput); REQUIRE(mySlice->getOperator()->output(0) == *expectedOutput);
REQUIRE(mySlice->getOperator()->output(0).dims() == expectedOutput->dims());
REQUIRE(mySlice->getOperator()->output(0).dataType() == expectedOutput->dataType());
} }
SECTION("2D Tensor") { SECTION("2D Tensor") {
...@@ -57,8 +59,10 @@ TEST_CASE("[cpu/operator] Slice(forward)", "[Slice]") { ...@@ -57,8 +59,10 @@ TEST_CASE("[cpu/operator] Slice(forward)", "[Slice]") {
mySlice->getOperator()->associateInput(0,input0); mySlice->getOperator()->associateInput(0,input0);
mySlice->getOperator()->computeOutputDims(); mySlice->getOperator()->computeOutputDims();
mySlice->forward(); mySlice->forward();
mySlice->getOperator()->output(0).print(); // mySlice->getOperator()->output(0).print();
REQUIRE(*mySlice->getOperator()->getOutput(0) == *expectedOutput); REQUIRE(*mySlice->getOperator()->getOutput(0) == *expectedOutput);
REQUIRE(mySlice->getOperator()->output(0).dims() == expectedOutput->dims());
REQUIRE(mySlice->getOperator()->output(0).dataType() == expectedOutput->dataType());
} }
SECTION("3D Tensor") { SECTION("3D Tensor") {
...@@ -88,8 +92,10 @@ TEST_CASE("[cpu/operator] Slice(forward)", "[Slice]") { ...@@ -88,8 +92,10 @@ TEST_CASE("[cpu/operator] Slice(forward)", "[Slice]") {
mySlice->getOperator()->associateInput(0,input0); mySlice->getOperator()->associateInput(0,input0);
mySlice->getOperator()->computeOutputDims(); mySlice->getOperator()->computeOutputDims();
mySlice->forward(); mySlice->forward();
mySlice->getOperator()->output(0).print(); // mySlice->getOperator()->output(0).print();
REQUIRE(mySlice->getOperator()->output(0) == *expectedOutput); REQUIRE(mySlice->getOperator()->output(0) == *expectedOutput);
REQUIRE(mySlice->getOperator()->output(0).dims() == expectedOutput->dims());
REQUIRE(mySlice->getOperator()->output(0).dataType() == expectedOutput->dataType());
} }
SECTION("4D Tensor") { SECTION("4D Tensor") {
...@@ -148,7 +154,9 @@ TEST_CASE("[cpu/operator] Slice(forward)", "[Slice]") { ...@@ -148,7 +154,9 @@ TEST_CASE("[cpu/operator] Slice(forward)", "[Slice]") {
mySlice->getOperator()->associateInput(0,input0); mySlice->getOperator()->associateInput(0,input0);
mySlice->getOperator()->computeOutputDims(); mySlice->getOperator()->computeOutputDims();
mySlice->forward(); mySlice->forward();
mySlice->getOperator()->output(0).print(); // mySlice->getOperator()->output(0).print();
REQUIRE(mySlice->getOperator()->output(0) == *expectedOutput); REQUIRE(mySlice->getOperator()->output(0) == *expectedOutput);
REQUIRE(mySlice->getOperator()->output(0).dims() == expectedOutput->dims());
REQUIRE(mySlice->getOperator()->output(0).dataType() == expectedOutput->dataType());
} }
} }
\ No newline at end of file
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment