pytorch/tools/codegen/api
Will Constable 69389fb542 Sync lazy_tensor_staging back to master (#72875)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/72875

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

It also resolves lint issues and in particular cleans up the following:
* {Type}s shouldn't be passed into isValueType, and using the catch-all base class of CType is nicer than specifying a list of types.

Fixes #72852

Test Plan: test manually on lazy_tensor_staging branch

Reviewed By: shunting314

Differential Revision: D34250357

fbshipit-source-id: aa7d589f605055d5d02bc77c77fa6f1182ff7497
(cherry picked from commit 2f8f5e4971)
2022-02-18 03:49:46 +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 Sync lazy_tensor_staging back to master (#72875) 2022-02-18 03:49:46 +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