diff --git a/unit_tests/Test_Convert.cpp b/unit_tests/Test_Convert.cpp index aff855f372fbc8f7e41c05c27476b1ab0dbca9e0..82256da21113cb1acc8f58da62ed5483c12802ca 100644 --- a/unit_tests/Test_Convert.cpp +++ b/unit_tests/Test_Convert.cpp @@ -75,8 +75,8 @@ TEST_CASE("[cuda/convert] Convert(forward)") { g->getNode("conv3")->getOperator()->setInput(2, bias3); // input->addChild(g); - g->setDataType(Aidge::DataType::Int32); - g->getNode("conv1")->getOperator()->setDataType(DataType::Float32); + g->setDataType(Aidge::DataType::Float32); + g->getNode("conv1")->getOperator()->setDataType(DataType::Float16); g->getNode("conv3")->getOperator()->setDataType(DataType::Float64); g->setBackend("cuda"); @@ -157,8 +157,8 @@ TEST_CASE("[cuda/convert] Convert(forward)") { g->getNode("conv3")->getOperator()->setInput(2, bias3); // input->addChild(g); - g->setDataType(Aidge::DataType::Int32); - g->getNode("conv1")->getOperator()->setDataType(DataType::Float32); + g->setDataType(Aidge::DataType::Float32); + g->getNode("conv1")->getOperator()->setDataType(DataType::Float16); g->getNode("conv3")->getOperator()->setDataType(DataType::Float64); explicitConvert(g);