commit 6185478edbe3b7cb58dd18f7479ef6d0baba031e
author Brian Hirsh <hirsheybar@fb.com> Tue May 24 08:30:36 2022 -0700
committer PyTorch MergeBot <pytorchmergebot@users.noreply.github.com> Tue May 24 19:56:43 2022 +0000
tree 4ddccbca62549d3ff2fe192726af8c5ef5f00d68
parent 2eea5eff622df108f2f77dda61ddec3a62b51e51

make logsumexp composite compliant

Pull Request resolved: https://github.com/pytorch/pytorch/pull/77130

 composite

Approved by: https://github.com/ezyang

aten/src/ATen/native/ReduceOps.cpp

diff --git a/aten/src/ATen/native/ReduceOps.cpp b/aten/src/ATen/native/ReduceOps.cpp
index 016d532..2015c16 100644
--- a/aten/src/ATen/native/ReduceOps.cpp
+++ b/aten/src/ATen/native/ReduceOps.cpp
@@ -1331,7 +1331,7 @@
     result_options = self.options();
   }
   auto result = at::empty({0}, result_options);
-  return at::native::logsumexp_out(self, dims, keepdim, result);
+  return at::logsumexp_outf(self, dims, keepdim, result);
 }
 
 Tensor logsumexp(const Tensor& self, DimnameList dims, bool keepdim) {