Brian Hirsh ba86dfcd83 AOTDispatch subclass (#104483) ... This is a PoC of AOTDispatch support. This PR actually works on basic examples, and I'm working on testing it out on `DTensor` (with @wanchaol), `SemiStructuredSparsityTensor` (with @jcaip), and `FP8Tensor`. There are some design decisions baked into the PR that I think we need consensus on though - so I'm planning on writing a larger design doc to go over the changes. Pull Request resolved: https://github.com/pytorch/pytorch/pull/104483 Approved by: https://github.com/ezyang		2023-10-10 16:13:16 +00:00
..
__init__.py	Revert "Replace _dynamo.config with an object instead of module (#96455)"	2023-06-05 07:06:14 +00:00
aot_autograd.py	AOTDispatch subclass (#104483)	2023-10-10 16:13:16 +00:00
apis.py	[pt2] support vmap (#101707)	2023-08-09 03:39:33 +00:00
autograd_function.py	assign var for "not populated" str (#108844)	2023-09-12 20:53:48 +00:00
batch_norm_replacement.py	[functorch] move batch_norm_replacement to torch.func (#91412)	2023-01-12 19:15:41 +00:00
benchmark_utils.py	Improve support for contextlib.nullcontext (#98111)	2023-04-02 02:33:14 +00:00
compile_utils.py	inductor: fix CSE issue when have symbolic shape input at the freezing path (#105651)	2023-07-26 08:07:31 +00:00
compilers.py	Fix flake8 lint errors - part 2 - manual fixes (#99799)	2023-04-24 06:03:26 +00:00
config.py	Revert "Re-enable low memory dropout (#103330)"	2023-07-05 19:00:40 +00:00
deprecated.py	[pt2] support vmap (#101707)	2023-08-09 03:39:33 +00:00
eager_transforms.py	python functionalization: support higher order ops (#108656)	2023-09-20 04:37:31 +00:00
functional_call.py	[test][docs] Fix doctest warnings for syntax errors (#110517)	2023-10-05 00:00:06 +00:00
fx_minifier.py	Add function to port FX minified graph to HLO via StableHLO (#109084)	2023-10-02 19:36:04 +00:00
make_functional.py	[BE] [2/3] Rewrite super() calls in functorch and torch (#94588)	2023-02-10 21:16:33 +00:00
partitioners.py	Made pattern-matcher diagnostics lazily reported + added TORCH_COMPILE_CPROFILE (#110504)	2023-10-05 15:47:30 +00:00
pyfunctorch.py	[BE]: Bugfix functorch and some generic typing improvements (#101337)	2023-05-14 14:20:56 +00:00
python_key.py	[Reland] Move functorch/_src to torch/_functorch (#88756) (#90091)	2022-12-03 14:17:15 +00:00
pytree_hacks.py	[BE] Enable ruff's UP rules and autoformat dynamo / functorch and refs (#105432)	2023-07-19 13:48:44 +00:00
top_operators_github_usage.py	[BE]: Update ruff to v0.0.290 (#109435)	2023-09-16 18:43:34 +00:00
utils.py	[pt2] support vmap (#101707)	2023-08-09 03:39:33 +00:00
vmap.py	[pt2] support vmap (#101707)	2023-08-09 03:39:33 +00:00