pytorch/functorch/codegen
Richard Zou f67b293bfa Fix some mypy issues in functorch/codegen/gen_vmap_plumbing.py (#82314)
Necessary to make lint happy in the next PR in the stack

Test Plan
- `mypy functorch/codegen/gen_vmap_plumbing.py`
Pull Request resolved: https://github.com/pytorch/pytorch/pull/82314
Approved by: https://github.com/ezyang
2022-07-27 18:11:02 +00:00
..
gen_functorch_lagging_op_db.py [functorch] Added softmax batching rule (tested through local OpInfo, to be upstreamed) 2022-07-21 13:40:57 -07:00
gen_vmap_plumbing.py Fix some mypy issues in functorch/codegen/gen_vmap_plumbing.py (#82314) 2022-07-27 18:11:02 +00:00
gen.py [functorch] Align functorch's flake8 config with pytorch's (pytorch/functorch#963) 2022-07-21 13:41:37 -07:00