diff --git a/src/operator/nn/dropout-inl.h b/src/operator/nn/dropout-inl.h index 82bdda69dcd0..01611dfce191 100644 --- a/src/operator/nn/dropout-inl.h +++ b/src/operator/nn/dropout-inl.h @@ -39,7 +39,10 @@ #include "../random/sampler.h" #include "../tensor/elemwise_binary_broadcast_op.h" -#define MXNET_USE_MKL_DROPOUT defined(USE_MKL) && defined(_OPENMP) && !defined(__CUDACC__) +#if defined(USE_MKL) && defined(_OPENMP) && !defined(__CUDACC__) +#define MXNET_USE_MKL_DROPOUT 1 +#endif + #if MXNET_USE_MKL_DROPOUT #include