Merge branch 'main' into chunk-cat

intel · Nov 19, 2024 · ae4308c · ae4308c
2 parents 131e546 + 000d343
commit ae4308c
Show file tree

Hide file tree

Showing 5 changed files with 817 additions and 837 deletions.
diff --git a/src/ATen/native/xpu/ForeachReduceOp.cpp b/src/ATen/native/xpu/ForeachReduceOp.cpp
@@ -61,7 +61,8 @@ std::vector<Tensor> foreach_tensor_norm_xpu(
             at::isComplexType(scalar_type);
       });
   if (!at::native::can_use_fast_route(tensors) || has_int_or_complex ||
-      !(p == static_cast<double>(1) || p == static_cast<double>(2))) {
+      !(p == static_cast<double>(1) || p == static_cast<double>(2) ||
+        p == std::numeric_limits<double>::infinity())) {
     return at::native::foreach_tensor_norm_slow(tensors, ord, dtype);
   }
   check_foreach_norm_dtype(