From 4e00cee05319b6d7a9c4641c7353eb29969ff870 Mon Sep 17 00:00:00 2001
From: Olivier BICHLER <olivier.bichler@cea.fr>
Date: Sun, 10 Dec 2023 19:03:57 +0100
Subject: [PATCH] Fixed unsupported CuDNN type
---
unit_tests/Test_Convert.cpp | 8 ++++----
1 file changed, 4 insertions(+), 4 deletions(-)
diff --git a/unit_tests/Test_Convert.cpp b/unit_tests/Test_Convert.cpp
index aff855f..82256da 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);
--
GitLab