pytorch/test/cpp
Nick Gibson 0b30a8d007 [NNC] Simplify and fix some bugs in Bounds Inference (#47450)
Summary:
Refactors NNC bounds inference to use the dependency analysis added in https://github.com/pytorch/pytorch/issues/46952. This ends up being a pretty good simplification because we no longer need the complicated bound merging code that we used to determine contiguous ranges. There were no usages of that code and the memory dependency analyzer is closer to what we want for those use cases anyway.

Added tests for a few cases uncovered by the existing bounds inference test - much of the coverage for this feature is in tests of it's uses: rfactor, computeAt and cacheAccesses.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/47450

Reviewed By: heitorschueroff

Differential Revision: D24834458

Pulled By: nickgg

fbshipit-source-id: f93e40b09c0745dcc46c7e34359db594436d04f0
2020-11-09 21:37:04 -08:00
..
api Add input argument to autograd.backward() cpp api (#47214) 2020-11-04 14:43:59 -08:00
common
dist_autograd Fix Windows build failure after DDP PR merged (#45335) 2020-09-25 12:37:50 -07:00
jit Support extra files in _load_for_mobile (#47425) 2020-11-06 20:26:54 -08:00
rpc Remove lock from GraphTask::set_exception_without_signal. (#45867) 2020-10-05 20:02:29 -07:00
tensorexpr [NNC] Simplify and fix some bugs in Bounds Inference (#47450) 2020-11-09 21:37:04 -08:00
__init__.py