[CodeExtractor] Remove stale llvm.assume calls from extracted region
authorVedant Kumar <vsk@apple.com>
Wed, 29 Jan 2020 01:03:39 +0000 (17:03 -0800)
committerVedant Kumar <vsk@apple.com>
Wed, 29 Jan 2020 01:18:01 +0000 (17:18 -0800)
commit8359511c62b727e360f93d9f5dab5893548227c8
treeae5cd374798ac40a2fc22bbd3535e4800c9cd757
parent2d92336db0087ad295401865d7749d4d1cfe4846
[CodeExtractor] Remove stale llvm.assume calls from extracted region

During extraction, stale llvm.assume handles may be retained in the
original function. The setup is:

1) CodeExtractor unregisters assumptions in the blocks that are to be
   extracted.

2) Extraction happens. There are now two functions: f1 and f1.extracted.

3) Leftover assumptions in f1 (/not/ removed as they were not in the set of
   blocks to be extracted) now have affected-value llvm.assume handles in
   f1.extracted.

When assumptions for a value used in f1 are looked up, ValueTracking can assert
as some of the handles are in the wrong function. To fix this, simply erase the
llvm.assume calls in the extracted function.

Alternatives include flushing the assumption cache in the original function, or
walking all values used in the original function to prune stale affected-value
handles. Both seem more expensive.

Testing: check-llvm, LNT run with -mllvm -hot-cold-split enabled

rdar://58460728
llvm/include/llvm/Transforms/Utils/CodeExtractor.h
llvm/lib/Transforms/Utils/CodeExtractor.cpp
llvm/test/Transforms/CodeExtractor/extract-assume.ll
llvm/test/Transforms/HotColdSplit/assumption-cache-invalidation.ll
llvm/test/Transforms/HotColdSplit/stale-assume-in-original-func.ll [new file with mode: 0644]
llvm/unittests/Transforms/Utils/CodeExtractorTest.cpp