Align the semantic of accumulate type in the renorm operator with sto…

…ck PyTorch (#823) Align the semantic of accumulate type in the renorm operator with stock PyTorch.
intel · Aug 29, 2024 · 4eddbe9 · 4eddbe9
1 parent 52c8079
commit 4eddbe9
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/ATen/native/xpu/Normalization.cpp b/src/ATen/native/xpu/Normalization.cpp
@@ -47,7 +47,10 @@ Tensor& renorm_impl(
   reduce_dims.erase(reduce_dims.begin() + dim);
 
   auto dtype = self.scalar_type();
-  auto acc_type = at::toAccumulateType(dtype, c10::DeviceType::XPU);
+
+  // This is a device-independent accumulate type, and we follow PyTorch's design.
+  auto acc_type = at::toAccumulateType(dtype, true); 
+
   Tensor norm;
   if (acc_type != dtype) {
     norm = at::linalg_vector_norm(