Fixed types

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());