This is an automated email from the ASF dual-hosted git repository.
patriczhao pushed a commit to branch mkldnn-v1.0
in repository https://gitbox.apache.org/repos/asf/incubator-mxnet.git
The following commit(s) were added to refs/heads/mkldnn-v1.0 by this push:
new 9f77575 [mkldnn-v1.0] Update enabling flag for MKL dropout (#16433)
9f77575 is described below
commit 9f77575c7aeaf0643dee88ae0b18a00b9a62d671
Author: Wuxun Zhang <[email protected]>
AuthorDate: Sun Oct 13 17:17:21 2019 +0800
[mkldnn-v1.0] Update enabling flag for MKL dropout (#16433)
* use MSHADOW_USE_MKL to determine whther to use mkl optimized dropout
* rebase code
---
src/operator/nn/dropout-inl.h | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/src/operator/nn/dropout-inl.h b/src/operator/nn/dropout-inl.h
index cb1cd42..e8ee98c 100644
--- a/src/operator/nn/dropout-inl.h
+++ b/src/operator/nn/dropout-inl.h
@@ -39,7 +39,7 @@
#include "../random/sampler.h"
#include "../tensor/elemwise_binary_broadcast_op.h"
-#if defined(USE_MKL) && defined(_OPENMP) && !defined(__CUDACC__)
+#if (MSHADOW_USE_MKL == 1) && defined(_OPENMP) && !defined(__CUDACC__)
#define MXNET_USE_MKL_DROPOUT 1
#endif
