pytorch/torchgen/api
Brian Vaughan dbb96ef30d improve annotation device parameters where a device ordinal is allowed (#113647)
Using mypy in code that depends on pytorch, I noticed that the type annotation doesn't allow a device ordinal.

`error: Argument "device" to "to_empty" of "Module" has incompatible type "int"; expected "str | device"  [arg-type]`

Pull Request resolved: https://github.com/pytorch/pytorch/pull/113647
Approved by: https://github.com/albanD
2023-11-17 14:41:22 +00:00
..
types Add support for float8_e4m3fnuz and _e5m2fnuz (#107586) 2023-11-15 15:01:11 +00:00
__init__.py
autograd.py Add 0dim Tensor overload for _foreach_div (#113688) 2023-11-15 20:59:32 +00:00
cpp.py [torchgen] Add support for optional tensor (#112938) 2023-11-06 20:03:05 +00:00
dispatcher.py Use presence of _symint in kernel name to generate symint sig or not (#84579) 2022-09-09 18:31:56 +00:00
functionalization.py Bump black version to 23.1.0 (#96578) 2023-03-15 06:27:59 +00:00
lazy.py Revert "Add support for torch.Generator type in TorchScript (#110413)" 2023-11-15 00:51:23 +00:00
meta.py
native.py Reland 2 Many symintifications (#87604) (#87980) 2022-10-28 13:40:11 +00:00
python.py improve annotation device parameters where a device ordinal is allowed (#113647) 2023-11-17 14:41:22 +00:00
structured.py Fix typo under torchgen directory (#111154) 2023-10-13 16:43:46 +00:00
translate.py Bump black version to 23.1.0 (#96578) 2023-03-15 06:27:59 +00:00
ufunc.py Back out "Revert D38984222: Don't introduce new overload for SymInt (#83628)" (#84173) 2022-08-29 18:01:07 +00:00
unboxing.py [BE]: remove unnecessary enumerate calls (#111690) 2023-10-20 23:20:29 +00:00