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

Merge branch 'unified_params' into 'main'

Changes following MR

See merge request !1
parents ebef5c34 ee232ad7
No related branches found
No related tags found
1 merge request!1Changes following MR
Pipeline #32493 passed
......@@ -58,7 +58,11 @@ class TensorImpl_cuda : public TensorImpl, public TensorImpl_cuda_ {
void *rawPtr() override {
lazyInit(reinterpret_cast<void**>(&mData));
return mData;
};
}
void* getRaw(std::size_t idx) {
return static_cast<void*>(static_cast<T*>(rawPtr()) + idx);
}
const cudnnTensorDescriptor_t& getCudnnTensorDesc() const override {
if (mCudnnTensor == nullptr) {
......
......@@ -80,9 +80,9 @@ void Aidge::ConvImpl_cuda<DIM>::forward() {
// Lazy-initialize CuDNN convolution descriptor
if (mConvDesc == nullptr) {
const std::vector<int> strides(mOp.template get<ConvParam::StrideDims>().begin(), mOp.template get<ConvParam::StrideDims>().end());
const std::vector<int> strides(mOp.template getAttr<ConvAttr::StrideDims>().begin(), mOp.template getAttr<ConvAttr::StrideDims>().end());
const std::vector<int> paddings(DIM, 0);
const std::vector<int> upscales(mOp.template get<ConvParam::DilationDims>().begin(), mOp.template get<ConvParam::DilationDims>().end());
const std::vector<int> upscales(mOp.template getAttr<ConvAttr::DilationDims>().begin(), mOp.template getAttr<ConvAttr::DilationDims>().end());
CHECK_CUDNN_STATUS(cudnnCreateConvolutionDescriptor(&mConvDesc));
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