Skip to content
This repository was archived by the owner on Nov 17, 2023. It is now read-only.

Commit 9f77575

Browse files
wuxun-zhangpengzhao-intel
authored andcommitted
[mkldnn-v1.0] Update enabling flag for MKL dropout (#16433)
* use MSHADOW_USE_MKL to determine whther to use mkl optimized dropout * rebase code
1 parent 2b363a0 commit 9f77575

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/operator/nn/dropout-inl.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -39,7 +39,7 @@
3939
#include "../random/sampler.h"
4040
#include "../tensor/elemwise_binary_broadcast_op.h"
4141

42-
#if defined(USE_MKL) && defined(_OPENMP) && !defined(__CUDACC__)
42+
#if (MSHADOW_USE_MKL == 1) && defined(_OPENMP) && !defined(__CUDACC__)
4343
#define MXNET_USE_MKL_DROPOUT 1
4444
#endif
4545

0 commit comments

Comments
 (0)