AojunZhou / Incremental-Network-Quantization

Caffe Implementation for Incremental network quantization
Other
191 stars 74 forks source link

For CuDNN higher version support #33

Open mdamircoder opened 6 years ago

mdamircoder commented 6 years ago

Check for setConvolutionDesc() in cudnn.hpp It should be

inline void setConvolutionDesc(cudnnConvolutionDescriptor_t* conv, cudnnTensorDescriptor_t bottom, cudnnFilterDescriptor_t filter, int pad_h, int pad_w, int stride_h, int stride_w) {

if CUDNN_VERSION_MIN(6, 0, 0) //Added from here (amir)

// For CuDNN higher version

CUDNN_CHECK(cudnnSetConvolution2dDescriptor(*conv, pad_h, pad_w, stride_h, stride_w, 1, 1, CUDNN_CROSS_CORRELATION, dataType::type));

else

CUDNN_CHECK(cudnnSetConvolution2dDescriptor(*conv,
  pad_h, pad_w, stride_h, stride_w, 1, 1, CUDNN_CROSS_CORRELATION));

endif //Added upto this (amir)

}