Fix warning on macro expansion using defined.

warning: macro expansion producing 'defined' has undefined behavior [-Wexpansion-to-defined]
apache · Apr 9, 2019 · ad3df98 · ad3df98
1 parent 733e54c
commit ad3df98
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>