fix the bug when the dim of slice is -1 (#26)

AlibabaPAI · Dec 11, 2024 · 294a91c · 294a91c
1 parent 0a684a8
commit 294a91c
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/torch_xla/csrc/aten_xla_type.cpp b/torch_xla/csrc/aten_xla_type.cpp
@@ -2855,6 +2855,8 @@ at::Tensor XLANativeFunctions::slice_copy_symint(
     const at::Tensor& self, int64_t dim, c10::optional<c10::SymInt> start,
     c10::optional<c10::SymInt> end, c10::SymInt step) {
   TORCH_LAZY_FN_COUNTER_TIMED_TRACING("xla::");
+  // dim maybe -1
+  dim = torch::lazy::GetCanonicalDimensionIndex(dim, self.dim());
   c10::SymInt start_val = start.has_value() ? start.value() : 0;
   c10::SymInt end_val = end.has_value() ? end.value() : self.sym_sizes()[dim];
   if (!start_val.is_symbolic() && start_val < 0) {