Try add eval() to speedup the eigen performance. (#48855)

PaddlePaddle · Dec 8, 2022 · e89a50c · e89a50c
1 parent d3e9e73
commit e89a50c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/paddle/phi/kernels/impl/logsumexp_kernel_impl.h b/paddle/phi/kernels/impl/logsumexp_kernel_impl.h
@@ -51,7 +51,7 @@ struct LogsumexpFunctor {
 
     auto x_mt = (*x).template cast<MT>();
     auto y_dim = y->dimensions();
-    auto x_max = x_mt.maximum(dim);
+    auto x_max = x_mt.maximum(dim).eval();
     y->device(place) =
         (x_max +
          (x_mt - x_max.reshape(t_dim).broadcast(r_dim)).exp().sum(dim).log())