pytorch/tools/codegen/api
Will Constable 397183f44c Add Lazy Tensor codegen infra (#69020)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/69020

Merges the lazy tensor codegen infra which has already been used on lazy_tensor_staging.

Test Plan: Test via lazy_tensor_staging branch

Reviewed By: alanwaketan, bdhirsh

Differential Revision: D32570613

fbshipit-source-id: 2cd5698644398bda69669683f8de79fd3b6639b5
2021-12-02 07:51:52 -08:00
..
__init__.py Rewrite of ATen code generator (#42629) 2020-08-31 09:00:22 -07:00
autograd.py Make autograd codegen for differentiable outputs safer to use (#65823) 2021-10-01 07:27:57 -07:00
cpp.py Add a Functionalization pass in core (#64432) 2021-10-28 10:51:17 -07:00
dispatcher.py Add a Functionalization pass in core (#64432) 2021-10-28 10:51:17 -07:00
functionalization.py Add a Functionalization pass in core (#64432) 2021-10-28 10:51:17 -07:00
lazy.py Add Lazy Tensor codegen infra (#69020) 2021-12-02 07:51:52 -08:00
meta.py Un-ignore F403 in .flake8 (#55838) 2021-04-13 09:24:07 -07:00
native.py move some codegen utilities into utils.py (#63094) 2021-10-28 10:49:17 -07:00
python.py Add an optional Device parameter to pin_memory/is_pinned that does nothing (#60201) 2021-06-28 10:38:52 -07:00
structured.py move some codegen utilities into utils.py (#63094) 2021-10-28 10:49:17 -07:00
translate.py Add a Functionalization pass in core (#64432) 2021-10-28 10:51:17 -07:00
types.py Add a Functionalization pass in core (#64432) 2021-10-28 10:51:17 -07:00