revise

intel · Jul 22, 2024 · 1060b79 · 1060b79
1 parent 2e064ef
commit 1060b79
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/src/ATen/native/xpu/ReduceOps.cpp b/src/ATen/native/xpu/ReduceOps.cpp
@@ -290,9 +290,8 @@ Tensor XPUNativeFunctions::prod(
     const Tensor& self,
     std::optional<ScalarType> opt_dtype) {
   auto dtype = at::native::get_dtype_from_self(self, opt_dtype, true);
-  // auto shape = at::native::meta::get_reduction_shape(self, {}, false);
-  // Tensor out = at::empty(shape, self.options().dtype(dtype));
-  Tensor out;
+  auto shape = at::meta::get_reduction_shape(self, {}, false);
+  Tensor out = at::empty(shape, self.options().dtype(dtype));
   out = prod_meta(self, {}, false, dtype, out);
   return out;
 }