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

Fixed getAttr

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