commit 69f7b409495e83a10ee86ecabe92c77210929651
author Nikita Karetnikov <nikita@karetnikov.org> Tue May 23 12:03:14 2023 +0200
committer PyTorch MergeBot <pytorchmergebot@users.noreply.github.com> Wed May 24 00:21:52 2023 +0000
tree cf461ef7a34bd813dfc7e9a81a4c1b48ff3faa40
parent 42b974e8f7fe14610ba0df55dcbb257de9961890

[pt2] add `SymInt` support for `eye` (#101955)

Pull Request resolved: https://github.com/pytorch/pytorch/pull/101955
Approved by: https://github.com/Skylion007, https://github.com/ezyang

aten/src/ATen/native/native_functions.yaml

diff --git a/aten/src/ATen/native/native_functions.yaml b/aten/src/ATen/native/native_functions.yaml
index 1072cb2..c59104c 100644
--- a/aten/src/ATen/native/native_functions.yaml
+++ b/aten/src/ATen/native/native_functions.yaml
@@ -2521,21 +2521,21 @@
   device_guard: False
 
 # decomposes to eye.m
-- func: eye(int n, *, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None) -> Tensor
+- func: eye(SymInt n, *, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None) -> Tensor
   dispatch:
     CompositeExplicitAutograd: eye
 
-- func: eye.m(int n, int m, *, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None) -> Tensor
+- func: eye.m(SymInt n, SymInt m, *, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None) -> Tensor
   dispatch:
     CompositeExplicitAutograd: eye
 
-- func: eye.out(int n, *, Tensor(a!) out) -> Tensor(a!)
+- func: eye.out(SymInt n, *, Tensor(a!) out) -> Tensor(a!)
   dispatch:
     CPU, Meta: eye_out_cpu
     CUDA: eye_out_cuda
     MPS: eye_out_mps
 
-- func: eye.m_out(int n, int m, *, Tensor(a!) out) -> Tensor(a!)
+- func: eye.m_out(SymInt n, SymInt m, *, Tensor(a!) out) -> Tensor(a!)
   dispatch:
     CPU, Meta: eye_out_cpu
     CUDA: eye_out_cuda