pytorch/test/cpp
Bert Maher 565d8235e5 [nnc] Test cases for uneven split + reorder (#53091)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/53091

Split with tail followed by reorder causes a segfault in NNC
Split with mask followed by reorder generates invalid code that writes out of
bounds
ghstack-source-id: 122870733

Test Plan: LoopNest.ColReduceSplit*

Reviewed By: navahgar

Differential Revision: D26746254

fbshipit-source-id: f8a0de18531b34d2bf06ccaa35d9c98b81b5c600
2021-03-02 20:36:48 -08:00
..
api Huber loss (#50553) 2021-03-02 17:30:45 -08:00
common
dist_autograd
jit Add default arguments to cuda stream and events (#53025) 2021-03-02 14:37:24 -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] Test cases for uneven split + reorder (#53091) 2021-03-02 20:36:48 -08:00
__init__.py