Skip to content
Snippets Groups Projects
Commit e8564e81 authored by Houssem ROUIS's avatar Houssem ROUIS
Browse files

Merge branch 'dev' of gitlab.eclipse.org:eclipse/aidge/aidge_backend_cuda into mobilenet_operators

parents 01340ecd 4a03726d
No related branches found
No related tags found
2 merge requests!32version 0.2.1,!14MobileNet operators
......@@ -111,18 +111,11 @@ void Aidge::ConvImpl_cuda<DIM>::forward() {
// Do the actual forward computation
// Template is only for scaling parameters, which are always in float
// excepted when the convolution is performed in double precision.
switch(op.getOutput(0)->dataType()) {
case DataType::Float64:
forward_<double>(input0, input1, input2);
break;
case DataType::Float32:
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");
if (op.getOutput(0)->dataType() == DataType::Float64) {
forward_<double>(input0, input1, input2);
}
else {
forward_<float>(input0, input1, input2);
}
}
......
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