Skip to content
Snippets Groups Projects
Commit fd753f2c authored by Olivier BICHLER's avatar Olivier BICHLER
Browse files

Merge branch 'dev' into forwarddims

parents 00a1e999 4a03726d
No related branches found
No related tags found
2 merge requests!32version 0.2.1,!16Make forwardDims() optional and handle data dependency
Pipeline #44283 failed
...@@ -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