Skip to content

Instantly share code, notes, and snippets.

@prabindh

prabindh/convolutional_layer.c

Created August 22, 2017 01:53
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save prabindh/6e63fd67816fe63991595d25b931ab04 to your computer and use it in GitHub Desktop.
Save prabindh/6e63fd67816fe63991595d25b931ab04 to your computer and use it in GitHub Desktop.
Download ZIP
Fix for CUDA 8 build error
Raw
convolutional_layer.c
#if CUDNN_MAJOR >= 6
cudnnSetConvolution2dDescriptor(l->convDesc, l->pad, l->pad, l->stride, l->stride, 1, 1, CUDNN_CROSS_CORRELATION, CUDNN_DATA_FLOAT);
#else
cudnnSetConvolution2dDescriptor(l->convDesc, l->pad, l->pad, l->stride, l->stride, 1, 1, CUDNN_CROSS_CORRELATION);
#endif
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment