Update on "Change flash attention outputs to be SymInt instead of int"

ezyang · ezyang · commit 02e5f2e44370 · 2023-10-04T13:50:33.000-07:00
Fixes #110322 Signed-off-by: Edward Z. Yang <ezyangmeta.com> cc voznesenskym penguinwu EikanWang jgong5 Guobing-Chen XiaobingSuper zhuhaozhe blzheng Xia-Weiwen wenzhe-nrv jiayisunx peterbell10 ipiszy yf225 chenyang78 kadeng muchulee8 aakhundov ColinPeppler [ghstack-poisoned]
diff --git a/torch/_C/return_types.pyi.in b/torch/_C/return_types.pyi.in
@@ -16,7 +16,7 @@ from typing import (
     Union,
 )
 
-from torch import contiguous_format, Generator, inf, memory_format, strided, Tensor
+from torch import contiguous_format, Generator, inf, memory_format, strided, Tensor, SymInt
 from torch.types import (
     _bool,
     _device,

Original file line number	Diff line number	Diff line change
`@@ -16,7 +16,7 @@ from typing import (`
`16`	`16`	`Union,`
`17`	`17`	`)`
`18`	`18`
`19`		`-from torch import contiguous_format, Generator, inf, memory_format, strided, Tensor`
	`19`	`+from torch import contiguous_format, Generator, inf, memory_format, strided, Tensor, SymInt`
`20`	`20`	`from torch.types import (`
`21`	`21`	`_bool,`
`22`	`22`	`_device,`