提交 a28cb169 authored 作者: carriepl's avatar carriepl 提交者: Frederic

Add missing syntax

上级 757b04c0
...@@ -18,11 +18,11 @@ static inline cudnnStatus_t cudnnSetConvolutionNdDescriptor_v3( ...@@ -18,11 +18,11 @@ static inline cudnnStatus_t cudnnSetConvolutionNdDescriptor_v3(
int filterStrideA[] int filterStrideA[]
int upscaleA[], int upscaleA[],
cudnnConvolutionMode_t mode, cudnnConvolutionMode_t mode,
cudnn_dataType_t dataType) cudnn_dataType_t dataType) {
return cudnnSetConvolutionNdDescriptor(convDesc, arrayLength, padA, return cudnnSetConvolutionNdDescriptor(convDesc, arrayLength, padA,
filterStrideA, upscaleA, mode); filterStrideA, upscaleA, mode);
) }
#endif #endif
......
...@@ -118,10 +118,11 @@ static inline cudnnStatus_t cudnnSetConvolutionNdDescriptor_v3( ...@@ -118,10 +118,11 @@ static inline cudnnStatus_t cudnnSetConvolutionNdDescriptor_v3(
int filterStrideA[] int filterStrideA[]
int upscaleA[], int upscaleA[],
cudnnConvolutionMode_t mode, cudnnConvolutionMode_t mode,
cudnn_dataType_t dataType) cudnn_dataType_t dataType) {
return cudnnSetConvolutionNdDescriptor(convDesc, arrayLength, padA, return cudnnSetConvolutionNdDescriptor(convDesc, arrayLength, padA,
filterStrideA, upscaleA, mode); filterStrideA, upscaleA, mode);
}
#endif #endif
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论