Skip to content
Snippets Groups Projects
Commit 6605927b authored by Olivier BICHLER's avatar Olivier BICHLER Committed by Cyril Moineau
Browse files

Reverted erroneous change: please read the comment and CuDNN doc!

parent dc0dcb88
No related branches found
No related tags found
1 merge request!200.2.1
...@@ -102,18 +102,11 @@ void Aidge::ConvImpl_cuda<DIM>::forward() { ...@@ -102,18 +102,11 @@ void Aidge::ConvImpl_cuda<DIM>::forward() {
// Do the actual forward computation // Do the actual forward computation
// Template is only for scaling parameters, which are always in float // Template is only for scaling parameters, which are always in float
// excepted when the convolution is performed in double precision. // excepted when the convolution is performed in double precision.
switch(op.getOutput(0)->dataType()) { if (op.getOutput(0)->dataType() == DataType::Float64) {
case DataType::Float64: forward_<double>(input0, input1, input2);
forward_<double>(input0, input1, input2); }
break; else {
case DataType::Float32: forward_<float>(input0, input1, input2);
forward_<float>(input0, input1, input2);
break;
case DataType::Float16:
forward_<half>(input0, input1, input2);
break;
default:
AIDGE_THROW_OR_ABORT(std::runtime_error, "Data type is not supported by Backend Cuda");
} }
} }
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment