提交 2c581ec1 authored 作者: Frederic's avatar Frederic

small crash fix to the cudnn r2 rc2 fix

上级 08b61cbb
......@@ -152,7 +152,7 @@ cudnnConvolutionBackwardData_v2(
// r2 rc1 and rc2 do not have the same macro defined
// I didn't checked if this the right combination, but as we do not wrap the padding interface, it is fine for now.
# define CUDNN_POOLING_AVERAGE_COUNT_INCLUDE_PADDING 1
# define CUDNN_POOLING_AVERAGE_COUNT_INCLUDE_PADDING ((cudnnPoolingMode_t)1)
#endif
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论