fb64da0791
[2/N] Use "is" in python type comparison ( #165142 )
...
This is follow-up of #165037 . It generally recommended to use `is/is not` to compare types. Therefore this series of changes apply this suggestion in the code base, and it aims to finally enabling related linter checks.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/165142
Approved by: https://github.com/albanD
2025-10-10 15:36:44 +00:00
3fe3c23d4e
[cond] support gen_schema for cond ( #154193 )
...
Pull Request resolved: https://github.com/pytorch/pytorch/pull/154193
Approved by: https://github.com/zou3519
2025-08-15 17:28:13 +00:00
7f14b42adf
[BE][2/16] fix typos in torch/ (torch/_*/) ( #156312 )
...
Pull Request resolved: https://github.com/pytorch/pytorch/pull/156312
Approved by: https://github.com/albanD
2025-07-12 05:47:06 +00:00
e15f4248ad
Revert "[BE][2/16] fix typos in torch/ (torch/_*/) ( #156312 )"
...
This reverts commit 7a92b5119654c07d15f5c0818e6ae804b01e836c.
Reverted https://github.com/pytorch/pytorch/pull/156312 on behalf of https://github.com/XuehaiPan due to landrace ([comment](https://github.com/pytorch/pytorch/pull/156312#issuecomment-3064672250 ))
2025-07-12 04:40:52 +00:00
7a92b51196
[BE][2/16] fix typos in torch/ (torch/_*/) ( #156312 )
...
Pull Request resolved: https://github.com/pytorch/pytorch/pull/156312
Approved by: https://github.com/albanD
2025-07-12 01:47:22 +00:00
21990fbad9
Revert "[cond] support gen_schema for cond ( #154193 )"
...
This reverts commit 6de41ce0f899604c3f8b33e1f8d37eb89b3a963e.
Reverted https://github.com/pytorch/pytorch/pull/154193 on behalf of https://github.com/Camyll due to issue landing internally, discussed with Yidi offline ([comment](https://github.com/pytorch/pytorch/pull/154193#issuecomment-3009160081 ))
2025-06-26 17:10:00 +00:00
6de41ce0f8
[cond] support gen_schema for cond ( #154193 )
...
Pull Request resolved: https://github.com/pytorch/pytorch/pull/154193
Approved by: https://github.com/zou3519
ghstack dependencies: #155644
2025-06-25 21:19:58 +00:00
162ca185ff
[BE][PYFMT] migrate PYFMT for torch/_[a-h]*/ to ruff format ( #144551 )
...
Pull Request resolved: https://github.com/pytorch/pytorch/pull/144551
Approved by: https://github.com/ezyang
ghstack dependencies: #148186
2025-06-25 06:16:06 +00:00
d6be87648f
[hop schema] add schema.tree_spec to support pytree inputs ( #154191 )
...
Pull Request resolved: https://github.com/pytorch/pytorch/pull/154191
Approved by: https://github.com/zou3519
ghstack dependencies: #155261 , #154072
2025-06-11 22:52:37 +00:00
8e6e79fc1b
[hop_schema] support gen_schema for invoke_subgraph ( #152984 )
...
Pull Request resolved: https://github.com/pytorch/pytorch/pull/152984
Approved by: https://github.com/zou3519
ghstack dependencies: #151067 , #152974
2025-05-21 18:55:46 +00:00
9c33899196
[hop_schema] add HopSchemaGenerator to make it easier to create hop schema ( #152974 )
...
Pull Request resolved: https://github.com/pytorch/pytorch/pull/152974
Approved by: https://github.com/zou3519
ghstack dependencies: #151067
2025-05-21 18:55:46 +00:00
1e0f19e173
auto functionalize base_hop ( #151067 )
...
Pull Request resolved: https://github.com/pytorch/pytorch/pull/151067
Approved by: https://github.com/zou3519
2025-05-21 18:55:46 +00:00
a776a566db
[hop][schema] allow adding kw_only info to schema argument ( #152246 )
...
Pull Request resolved: https://github.com/pytorch/pytorch/pull/152246
Approved by: https://github.com/zou3519
ghstack dependencies: #152072 , #152073 , #152244 , #152245
2025-05-02 02:08:06 +00:00
69e41cee04
move find_hop_schema into _higher_order_ops/schema.py ( #151147 )
...
Pull Request resolved: https://github.com/pytorch/pytorch/pull/151147
Approved by: https://github.com/zou3519
2025-04-23 18:26:37 +00:00
c714d2fc0e
[hop] support base_hop._gen_schema ( #149688 )
...
This PR creates two utils for generating a schema for hops from example inputs and use base hop as an exmaple.
1. HopArgumentInfoGen creates an argument or an output schema with mutation information.
2. CFuncitonSchemaGen piece together the argument info of inputs and outputs and produces torch._C.FunctionSchema.
is_write attribute of argument info can be computed. Note that the is_write annotation only works when the inputs are flattened (e.g. cannot support mutation inside tuple). We need special handling the case where we have tuple inputs like cond.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/149688
Approved by: https://github.com/zou3519
2025-04-09 16:42:55 +00:00
