Elias Ellison d9f0774f98 [JIT] Allow registering Decompositions ... - Allow registering custom decompositions - Add easier API for invoking decompositions - Shorten API names (no users yet) I am doing these as one pr because they are fairly short/simple and because github first does not support ghstack yet. cc @Chillee @zou3519 Pull Request resolved: https://github.com/pytorch/pytorch/pull/76252 Approved by: https://github.com/davidberard98		2022-04-26 03:00:35 +00:00
..
api	Extend jit::load to work on flatbuffer file; Take 2 (#75256)	2022-04-06 17:54:01 +00:00
c10d	Add cuda-11.3+clang9 build workflow (take 2)	2022-04-11 17:13:01 +00:00
common
dist_autograd
jit	[JIT] Allow registering Decompositions	2022-04-26 03:00:35 +00:00
lazy	E2E SymInt example narrow_copy	2022-04-26 02:40:27 +00:00
lite_interpreter_runtime	[PyTorch] Add codegen unboxing ability (#69881)	2022-03-01 23:28:13 +00:00
monitor	torch/monitor: merge Interval and FixedCount stats (#72009)	2022-01-30 23:21:59 +00:00
profiler	[RecordFunction] More effecient machinery to determine which callbacks to run. (#75807)	2022-04-19 20:46:16 +00:00
rpc
tensorexpr	Revert "[nnc] Strides to Tensor (#72962)"	2022-04-25 19:50:00 +00:00
__init__.py