提交 5a015a8d authored 作者: carriepl's avatar carriepl 提交者: Frederic

Add definition for new implementation in cudnn_helper.h (cuda backend)

上级 330da07d
......@@ -31,6 +31,7 @@ static inline cudnnStatus_t cudnnSetConvolutionNdDescriptor_v3(
#if defined(CUDNN_VERSION) && CUDNN_VERSION < 4000
#define CUDNN_CONVOLUTION_FWD_ALGO_FFT_TILING 5
#define CUDNN_CONVOLUTION_BWD_FILTER_ALGO_3 3
#define CUDNN_CONVOLUTION_BWD_DATA_ALGO_FFT_TILING 3
#endif
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论