1 | //===- MemCpyOptimizer.h - memcpy optimization ------------------*- C++ -*-===// |
2 | // |
3 | // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. |
4 | // See https://llvm.org/LICENSE.txt for license information. |
5 | // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception |
6 | // |
7 | //===----------------------------------------------------------------------===// |
8 | // |
9 | // This pass performs various transformations related to eliminating memcpy |
10 | // calls, or transforming sets of stores into memset's. |
11 | // |
12 | //===----------------------------------------------------------------------===// |
13 | |
14 | #ifndef LLVM_TRANSFORMS_SCALAR_MEMCPYOPTIMIZER_H |
15 | #define LLVM_TRANSFORMS_SCALAR_MEMCPYOPTIMIZER_H |
16 | |
17 | #include "llvm/IR/BasicBlock.h" |
18 | #include "llvm/IR/PassManager.h" |
19 | |
20 | namespace llvm { |
21 | |
22 | class AAResults; |
23 | class AllocaInst; |
24 | class BatchAAResults; |
25 | class AssumptionCache; |
26 | class CallBase; |
27 | class CallInst; |
28 | class DominatorTree; |
29 | class Function; |
30 | class Instruction; |
31 | class LoadInst; |
32 | class MemCpyInst; |
33 | class MemMoveInst; |
34 | class MemorySSA; |
35 | class MemorySSAUpdater; |
36 | class MemSetInst; |
37 | class PostDominatorTree; |
38 | class StoreInst; |
39 | class TargetLibraryInfo; |
40 | class Value; |
41 | |
42 | class MemCpyOptPass : public PassInfoMixin<MemCpyOptPass> { |
43 | TargetLibraryInfo *TLI = nullptr; |
44 | AAResults *AA = nullptr; |
45 | AssumptionCache *AC = nullptr; |
46 | DominatorTree *DT = nullptr; |
47 | PostDominatorTree *PDT = nullptr; |
48 | MemorySSA *MSSA = nullptr; |
49 | MemorySSAUpdater *MSSAU = nullptr; |
50 | |
51 | public: |
52 | MemCpyOptPass() = default; |
53 | |
54 | PreservedAnalyses run(Function &F, FunctionAnalysisManager &AM); |
55 | |
56 | // Glue for the old PM. |
57 | bool runImpl(Function &F, TargetLibraryInfo *TLI, AAResults *AA, |
58 | AssumptionCache *AC, DominatorTree *DT, PostDominatorTree *PDT, |
59 | MemorySSA *MSSA); |
60 | |
61 | private: |
62 | // Helper functions |
63 | bool processStore(StoreInst *SI, BasicBlock::iterator &BBI); |
64 | bool processStoreOfLoad(StoreInst *SI, LoadInst *LI, const DataLayout &DL, |
65 | BasicBlock::iterator &BBI); |
66 | bool processMemSet(MemSetInst *SI, BasicBlock::iterator &BBI); |
67 | bool processMemCpy(MemCpyInst *M, BasicBlock::iterator &BBI); |
68 | bool processMemMove(MemMoveInst *M); |
69 | bool performCallSlotOptzn(Instruction *cpyLoad, Instruction *cpyStore, |
70 | Value *cpyDst, Value *cpySrc, TypeSize cpyLen, |
71 | Align cpyAlign, BatchAAResults &BAA, |
72 | std::function<CallInst *()> GetC); |
73 | bool processMemCpyMemCpyDependence(MemCpyInst *M, MemCpyInst *MDep, |
74 | BatchAAResults &BAA); |
75 | bool processMemSetMemCpyDependence(MemCpyInst *MemCpy, MemSetInst *MemSet, |
76 | BatchAAResults &BAA); |
77 | bool performMemCpyToMemSetOptzn(MemCpyInst *MemCpy, MemSetInst *MemSet, |
78 | BatchAAResults &BAA); |
79 | bool processByValArgument(CallBase &CB, unsigned ArgNo); |
80 | bool processImmutArgument(CallBase &CB, unsigned ArgNo); |
81 | Instruction *tryMergingIntoMemset(Instruction *I, Value *StartPtr, |
82 | Value *ByteVal); |
83 | bool moveUp(StoreInst *SI, Instruction *P, const LoadInst *LI); |
84 | bool performStackMoveOptzn(Instruction *Load, Instruction *Store, |
85 | AllocaInst *DestAlloca, AllocaInst *SrcAlloca, |
86 | TypeSize Size, BatchAAResults &BAA); |
87 | |
88 | void eraseInstruction(Instruction *I); |
89 | bool iterateOnFunction(Function &F); |
90 | }; |
91 | |
92 | } // end namespace llvm |
93 | |
94 | #endif // LLVM_TRANSFORMS_SCALAR_MEMCPYOPTIMIZER_H |
95 | |