зеркало из 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) {
|
||||
CUDNN_CALL(cudnnCreateConvolutionDescriptor(&convDesc_));
|
||||
|
||||
#if CUDNN_MAJOR > 5
|
||||
CUDNN_CALL(cudnnSetConvolution2dDescriptor(convDesc_,
|
||||
hPad,
|
||||
wPad,
|
||||
|
@ -177,10 +178,16 @@ void ConvolutionWrapper::setConvDescriptor(int hPad,
|
|||
wStride,
|
||||
1,
|
||||
1, // upscales
|
||||
#if CUDNN_MAJOR > 5
|
||||
CUDNN_CROSS_CORRELATION,
|
||||
CUDNN_DATA_FLOAT));
|
||||
#else
|
||||
CUDNN_CALL(cudnnSetConvolution2dDescriptor(convDesc_,
|
||||
hPad,
|
||||
wPad,
|
||||
hStride,
|
||||
wStride,
|
||||
1,
|
||||
1, // upscales
|
||||
CUDNN_CROSS_CORRELATION));
|
||||
#endif
|
||||
}
|
||||
|
|
Загрузка…
Ссылка в новой задаче