You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@mxnet.apache.org by ma...@apache.org on 2018/03/31 14:26:26 UTC
[incubator-mxnet] branch master updated: [MXNET-240] Fix CUDA
compile error (#10302)
This is an automated email from the ASF dual-hosted git repository.
marcoabreu 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 453e6e9 [MXNET-240] Fix CUDA compile error (#10302)
453e6e9 is described below
commit 453e6e9ee1b3388a7b15f32f3aae7a1957b5409b
Author: Da Zheng <zh...@gmail.com>
AuthorDate: Sat Mar 31 07:26:21 2018 -0700
[MXNET-240] Fix CUDA compile error (#10302)
* Fix a compile error.
* Disable CuDNN on CentOS.
* Update runtime_functions.sh
---
src/operator/nn/batch_norm.cu | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/src/operator/nn/batch_norm.cu b/src/operator/nn/batch_norm.cu
index 703ed39..03962cb 100644
--- a/src/operator/nn/batch_norm.cu
+++ b/src/operator/nn/batch_norm.cu
@@ -708,7 +708,7 @@ void BatchNormGradCompute<gpu>(const nnvm::NodeAttrs& attrs,
})
}
#else
- MSHADOW_REAL_TYPE_SWITCH_EX(out_grad[0].type_flag_, DType, AccReal, {
+ MSHADOW_REAL_TYPE_SWITCH_EX(dtype, DType, AccReal, {
BatchNormBackward<gpu, DType, AccReal>(ctx, param, inputs, req, outputs);
});
#endif
--
To stop receiving notification emails like this one, please contact
marcoabreu@apache.org.