You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@mxnet.apache.org by jx...@apache.org on 2017/08/21 23:15:58 UTC
[incubator-mxnet] branch master updated: Move usage of persistent
BN to cuDNN 7.0.3 (#7543)
This is an automated email from the ASF dual-hosted git repository.
jxie pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-mxnet.git
The following commit(s) were added to refs/heads/master by this push:
new ab14867 Move usage of persistent BN to cuDNN 7.0.3 (#7543)
ab14867 is described below
commit ab1486704a7e0fdf1c7b0619306f4899a6f1e8de
Author: Przemyslaw Tredak <pt...@gmail.com>
AuthorDate: Mon Aug 21 16:15:55 2017 -0700
Move usage of persistent BN to cuDNN 7.0.3 (#7543)
---
src/operator/cudnn_batch_norm-inl.h | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/src/operator/cudnn_batch_norm-inl.h b/src/operator/cudnn_batch_norm-inl.h
index b0c5f43..258bed5 100644
--- a/src/operator/cudnn_batch_norm-inl.h
+++ b/src/operator/cudnn_batch_norm-inl.h
@@ -112,7 +112,7 @@ class CuDNNBatchNormOp : public Operator {
Tensor<gpu, 4, DType> y =
out_data[cudnnbatchnorm::kOut].get_with_shape<gpu, 4, DType>(shape_, s);
-#if CUDNN_VERSION >= 7000
+#if CUDNN_VERSION >= 7003
auto mode = CUDNN_BATCHNORM_SPATIAL_PERSISTENT;
#else
auto mode = CUDNN_BATCHNORM_SPATIAL;
@@ -201,7 +201,7 @@ class CuDNNBatchNormOp : public Operator {
out_grad[cudnnbatchnorm::kOut].get_with_shape<gpu, 4, DType>(shape_, s);
#if CUDNN_VERSION >= 4007
-#if CUDNN_VERSION >= 7000
+#if CUDNN_VERSION >= 7003
auto mode = CUDNN_BATCHNORM_SPATIAL_PERSISTENT;
#else
auto mode = CUDNN_BATCHNORM_SPATIAL;
--
To stop receiving notification emails like this one, please contact
['"commits@mxnet.apache.org" <co...@mxnet.apache.org>'].