pytorch/tools/codegen/api
Will Constable 3842140fd5 Update lazy_ir.py from lazy_tensor_staging (#72730)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/72730

This diff contains changes from several PRs landed to lazy_tensor_staging branch.
- generating 'fallback' overrides for each codegenned op, useful for debugging
- supports operators which are missing aten:: symbols for op names, instead using their string counterpart
- makes the IR class a base class instead of hardcoding the assumption of TS

Test Plan: tested on lazy_tensor_staging branch

Reviewed By: desertfire

Differential Revision: D34178476

fbshipit-source-id: 7190b2e0d82b4eb1f4510c858c24446c6df3f9d0
(cherry picked from commit 6713d3f0ef)
2022-02-16 18:33:31 +00:00
..
__init__.py
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 Update lazy_ir.py from lazy_tensor_staging (#72730) 2022-02-16 18:33:31 +00: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 Structured Kernel Precompute codegen handle fields without replacement (#71368) 2022-02-08 03:56:56 +00: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