зеркало из https://github.com/mozilla/marian.git
Fix fatal error C1019: unexpected #else
This commit is contained in:
Родитель
112118fce3
Коммит
5d41dff700
|
@ -170,6 +170,7 @@ void ConvolutionWrapper::setConvDescriptor(int hPad,
|
||||||
int wStride) {
|
int wStride) {
|
||||||
CUDNN_CALL(cudnnCreateConvolutionDescriptor(&convDesc_));
|
CUDNN_CALL(cudnnCreateConvolutionDescriptor(&convDesc_));
|
||||||
|
|
||||||
|
#if CUDNN_MAJOR > 5
|
||||||
CUDNN_CALL(cudnnSetConvolution2dDescriptor(convDesc_,
|
CUDNN_CALL(cudnnSetConvolution2dDescriptor(convDesc_,
|
||||||
hPad,
|
hPad,
|
||||||
wPad,
|
wPad,
|
||||||
|
@ -177,10 +178,16 @@ void ConvolutionWrapper::setConvDescriptor(int hPad,
|
||||||
wStride,
|
wStride,
|
||||||
1,
|
1,
|
||||||
1, // upscales
|
1, // upscales
|
||||||
#if CUDNN_MAJOR > 5
|
|
||||||
CUDNN_CROSS_CORRELATION,
|
CUDNN_CROSS_CORRELATION,
|
||||||
CUDNN_DATA_FLOAT));
|
CUDNN_DATA_FLOAT));
|
||||||
#else
|
#else
|
||||||
|
CUDNN_CALL(cudnnSetConvolution2dDescriptor(convDesc_,
|
||||||
|
hPad,
|
||||||
|
wPad,
|
||||||
|
hStride,
|
||||||
|
wStride,
|
||||||
|
1,
|
||||||
|
1, // upscales
|
||||||
CUDNN_CROSS_CORRELATION));
|
CUDNN_CROSS_CORRELATION));
|
||||||
#endif
|
#endif
|
||||||
}
|
}
|
||||||
|
|
Загрузка…
Ссылка в новой задаче