pytorch/test/cpp
Animesh Jain cc24e4e5d0 [NNC] Normalize loops in SplitWithTail (#66242)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/66242

While working on random test generation, I observed that many simple transformations were upsetting vectorization. Digging deeper, I found that it calls SplitWithTail which incorrectly splits the loop when the loop start is not zero. This path normalizes the loop before we start splitting it.

Test Plan: Imported from OSS

Reviewed By: ZolotukhinM

Differential Revision: D31506853

Pulled By: anijain2305

fbshipit-source-id: 5c5f2568ce0a239bfaa515458be52541eafd23b1
2021-10-11 13:44:05 -07:00
..
api Compile without -Wno-unused-variable (take 2) (#66041) 2021-10-04 20:39:39 -07:00
c10d Update CMake and use native CUDA language support (#62445) 2021-10-11 09:05:48 -07:00
common
dist_autograd Fix distributed autograd gradients synchronization (#57792) 2021-05-09 17:32:59 -07:00
jit Revert D30710710: [Pytorch Edge] Support profiling kineto events from external source 2021-10-08 17:46:18 -07:00
lazy [pytorch][PR] Add hash and int128 utils for Lazy Tensor Core" (#66181) 2021-10-07 10:05:26 -07:00
lite_interpreter_runtime Revert D30710710: [Pytorch Edge] Support profiling kineto events from external source 2021-10-08 17:46:18 -07:00
rpc Remove process_group_agent and faulty_process_group_agent files (#62985) 2021-08-10 15:57:39 -07:00
tensorexpr [NNC] Normalize loops in SplitWithTail (#66242) 2021-10-11 13:44:05 -07:00
__init__.py remediation of S205607 2020-07-17 17:19:47 -07:00