Fix warning on macro expansion using defined.

warning: macro expansion producing 'defined' has undefined behavior [-Wexpansion-to-defined]
apache · Apr 5, 2019 · ed0dfd5 · ed0dfd5
1 parent d5d1d7a
commit ed0dfd5
Showing 1 changed file with 4 additions and 1 deletion.
diff --git 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 <omp.h>