diff --git a/unit_tests/Test_Convert.cpp b/unit_tests/Test_Convert.cpp index 109763046f659d987a1190bcdef4afb28ca3d862..74dd68ddd0d56a9a03e0a258ed7ab0b9a158f8de 100644 --- a/unit_tests/Test_Convert.cpp +++ b/unit_tests/Test_Convert.cpp @@ -115,19 +115,19 @@ TEST_CASE("[cuda/convert] Convert(forward)") { Tensor hostOther1(other1->dataType()); hostOther1.setBackend("cpu"); hostOther1.copyCastFrom(*other1); - REQUIRE(approxEq<half_float::half, int>(hostOther1, *expectedOutput1, 0.0, 1.0e-12)); + REQUIRE(approxEq<half_float::half, int>(hostOther1, *expectedOutput1, 0.001, 0.0)); std::shared_ptr<Tensor> other2 = std::static_pointer_cast<OperatorTensor>(g->getNode("conv2")->getOperator())->getOutput(0); Tensor hostOther2(other2->dataType()); hostOther2.setBackend("cpu"); hostOther2.copyCastFrom(*other2); - REQUIRE(approxEq<float, int>(hostOther2, *expectedOutput2, 0.0, 1.0e-12)); + REQUIRE(approxEq<float, int>(hostOther2, *expectedOutput2, 0.001, 0.0)); std::shared_ptr<Tensor> other3 = std::static_pointer_cast<OperatorTensor>(g->getNode("conv3")->getOperator())->getOutput(0); Tensor hostOther3(other3->dataType()); hostOther3.setBackend("cpu"); hostOther3.copyCastFrom(*other3); - REQUIRE(approxEq<double, int>(hostOther3, *expectedOutput3, 0.0, 1.0e-12)); + REQUIRE(approxEq<double, int>(hostOther3, *expectedOutput3, 0.001, 0.0)); } SECTION("Test explicit") { @@ -202,18 +202,18 @@ TEST_CASE("[cuda/convert] Convert(forward)") { Tensor hostOther1(other1->dataType()); hostOther1.setBackend("cpu"); hostOther1.copyCastFrom(*other1); - REQUIRE(approxEq<half_float::half, int>(hostOther1, *expectedOutput1, 0.0, 1.0e-12)); + REQUIRE(approxEq<half_float::half, int>(hostOther1, *expectedOutput1, 0.001, 0.0)); std::shared_ptr<Tensor> other2 = std::static_pointer_cast<OperatorTensor>(g->getNode("conv2")->getOperator())->getOutput(0); Tensor hostOther2(other2->dataType()); hostOther2.setBackend("cpu"); hostOther2.copyCastFrom(*other2); - REQUIRE(approxEq<float, int>(hostOther2, *expectedOutput2, 0.0, 1.0e-12)); + REQUIRE(approxEq<float, int>(hostOther2, *expectedOutput2, 0.001, 0.0)); std::shared_ptr<Tensor> other3 = std::static_pointer_cast<OperatorTensor>(g->getNode("conv3")->getOperator())->getOutput(0); Tensor hostOther3(other3->dataType()); hostOther3.setBackend("cpu"); hostOther3.copyCastFrom(*other3); - REQUIRE(approxEq<double, int>(hostOther3, *expectedOutput3, 0.0, 1.0e-12)); + REQUIRE(approxEq<double, int>(hostOther3, *expectedOutput3, 0.001, 0.0)); } }