diff --git a/paddle/fluid/operators/math/fc.cc b/paddle/fluid/operators/math/fc.cc index 27a75f7631aa50141de57da3ba4d6a71dd1b5ed5..9309a992f73a0951d907ed93437c9ef762820e85 100644 --- a/paddle/fluid/operators/math/fc.cc +++ b/paddle/fluid/operators/math/fc.cc @@ -61,27 +61,20 @@ class FCFunctor { "When bias is NULL, relu can not be true.")); return; } - if (relu) { - auto compute = - jit::KernelFuncs, platform::CPUPlace>::Cache() - .At(N); - for (int i = 0; i < M; i++) { - T* dst = Y + i * N; - T* src = (padding_weights) ? Y1_data + i * (N + 4) : dst; - compute(B, src, dst, N); - } - } else { - auto compute = - jit::KernelFuncs, platform::CPUPlace>::Cache().At( - N); + auto compute = + relu + ? jit::KernelFuncs, + platform::CPUPlace>::Cache() + .At(N) + : jit::KernelFuncs, platform::CPUPlace>::Cache() + .At(N); #ifdef PADDLE_WITH_MKLML #pragma omp parallel for #endif - for (int i = 0; i < M; i++) { - T* dst = Y + i * N; - T* src = (padding_weights) ? Y1_data + i * (N + 4) : dst; - compute(B, src, dst, N); - } + for (int i = 0; i < M; i++) { + T* dst = Y + i * N; + T* src = (padding_weights) ? Y1_data + i * (N + 4) : dst; + compute(B, src, dst, N); } } };