change macro, can use omp when paddle use mklml

8 years ago · 40450401a6
parent 00a51557ee
commit 40450401a6
1 changed files with 1 additions and 1 deletions
--- a/paddle/parameter/ParameterUpdateFunctions.cpp
+++ b/paddle/parameter/ParameterUpdateFunctions.cpp
@ -30,7 +30,7 @@ void sgdUpdateCpu(real learningRate,
                  const real* grad,
                  real* momentumVec) {
  decayRate *= learningRate;
-#ifdef PADDLE_USE_MKLDNN
+#ifdef PADDLE_USE_MKLML
 #pragma omp parallel for
 #endif
  for (size_t i = 0; i < size; ++i) {