diff --git a/src/ATen/native/xpu/ReduceOps.cpp b/src/ATen/native/xpu/ReduceOps.cpp index 388831282..83f1a73a7 100644 --- a/src/ATen/native/xpu/ReduceOps.cpp +++ b/src/ATen/native/xpu/ReduceOps.cpp @@ -320,7 +320,8 @@ inline TensorIterator get_allany_iter( const Tensor& result, OptionalIntArrayRef dims, bool keepdim) { - return meta::make_reduction(self, result, dims, keepdim, self.scalar_type()); + return meta::make_reduction_from_out_ty( + self, result, dims, keepdim, self.scalar_type()); } template