Companion PR for 'SymIntify _copy functionalization kernels (and _copy_out too)' (#4164)

* Companion PR for 'SymIntify _copy functionalization kernels (and _copy_out too)'

Upstream at pytorch/pytorch#88572

Signed-off-by: Edward Z. Yang <[email protected]>

* fix lint

Signed-off-by: Edward Z. Yang <[email protected]>

* Delete .torch_pin

Signed-off-by: Edward Z. Yang <[email protected]>

Author: ezyang

torch_xla/csrc/aten_xla_type.cpp

@@ -3067,9 +3067,11 @@ at::Tensor XLANativeFunctions::new_empty_strided_symint(
                                               device, pin_memory);
 }
 
-at::Tensor XLANativeFunctions::narrow_copy(const at::Tensor& self, int64_t dim,
-                                           int64_t start, int64_t length) {
-  return at::native::narrow_copy_dense(self, dim, start, length);
+at::Tensor XLANativeFunctions::narrow_copy_symint(const at::Tensor& self,
+                                                  int64_t dim,
+                                                  c10::SymInt start,
+                                                  c10::SymInt length) {
+  return at::native::narrow_copy_dense_symint(self, dim, start, length);
 }
 at::Tensor XLANativeFunctions::pixel_shuffle(const at::Tensor& self,
                                              int64_t upscale_factor) {

xla_native_functions.yaml

@@ -355,6 +355,7 @@ symint:
   - new_empty_strided
   - view
   - diagonal_backward
+  - narrow_copy
 autograd:
   - einsum
   - max_pool2d