Mike Iovine 1385f9fb12 [JIT] Add variadic stack op (#63578) ... Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/63578 Added a new op `prim::VarStack` and a pass that transforms instances of `aten::stack(list, dim)` into `prim::VarStack(list[0], ..., list[n], dim)`. Also provided a JIT interpreter implementation. Most of the implementation/tests are the same as `prim::VarConcat`. Test Plan: `buck test caffe2/test/cpp/jit:jit -- TestStackOpt` Reviewed By: navahgar Differential Revision: D30426232 fbshipit-source-id: 9829a7db6e0a5038c9b7528c43c25b0c221aa2ce		2021-08-24 08:20:54 -07:00
..
api	Revert D30279364: [codemod][lint][fbcode/c*] Enable BLACK by default	2021-08-12 11:45:01 -07:00
c10d	[Pytorch Profiler] Add support for adding module hierarchy to (#61792)	2021-08-13 21:39:10 -07:00
common
dist_autograd	Fix distributed autograd gradients synchronization (#57792)	2021-05-09 17:32:59 -07:00
jit	[JIT] Add variadic stack op (#63578)	2021-08-24 08:20:54 -07:00
lite_interpreter_runtime	[Pytorch Edge] Enable kineto profiler on mobile via EdgeKinetoProfiler (#62419)	2021-08-13 21:40:19 -07:00
rpc	Remove process_group_agent and faulty_process_group_agent files (#62985)	2021-08-10 15:57:39 -07:00
tensorexpr	[TensorExpr] Nuke KernelArena and KernelScope. (#63587)	2021-08-24 00:32:16 -07:00
__init__.py	remediation of S205607	2020-07-17 17:19:47 -07:00