code optimize

test=develop
6 years ago · 936926aadd
parent 9fd78df71c
commit 936926aadd
1 changed files with 1 additions and 1 deletions
--- a/paddle/fluid/operators/math/selected_rows_functor.cc
+++ b/paddle/fluid/operators/math/selected_rows_functor.cc
@ -300,11 +300,11 @@ struct MergeAdd<platform::CPUDeviceContext, T> {

    auto* out_data = out.mutable_value()->data<T>();

+    auto blas = math::GetBlas<platform::CPUDeviceContext, T>(context);
    for (auto* input : inputs) {
      auto* input_data = input->value().data<T>();
      auto& input_rows = input->rows();

-      auto blas = math::GetBlas<platform::CPUDeviceContext, T>(context);
      for (size_t i = 0; i < input_rows.size(); i++) {
        size_t out_i = rows_to_id[input_rows[i]];
        elementwise_add<platform::CPUDeviceContext, T>(