Fix warning on macro expansion using defined. (#14598)

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