Fix 0-sized size and stride case.

pytorch · Feb 28, 2024 · 9be6084 · 9be6084
1 parent f2189d3
commit 9be6084
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/torch_xla/csrc/aten_xla_type.cpp b/torch_xla/csrc/aten_xla_type.cpp
@@ -711,8 +711,7 @@ at::Tensor XLANativeFunctions::as_strided_copy(
   if (dim == 0 && tensor.numel() > 0) {
     // If there's no specified dimension, return the first element of the
     // storage. This behavior is consistent with eager.
-    return take(tensor,
-                at::tensor({0}, at::TensorOptions().device(tensor.device())));
+    return select_copy(view_copy_symint(tensor, {tensor.numel()}), 0, 0);
   }
 
   if (storage_size == 0) {