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

Fixed missing template parameter

parent ad0c5db0
No related branches found
No related tags found
1 merge request!3Update with default operator impl
Pipeline #33246 passed
......@@ -30,7 +30,7 @@ void Aidge::ConvImpl_cuda<DIM>::forward() {
// Lazy-initialize CuDNN convolution descriptor
if (mConvDesc == nullptr) {
const Conv_Op& convOp = static_cast<const Conv_Op&>(mOp);
const Conv_Op<DIM>& convOp = static_cast<const Conv_Op<DIM>&>(mOp);
const std::vector<int> strides(convOp.template getAttr<ConvAttr::StrideDims>().begin(), convOp.template getAttr<ConvAttr::StrideDims>().end());
const std::vector<int> paddings(DIM, 0);
const std::vector<int> upscales(convOp.template getAttr<ConvAttr::DilationDims>().begin(), convOp.template getAttr<ConvAttr::DilationDims>().end());
......
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