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

Commit 32c3d8c

Browse files
committed
use MSHADOW_USE_MKL to determine whther to use mkl optimized dropout
1 parent 458bb73 commit 32c3d8c

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)