提交 93f6f441 authored 作者: carriepl's avatar carriepl 提交者: Frederic

Define new V4 elements from V3 elements in cudnn_helper (gpua backend)

上级 674cd4f1
......@@ -125,5 +125,14 @@ static inline cudnnStatus_t cudnnSetConvolutionNdDescriptor_v3(
#endif
// If needed, define element of the V4 interface in terms of elements of
// previous versions
#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
#endif
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论