use MSHADOW_USE_MKL to determine whther to use mkl optimized dropout

apache · Oct 11, 2019 · 32c3d8c · 32c3d8c
1 parent 458bb73
commit 32c3d8c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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