rzou ca2d424c6e Tighten torch.library.infer_schema input types (#130705) ...

Made the following changes:
- mutates_args is now keyword-only and mandatory. This is to align with
  torch.library.custom_op (which makes it mandatory because it's easy to
  miss)
- op_name is now keyword-only. This helps the readability of the API
- updated all usages of infer_schema

This change is not BC-breaking because we introduced
torch.library.infer_schema a couple of days ago.

Test Plan:
- tests
Pull Request resolved: https://github.com/pytorch/pytorch/pull/130705
Approved by: https://github.com/yushangdi

2024-07-15 16:43:57 +00:00

..

__init__.py

[custom_op] Create a new torch._custom_op namespace (#101823)

2023-05-23 18:31:29 +00:00

autograd.py

Flip default value for mypy disallow_untyped_defs [1/11] (#127838)

2024-06-08 18:16:33 +00:00

functional.py

Flip default value for mypy disallow_untyped_defs [1/11] (#127838)

2024-06-08 18:16:33 +00:00

impl.py

Tighten torch.library.infer_schema input types (#130705)

2024-07-15 16:43:57 +00:00