Nick Gibson db2e9c1e7f [NNC] Intermediate allocs flattened and dependency support (#49554) ... Summary: Makes two changes in NNC for intermediate buffer allocations: 1. Flattens dimensions of buffers allocated in LoopNest::prepareForCodegen() to match their flattened usages. 2. Adds support for tracking memory dependencies of Alloc/Free to the MemDependencyChecker, which will allow us to check safety of accesses to intermediate buffers (coming in a future diff). I didn't add any new tests as the mem dependency checker tests already cover it pretty well, particularly the GEMM test. Pull Request resolved: https://github.com/pytorch/pytorch/pull/49554 Reviewed By: VitalyFedyunin Differential Revision: D25643133 Pulled By: nickgg fbshipit-source-id: 66be3054eb36f0a4279d0c36562e63aa2dae371c		2020-12-21 10:35:15 -08:00
..
api	Fix CustomAutogradTest.ReentrantPriority rerun failures (#49581)	2020-12-18 00:34:06 -08:00
common	Trim libshm deps, move tempfile.h to c10 (#17019)	2019-02-13 19:38:35 -08:00
dist_autograd	Fix Windows build failure after DDP PR merged (#45335)	2020-09-25 12:37:50 -07:00
jit	[PyTorch Mobile] Export Operator List from Mobile CompilationUnit instead of from TorchScript Model (#49385)	2020-12-18 11:17:57 -08:00
rpc	[c10d] switch ProcessGroup to be managed by intrusive_ptr (#47343)	2020-11-12 07:36:23 -08:00
tensorexpr	[NNC] Intermediate allocs flattened and dependency support (#49554)	2020-12-21 10:35:15 -08:00
__init__.py	remediation of S205607	2020-07-17 17:19:47 -07:00