pytorch/test/cpp
Raghavan Raman aae188c529 [NNC] Handle non literal constant bounds in Unroll. (#53029)
Summary:
Fixes https://github.com/pytorch/pytorch/issues/53000

Also added test to confirm this case works in FlattenLoop as well.

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

Reviewed By: bertmaher

Differential Revision: D26742705

Pulled By: navahgar

fbshipit-source-id: d87a0f9698411026b5b6e55eee7c2b9fb123d06b
2021-03-02 00:35:27 -08:00
..
api Fix autograd when inputs contains tensors without materialized grad_fn (#51940) 2021-02-11 09:22:15 -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 Extend subgraph utils to cover merging a node following a subgraph (#52513) 2021-03-01 21:22:43 -08:00
lite_interpreter_runtime [PyTorch] update CMake to build libtorch lite (#51419) 2021-02-21 01:43:54 -08:00
rpc [RPC] Refactor rref_context to not use utils::Future (#51697) 2021-02-19 00:59:38 -08:00
tensorexpr [NNC] Handle non literal constant bounds in Unroll. (#53029) 2021-03-02 00:35:27 -08:00
__init__.py remediation of S205607 2020-07-17 17:19:47 -07:00