diff --git a/unit_tests/Test_Convert.cpp b/unit_tests/Test_Convert.cpp index 82256da21113cb1acc8f58da62ed5483c12802ca..21f9d43fa94b541809f23356c68ca624d3434eb4 100644 --- a/unit_tests/Test_Convert.cpp +++ b/unit_tests/Test_Convert.cpp @@ -114,12 +114,12 @@ TEST_CASE("[cuda/convert] Convert(forward)") { std::shared_ptr<Tensor> other1 = std::static_pointer_cast<OperatorTensor>(g->getNode("conv1")->getOperator())->getOutput(0); Tensor hostOther1(other1->dataType()); hostOther1.copyCastFrom(*other1); - REQUIRE(approxEq<float, int>(hostOther1, *expectedOutput1, 0.0, 1.0e-12)); + REQUIRE(approxEq<half_float::half, int>(hostOther1, *expectedOutput1, 0.0, 1.0e-12)); std::shared_ptr<Tensor> other2 = std::static_pointer_cast<OperatorTensor>(g->getNode("conv2")->getOperator())->getOutput(0); Tensor hostOther2(other2->dataType()); hostOther2.copyCastFrom(*other2); - REQUIRE(approxEq<int>(hostOther2, *expectedOutput2, 0.0, 1.0e-12)); + REQUIRE(approxEq<float, int>(hostOther2, *expectedOutput2, 0.0, 1.0e-12)); std::shared_ptr<Tensor> other3 = std::static_pointer_cast<OperatorTensor>(g->getNode("conv3")->getOperator())->getOutput(0); Tensor hostOther3(other3->dataType()); @@ -198,12 +198,12 @@ TEST_CASE("[cuda/convert] Convert(forward)") { std::shared_ptr<Tensor> other1 = std::static_pointer_cast<OperatorTensor>(g->getNode("conv1")->getOperator())->getOutput(0); Tensor hostOther1(other1->dataType()); hostOther1.copyCastFrom(*other1); - REQUIRE(approxEq<float, int>(hostOther1, *expectedOutput1, 0.0, 1.0e-12)); + REQUIRE(approxEq<half_float::half, int>(hostOther1, *expectedOutput1, 0.0, 1.0e-12)); std::shared_ptr<Tensor> other2 = std::static_pointer_cast<OperatorTensor>(g->getNode("conv2")->getOperator())->getOutput(0); Tensor hostOther2(other2->dataType()); hostOther2.copyCastFrom(*other2); - REQUIRE(approxEq<int>(hostOther2, *expectedOutput2, 0.0, 1.0e-12)); + REQUIRE(approxEq<float, int>(hostOther2, *expectedOutput2, 0.0, 1.0e-12)); std::shared_ptr<Tensor> other3 = std::static_pointer_cast<OperatorTensor>(g->getNode("conv3")->getOperator())->getOutput(0); Tensor hostOther3(other3->dataType());