1de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel//===- BBVectorize.cpp - A Basic-Block Vectorizer -------------------------===//
2de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel//
3de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel//                     The LLVM Compiler Infrastructure
4de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel//
5de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel// This file is distributed under the University of Illinois Open Source
6de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel// License. See LICENSE.TXT for details.
7de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel//
8de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel//===----------------------------------------------------------------------===//
9de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel//
10de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel// This file implements a basic-block vectorization pass. The algorithm was
11de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel// inspired by that used by the Vienna MAP Vectorizor by Franchetti and Kral,
12de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel// et al. It works by looking for chains of pairable operations and then
13de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel// pairing them.
14de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel//
15de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel//===----------------------------------------------------------------------===//
16de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
17de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel#define BBV_NAME "bb-vectorize"
18de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel#define DEBUG_TYPE BBV_NAME
19d04a8d4b33ff316ca4cf961e06c9e312eff8e64fChandler Carruth#include "llvm/Transforms/Vectorize.h"
20de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel#include "llvm/ADT/DenseMap.h"
21de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel#include "llvm/ADT/DenseSet.h"
22d04a8d4b33ff316ca4cf961e06c9e312eff8e64fChandler Carruth#include "llvm/ADT/STLExtras.h"
2386c88c938aec8006d2ce83325ec1f31e1154620bHal Finkel#include "llvm/ADT/SmallSet.h"
24de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel#include "llvm/ADT/SmallVector.h"
25de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel#include "llvm/ADT/Statistic.h"
26de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel#include "llvm/ADT/StringExtras.h"
27de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel#include "llvm/Analysis/AliasAnalysis.h"
28de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel#include "llvm/Analysis/AliasSetTracker.h"
29e29c19091cca58db668407dfc5dd86c70e8b3d49Hal Finkel#include "llvm/Analysis/Dominators.h"
30de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel#include "llvm/Analysis/ScalarEvolution.h"
31de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel#include "llvm/Analysis/ScalarEvolutionExpressions.h"
32be04929f7fd76a921540e9901f24563e51dc1219Chandler Carruth#include "llvm/Analysis/TargetTransformInfo.h"
33de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel#include "llvm/Analysis/ValueTracking.h"
340b8c9a80f20772c3793201ab5b251d3520b9cea3Chandler Carruth#include "llvm/IR/Constants.h"
350b8c9a80f20772c3793201ab5b251d3520b9cea3Chandler Carruth#include "llvm/IR/DataLayout.h"
360b8c9a80f20772c3793201ab5b251d3520b9cea3Chandler Carruth#include "llvm/IR/DerivedTypes.h"
370b8c9a80f20772c3793201ab5b251d3520b9cea3Chandler Carruth#include "llvm/IR/Function.h"
380b8c9a80f20772c3793201ab5b251d3520b9cea3Chandler Carruth#include "llvm/IR/Instructions.h"
390b8c9a80f20772c3793201ab5b251d3520b9cea3Chandler Carruth#include "llvm/IR/IntrinsicInst.h"
400b8c9a80f20772c3793201ab5b251d3520b9cea3Chandler Carruth#include "llvm/IR/Intrinsics.h"
410b8c9a80f20772c3793201ab5b251d3520b9cea3Chandler Carruth#include "llvm/IR/LLVMContext.h"
420b8c9a80f20772c3793201ab5b251d3520b9cea3Chandler Carruth#include "llvm/IR/Metadata.h"
430b8c9a80f20772c3793201ab5b251d3520b9cea3Chandler Carruth#include "llvm/IR/Type.h"
44d04a8d4b33ff316ca4cf961e06c9e312eff8e64fChandler Carruth#include "llvm/Pass.h"
45de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel#include "llvm/Support/CommandLine.h"
46de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel#include "llvm/Support/Debug.h"
47de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel#include "llvm/Support/ValueHandle.h"
48d04a8d4b33ff316ca4cf961e06c9e312eff8e64fChandler Carruth#include "llvm/Support/raw_ostream.h"
4964e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel#include "llvm/Transforms/Utils/Local.h"
50de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel#include <algorithm>
51de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkelusing namespace llvm;
52de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
5365309660fa61a837cc05323f69c618a7d8134d56Hal Finkelstatic cl::opt<bool>
5465309660fa61a837cc05323f69c618a7d8134d56Hal FinkelIgnoreTargetInfo("bb-vectorize-ignore-target-info",  cl::init(false),
5565309660fa61a837cc05323f69c618a7d8134d56Hal Finkel  cl::Hidden, cl::desc("Ignore target information"));
5665309660fa61a837cc05323f69c618a7d8134d56Hal Finkel
57de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkelstatic cl::opt<unsigned>
58de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal FinkelReqChainDepth("bb-vectorize-req-chain-depth", cl::init(6), cl::Hidden,
59de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  cl::desc("The required chain depth for vectorization"));
60de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
6178fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkelstatic cl::opt<bool>
6278fd353d5e5daedc47ecc31b6193ca48793c249cHal FinkelUseChainDepthWithTI("bb-vectorize-use-chain-depth",  cl::init(false),
6378fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel  cl::Hidden, cl::desc("Use the chain depth requirement with"
6478fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel                       " target information"));
6578fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel
66de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkelstatic cl::opt<unsigned>
67de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal FinkelSearchLimit("bb-vectorize-search-limit", cl::init(400), cl::Hidden,
68de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  cl::desc("The maximum search distance for instruction pairs"));
69de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
70de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkelstatic cl::opt<bool>
71de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal FinkelSplatBreaksChain("bb-vectorize-splat-breaks-chain", cl::init(false), cl::Hidden,
72de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  cl::desc("Replicating one element to a pair breaks the chain"));
73de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
74de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkelstatic cl::opt<unsigned>
75de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal FinkelVectorBits("bb-vectorize-vector-bits", cl::init(128), cl::Hidden,
76de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  cl::desc("The size of the native vector registers"));
77de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
78de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkelstatic cl::opt<unsigned>
79de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal FinkelMaxIter("bb-vectorize-max-iter", cl::init(0), cl::Hidden,
80de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  cl::desc("The maximum number of pairing iterations"));
81de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
8264e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkelstatic cl::opt<bool>
8364e1b28643d87e70734deb5f3d2d298e859c2fd2Hal FinkelPow2LenOnly("bb-vectorize-pow2-len-only", cl::init(false), cl::Hidden,
8464e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel  cl::desc("Don't try to form non-2^n-length vectors"));
8564e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel
86de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkelstatic cl::opt<unsigned>
875d4e18bc39fea892f523d960213906d296d3cb38Hal FinkelMaxInsts("bb-vectorize-max-instr-per-group", cl::init(500), cl::Hidden,
885d4e18bc39fea892f523d960213906d296d3cb38Hal Finkel  cl::desc("The maximum number of pairable instructions per group"));
895d4e18bc39fea892f523d960213906d296d3cb38Hal Finkel
905d4e18bc39fea892f523d960213906d296d3cb38Hal Finkelstatic cl::opt<unsigned>
91ab90084bca42b74a5b5edad9b416bd81e105dad0Hal FinkelMaxPairs("bb-vectorize-max-pairs-per-group", cl::init(3000), cl::Hidden,
92ab90084bca42b74a5b5edad9b416bd81e105dad0Hal Finkel  cl::desc("The maximum number of candidate instruction pairs per group"));
93ab90084bca42b74a5b5edad9b416bd81e105dad0Hal Finkel
94ab90084bca42b74a5b5edad9b416bd81e105dad0Hal Finkelstatic cl::opt<unsigned>
95de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal FinkelMaxCandPairsForCycleCheck("bb-vectorize-max-cycle-check-pairs", cl::init(200),
96de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  cl::Hidden, cl::desc("The maximum number of candidate pairs with which to use"
97de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel                       " a full cycle check"));
98de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
99de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkelstatic cl::opt<bool>
100768edf3cd037aab10391abc279f71470df8e3156Hal FinkelNoBools("bb-vectorize-no-bools", cl::init(false), cl::Hidden,
101768edf3cd037aab10391abc279f71470df8e3156Hal Finkel  cl::desc("Don't try to vectorize boolean (i1) values"));
102768edf3cd037aab10391abc279f71470df8e3156Hal Finkel
103768edf3cd037aab10391abc279f71470df8e3156Hal Finkelstatic cl::opt<bool>
104de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal FinkelNoInts("bb-vectorize-no-ints", cl::init(false), cl::Hidden,
105de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  cl::desc("Don't try to vectorize integer values"));
106de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
107de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkelstatic cl::opt<bool>
108de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal FinkelNoFloats("bb-vectorize-no-floats", cl::init(false), cl::Hidden,
109de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  cl::desc("Don't try to vectorize floating-point values"));
110de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
111822ab00847da841a63be4e3883cb5f442dc69069Hal Finkel// FIXME: This should default to false once pointer vector support works.
112de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkelstatic cl::opt<bool>
113822ab00847da841a63be4e3883cb5f442dc69069Hal FinkelNoPointers("bb-vectorize-no-pointers", cl::init(/*false*/ true), cl::Hidden,
114f3f5a1e6f77a842ccb24cc81766437da5197d712Hal Finkel  cl::desc("Don't try to vectorize pointer values"));
115f3f5a1e6f77a842ccb24cc81766437da5197d712Hal Finkel
116f3f5a1e6f77a842ccb24cc81766437da5197d712Hal Finkelstatic cl::opt<bool>
117de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal FinkelNoCasts("bb-vectorize-no-casts", cl::init(false), cl::Hidden,
118de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  cl::desc("Don't try to vectorize casting (conversion) operations"));
119de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
120de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkelstatic cl::opt<bool>
121de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal FinkelNoMath("bb-vectorize-no-math", cl::init(false), cl::Hidden,
122de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  cl::desc("Don't try to vectorize floating-point math intrinsics"));
123de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
124de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkelstatic cl::opt<bool>
125de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal FinkelNoFMA("bb-vectorize-no-fma", cl::init(false), cl::Hidden,
126de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  cl::desc("Don't try to vectorize the fused-multiply-add intrinsic"));
127de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
128de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkelstatic cl::opt<bool>
129fc3665c87519850f629c9565535e3be447e10addHal FinkelNoSelect("bb-vectorize-no-select", cl::init(false), cl::Hidden,
130fc3665c87519850f629c9565535e3be447e10addHal Finkel  cl::desc("Don't try to vectorize select instructions"));
131fc3665c87519850f629c9565535e3be447e10addHal Finkel
132fc3665c87519850f629c9565535e3be447e10addHal Finkelstatic cl::opt<bool>
133e415f96b6a43ac8861148a11a4258bc38c247e8fHal FinkelNoCmp("bb-vectorize-no-cmp", cl::init(false), cl::Hidden,
134e415f96b6a43ac8861148a11a4258bc38c247e8fHal Finkel  cl::desc("Don't try to vectorize comparison instructions"));
135e415f96b6a43ac8861148a11a4258bc38c247e8fHal Finkel
136e415f96b6a43ac8861148a11a4258bc38c247e8fHal Finkelstatic cl::opt<bool>
137f3f5a1e6f77a842ccb24cc81766437da5197d712Hal FinkelNoGEP("bb-vectorize-no-gep", cl::init(false), cl::Hidden,
138f3f5a1e6f77a842ccb24cc81766437da5197d712Hal Finkel  cl::desc("Don't try to vectorize getelementptr instructions"));
139f3f5a1e6f77a842ccb24cc81766437da5197d712Hal Finkel
140f3f5a1e6f77a842ccb24cc81766437da5197d712Hal Finkelstatic cl::opt<bool>
141de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal FinkelNoMemOps("bb-vectorize-no-mem-ops", cl::init(false), cl::Hidden,
142de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  cl::desc("Don't try to vectorize loads and stores"));
143de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
144de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkelstatic cl::opt<bool>
145de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal FinkelAlignedOnly("bb-vectorize-aligned-only", cl::init(false), cl::Hidden,
146de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  cl::desc("Only generate aligned loads and stores"));
147de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
148de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkelstatic cl::opt<bool>
149edc8db87dc2ed4d2971e7f50464f5f4d0fead537Hal FinkelNoMemOpBoost("bb-vectorize-no-mem-op-boost",
150edc8db87dc2ed4d2971e7f50464f5f4d0fead537Hal Finkel  cl::init(false), cl::Hidden,
151edc8db87dc2ed4d2971e7f50464f5f4d0fead537Hal Finkel  cl::desc("Don't boost the chain-depth contribution of loads and stores"));
152edc8db87dc2ed4d2971e7f50464f5f4d0fead537Hal Finkel
153edc8db87dc2ed4d2971e7f50464f5f4d0fead537Hal Finkelstatic cl::opt<bool>
154de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal FinkelFastDep("bb-vectorize-fast-dep", cl::init(false), cl::Hidden,
155de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  cl::desc("Use a fast instruction dependency analysis"));
156de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
157de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel#ifndef NDEBUG
158de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkelstatic cl::opt<bool>
159de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal FinkelDebugInstructionExamination("bb-vectorize-debug-instruction-examination",
160de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  cl::init(false), cl::Hidden,
161de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  cl::desc("When debugging is enabled, output information on the"
162de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel           " instruction-examination process"));
163de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkelstatic cl::opt<bool>
164de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal FinkelDebugCandidateSelection("bb-vectorize-debug-candidate-selection",
165de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  cl::init(false), cl::Hidden,
166de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  cl::desc("When debugging is enabled, output information on the"
167de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel           " candidate-selection process"));
168de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkelstatic cl::opt<bool>
169de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal FinkelDebugPairSelection("bb-vectorize-debug-pair-selection",
170de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  cl::init(false), cl::Hidden,
171de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  cl::desc("When debugging is enabled, output information on the"
172de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel           " pair-selection process"));
173de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkelstatic cl::opt<bool>
174de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal FinkelDebugCycleCheck("bb-vectorize-debug-cycle-check",
175de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  cl::init(false), cl::Hidden,
176de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  cl::desc("When debugging is enabled, output information on the"
177de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel           " cycle-checking process"));
17872465ea23d010507d3746adc126d719005981e05Hal Finkel
17972465ea23d010507d3746adc126d719005981e05Hal Finkelstatic cl::opt<bool>
18072465ea23d010507d3746adc126d719005981e05Hal FinkelPrintAfterEveryPair("bb-vectorize-debug-print-after-every-pair",
18172465ea23d010507d3746adc126d719005981e05Hal Finkel  cl::init(false), cl::Hidden,
18272465ea23d010507d3746adc126d719005981e05Hal Finkel  cl::desc("When debugging is enabled, dump the basic block after"
18372465ea23d010507d3746adc126d719005981e05Hal Finkel           " every pair is fused"));
184de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel#endif
185de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
186de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal FinkelSTATISTIC(NumFusedOps, "Number of operations fused by bb-vectorize");
187de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
188de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkelnamespace {
189de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  struct BBVectorize : public BasicBlockPass {
190de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    static char ID; // Pass identification, replacement for typeid
191bef377b7d7ce31edb40c87f8786d1b7bb6cdd6b1Hongbin Zheng
192940371bc65570ec0add1ede4f4d9f0a41ba25e09Hongbin Zheng    const VectorizeConfig Config;
193bef377b7d7ce31edb40c87f8786d1b7bb6cdd6b1Hongbin Zheng
194bef377b7d7ce31edb40c87f8786d1b7bb6cdd6b1Hongbin Zheng    BBVectorize(const VectorizeConfig &C = VectorizeConfig())
195bef377b7d7ce31edb40c87f8786d1b7bb6cdd6b1Hongbin Zheng      : BasicBlockPass(ID), Config(C) {
196de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      initializeBBVectorizePass(*PassRegistry::getPassRegistry());
197de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    }
198de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
199bef377b7d7ce31edb40c87f8786d1b7bb6cdd6b1Hongbin Zheng    BBVectorize(Pass *P, const VectorizeConfig &C)
200bef377b7d7ce31edb40c87f8786d1b7bb6cdd6b1Hongbin Zheng      : BasicBlockPass(ID), Config(C) {
20187825e7970a361ce5a8bab19bc880ff7f6242ca2Hongbin Zheng      AA = &P->getAnalysis<AliasAnalysis>();
202e29c19091cca58db668407dfc5dd86c70e8b3d49Hal Finkel      DT = &P->getAnalysis<DominatorTree>();
20387825e7970a361ce5a8bab19bc880ff7f6242ca2Hongbin Zheng      SE = &P->getAnalysis<ScalarEvolution>();
2043574eca1b02600bac4e625297f4ecf745f4c4f32Micah Villmow      TD = P->getAnalysisIfAvailable<DataLayout>();
2058bd6c52396ab6e7955fdcc1bce099b7cba29a308Chandler Carruth      TTI = IgnoreTargetInfo ? 0 : &P->getAnalysis<TargetTransformInfo>();
20687825e7970a361ce5a8bab19bc880ff7f6242ca2Hongbin Zheng    }
20787825e7970a361ce5a8bab19bc880ff7f6242ca2Hongbin Zheng
208de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    typedef std::pair<Value *, Value *> ValuePair;
20965309660fa61a837cc05323f69c618a7d8134d56Hal Finkel    typedef std::pair<ValuePair, int> ValuePairWithCost;
210de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    typedef std::pair<ValuePair, size_t> ValuePairWithDepth;
211de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    typedef std::pair<ValuePair, ValuePair> VPPair; // A ValuePair pair
21272465ea23d010507d3746adc126d719005981e05Hal Finkel    typedef std::pair<VPPair, unsigned> VPPairWithType;
213de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
214de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    AliasAnalysis *AA;
215e29c19091cca58db668407dfc5dd86c70e8b3d49Hal Finkel    DominatorTree *DT;
216de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    ScalarEvolution *SE;
2173574eca1b02600bac4e625297f4ecf745f4c4f32Micah Villmow    DataLayout *TD;
218abc227d9b39baed266c453f459ca14242f5c1eeaChandler Carruth    const TargetTransformInfo *TTI;
219de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
220de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    // FIXME: const correct?
221de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
22264e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel    bool vectorizePairs(BasicBlock &BB, bool NonPow2Len = false);
223de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
2245d4e18bc39fea892f523d960213906d296d3cb38Hal Finkel    bool getCandidatePairs(BasicBlock &BB,
2255d4e18bc39fea892f523d960213906d296d3cb38Hal Finkel                       BasicBlock::iterator &Start,
2266ca6d3b1eac5b8611f3a9e2c270c2e794d37e1f5Hal Finkel                       DenseMap<Value *, std::vector<Value *> > &CandidatePairs,
227a9779bfbc9ab0cf3f157453fd0afd110b04a9fdcHal Finkel                       DenseSet<ValuePair> &FixedOrderPairs,
22865309660fa61a837cc05323f69c618a7d8134d56Hal Finkel                       DenseMap<ValuePair, int> &CandidatePairCostSavings,
22964e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel                       std::vector<Value *> &PairableInsts, bool NonPow2Len);
230de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
23178fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel    // FIXME: The current implementation does not account for pairs that
23278fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel    // are connected in multiple ways. For example:
23378fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel    //   C1 = A1 / A2; C2 = A2 / A1 (which may be both direct and a swap)
23472465ea23d010507d3746adc126d719005981e05Hal Finkel    enum PairConnectionType {
23572465ea23d010507d3746adc126d719005981e05Hal Finkel      PairConnectionDirect,
23672465ea23d010507d3746adc126d719005981e05Hal Finkel      PairConnectionSwap,
23772465ea23d010507d3746adc126d719005981e05Hal Finkel      PairConnectionSplat
23872465ea23d010507d3746adc126d719005981e05Hal Finkel    };
23972465ea23d010507d3746adc126d719005981e05Hal Finkel
24097a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel    void computeConnectedPairs(
24197a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel             DenseMap<Value *, std::vector<Value *> > &CandidatePairs,
24297a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel             DenseSet<ValuePair> &CandidatePairsSet,
24397a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel             std::vector<Value *> &PairableInsts,
24497a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel             DenseMap<ValuePair, std::vector<ValuePair> > &ConnectedPairs,
24597a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel             DenseMap<VPPair, unsigned> &PairConnectionTypes);
246de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
247de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    void buildDepMap(BasicBlock &BB,
24897a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel             DenseMap<Value *, std::vector<Value *> > &CandidatePairs,
24997a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel             std::vector<Value *> &PairableInsts,
25097a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel             DenseSet<ValuePair> &PairableInstUsers);
251de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
2526ca6d3b1eac5b8611f3a9e2c270c2e794d37e1f5Hal Finkel    void choosePairs(DenseMap<Value *, std::vector<Value *> > &CandidatePairs,
25397a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel             DenseSet<ValuePair> &CandidatePairsSet,
25497a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel             DenseMap<ValuePair, int> &CandidatePairCostSavings,
25597a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel             std::vector<Value *> &PairableInsts,
25697a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel             DenseSet<ValuePair> &FixedOrderPairs,
25797a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel             DenseMap<VPPair, unsigned> &PairConnectionTypes,
25897a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel             DenseMap<ValuePair, std::vector<ValuePair> > &ConnectedPairs,
25997a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel             DenseMap<ValuePair, std::vector<ValuePair> > &ConnectedPairDeps,
26097a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel             DenseSet<ValuePair> &PairableInstUsers,
26197a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel             DenseMap<Value *, Value *>& ChosenPairs);
262de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
263de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    void fuseChosenPairs(BasicBlock &BB,
26497a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel             std::vector<Value *> &PairableInsts,
26597a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel             DenseMap<Value *, Value *>& ChosenPairs,
26697a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel             DenseSet<ValuePair> &FixedOrderPairs,
26797a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel             DenseMap<VPPair, unsigned> &PairConnectionTypes,
26897a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel             DenseMap<ValuePair, std::vector<ValuePair> > &ConnectedPairs,
26997a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel             DenseMap<ValuePair, std::vector<ValuePair> > &ConnectedPairDeps);
27072465ea23d010507d3746adc126d719005981e05Hal Finkel
271de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
272de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    bool isInstVectorizable(Instruction *I, bool &IsSimpleLoadStore);
273de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
274de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    bool areInstsCompatible(Instruction *I, Instruction *J,
27565309660fa61a837cc05323f69c618a7d8134d56Hal Finkel                       bool IsSimpleLoadStore, bool NonPow2Len,
276a9779bfbc9ab0cf3f157453fd0afd110b04a9fdcHal Finkel                       int &CostSavings, int &FixedOrder);
277de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
278de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    bool trackUsesOfI(DenseSet<Value *> &Users,
279de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel                      AliasSetTracker &WriteSet, Instruction *I,
280de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel                      Instruction *J, bool UpdateUsers = true,
2812f0e63cc16feb39480805bd00f53bbe5e3031d29Hal Finkel                      DenseSet<ValuePair> *LoadMoveSetPairs = 0);
2821230ad6e8cb7977527ac64dcf5005464d7d6c20bSebastian Pop
28397a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel  void computePairsConnectedTo(
28497a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel             DenseMap<Value *, std::vector<Value *> > &CandidatePairs,
28597a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel             DenseSet<ValuePair> &CandidatePairsSet,
28697a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel             std::vector<Value *> &PairableInsts,
28797a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel             DenseMap<ValuePair, std::vector<ValuePair> > &ConnectedPairs,
28897a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel             DenseMap<VPPair, unsigned> &PairConnectionTypes,
28997a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel             ValuePair P);
290de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
291de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    bool pairsConflict(ValuePair P, ValuePair Q,
29297a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel             DenseSet<ValuePair> &PairableInstUsers,
29397a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel             DenseMap<ValuePair, std::vector<ValuePair> >
29497a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel               *PairableInstUserMap = 0,
29597a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel             DenseSet<VPPair> *PairableInstUserPairSet = 0);
296de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
297de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    bool pairWillFormCycle(ValuePair P,
29897a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel             DenseMap<ValuePair, std::vector<ValuePair> > &PairableInstUsers,
29997a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel             DenseSet<ValuePair> &CurrentPairs);
300de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
301f64a7a83bea5f1d2ab1e71231616c6cb0487d56eHal Finkel    void pruneDAGFor(
30297a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel             DenseMap<Value *, std::vector<Value *> > &CandidatePairs,
30397a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel             std::vector<Value *> &PairableInsts,
30497a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel             DenseMap<ValuePair, std::vector<ValuePair> > &ConnectedPairs,
30597a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel             DenseSet<ValuePair> &PairableInstUsers,
30697a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel             DenseMap<ValuePair, std::vector<ValuePair> > &PairableInstUserMap,
30797a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel             DenseSet<VPPair> &PairableInstUserPairSet,
30897a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel             DenseMap<Value *, Value *> &ChosenPairs,
309f64a7a83bea5f1d2ab1e71231616c6cb0487d56eHal Finkel             DenseMap<ValuePair, size_t> &DAG,
310f64a7a83bea5f1d2ab1e71231616c6cb0487d56eHal Finkel             DenseSet<ValuePair> &PrunedDAG, ValuePair J,
31197a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel             bool UseCycleCheck);
312de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
313f64a7a83bea5f1d2ab1e71231616c6cb0487d56eHal Finkel    void buildInitialDAGFor(
31497a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel             DenseMap<Value *, std::vector<Value *> > &CandidatePairs,
31597a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel             DenseSet<ValuePair> &CandidatePairsSet,
31697a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel             std::vector<Value *> &PairableInsts,
31797a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel             DenseMap<ValuePair, std::vector<ValuePair> > &ConnectedPairs,
31897a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel             DenseSet<ValuePair> &PairableInstUsers,
31997a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel             DenseMap<Value *, Value *> &ChosenPairs,
320f64a7a83bea5f1d2ab1e71231616c6cb0487d56eHal Finkel             DenseMap<ValuePair, size_t> &DAG, ValuePair J);
321de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
322f64a7a83bea5f1d2ab1e71231616c6cb0487d56eHal Finkel    void findBestDAGFor(
32397a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel             DenseMap<Value *, std::vector<Value *> > &CandidatePairs,
32497a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel             DenseSet<ValuePair> &CandidatePairsSet,
32597a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel             DenseMap<ValuePair, int> &CandidatePairCostSavings,
32697a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel             std::vector<Value *> &PairableInsts,
32797a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel             DenseSet<ValuePair> &FixedOrderPairs,
32897a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel             DenseMap<VPPair, unsigned> &PairConnectionTypes,
32997a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel             DenseMap<ValuePair, std::vector<ValuePair> > &ConnectedPairs,
33097a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel             DenseMap<ValuePair, std::vector<ValuePair> > &ConnectedPairDeps,
33197a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel             DenseSet<ValuePair> &PairableInstUsers,
33297a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel             DenseMap<ValuePair, std::vector<ValuePair> > &PairableInstUserMap,
33397a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel             DenseSet<VPPair> &PairableInstUserPairSet,
33497a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel             DenseMap<Value *, Value *> &ChosenPairs,
335f64a7a83bea5f1d2ab1e71231616c6cb0487d56eHal Finkel             DenseSet<ValuePair> &BestDAG, size_t &BestMaxDepth,
33697a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel             int &BestEffSize, Value *II, std::vector<Value *>&JJ,
33797a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel             bool UseCycleCheck);
338de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
339de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    Value *getReplacementPointerInput(LLVMContext& Context, Instruction *I,
340202d1cb8a587a9513d8bb65bf4a3d88a55132860Hal Finkel                     Instruction *J, unsigned o);
341de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
342de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    void fillNewShuffleMask(LLVMContext& Context, Instruction *J,
34364e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel                     unsigned MaskOffset, unsigned NumInElem,
34464e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel                     unsigned NumInElem1, unsigned IdxOffset,
34564e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel                     std::vector<Constant*> &Mask);
346de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
347de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    Value *getReplacementShuffleMask(LLVMContext& Context, Instruction *I,
348de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel                     Instruction *J);
349de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
35064e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel    bool expandIEChain(LLVMContext& Context, Instruction *I, Instruction *J,
35164e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel                       unsigned o, Value *&LOp, unsigned numElemL,
35272465ea23d010507d3746adc126d719005981e05Hal Finkel                       Type *ArgTypeL, Type *ArgTypeR, bool IBeforeJ,
35364e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel                       unsigned IdxOff = 0);
35464e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel
355de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    Value *getReplacementInput(LLVMContext& Context, Instruction *I,
35672465ea23d010507d3746adc126d719005981e05Hal Finkel                     Instruction *J, unsigned o, bool IBeforeJ);
357de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
358de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    void getReplacementInputsForPair(LLVMContext& Context, Instruction *I,
35972465ea23d010507d3746adc126d719005981e05Hal Finkel                     Instruction *J, SmallVector<Value *, 3> &ReplacedOperands,
36072465ea23d010507d3746adc126d719005981e05Hal Finkel                     bool IBeforeJ);
361de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
362de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    void replaceOutputsOfPair(LLVMContext& Context, Instruction *I,
363de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel                     Instruction *J, Instruction *K,
364de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel                     Instruction *&InsertionPt, Instruction *&K1,
365202d1cb8a587a9513d8bb65bf4a3d88a55132860Hal Finkel                     Instruction *&K2);
366de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
367de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    void collectPairLoadMoveSet(BasicBlock &BB,
368de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel                     DenseMap<Value *, Value *> &ChosenPairs,
36997a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel                     DenseMap<Value *, std::vector<Value *> > &LoadMoveSet,
3702f0e63cc16feb39480805bd00f53bbe5e3031d29Hal Finkel                     DenseSet<ValuePair> &LoadMoveSetPairs,
371de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel                     Instruction *I);
372de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
373de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    void collectLoadMoveSet(BasicBlock &BB,
374de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel                     std::vector<Value *> &PairableInsts,
375de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel                     DenseMap<Value *, Value *> &ChosenPairs,
37697a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel                     DenseMap<Value *, std::vector<Value *> > &LoadMoveSet,
3772f0e63cc16feb39480805bd00f53bbe5e3031d29Hal Finkel                     DenseSet<ValuePair> &LoadMoveSetPairs);
378de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
379de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    bool canMoveUsesOfIAfterJ(BasicBlock &BB,
3802f0e63cc16feb39480805bd00f53bbe5e3031d29Hal Finkel                     DenseSet<ValuePair> &LoadMoveSetPairs,
381de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel                     Instruction *I, Instruction *J);
382de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
383de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    void moveUsesOfIAfterJ(BasicBlock &BB,
3842f0e63cc16feb39480805bd00f53bbe5e3031d29Hal Finkel                     DenseSet<ValuePair> &LoadMoveSetPairs,
385de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel                     Instruction *&InsertionPt,
386de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel                     Instruction *I, Instruction *J);
387de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
388ab4684e26fe21857d8c8bc6ba7a5234c35117c83Hal Finkel    void combineMetadata(Instruction *K, const Instruction *J);
389ab4684e26fe21857d8c8bc6ba7a5234c35117c83Hal Finkel
39087825e7970a361ce5a8bab19bc880ff7f6242ca2Hongbin Zheng    bool vectorizeBB(BasicBlock &BB) {
391e29c19091cca58db668407dfc5dd86c70e8b3d49Hal Finkel      if (!DT->isReachableFromEntry(&BB)) {
392e29c19091cca58db668407dfc5dd86c70e8b3d49Hal Finkel        DEBUG(dbgs() << "BBV: skipping unreachable " << BB.getName() <<
393e29c19091cca58db668407dfc5dd86c70e8b3d49Hal Finkel              " in " << BB.getParent()->getName() << "\n");
394e29c19091cca58db668407dfc5dd86c70e8b3d49Hal Finkel        return false;
395e29c19091cca58db668407dfc5dd86c70e8b3d49Hal Finkel      }
396e29c19091cca58db668407dfc5dd86c70e8b3d49Hal Finkel
397abc227d9b39baed266c453f459ca14242f5c1eeaChandler Carruth      DEBUG(if (TTI) dbgs() << "BBV: using target information\n");
39865309660fa61a837cc05323f69c618a7d8134d56Hal Finkel
399de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      bool changed = false;
400de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      // Iterate a sufficient number of times to merge types of size 1 bit,
401de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      // then 2 bits, then 4, etc. up to half of the target vector width of the
402de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      // target vector register.
40364e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel      unsigned n = 1;
40464e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel      for (unsigned v = 2;
405abc227d9b39baed266c453f459ca14242f5c1eeaChandler Carruth           (TTI || v <= Config.VectorBits) &&
40665309660fa61a837cc05323f69c618a7d8134d56Hal Finkel           (!Config.MaxIter || n <= Config.MaxIter);
407de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel           v *= 2, ++n) {
408bef377b7d7ce31edb40c87f8786d1b7bb6cdd6b1Hongbin Zheng        DEBUG(dbgs() << "BBV: fusing loop #" << n <<
409de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel              " for " << BB.getName() << " in " <<
410de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel              BB.getParent()->getName() << "...\n");
411de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        if (vectorizePairs(BB))
412de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel          changed = true;
413de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        else
414de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel          break;
415de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      }
416de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
41764e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel      if (changed && !Pow2LenOnly) {
41864e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        ++n;
41964e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        for (; !Config.MaxIter || n <= Config.MaxIter; ++n) {
42064e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel          DEBUG(dbgs() << "BBV: fusing for non-2^n-length vectors loop #: " <<
42164e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel                n << " for " << BB.getName() << " in " <<
42264e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel                BB.getParent()->getName() << "...\n");
42364e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel          if (!vectorizePairs(BB, true)) break;
42464e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        }
42564e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel      }
42664e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel
427de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      DEBUG(dbgs() << "BBV: done!\n");
428de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      return changed;
429de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    }
430de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
43187825e7970a361ce5a8bab19bc880ff7f6242ca2Hongbin Zheng    virtual bool runOnBasicBlock(BasicBlock &BB) {
43287825e7970a361ce5a8bab19bc880ff7f6242ca2Hongbin Zheng      AA = &getAnalysis<AliasAnalysis>();
433e29c19091cca58db668407dfc5dd86c70e8b3d49Hal Finkel      DT = &getAnalysis<DominatorTree>();
43487825e7970a361ce5a8bab19bc880ff7f6242ca2Hongbin Zheng      SE = &getAnalysis<ScalarEvolution>();
4353574eca1b02600bac4e625297f4ecf745f4c4f32Micah Villmow      TD = getAnalysisIfAvailable<DataLayout>();
4368bd6c52396ab6e7955fdcc1bce099b7cba29a308Chandler Carruth      TTI = IgnoreTargetInfo ? 0 : &getAnalysis<TargetTransformInfo>();
43787825e7970a361ce5a8bab19bc880ff7f6242ca2Hongbin Zheng
43887825e7970a361ce5a8bab19bc880ff7f6242ca2Hongbin Zheng      return vectorizeBB(BB);
43987825e7970a361ce5a8bab19bc880ff7f6242ca2Hongbin Zheng    }
44087825e7970a361ce5a8bab19bc880ff7f6242ca2Hongbin Zheng
441de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    virtual void getAnalysisUsage(AnalysisUsage &AU) const {
442de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      BasicBlockPass::getAnalysisUsage(AU);
443de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      AU.addRequired<AliasAnalysis>();
444e29c19091cca58db668407dfc5dd86c70e8b3d49Hal Finkel      AU.addRequired<DominatorTree>();
445de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      AU.addRequired<ScalarEvolution>();
4468bd6c52396ab6e7955fdcc1bce099b7cba29a308Chandler Carruth      AU.addRequired<TargetTransformInfo>();
447de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      AU.addPreserved<AliasAnalysis>();
448e29c19091cca58db668407dfc5dd86c70e8b3d49Hal Finkel      AU.addPreserved<DominatorTree>();
449de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      AU.addPreserved<ScalarEvolution>();
4507e004d177fe76145f75a9417ed2e281f1b9abaf7Hal Finkel      AU.setPreservesCFG();
451de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    }
452de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
45364e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel    static inline VectorType *getVecTypeForPair(Type *ElemTy, Type *Elem2Ty) {
45464e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel      assert(ElemTy->getScalarType() == Elem2Ty->getScalarType() &&
45564e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel             "Cannot form vector from incompatible scalar types");
45664e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel      Type *STy = ElemTy->getScalarType();
45764e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel
45864e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel      unsigned numElem;
459de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      if (VectorType *VTy = dyn_cast<VectorType>(ElemTy)) {
46064e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        numElem = VTy->getNumElements();
46164e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel      } else {
46264e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        numElem = 1;
46364e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel      }
46464e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel
46564e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel      if (VectorType *VTy = dyn_cast<VectorType>(Elem2Ty)) {
46664e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        numElem += VTy->getNumElements();
46764e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel      } else {
46864e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        numElem += 1;
469de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      }
4707e004d177fe76145f75a9417ed2e281f1b9abaf7Hal Finkel
47164e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel      return VectorType::get(STy, numElem);
47264e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel    }
47364e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel
47464e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel    static inline void getInstructionTypes(Instruction *I,
47564e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel                                           Type *&T1, Type *&T2) {
4763fc1e4aa159ec15058bb26acbec39f6e09990207Hal Finkel      if (StoreInst *SI = dyn_cast<StoreInst>(I)) {
47764e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        // For stores, it is the value type, not the pointer type that matters
47864e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        // because the value is what will come from a vector register.
47964e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel
4803fc1e4aa159ec15058bb26acbec39f6e09990207Hal Finkel        Value *IVal = SI->getValueOperand();
48164e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        T1 = IVal->getType();
48264e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel      } else {
48364e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        T1 = I->getType();
48464e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel      }
48564e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel
4863fc1e4aa159ec15058bb26acbec39f6e09990207Hal Finkel      if (CastInst *CI = dyn_cast<CastInst>(I))
4873fc1e4aa159ec15058bb26acbec39f6e09990207Hal Finkel        T2 = CI->getSrcTy();
48864e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel      else
48964e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        T2 = T1;
49065309660fa61a837cc05323f69c618a7d8134d56Hal Finkel
49165309660fa61a837cc05323f69c618a7d8134d56Hal Finkel      if (SelectInst *SI = dyn_cast<SelectInst>(I)) {
49265309660fa61a837cc05323f69c618a7d8134d56Hal Finkel        T2 = SI->getCondition()->getType();
4938b9796f4f83bea2bbefcd2822eb574abdb7f3d1bHal Finkel      } else if (ShuffleVectorInst *SI = dyn_cast<ShuffleVectorInst>(I)) {
4948b9796f4f83bea2bbefcd2822eb574abdb7f3d1bHal Finkel        T2 = SI->getOperand(0)->getType();
4955094257518ea7b615d87ef5bea657625ffa81991Hal Finkel      } else if (CmpInst *CI = dyn_cast<CmpInst>(I)) {
4965094257518ea7b615d87ef5bea657625ffa81991Hal Finkel        T2 = CI->getOperand(0)->getType();
49765309660fa61a837cc05323f69c618a7d8134d56Hal Finkel      }
498de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    }
499de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
500de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    // Returns the weight associated with the provided value. A chain of
501de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    // candidate pairs has a length given by the sum of the weights of its
502de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    // members (one weight per pair; the weight of each member of the pair
503de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    // is assumed to be the same). This length is then compared to the
504de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    // chain-length threshold to determine if a given chain is significant
505de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    // enough to be vectorized. The length is also used in comparing
506de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    // candidate chains where longer chains are considered to be better.
507de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    // Note: when this function returns 0, the resulting instructions are
508de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    // not actually fused.
509bef377b7d7ce31edb40c87f8786d1b7bb6cdd6b1Hongbin Zheng    inline size_t getDepthFactor(Value *V) {
510de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      // InsertElement and ExtractElement have a depth factor of zero. This is
511de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      // for two reasons: First, they cannot be usefully fused. Second, because
512de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      // the pass generates a lot of these, they can confuse the simple metric
513f64a7a83bea5f1d2ab1e71231616c6cb0487d56eHal Finkel      // used to compare the dags in the next iteration. Thus, giving them a
514de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      // weight of zero allows the pass to essentially ignore them in
515de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      // subsequent iterations when looking for vectorization opportunities
516de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      // while still tracking dependency chains that flow through those
517de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      // instructions.
518de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      if (isa<InsertElementInst>(V) || isa<ExtractElementInst>(V))
519de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        return 0;
520de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
521edc8db87dc2ed4d2971e7f50464f5f4d0fead537Hal Finkel      // Give a load or store half of the required depth so that load/store
522edc8db87dc2ed4d2971e7f50464f5f4d0fead537Hal Finkel      // pairs will vectorize.
523bef377b7d7ce31edb40c87f8786d1b7bb6cdd6b1Hongbin Zheng      if (!Config.NoMemOpBoost && (isa<LoadInst>(V) || isa<StoreInst>(V)))
524bef377b7d7ce31edb40c87f8786d1b7bb6cdd6b1Hongbin Zheng        return Config.ReqChainDepth/2;
525edc8db87dc2ed4d2971e7f50464f5f4d0fead537Hal Finkel
526de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      return 1;
527de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    }
528de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
529abc227d9b39baed266c453f459ca14242f5c1eeaChandler Carruth    // Returns the cost of the provided instruction using TTI.
53046fb81cf4009cc34af97c5a1c0e824e2633fb4e4Hal Finkel    // This does not handle loads and stores.
53146fb81cf4009cc34af97c5a1c0e824e2633fb4e4Hal Finkel    unsigned getInstrCost(unsigned Opcode, Type *T1, Type *T2) {
53246fb81cf4009cc34af97c5a1c0e824e2633fb4e4Hal Finkel      switch (Opcode) {
53346fb81cf4009cc34af97c5a1c0e824e2633fb4e4Hal Finkel      default: break;
53446fb81cf4009cc34af97c5a1c0e824e2633fb4e4Hal Finkel      case Instruction::GetElementPtr:
53546fb81cf4009cc34af97c5a1c0e824e2633fb4e4Hal Finkel        // We mark this instruction as zero-cost because scalar GEPs are usually
53646fb81cf4009cc34af97c5a1c0e824e2633fb4e4Hal Finkel        // lowered to the intruction addressing mode. At the moment we don't
53746fb81cf4009cc34af97c5a1c0e824e2633fb4e4Hal Finkel        // generate vector GEPs.
53846fb81cf4009cc34af97c5a1c0e824e2633fb4e4Hal Finkel        return 0;
53946fb81cf4009cc34af97c5a1c0e824e2633fb4e4Hal Finkel      case Instruction::Br:
540abc227d9b39baed266c453f459ca14242f5c1eeaChandler Carruth        return TTI->getCFInstrCost(Opcode);
54146fb81cf4009cc34af97c5a1c0e824e2633fb4e4Hal Finkel      case Instruction::PHI:
54246fb81cf4009cc34af97c5a1c0e824e2633fb4e4Hal Finkel        return 0;
54346fb81cf4009cc34af97c5a1c0e824e2633fb4e4Hal Finkel      case Instruction::Add:
54446fb81cf4009cc34af97c5a1c0e824e2633fb4e4Hal Finkel      case Instruction::FAdd:
54546fb81cf4009cc34af97c5a1c0e824e2633fb4e4Hal Finkel      case Instruction::Sub:
54646fb81cf4009cc34af97c5a1c0e824e2633fb4e4Hal Finkel      case Instruction::FSub:
54746fb81cf4009cc34af97c5a1c0e824e2633fb4e4Hal Finkel      case Instruction::Mul:
54846fb81cf4009cc34af97c5a1c0e824e2633fb4e4Hal Finkel      case Instruction::FMul:
54946fb81cf4009cc34af97c5a1c0e824e2633fb4e4Hal Finkel      case Instruction::UDiv:
55046fb81cf4009cc34af97c5a1c0e824e2633fb4e4Hal Finkel      case Instruction::SDiv:
55146fb81cf4009cc34af97c5a1c0e824e2633fb4e4Hal Finkel      case Instruction::FDiv:
55246fb81cf4009cc34af97c5a1c0e824e2633fb4e4Hal Finkel      case Instruction::URem:
55346fb81cf4009cc34af97c5a1c0e824e2633fb4e4Hal Finkel      case Instruction::SRem:
55446fb81cf4009cc34af97c5a1c0e824e2633fb4e4Hal Finkel      case Instruction::FRem:
55546fb81cf4009cc34af97c5a1c0e824e2633fb4e4Hal Finkel      case Instruction::Shl:
55646fb81cf4009cc34af97c5a1c0e824e2633fb4e4Hal Finkel      case Instruction::LShr:
55746fb81cf4009cc34af97c5a1c0e824e2633fb4e4Hal Finkel      case Instruction::AShr:
55846fb81cf4009cc34af97c5a1c0e824e2633fb4e4Hal Finkel      case Instruction::And:
55946fb81cf4009cc34af97c5a1c0e824e2633fb4e4Hal Finkel      case Instruction::Or:
56046fb81cf4009cc34af97c5a1c0e824e2633fb4e4Hal Finkel      case Instruction::Xor:
561abc227d9b39baed266c453f459ca14242f5c1eeaChandler Carruth        return TTI->getArithmeticInstrCost(Opcode, T1);
56246fb81cf4009cc34af97c5a1c0e824e2633fb4e4Hal Finkel      case Instruction::Select:
56346fb81cf4009cc34af97c5a1c0e824e2633fb4e4Hal Finkel      case Instruction::ICmp:
56446fb81cf4009cc34af97c5a1c0e824e2633fb4e4Hal Finkel      case Instruction::FCmp:
565abc227d9b39baed266c453f459ca14242f5c1eeaChandler Carruth        return TTI->getCmpSelInstrCost(Opcode, T1, T2);
56646fb81cf4009cc34af97c5a1c0e824e2633fb4e4Hal Finkel      case Instruction::ZExt:
56746fb81cf4009cc34af97c5a1c0e824e2633fb4e4Hal Finkel      case Instruction::SExt:
56846fb81cf4009cc34af97c5a1c0e824e2633fb4e4Hal Finkel      case Instruction::FPToUI:
56946fb81cf4009cc34af97c5a1c0e824e2633fb4e4Hal Finkel      case Instruction::FPToSI:
57046fb81cf4009cc34af97c5a1c0e824e2633fb4e4Hal Finkel      case Instruction::FPExt:
57146fb81cf4009cc34af97c5a1c0e824e2633fb4e4Hal Finkel      case Instruction::PtrToInt:
57246fb81cf4009cc34af97c5a1c0e824e2633fb4e4Hal Finkel      case Instruction::IntToPtr:
57346fb81cf4009cc34af97c5a1c0e824e2633fb4e4Hal Finkel      case Instruction::SIToFP:
57446fb81cf4009cc34af97c5a1c0e824e2633fb4e4Hal Finkel      case Instruction::UIToFP:
57546fb81cf4009cc34af97c5a1c0e824e2633fb4e4Hal Finkel      case Instruction::Trunc:
57646fb81cf4009cc34af97c5a1c0e824e2633fb4e4Hal Finkel      case Instruction::FPTrunc:
57746fb81cf4009cc34af97c5a1c0e824e2633fb4e4Hal Finkel      case Instruction::BitCast:
57886ccc55c82651f91fd6a312c5f6a4b511bcd1aecHal Finkel      case Instruction::ShuffleVector:
579abc227d9b39baed266c453f459ca14242f5c1eeaChandler Carruth        return TTI->getCastInstrCost(Opcode, T1, T2);
58046fb81cf4009cc34af97c5a1c0e824e2633fb4e4Hal Finkel      }
58146fb81cf4009cc34af97c5a1c0e824e2633fb4e4Hal Finkel
58246fb81cf4009cc34af97c5a1c0e824e2633fb4e4Hal Finkel      return 1;
58346fb81cf4009cc34af97c5a1c0e824e2633fb4e4Hal Finkel    }
58446fb81cf4009cc34af97c5a1c0e824e2633fb4e4Hal Finkel
585de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    // This determines the relative offset of two loads or stores, returning
586de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    // true if the offset could be determined to be some constant value.
587de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    // For example, if OffsetInElmts == 1, then J accesses the memory directly
588de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    // after I; if OffsetInElmts == -1 then I accesses the memory
58964e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel    // directly after J.
590de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    bool getPairPtrInfo(Instruction *I, Instruction *J,
591de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        Value *&IPtr, Value *&JPtr, unsigned &IAlignment, unsigned &JAlignment,
59265309660fa61a837cc05323f69c618a7d8134d56Hal Finkel        unsigned &IAddressSpace, unsigned &JAddressSpace,
59393f6f457614299eee3d22f376ab8f42a130f1912Hal Finkel        int64_t &OffsetInElmts, bool ComputeOffset = true) {
594de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      OffsetInElmts = 0;
59565309660fa61a837cc05323f69c618a7d8134d56Hal Finkel      if (LoadInst *LI = dyn_cast<LoadInst>(I)) {
59665309660fa61a837cc05323f69c618a7d8134d56Hal Finkel        LoadInst *LJ = cast<LoadInst>(J);
59765309660fa61a837cc05323f69c618a7d8134d56Hal Finkel        IPtr = LI->getPointerOperand();
59865309660fa61a837cc05323f69c618a7d8134d56Hal Finkel        JPtr = LJ->getPointerOperand();
59965309660fa61a837cc05323f69c618a7d8134d56Hal Finkel        IAlignment = LI->getAlignment();
60065309660fa61a837cc05323f69c618a7d8134d56Hal Finkel        JAlignment = LJ->getAlignment();
60165309660fa61a837cc05323f69c618a7d8134d56Hal Finkel        IAddressSpace = LI->getPointerAddressSpace();
60265309660fa61a837cc05323f69c618a7d8134d56Hal Finkel        JAddressSpace = LJ->getPointerAddressSpace();
603de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      } else {
60465309660fa61a837cc05323f69c618a7d8134d56Hal Finkel        StoreInst *SI = cast<StoreInst>(I), *SJ = cast<StoreInst>(J);
60565309660fa61a837cc05323f69c618a7d8134d56Hal Finkel        IPtr = SI->getPointerOperand();
60665309660fa61a837cc05323f69c618a7d8134d56Hal Finkel        JPtr = SJ->getPointerOperand();
60765309660fa61a837cc05323f69c618a7d8134d56Hal Finkel        IAlignment = SI->getAlignment();
60865309660fa61a837cc05323f69c618a7d8134d56Hal Finkel        JAlignment = SJ->getAlignment();
60965309660fa61a837cc05323f69c618a7d8134d56Hal Finkel        IAddressSpace = SI->getPointerAddressSpace();
61065309660fa61a837cc05323f69c618a7d8134d56Hal Finkel        JAddressSpace = SJ->getPointerAddressSpace();
611de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      }
612de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
61393f6f457614299eee3d22f376ab8f42a130f1912Hal Finkel      if (!ComputeOffset)
61493f6f457614299eee3d22f376ab8f42a130f1912Hal Finkel        return true;
61593f6f457614299eee3d22f376ab8f42a130f1912Hal Finkel
616de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      const SCEV *IPtrSCEV = SE->getSCEV(IPtr);
617de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      const SCEV *JPtrSCEV = SE->getSCEV(JPtr);
618de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
619de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      // If this is a trivial offset, then we'll get something like
620de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      // 1*sizeof(type). With target data, which we need anyway, this will get
621de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      // constant folded into a number.
622de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      const SCEV *OffsetSCEV = SE->getMinusSCEV(JPtrSCEV, IPtrSCEV);
623de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      if (const SCEVConstant *ConstOffSCEV =
624de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel            dyn_cast<SCEVConstant>(OffsetSCEV)) {
625de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        ConstantInt *IntOff = ConstOffSCEV->getValue();
626de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        int64_t Offset = IntOff->getSExtValue();
627de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
628de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        Type *VTy = cast<PointerType>(IPtr->getType())->getElementType();
629de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        int64_t VTyTSS = (int64_t) TD->getTypeStoreSize(VTy);
630de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
63164e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        Type *VTy2 = cast<PointerType>(JPtr->getType())->getElementType();
63264e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        if (VTy != VTy2 && Offset < 0) {
63364e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel          int64_t VTy2TSS = (int64_t) TD->getTypeStoreSize(VTy2);
63464e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel          OffsetInElmts = Offset/VTy2TSS;
63564e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel          return (abs64(Offset) % VTy2TSS) == 0;
63664e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        }
637de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
638de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        OffsetInElmts = Offset/VTyTSS;
639de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        return (abs64(Offset) % VTyTSS) == 0;
640de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      }
641de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
642de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      return false;
643de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    }
644de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
645de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    // Returns true if the provided CallInst represents an intrinsic that can
646de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    // be vectorized.
647de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    bool isVectorizableIntrinsic(CallInst* I) {
648de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      Function *F = I->getCalledFunction();
649de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      if (!F) return false;
650de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
651a77728415857196035c0090f7b2749d7971811a2Hal Finkel      Intrinsic::ID IID = (Intrinsic::ID) F->getIntrinsicID();
652de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      if (!IID) return false;
653de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
654de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      switch(IID) {
655de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      default:
656de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        return false;
657de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      case Intrinsic::sqrt:
658de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      case Intrinsic::powi:
659de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      case Intrinsic::sin:
660de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      case Intrinsic::cos:
661de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      case Intrinsic::log:
662de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      case Intrinsic::log2:
663de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      case Intrinsic::log10:
664de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      case Intrinsic::exp:
665de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      case Intrinsic::exp2:
666de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      case Intrinsic::pow:
66786312cc15f29ce2bbd9647b94862e068045280c3Hongbin Zheng        return Config.VectorizeMath;
668de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      case Intrinsic::fma:
66964a7a24edf719bb6ffacc030c23f4cd99312f3fbHal Finkel      case Intrinsic::fmuladd:
67086312cc15f29ce2bbd9647b94862e068045280c3Hongbin Zheng        return Config.VectorizeFMA;
671de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      }
672de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    }
673de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
674b2b2469a9178f7e22cd7a69f3093e54d67d6b712Hal Finkel    bool isPureIEChain(InsertElementInst *IE) {
675b2b2469a9178f7e22cd7a69f3093e54d67d6b712Hal Finkel      InsertElementInst *IENext = IE;
676b2b2469a9178f7e22cd7a69f3093e54d67d6b712Hal Finkel      do {
677b2b2469a9178f7e22cd7a69f3093e54d67d6b712Hal Finkel        if (!isa<UndefValue>(IENext->getOperand(0)) &&
678b2b2469a9178f7e22cd7a69f3093e54d67d6b712Hal Finkel            !isa<InsertElementInst>(IENext->getOperand(0))) {
679b2b2469a9178f7e22cd7a69f3093e54d67d6b712Hal Finkel          return false;
680b2b2469a9178f7e22cd7a69f3093e54d67d6b712Hal Finkel        }
681b2b2469a9178f7e22cd7a69f3093e54d67d6b712Hal Finkel      } while ((IENext =
682b2b2469a9178f7e22cd7a69f3093e54d67d6b712Hal Finkel                 dyn_cast<InsertElementInst>(IENext->getOperand(0))));
683b2b2469a9178f7e22cd7a69f3093e54d67d6b712Hal Finkel
684b2b2469a9178f7e22cd7a69f3093e54d67d6b712Hal Finkel      return true;
685b2b2469a9178f7e22cd7a69f3093e54d67d6b712Hal Finkel    }
686de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  };
687de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
688de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  // This function implements one vectorization iteration on the provided
689de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  // basic block. It returns true if the block is changed.
69064e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel  bool BBVectorize::vectorizePairs(BasicBlock &BB, bool NonPow2Len) {
6915d4e18bc39fea892f523d960213906d296d3cb38Hal Finkel    bool ShouldContinue;
6925d4e18bc39fea892f523d960213906d296d3cb38Hal Finkel    BasicBlock::iterator Start = BB.getFirstInsertionPt();
6935d4e18bc39fea892f523d960213906d296d3cb38Hal Finkel
6945d4e18bc39fea892f523d960213906d296d3cb38Hal Finkel    std::vector<Value *> AllPairableInsts;
6955d4e18bc39fea892f523d960213906d296d3cb38Hal Finkel    DenseMap<Value *, Value *> AllChosenPairs;
696a9779bfbc9ab0cf3f157453fd0afd110b04a9fdcHal Finkel    DenseSet<ValuePair> AllFixedOrderPairs;
69772465ea23d010507d3746adc126d719005981e05Hal Finkel    DenseMap<VPPair, unsigned> AllPairConnectionTypes;
69897a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel    DenseMap<ValuePair, std::vector<ValuePair> > AllConnectedPairs,
69997a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel                                                 AllConnectedPairDeps;
7005d4e18bc39fea892f523d960213906d296d3cb38Hal Finkel
7015d4e18bc39fea892f523d960213906d296d3cb38Hal Finkel    do {
7025d4e18bc39fea892f523d960213906d296d3cb38Hal Finkel      std::vector<Value *> PairableInsts;
7036ca6d3b1eac5b8611f3a9e2c270c2e794d37e1f5Hal Finkel      DenseMap<Value *, std::vector<Value *> > CandidatePairs;
704a9779bfbc9ab0cf3f157453fd0afd110b04a9fdcHal Finkel      DenseSet<ValuePair> FixedOrderPairs;
70565309660fa61a837cc05323f69c618a7d8134d56Hal Finkel      DenseMap<ValuePair, int> CandidatePairCostSavings;
7065d4e18bc39fea892f523d960213906d296d3cb38Hal Finkel      ShouldContinue = getCandidatePairs(BB, Start, CandidatePairs,
707a9779bfbc9ab0cf3f157453fd0afd110b04a9fdcHal Finkel                                         FixedOrderPairs,
70865309660fa61a837cc05323f69c618a7d8134d56Hal Finkel                                         CandidatePairCostSavings,
70964e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel                                         PairableInsts, NonPow2Len);
7105d4e18bc39fea892f523d960213906d296d3cb38Hal Finkel      if (PairableInsts.empty()) continue;
7113706ac7aa83ab0aed9e2da7d5fc2386ac1f035f5Sebastian Pop
712b1a82589339fed148c12b052d30861a539552f1aHal Finkel      // Build the candidate pair set for faster lookups.
713b1a82589339fed148c12b052d30861a539552f1aHal Finkel      DenseSet<ValuePair> CandidatePairsSet;
7146ca6d3b1eac5b8611f3a9e2c270c2e794d37e1f5Hal Finkel      for (DenseMap<Value *, std::vector<Value *> >::iterator I =
7156ca6d3b1eac5b8611f3a9e2c270c2e794d37e1f5Hal Finkel           CandidatePairs.begin(), E = CandidatePairs.end(); I != E; ++I)
7166ca6d3b1eac5b8611f3a9e2c270c2e794d37e1f5Hal Finkel        for (std::vector<Value *>::iterator J = I->second.begin(),
7176ca6d3b1eac5b8611f3a9e2c270c2e794d37e1f5Hal Finkel             JE = I->second.end(); J != JE; ++J)
7186ca6d3b1eac5b8611f3a9e2c270c2e794d37e1f5Hal Finkel          CandidatePairsSet.insert(ValuePair(I->first, *J));
719b1a82589339fed148c12b052d30861a539552f1aHal Finkel
7205d4e18bc39fea892f523d960213906d296d3cb38Hal Finkel      // Now we have a map of all of the pairable instructions and we need to
7215d4e18bc39fea892f523d960213906d296d3cb38Hal Finkel      // select the best possible pairing. A good pairing is one such that the
7225d4e18bc39fea892f523d960213906d296d3cb38Hal Finkel      // users of the pair are also paired. This defines a (directed) forest
72394c22716d60ff5edf6a98a3c67e0faa001be1142Sylvestre Ledru      // over the pairs such that two pairs are connected iff the second pair
7245d4e18bc39fea892f523d960213906d296d3cb38Hal Finkel      // uses the first.
7253706ac7aa83ab0aed9e2da7d5fc2386ac1f035f5Sebastian Pop
7265d4e18bc39fea892f523d960213906d296d3cb38Hal Finkel      // Note that it only matters that both members of the second pair use some
7275d4e18bc39fea892f523d960213906d296d3cb38Hal Finkel      // element of the first pair (to allow for splatting).
7283706ac7aa83ab0aed9e2da7d5fc2386ac1f035f5Sebastian Pop
72997a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel      DenseMap<ValuePair, std::vector<ValuePair> > ConnectedPairs,
73097a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel                                                   ConnectedPairDeps;
73172465ea23d010507d3746adc126d719005981e05Hal Finkel      DenseMap<VPPair, unsigned> PairConnectionTypes;
732b1a82589339fed148c12b052d30861a539552f1aHal Finkel      computeConnectedPairs(CandidatePairs, CandidatePairsSet,
733b1a82589339fed148c12b052d30861a539552f1aHal Finkel                            PairableInsts, ConnectedPairs, PairConnectionTypes);
7345d4e18bc39fea892f523d960213906d296d3cb38Hal Finkel      if (ConnectedPairs.empty()) continue;
7353706ac7aa83ab0aed9e2da7d5fc2386ac1f035f5Sebastian Pop
73697a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel      for (DenseMap<ValuePair, std::vector<ValuePair> >::iterator
73772465ea23d010507d3746adc126d719005981e05Hal Finkel           I = ConnectedPairs.begin(), IE = ConnectedPairs.end();
73897a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel           I != IE; ++I)
73997a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel        for (std::vector<ValuePair>::iterator J = I->second.begin(),
74097a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel             JE = I->second.end(); J != JE; ++J)
74197a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel          ConnectedPairDeps[*J].push_back(I->first);
74272465ea23d010507d3746adc126d719005981e05Hal Finkel
7435d4e18bc39fea892f523d960213906d296d3cb38Hal Finkel      // Build the pairable-instruction dependency map
7445d4e18bc39fea892f523d960213906d296d3cb38Hal Finkel      DenseSet<ValuePair> PairableInstUsers;
7455d4e18bc39fea892f523d960213906d296d3cb38Hal Finkel      buildDepMap(BB, CandidatePairs, PairableInsts, PairableInstUsers);
7463706ac7aa83ab0aed9e2da7d5fc2386ac1f035f5Sebastian Pop
74735564dc3ae1c377abad425cb09928eaf676dcb3cHal Finkel      // There is now a graph of the connected pairs. For each variable, pick
748f64a7a83bea5f1d2ab1e71231616c6cb0487d56eHal Finkel      // the pairing with the largest dag meeting the depth requirement on at
749f64a7a83bea5f1d2ab1e71231616c6cb0487d56eHal Finkel      // least one branch. Then select all pairings that are part of that dag
75035564dc3ae1c377abad425cb09928eaf676dcb3cHal Finkel      // and remove them from the list of available pairings and pairable
75135564dc3ae1c377abad425cb09928eaf676dcb3cHal Finkel      // variables.
7523706ac7aa83ab0aed9e2da7d5fc2386ac1f035f5Sebastian Pop
7535d4e18bc39fea892f523d960213906d296d3cb38Hal Finkel      DenseMap<Value *, Value *> ChosenPairs;
754b1a82589339fed148c12b052d30861a539552f1aHal Finkel      choosePairs(CandidatePairs, CandidatePairsSet,
755b1a82589339fed148c12b052d30861a539552f1aHal Finkel        CandidatePairCostSavings,
75686ccc55c82651f91fd6a312c5f6a4b511bcd1aecHal Finkel        PairableInsts, FixedOrderPairs, PairConnectionTypes,
75786ccc55c82651f91fd6a312c5f6a4b511bcd1aecHal Finkel        ConnectedPairs, ConnectedPairDeps,
7585d4e18bc39fea892f523d960213906d296d3cb38Hal Finkel        PairableInstUsers, ChosenPairs);
7593706ac7aa83ab0aed9e2da7d5fc2386ac1f035f5Sebastian Pop
7605d4e18bc39fea892f523d960213906d296d3cb38Hal Finkel      if (ChosenPairs.empty()) continue;
7615d4e18bc39fea892f523d960213906d296d3cb38Hal Finkel      AllPairableInsts.insert(AllPairableInsts.end(), PairableInsts.begin(),
7625d4e18bc39fea892f523d960213906d296d3cb38Hal Finkel                              PairableInsts.end());
7635d4e18bc39fea892f523d960213906d296d3cb38Hal Finkel      AllChosenPairs.insert(ChosenPairs.begin(), ChosenPairs.end());
764a9779bfbc9ab0cf3f157453fd0afd110b04a9fdcHal Finkel
76572465ea23d010507d3746adc126d719005981e05Hal Finkel      // Only for the chosen pairs, propagate information on fixed-order pairs,
76672465ea23d010507d3746adc126d719005981e05Hal Finkel      // pair connections, and their types to the data structures used by the
76772465ea23d010507d3746adc126d719005981e05Hal Finkel      // pair fusion procedures.
768a9779bfbc9ab0cf3f157453fd0afd110b04a9fdcHal Finkel      for (DenseMap<Value *, Value *>::iterator I = ChosenPairs.begin(),
769a9779bfbc9ab0cf3f157453fd0afd110b04a9fdcHal Finkel           IE = ChosenPairs.end(); I != IE; ++I) {
770a9779bfbc9ab0cf3f157453fd0afd110b04a9fdcHal Finkel        if (FixedOrderPairs.count(*I))
771a9779bfbc9ab0cf3f157453fd0afd110b04a9fdcHal Finkel          AllFixedOrderPairs.insert(*I);
772a9779bfbc9ab0cf3f157453fd0afd110b04a9fdcHal Finkel        else if (FixedOrderPairs.count(ValuePair(I->second, I->first)))
773a9779bfbc9ab0cf3f157453fd0afd110b04a9fdcHal Finkel          AllFixedOrderPairs.insert(ValuePair(I->second, I->first));
77472465ea23d010507d3746adc126d719005981e05Hal Finkel
77572465ea23d010507d3746adc126d719005981e05Hal Finkel        for (DenseMap<Value *, Value *>::iterator J = ChosenPairs.begin();
77672465ea23d010507d3746adc126d719005981e05Hal Finkel             J != IE; ++J) {
77772465ea23d010507d3746adc126d719005981e05Hal Finkel          DenseMap<VPPair, unsigned>::iterator K =
77872465ea23d010507d3746adc126d719005981e05Hal Finkel            PairConnectionTypes.find(VPPair(*I, *J));
77972465ea23d010507d3746adc126d719005981e05Hal Finkel          if (K != PairConnectionTypes.end()) {
78072465ea23d010507d3746adc126d719005981e05Hal Finkel            AllPairConnectionTypes.insert(*K);
78172465ea23d010507d3746adc126d719005981e05Hal Finkel          } else {
78272465ea23d010507d3746adc126d719005981e05Hal Finkel            K = PairConnectionTypes.find(VPPair(*J, *I));
78372465ea23d010507d3746adc126d719005981e05Hal Finkel            if (K != PairConnectionTypes.end())
78472465ea23d010507d3746adc126d719005981e05Hal Finkel              AllPairConnectionTypes.insert(*K);
78572465ea23d010507d3746adc126d719005981e05Hal Finkel          }
78672465ea23d010507d3746adc126d719005981e05Hal Finkel        }
78772465ea23d010507d3746adc126d719005981e05Hal Finkel      }
78872465ea23d010507d3746adc126d719005981e05Hal Finkel
78997a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel      for (DenseMap<ValuePair, std::vector<ValuePair> >::iterator
79072465ea23d010507d3746adc126d719005981e05Hal Finkel           I = ConnectedPairs.begin(), IE = ConnectedPairs.end();
79197a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel           I != IE; ++I)
79297a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel        for (std::vector<ValuePair>::iterator J = I->second.begin(),
79397a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel          JE = I->second.end(); J != JE; ++J)
79497a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel          if (AllPairConnectionTypes.count(VPPair(I->first, *J))) {
79597a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel            AllConnectedPairs[I->first].push_back(*J);
79697a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel            AllConnectedPairDeps[*J].push_back(I->first);
79797a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel          }
7985d4e18bc39fea892f523d960213906d296d3cb38Hal Finkel    } while (ShouldContinue);
7995d4e18bc39fea892f523d960213906d296d3cb38Hal Finkel
8005d4e18bc39fea892f523d960213906d296d3cb38Hal Finkel    if (AllChosenPairs.empty()) return false;
8015d4e18bc39fea892f523d960213906d296d3cb38Hal Finkel    NumFusedOps += AllChosenPairs.size();
8023706ac7aa83ab0aed9e2da7d5fc2386ac1f035f5Sebastian Pop
803de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    // A set of pairs has now been selected. It is now necessary to replace the
804de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    // paired instructions with vector instructions. For this procedure each
80543ec0f4921e315dd9507be7467e633a837ad23dbSebastian Pop    // operand must be replaced with a vector operand. This vector is formed
806de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    // by using build_vector on the old operands. The replaced values are then
807de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    // replaced with a vector_extract on the result.  Subsequent optimization
808de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    // passes should coalesce the build/extract combinations.
8093706ac7aa83ab0aed9e2da7d5fc2386ac1f035f5Sebastian Pop
81072465ea23d010507d3746adc126d719005981e05Hal Finkel    fuseChosenPairs(BB, AllPairableInsts, AllChosenPairs, AllFixedOrderPairs,
81172465ea23d010507d3746adc126d719005981e05Hal Finkel                    AllPairConnectionTypes,
81272465ea23d010507d3746adc126d719005981e05Hal Finkel                    AllConnectedPairs, AllConnectedPairDeps);
81364e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel
81464e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel    // It is important to cleanup here so that future iterations of this
81564e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel    // function have less work to do.
8168e0d1c03ca7fd86e6879b4e37d0d7f0e982feef6Benjamin Kramer    (void) SimplifyInstructionsInBlock(&BB, TD, AA->getTargetLibraryInfo());
817de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    return true;
818de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  }
819de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
820de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  // This function returns true if the provided instruction is capable of being
821de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  // fused into a vector instruction. This determination is based only on the
822de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  // type and other attributes of the instruction.
823de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  bool BBVectorize::isInstVectorizable(Instruction *I,
824de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel                                         bool &IsSimpleLoadStore) {
825de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    IsSimpleLoadStore = false;
826de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
827de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    if (CallInst *C = dyn_cast<CallInst>(I)) {
828de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      if (!isVectorizableIntrinsic(C))
829de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        return false;
830de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    } else if (LoadInst *L = dyn_cast<LoadInst>(I)) {
831de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      // Vectorize simple loads if possbile:
832de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      IsSimpleLoadStore = L->isSimple();
83386312cc15f29ce2bbd9647b94862e068045280c3Hongbin Zheng      if (!IsSimpleLoadStore || !Config.VectorizeMemOps)
834de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        return false;
835de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    } else if (StoreInst *S = dyn_cast<StoreInst>(I)) {
836de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      // Vectorize simple stores if possbile:
837de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      IsSimpleLoadStore = S->isSimple();
83886312cc15f29ce2bbd9647b94862e068045280c3Hongbin Zheng      if (!IsSimpleLoadStore || !Config.VectorizeMemOps)
839de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        return false;
840de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    } else if (CastInst *C = dyn_cast<CastInst>(I)) {
841de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      // We can vectorize casts, but not casts of pointer types, etc.
84286312cc15f29ce2bbd9647b94862e068045280c3Hongbin Zheng      if (!Config.VectorizeCasts)
843de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        return false;
844de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
845de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      Type *SrcTy = C->getSrcTy();
846f3f5a1e6f77a842ccb24cc81766437da5197d712Hal Finkel      if (!SrcTy->isSingleValueType())
847de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        return false;
848de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
849de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      Type *DestTy = C->getDestTy();
850f3f5a1e6f77a842ccb24cc81766437da5197d712Hal Finkel      if (!DestTy->isSingleValueType())
851de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        return false;
852fc3665c87519850f629c9565535e3be447e10addHal Finkel    } else if (isa<SelectInst>(I)) {
853fc3665c87519850f629c9565535e3be447e10addHal Finkel      if (!Config.VectorizeSelect)
854fc3665c87519850f629c9565535e3be447e10addHal Finkel        return false;
855e415f96b6a43ac8861148a11a4258bc38c247e8fHal Finkel    } else if (isa<CmpInst>(I)) {
856e415f96b6a43ac8861148a11a4258bc38c247e8fHal Finkel      if (!Config.VectorizeCmp)
857e415f96b6a43ac8861148a11a4258bc38c247e8fHal Finkel        return false;
858f3f5a1e6f77a842ccb24cc81766437da5197d712Hal Finkel    } else if (GetElementPtrInst *G = dyn_cast<GetElementPtrInst>(I)) {
859f3f5a1e6f77a842ccb24cc81766437da5197d712Hal Finkel      if (!Config.VectorizeGEP)
860f3f5a1e6f77a842ccb24cc81766437da5197d712Hal Finkel        return false;
861f3f5a1e6f77a842ccb24cc81766437da5197d712Hal Finkel
862f3f5a1e6f77a842ccb24cc81766437da5197d712Hal Finkel      // Currently, vector GEPs exist only with one index.
863f3f5a1e6f77a842ccb24cc81766437da5197d712Hal Finkel      if (G->getNumIndices() != 1)
864f3f5a1e6f77a842ccb24cc81766437da5197d712Hal Finkel        return false;
865de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    } else if (!(I->isBinaryOp() || isa<ShuffleVectorInst>(I) ||
866de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        isa<ExtractElementInst>(I) || isa<InsertElementInst>(I))) {
867de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      return false;
868de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    }
869de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
870de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    // We can't vectorize memory operations without target data
871de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    if (TD == 0 && IsSimpleLoadStore)
872de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      return false;
873de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
874de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    Type *T1, *T2;
87564e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel    getInstructionTypes(I, T1, T2);
876de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
877de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    // Not every type can be vectorized...
878de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    if (!(VectorType::isValidElementType(T1) || T1->isVectorTy()) ||
879de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        !(VectorType::isValidElementType(T2) || T2->isVectorTy()))
880de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      return false;
881de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
88265309660fa61a837cc05323f69c618a7d8134d56Hal Finkel    if (T1->getScalarSizeInBits() == 1) {
883768edf3cd037aab10391abc279f71470df8e3156Hal Finkel      if (!Config.VectorizeBools)
884768edf3cd037aab10391abc279f71470df8e3156Hal Finkel        return false;
885768edf3cd037aab10391abc279f71470df8e3156Hal Finkel    } else {
88665309660fa61a837cc05323f69c618a7d8134d56Hal Finkel      if (!Config.VectorizeInts && T1->isIntOrIntVectorTy())
887768edf3cd037aab10391abc279f71470df8e3156Hal Finkel        return false;
888768edf3cd037aab10391abc279f71470df8e3156Hal Finkel    }
88965309660fa61a837cc05323f69c618a7d8134d56Hal Finkel
89065309660fa61a837cc05323f69c618a7d8134d56Hal Finkel    if (T2->getScalarSizeInBits() == 1) {
89165309660fa61a837cc05323f69c618a7d8134d56Hal Finkel      if (!Config.VectorizeBools)
89265309660fa61a837cc05323f69c618a7d8134d56Hal Finkel        return false;
89365309660fa61a837cc05323f69c618a7d8134d56Hal Finkel    } else {
89465309660fa61a837cc05323f69c618a7d8134d56Hal Finkel      if (!Config.VectorizeInts && T2->isIntOrIntVectorTy())
89565309660fa61a837cc05323f69c618a7d8134d56Hal Finkel        return false;
89665309660fa61a837cc05323f69c618a7d8134d56Hal Finkel    }
89765309660fa61a837cc05323f69c618a7d8134d56Hal Finkel
89886312cc15f29ce2bbd9647b94862e068045280c3Hongbin Zheng    if (!Config.VectorizeFloats
89986312cc15f29ce2bbd9647b94862e068045280c3Hongbin Zheng        && (T1->isFPOrFPVectorTy() || T2->isFPOrFPVectorTy()))
900de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      return false;
901de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
902e32e5440d6aaff8a77517e9d286846ae9e380770Hal Finkel    // Don't vectorize target-specific types.
903e32e5440d6aaff8a77517e9d286846ae9e380770Hal Finkel    if (T1->isX86_FP80Ty() || T1->isPPC_FP128Ty() || T1->isX86_MMXTy())
904e32e5440d6aaff8a77517e9d286846ae9e380770Hal Finkel      return false;
905e32e5440d6aaff8a77517e9d286846ae9e380770Hal Finkel    if (T2->isX86_FP80Ty() || T2->isPPC_FP128Ty() || T2->isX86_MMXTy())
906e32e5440d6aaff8a77517e9d286846ae9e380770Hal Finkel      return false;
907e32e5440d6aaff8a77517e9d286846ae9e380770Hal Finkel
90805bc5087a25bbcf59936d71ebfc878b545ef3e5cHal Finkel    if ((!Config.VectorizePointers || TD == 0) &&
90905bc5087a25bbcf59936d71ebfc878b545ef3e5cHal Finkel        (T1->getScalarType()->isPointerTy() ||
91005bc5087a25bbcf59936d71ebfc878b545ef3e5cHal Finkel         T2->getScalarType()->isPointerTy()))
911f3f5a1e6f77a842ccb24cc81766437da5197d712Hal Finkel      return false;
912f3f5a1e6f77a842ccb24cc81766437da5197d712Hal Finkel
913abc227d9b39baed266c453f459ca14242f5c1eeaChandler Carruth    if (!TTI && (T1->getPrimitiveSizeInBits() >= Config.VectorBits ||
914abc227d9b39baed266c453f459ca14242f5c1eeaChandler Carruth                 T2->getPrimitiveSizeInBits() >= Config.VectorBits))
915de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      return false;
916de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
917de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    return true;
918de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  }
919de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
920de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  // This function returns true if the two provided instructions are compatible
921de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  // (meaning that they can be fused into a vector instruction). This assumes
922de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  // that I has already been determined to be vectorizable and that J is not
923f64a7a83bea5f1d2ab1e71231616c6cb0487d56eHal Finkel  // in the use dag of I.
924de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  bool BBVectorize::areInstsCompatible(Instruction *I, Instruction *J,
92565309660fa61a837cc05323f69c618a7d8134d56Hal Finkel                       bool IsSimpleLoadStore, bool NonPow2Len,
926a9779bfbc9ab0cf3f157453fd0afd110b04a9fdcHal Finkel                       int &CostSavings, int &FixedOrder) {
927de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    DEBUG(if (DebugInstructionExamination) dbgs() << "BBV: looking at " << *I <<
928de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel                     " <-> " << *J << "\n");
929de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
93065309660fa61a837cc05323f69c618a7d8134d56Hal Finkel    CostSavings = 0;
931a9779bfbc9ab0cf3f157453fd0afd110b04a9fdcHal Finkel    FixedOrder = 0;
93265309660fa61a837cc05323f69c618a7d8134d56Hal Finkel
933de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    // Loads and stores can be merged if they have different alignments,
934de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    // but are otherwise the same.
93564e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel    if (!J->isSameOperationAs(I, Instruction::CompareIgnoringAlignment |
93664e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel                      (NonPow2Len ? Instruction::CompareUsingScalarTypes : 0)))
93764e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel      return false;
93864e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel
93964e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel    Type *IT1, *IT2, *JT1, *JT2;
94064e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel    getInstructionTypes(I, IT1, IT2);
94164e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel    getInstructionTypes(J, JT1, JT2);
94264e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel    unsigned MaxTypeBits = std::max(
94364e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel      IT1->getPrimitiveSizeInBits() + JT1->getPrimitiveSizeInBits(),
94464e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel      IT2->getPrimitiveSizeInBits() + JT2->getPrimitiveSizeInBits());
945abc227d9b39baed266c453f459ca14242f5c1eeaChandler Carruth    if (!TTI && MaxTypeBits > Config.VectorBits)
946de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      return false;
947ec4e85e3364f50802f2007e4b1e23661d4610366Hal Finkel
948de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    // FIXME: handle addsub-type operations!
949de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
950de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    if (IsSimpleLoadStore) {
951de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      Value *IPtr, *JPtr;
95265309660fa61a837cc05323f69c618a7d8134d56Hal Finkel      unsigned IAlignment, JAlignment, IAddressSpace, JAddressSpace;
953de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      int64_t OffsetInElmts = 0;
954de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      if (getPairPtrInfo(I, J, IPtr, JPtr, IAlignment, JAlignment,
95565309660fa61a837cc05323f69c618a7d8134d56Hal Finkel            IAddressSpace, JAddressSpace,
956de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel            OffsetInElmts) && abs64(OffsetInElmts) == 1) {
957a9779bfbc9ab0cf3f157453fd0afd110b04a9fdcHal Finkel        FixedOrder = (int) OffsetInElmts;
95865309660fa61a837cc05323f69c618a7d8134d56Hal Finkel        unsigned BottomAlignment = IAlignment;
95965309660fa61a837cc05323f69c618a7d8134d56Hal Finkel        if (OffsetInElmts < 0) BottomAlignment = JAlignment;
96065309660fa61a837cc05323f69c618a7d8134d56Hal Finkel
96165309660fa61a837cc05323f69c618a7d8134d56Hal Finkel        Type *aTypeI = isa<StoreInst>(I) ?
96265309660fa61a837cc05323f69c618a7d8134d56Hal Finkel          cast<StoreInst>(I)->getValueOperand()->getType() : I->getType();
96365309660fa61a837cc05323f69c618a7d8134d56Hal Finkel        Type *aTypeJ = isa<StoreInst>(J) ?
96465309660fa61a837cc05323f69c618a7d8134d56Hal Finkel          cast<StoreInst>(J)->getValueOperand()->getType() : J->getType();
96565309660fa61a837cc05323f69c618a7d8134d56Hal Finkel        Type *VType = getVecTypeForPair(aTypeI, aTypeJ);
96664e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel
96765309660fa61a837cc05323f69c618a7d8134d56Hal Finkel        if (Config.AlignedOnly) {
968de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel          // An aligned load or store is possible only if the instruction
969de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel          // with the lower offset has an alignment suitable for the
970de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel          // vector type.
9711230ad6e8cb7977527ac64dcf5005464d7d6c20bSebastian Pop
972de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel          unsigned VecAlignment = TD->getPrefTypeAlignment(VType);
973de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel          if (BottomAlignment < VecAlignment)
974de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel            return false;
975de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        }
97665309660fa61a837cc05323f69c618a7d8134d56Hal Finkel
977abc227d9b39baed266c453f459ca14242f5c1eeaChandler Carruth        if (TTI) {
978abc227d9b39baed266c453f459ca14242f5c1eeaChandler Carruth          unsigned ICost = TTI->getMemoryOpCost(I->getOpcode(), aTypeI,
979abc227d9b39baed266c453f459ca14242f5c1eeaChandler Carruth                                                IAlignment, IAddressSpace);
980abc227d9b39baed266c453f459ca14242f5c1eeaChandler Carruth          unsigned JCost = TTI->getMemoryOpCost(J->getOpcode(), aTypeJ,
981abc227d9b39baed266c453f459ca14242f5c1eeaChandler Carruth                                                JAlignment, JAddressSpace);
982abc227d9b39baed266c453f459ca14242f5c1eeaChandler Carruth          unsigned VCost = TTI->getMemoryOpCost(I->getOpcode(), VType,
983abc227d9b39baed266c453f459ca14242f5c1eeaChandler Carruth                                                BottomAlignment,
984abc227d9b39baed266c453f459ca14242f5c1eeaChandler Carruth                                                IAddressSpace);
9850cf5d396c14c71dd4fa1d102c2b3d178b1191436Hal Finkel
9860cf5d396c14c71dd4fa1d102c2b3d178b1191436Hal Finkel          ICost += TTI->getAddressComputationCost(aTypeI);
9870cf5d396c14c71dd4fa1d102c2b3d178b1191436Hal Finkel          JCost += TTI->getAddressComputationCost(aTypeJ);
9880cf5d396c14c71dd4fa1d102c2b3d178b1191436Hal Finkel          VCost += TTI->getAddressComputationCost(VType);
9890cf5d396c14c71dd4fa1d102c2b3d178b1191436Hal Finkel
99065309660fa61a837cc05323f69c618a7d8134d56Hal Finkel          if (VCost > ICost + JCost)
99165309660fa61a837cc05323f69c618a7d8134d56Hal Finkel            return false;
99282149a9106f221aa6a7271977c236b078e621f21Hal Finkel
993dc330f75b732b4ce1beace69ae7ed8e19d89bd9fHal Finkel          // We don't want to fuse to a type that will be split, even
99482149a9106f221aa6a7271977c236b078e621f21Hal Finkel          // if the two input types will also be split and there is no other
995dc330f75b732b4ce1beace69ae7ed8e19d89bd9fHal Finkel          // associated cost.
996abc227d9b39baed266c453f459ca14242f5c1eeaChandler Carruth          unsigned VParts = TTI->getNumberOfParts(VType);
997dc330f75b732b4ce1beace69ae7ed8e19d89bd9fHal Finkel          if (VParts > 1)
998dc330f75b732b4ce1beace69ae7ed8e19d89bd9fHal Finkel            return false;
999dc330f75b732b4ce1beace69ae7ed8e19d89bd9fHal Finkel          else if (!VParts && VCost == ICost + JCost)
100082149a9106f221aa6a7271977c236b078e621f21Hal Finkel            return false;
100182149a9106f221aa6a7271977c236b078e621f21Hal Finkel
100265309660fa61a837cc05323f69c618a7d8134d56Hal Finkel          CostSavings = ICost + JCost - VCost;
100365309660fa61a837cc05323f69c618a7d8134d56Hal Finkel        }
1004de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      } else {
1005de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        return false;
1006de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      }
1007abc227d9b39baed266c453f459ca14242f5c1eeaChandler Carruth    } else if (TTI) {
100846fb81cf4009cc34af97c5a1c0e824e2633fb4e4Hal Finkel      unsigned ICost = getInstrCost(I->getOpcode(), IT1, IT2);
100946fb81cf4009cc34af97c5a1c0e824e2633fb4e4Hal Finkel      unsigned JCost = getInstrCost(J->getOpcode(), JT1, JT2);
101065309660fa61a837cc05323f69c618a7d8134d56Hal Finkel      Type *VT1 = getVecTypeForPair(IT1, JT1),
101165309660fa61a837cc05323f69c618a7d8134d56Hal Finkel           *VT2 = getVecTypeForPair(IT2, JT2);
1012ec5c3c2bd37828e9fbd913f5ac7b7f75711ddd59Hal Finkel
1013ec5c3c2bd37828e9fbd913f5ac7b7f75711ddd59Hal Finkel      // Note that this procedure is incorrect for insert and extract element
1014ec5c3c2bd37828e9fbd913f5ac7b7f75711ddd59Hal Finkel      // instructions (because combining these often results in a shuffle),
1015ec5c3c2bd37828e9fbd913f5ac7b7f75711ddd59Hal Finkel      // but this cost is ignored (because insert and extract element
1016ec5c3c2bd37828e9fbd913f5ac7b7f75711ddd59Hal Finkel      // instructions are assigned a zero depth factor and are not really
1017ec5c3c2bd37828e9fbd913f5ac7b7f75711ddd59Hal Finkel      // fused in general).
101846fb81cf4009cc34af97c5a1c0e824e2633fb4e4Hal Finkel      unsigned VCost = getInstrCost(I->getOpcode(), VT1, VT2);
101965309660fa61a837cc05323f69c618a7d8134d56Hal Finkel
102065309660fa61a837cc05323f69c618a7d8134d56Hal Finkel      if (VCost > ICost + JCost)
102165309660fa61a837cc05323f69c618a7d8134d56Hal Finkel        return false;
102282149a9106f221aa6a7271977c236b078e621f21Hal Finkel
1023dc330f75b732b4ce1beace69ae7ed8e19d89bd9fHal Finkel      // We don't want to fuse to a type that will be split, even
102482149a9106f221aa6a7271977c236b078e621f21Hal Finkel      // if the two input types will also be split and there is no other
1025dc330f75b732b4ce1beace69ae7ed8e19d89bd9fHal Finkel      // associated cost.
1026abc227d9b39baed266c453f459ca14242f5c1eeaChandler Carruth      unsigned VParts1 = TTI->getNumberOfParts(VT1),
1027abc227d9b39baed266c453f459ca14242f5c1eeaChandler Carruth               VParts2 = TTI->getNumberOfParts(VT2);
10288b9796f4f83bea2bbefcd2822eb574abdb7f3d1bHal Finkel      if (VParts1 > 1 || VParts2 > 1)
1029dc330f75b732b4ce1beace69ae7ed8e19d89bd9fHal Finkel        return false;
10308b9796f4f83bea2bbefcd2822eb574abdb7f3d1bHal Finkel      else if ((!VParts1 || !VParts2) && VCost == ICost + JCost)
103182149a9106f221aa6a7271977c236b078e621f21Hal Finkel        return false;
103282149a9106f221aa6a7271977c236b078e621f21Hal Finkel
103365309660fa61a837cc05323f69c618a7d8134d56Hal Finkel      CostSavings = ICost + JCost - VCost;
1034de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    }
1035de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
10366173ed95daf2f209fe3883faee45967e4800ae75Hal Finkel    // The powi intrinsic is special because only the first argument is
10376173ed95daf2f209fe3883faee45967e4800ae75Hal Finkel    // vectorized, the second arguments must be equal.
10386173ed95daf2f209fe3883faee45967e4800ae75Hal Finkel    CallInst *CI = dyn_cast<CallInst>(I);
10396173ed95daf2f209fe3883faee45967e4800ae75Hal Finkel    Function *FI;
1040a77728415857196035c0090f7b2749d7971811a2Hal Finkel    if (CI && (FI = CI->getCalledFunction())) {
1041a77728415857196035c0090f7b2749d7971811a2Hal Finkel      Intrinsic::ID IID = (Intrinsic::ID) FI->getIntrinsicID();
1042a77728415857196035c0090f7b2749d7971811a2Hal Finkel      if (IID == Intrinsic::powi) {
1043a77728415857196035c0090f7b2749d7971811a2Hal Finkel        Value *A1I = CI->getArgOperand(1),
1044a77728415857196035c0090f7b2749d7971811a2Hal Finkel              *A1J = cast<CallInst>(J)->getArgOperand(1);
1045a77728415857196035c0090f7b2749d7971811a2Hal Finkel        const SCEV *A1ISCEV = SE->getSCEV(A1I),
1046a77728415857196035c0090f7b2749d7971811a2Hal Finkel                   *A1JSCEV = SE->getSCEV(A1J);
1047a77728415857196035c0090f7b2749d7971811a2Hal Finkel        return (A1ISCEV == A1JSCEV);
1048a77728415857196035c0090f7b2749d7971811a2Hal Finkel      }
1049a77728415857196035c0090f7b2749d7971811a2Hal Finkel
1050abc227d9b39baed266c453f459ca14242f5c1eeaChandler Carruth      if (IID && TTI) {
1051a77728415857196035c0090f7b2749d7971811a2Hal Finkel        SmallVector<Type*, 4> Tys;
1052a77728415857196035c0090f7b2749d7971811a2Hal Finkel        for (unsigned i = 0, ie = CI->getNumArgOperands(); i != ie; ++i)
1053a77728415857196035c0090f7b2749d7971811a2Hal Finkel          Tys.push_back(CI->getArgOperand(i)->getType());
1054abc227d9b39baed266c453f459ca14242f5c1eeaChandler Carruth        unsigned ICost = TTI->getIntrinsicInstrCost(IID, IT1, Tys);
1055a77728415857196035c0090f7b2749d7971811a2Hal Finkel
1056a77728415857196035c0090f7b2749d7971811a2Hal Finkel        Tys.clear();
1057a77728415857196035c0090f7b2749d7971811a2Hal Finkel        CallInst *CJ = cast<CallInst>(J);
1058a77728415857196035c0090f7b2749d7971811a2Hal Finkel        for (unsigned i = 0, ie = CJ->getNumArgOperands(); i != ie; ++i)
1059a77728415857196035c0090f7b2749d7971811a2Hal Finkel          Tys.push_back(CJ->getArgOperand(i)->getType());
1060abc227d9b39baed266c453f459ca14242f5c1eeaChandler Carruth        unsigned JCost = TTI->getIntrinsicInstrCost(IID, JT1, Tys);
1061a77728415857196035c0090f7b2749d7971811a2Hal Finkel
1062a77728415857196035c0090f7b2749d7971811a2Hal Finkel        Tys.clear();
1063a77728415857196035c0090f7b2749d7971811a2Hal Finkel        assert(CI->getNumArgOperands() == CJ->getNumArgOperands() &&
1064a77728415857196035c0090f7b2749d7971811a2Hal Finkel               "Intrinsic argument counts differ");
1065a77728415857196035c0090f7b2749d7971811a2Hal Finkel        for (unsigned i = 0, ie = CI->getNumArgOperands(); i != ie; ++i) {
1066a77728415857196035c0090f7b2749d7971811a2Hal Finkel          if (IID == Intrinsic::powi && i == 1)
1067a77728415857196035c0090f7b2749d7971811a2Hal Finkel            Tys.push_back(CI->getArgOperand(i)->getType());
1068a77728415857196035c0090f7b2749d7971811a2Hal Finkel          else
1069a77728415857196035c0090f7b2749d7971811a2Hal Finkel            Tys.push_back(getVecTypeForPair(CI->getArgOperand(i)->getType(),
1070a77728415857196035c0090f7b2749d7971811a2Hal Finkel                                            CJ->getArgOperand(i)->getType()));
1071a77728415857196035c0090f7b2749d7971811a2Hal Finkel        }
1072a77728415857196035c0090f7b2749d7971811a2Hal Finkel
1073a77728415857196035c0090f7b2749d7971811a2Hal Finkel        Type *RetTy = getVecTypeForPair(IT1, JT1);
1074abc227d9b39baed266c453f459ca14242f5c1eeaChandler Carruth        unsigned VCost = TTI->getIntrinsicInstrCost(IID, RetTy, Tys);
1075a77728415857196035c0090f7b2749d7971811a2Hal Finkel
1076a77728415857196035c0090f7b2749d7971811a2Hal Finkel        if (VCost > ICost + JCost)
1077a77728415857196035c0090f7b2749d7971811a2Hal Finkel          return false;
1078a77728415857196035c0090f7b2749d7971811a2Hal Finkel
1079a77728415857196035c0090f7b2749d7971811a2Hal Finkel        // We don't want to fuse to a type that will be split, even
1080a77728415857196035c0090f7b2749d7971811a2Hal Finkel        // if the two input types will also be split and there is no other
1081a77728415857196035c0090f7b2749d7971811a2Hal Finkel        // associated cost.
1082abc227d9b39baed266c453f459ca14242f5c1eeaChandler Carruth        unsigned RetParts = TTI->getNumberOfParts(RetTy);
1083a77728415857196035c0090f7b2749d7971811a2Hal Finkel        if (RetParts > 1)
1084a77728415857196035c0090f7b2749d7971811a2Hal Finkel          return false;
1085a77728415857196035c0090f7b2749d7971811a2Hal Finkel        else if (!RetParts && VCost == ICost + JCost)
1086a77728415857196035c0090f7b2749d7971811a2Hal Finkel          return false;
1087a77728415857196035c0090f7b2749d7971811a2Hal Finkel
1088a77728415857196035c0090f7b2749d7971811a2Hal Finkel        for (unsigned i = 0, ie = CI->getNumArgOperands(); i != ie; ++i) {
1089a77728415857196035c0090f7b2749d7971811a2Hal Finkel          if (!Tys[i]->isVectorTy())
1090a77728415857196035c0090f7b2749d7971811a2Hal Finkel            continue;
1091a77728415857196035c0090f7b2749d7971811a2Hal Finkel
1092abc227d9b39baed266c453f459ca14242f5c1eeaChandler Carruth          unsigned NumParts = TTI->getNumberOfParts(Tys[i]);
1093a77728415857196035c0090f7b2749d7971811a2Hal Finkel          if (NumParts > 1)
1094a77728415857196035c0090f7b2749d7971811a2Hal Finkel            return false;
1095a77728415857196035c0090f7b2749d7971811a2Hal Finkel          else if (!NumParts && VCost == ICost + JCost)
1096a77728415857196035c0090f7b2749d7971811a2Hal Finkel            return false;
1097a77728415857196035c0090f7b2749d7971811a2Hal Finkel        }
1098a77728415857196035c0090f7b2749d7971811a2Hal Finkel
1099a77728415857196035c0090f7b2749d7971811a2Hal Finkel        CostSavings = ICost + JCost - VCost;
1100a77728415857196035c0090f7b2749d7971811a2Hal Finkel      }
11016173ed95daf2f209fe3883faee45967e4800ae75Hal Finkel    }
11026173ed95daf2f209fe3883faee45967e4800ae75Hal Finkel
1103de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    return true;
1104de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  }
1105de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
1106de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  // Figure out whether or not J uses I and update the users and write-set
1107de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  // structures associated with I. Specifically, Users represents the set of
1108de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  // instructions that depend on I. WriteSet represents the set
1109de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  // of memory locations that are dependent on I. If UpdateUsers is true,
1110de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  // and J uses I, then Users is updated to contain J and WriteSet is updated
1111de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  // to contain any memory locations to which J writes. The function returns
1112de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  // true if J uses I. By default, alias analysis is used to determine
1113de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  // whether J reads from memory that overlaps with a location in WriteSet.
111497a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel  // If LoadMoveSet is not null, then it is a previously-computed map
1115de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  // where the key is the memory-based user instruction and the value is
1116de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  // the instruction to be compared with I. So, if LoadMoveSet is provided,
1117de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  // then the alias analysis is not used. This is necessary because this
1118de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  // function is called during the process of moving instructions during
1119de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  // vectorization and the results of the alias analysis are not stable during
1120de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  // that process.
1121de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  bool BBVectorize::trackUsesOfI(DenseSet<Value *> &Users,
1122de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel                       AliasSetTracker &WriteSet, Instruction *I,
1123de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel                       Instruction *J, bool UpdateUsers,
11242f0e63cc16feb39480805bd00f53bbe5e3031d29Hal Finkel                       DenseSet<ValuePair> *LoadMoveSetPairs) {
1125de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    bool UsesI = false;
1126de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
1127de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    // This instruction may already be marked as a user due, for example, to
1128de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    // being a member of a selected pair.
1129de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    if (Users.count(J))
1130de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      UsesI = true;
1131de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
1132de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    if (!UsesI)
11337e004d177fe76145f75a9417ed2e281f1b9abaf7Hal Finkel      for (User::op_iterator JU = J->op_begin(), JE = J->op_end();
11347e004d177fe76145f75a9417ed2e281f1b9abaf7Hal Finkel           JU != JE; ++JU) {
1135de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        Value *V = *JU;
1136de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        if (I == V || Users.count(V)) {
1137de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel          UsesI = true;
1138de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel          break;
1139de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        }
1140de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      }
1141de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    if (!UsesI && J->mayReadFromMemory()) {
11422f0e63cc16feb39480805bd00f53bbe5e3031d29Hal Finkel      if (LoadMoveSetPairs) {
11432f0e63cc16feb39480805bd00f53bbe5e3031d29Hal Finkel        UsesI = LoadMoveSetPairs->count(ValuePair(J, I));
1144de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      } else {
1145de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        for (AliasSetTracker::iterator W = WriteSet.begin(),
1146de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel             WE = WriteSet.end(); W != WE; ++W) {
114738a7f22445b8782682d1f8f253454ea0390d4ac5Hal Finkel          if (W->aliasesUnknownInst(J, *AA)) {
114838a7f22445b8782682d1f8f253454ea0390d4ac5Hal Finkel            UsesI = true;
114938a7f22445b8782682d1f8f253454ea0390d4ac5Hal Finkel            break;
1150de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel          }
1151de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        }
1152de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      }
1153de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    }
1154de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
1155de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    if (UsesI && UpdateUsers) {
1156de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      if (J->mayWriteToMemory()) WriteSet.add(J);
1157de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      Users.insert(J);
1158de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    }
1159de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
1160de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    return UsesI;
1161de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  }
1162de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
1163de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  // This function iterates over all instruction pairs in the provided
1164de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  // basic block and collects all candidate pairs for vectorization.
11655d4e18bc39fea892f523d960213906d296d3cb38Hal Finkel  bool BBVectorize::getCandidatePairs(BasicBlock &BB,
11665d4e18bc39fea892f523d960213906d296d3cb38Hal Finkel                       BasicBlock::iterator &Start,
11676ca6d3b1eac5b8611f3a9e2c270c2e794d37e1f5Hal Finkel                       DenseMap<Value *, std::vector<Value *> > &CandidatePairs,
1168a9779bfbc9ab0cf3f157453fd0afd110b04a9fdcHal Finkel                       DenseSet<ValuePair> &FixedOrderPairs,
116965309660fa61a837cc05323f69c618a7d8134d56Hal Finkel                       DenseMap<ValuePair, int> &CandidatePairCostSavings,
117064e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel                       std::vector<Value *> &PairableInsts, bool NonPow2Len) {
1171ab90084bca42b74a5b5edad9b416bd81e105dad0Hal Finkel    size_t TotalPairs = 0;
1172de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    BasicBlock::iterator E = BB.end();
11735d4e18bc39fea892f523d960213906d296d3cb38Hal Finkel    if (Start == E) return false;
11745d4e18bc39fea892f523d960213906d296d3cb38Hal Finkel
11755d4e18bc39fea892f523d960213906d296d3cb38Hal Finkel    bool ShouldContinue = false, IAfterStart = false;
11765d4e18bc39fea892f523d960213906d296d3cb38Hal Finkel    for (BasicBlock::iterator I = Start++; I != E; ++I) {
11775d4e18bc39fea892f523d960213906d296d3cb38Hal Finkel      if (I == Start) IAfterStart = true;
11785d4e18bc39fea892f523d960213906d296d3cb38Hal Finkel
1179de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      bool IsSimpleLoadStore;
1180de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      if (!isInstVectorizable(I, IsSimpleLoadStore)) continue;
1181de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
1182de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      // Look for an instruction with which to pair instruction *I...
1183de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      DenseSet<Value *> Users;
1184de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      AliasSetTracker WriteSet(*AA);
11855d4e18bc39fea892f523d960213906d296d3cb38Hal Finkel      bool JAfterStart = IAfterStart;
11865d4e18bc39fea892f523d960213906d296d3cb38Hal Finkel      BasicBlock::iterator J = llvm::next(I);
1187bef377b7d7ce31edb40c87f8786d1b7bb6cdd6b1Hongbin Zheng      for (unsigned ss = 0; J != E && ss <= Config.SearchLimit; ++J, ++ss) {
11885d4e18bc39fea892f523d960213906d296d3cb38Hal Finkel        if (J == Start) JAfterStart = true;
11895d4e18bc39fea892f523d960213906d296d3cb38Hal Finkel
1190de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        // Determine if J uses I, if so, exit the loop.
1191bef377b7d7ce31edb40c87f8786d1b7bb6cdd6b1Hongbin Zheng        bool UsesI = trackUsesOfI(Users, WriteSet, I, J, !Config.FastDep);
1192bef377b7d7ce31edb40c87f8786d1b7bb6cdd6b1Hongbin Zheng        if (Config.FastDep) {
1193de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel          // Note: For this heuristic to be effective, independent operations
1194de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel          // must tend to be intermixed. This is likely to be true from some
1195de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel          // kinds of grouped loop unrolling (but not the generic LLVM pass),
1196de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel          // but otherwise may require some kind of reordering pass.
1197de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
1198de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel          // When using fast dependency analysis,
1199de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel          // stop searching after first use:
1200de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel          if (UsesI) break;
1201de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        } else {
1202de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel          if (UsesI) continue;
1203de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        }
1204de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
1205de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        // J does not use I, and comes before the first use of I, so it can be
1206de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        // merged with I if the instructions are compatible.
1207a9779bfbc9ab0cf3f157453fd0afd110b04a9fdcHal Finkel        int CostSavings, FixedOrder;
120865309660fa61a837cc05323f69c618a7d8134d56Hal Finkel        if (!areInstsCompatible(I, J, IsSimpleLoadStore, NonPow2Len,
1209a9779bfbc9ab0cf3f157453fd0afd110b04a9fdcHal Finkel            CostSavings, FixedOrder)) continue;
1210de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
1211de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        // J is a candidate for merging with I.
1212de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        if (!PairableInsts.size() ||
1213de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel             PairableInsts[PairableInsts.size()-1] != I) {
1214de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel          PairableInsts.push_back(I);
1215de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        }
12165d4e18bc39fea892f523d960213906d296d3cb38Hal Finkel
12176ca6d3b1eac5b8611f3a9e2c270c2e794d37e1f5Hal Finkel        CandidatePairs[I].push_back(J);
1218ab90084bca42b74a5b5edad9b416bd81e105dad0Hal Finkel        ++TotalPairs;
1219abc227d9b39baed266c453f459ca14242f5c1eeaChandler Carruth        if (TTI)
122065309660fa61a837cc05323f69c618a7d8134d56Hal Finkel          CandidatePairCostSavings.insert(ValuePairWithCost(ValuePair(I, J),
122165309660fa61a837cc05323f69c618a7d8134d56Hal Finkel                                                            CostSavings));
12225d4e18bc39fea892f523d960213906d296d3cb38Hal Finkel
1223a9779bfbc9ab0cf3f157453fd0afd110b04a9fdcHal Finkel        if (FixedOrder == 1)
1224a9779bfbc9ab0cf3f157453fd0afd110b04a9fdcHal Finkel          FixedOrderPairs.insert(ValuePair(I, J));
1225a9779bfbc9ab0cf3f157453fd0afd110b04a9fdcHal Finkel        else if (FixedOrder == -1)
1226a9779bfbc9ab0cf3f157453fd0afd110b04a9fdcHal Finkel          FixedOrderPairs.insert(ValuePair(J, I));
1227a9779bfbc9ab0cf3f157453fd0afd110b04a9fdcHal Finkel
12285d4e18bc39fea892f523d960213906d296d3cb38Hal Finkel        // The next call to this function must start after the last instruction
12295d4e18bc39fea892f523d960213906d296d3cb38Hal Finkel        // selected during this invocation.
12305d4e18bc39fea892f523d960213906d296d3cb38Hal Finkel        if (JAfterStart) {
12315d4e18bc39fea892f523d960213906d296d3cb38Hal Finkel          Start = llvm::next(J);
12325d4e18bc39fea892f523d960213906d296d3cb38Hal Finkel          IAfterStart = JAfterStart = false;
12335d4e18bc39fea892f523d960213906d296d3cb38Hal Finkel        }
12345d4e18bc39fea892f523d960213906d296d3cb38Hal Finkel
1235de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        DEBUG(if (DebugCandidateSelection) dbgs() << "BBV: candidate pair "
123665309660fa61a837cc05323f69c618a7d8134d56Hal Finkel                     << *I << " <-> " << *J << " (cost savings: " <<
123765309660fa61a837cc05323f69c618a7d8134d56Hal Finkel                     CostSavings << ")\n");
12385d4e18bc39fea892f523d960213906d296d3cb38Hal Finkel
12395d4e18bc39fea892f523d960213906d296d3cb38Hal Finkel        // If we have already found too many pairs, break here and this function
12405d4e18bc39fea892f523d960213906d296d3cb38Hal Finkel        // will be called again starting after the last instruction selected
12415d4e18bc39fea892f523d960213906d296d3cb38Hal Finkel        // during this invocation.
1242ab90084bca42b74a5b5edad9b416bd81e105dad0Hal Finkel        if (PairableInsts.size() >= Config.MaxInsts ||
1243ab90084bca42b74a5b5edad9b416bd81e105dad0Hal Finkel            TotalPairs >= Config.MaxPairs) {
12445d4e18bc39fea892f523d960213906d296d3cb38Hal Finkel          ShouldContinue = true;
12455d4e18bc39fea892f523d960213906d296d3cb38Hal Finkel          break;
12465d4e18bc39fea892f523d960213906d296d3cb38Hal Finkel        }
1247de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      }
12485d4e18bc39fea892f523d960213906d296d3cb38Hal Finkel
12495d4e18bc39fea892f523d960213906d296d3cb38Hal Finkel      if (ShouldContinue)
12505d4e18bc39fea892f523d960213906d296d3cb38Hal Finkel        break;
1251de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    }
1252de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
1253de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    DEBUG(dbgs() << "BBV: found " << PairableInsts.size()
1254de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel           << " instructions with candidate pairs\n");
12555d4e18bc39fea892f523d960213906d296d3cb38Hal Finkel
12565d4e18bc39fea892f523d960213906d296d3cb38Hal Finkel    return ShouldContinue;
1257de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  }
1258de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
1259de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  // Finds candidate pairs connected to the pair P = <PI, PJ>. This means that
1260de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  // it looks for pairs such that both members have an input which is an
1261de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  // output of PI or PJ.
1262de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  void BBVectorize::computePairsConnectedTo(
126397a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel                  DenseMap<Value *, std::vector<Value *> > &CandidatePairs,
126497a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel                  DenseSet<ValuePair> &CandidatePairsSet,
126597a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel                  std::vector<Value *> &PairableInsts,
126697a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel                  DenseMap<ValuePair, std::vector<ValuePair> > &ConnectedPairs,
126797a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel                  DenseMap<VPPair, unsigned> &PairConnectionTypes,
126897a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel                  ValuePair P) {
1269bba23ed672c4cedd61a302497f45bf6f53fec7b2Hal Finkel    StoreInst *SI, *SJ;
1270bba23ed672c4cedd61a302497f45bf6f53fec7b2Hal Finkel
1271de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    // For each possible pairing for this variable, look at the uses of
1272de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    // the first value...
1273de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    for (Value::use_iterator I = P.first->use_begin(),
1274de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel         E = P.first->use_end(); I != E; ++I) {
1275bba23ed672c4cedd61a302497f45bf6f53fec7b2Hal Finkel      if (isa<LoadInst>(*I)) {
1276bba23ed672c4cedd61a302497f45bf6f53fec7b2Hal Finkel        // A pair cannot be connected to a load because the load only takes one
1277bba23ed672c4cedd61a302497f45bf6f53fec7b2Hal Finkel        // operand (the address) and it is a scalar even after vectorization.
1278bba23ed672c4cedd61a302497f45bf6f53fec7b2Hal Finkel        continue;
1279bba23ed672c4cedd61a302497f45bf6f53fec7b2Hal Finkel      } else if ((SI = dyn_cast<StoreInst>(*I)) &&
1280bba23ed672c4cedd61a302497f45bf6f53fec7b2Hal Finkel                 P.first == SI->getPointerOperand()) {
1281bba23ed672c4cedd61a302497f45bf6f53fec7b2Hal Finkel        // Similarly, a pair cannot be connected to a store through its
1282bba23ed672c4cedd61a302497f45bf6f53fec7b2Hal Finkel        // pointer operand.
1283bba23ed672c4cedd61a302497f45bf6f53fec7b2Hal Finkel        continue;
1284bba23ed672c4cedd61a302497f45bf6f53fec7b2Hal Finkel      }
1285bba23ed672c4cedd61a302497f45bf6f53fec7b2Hal Finkel
1286de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      // For each use of the first variable, look for uses of the second
1287de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      // variable...
1288de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      for (Value::use_iterator J = P.second->use_begin(),
1289de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel           E2 = P.second->use_end(); J != E2; ++J) {
1290bba23ed672c4cedd61a302497f45bf6f53fec7b2Hal Finkel        if ((SJ = dyn_cast<StoreInst>(*J)) &&
1291bba23ed672c4cedd61a302497f45bf6f53fec7b2Hal Finkel            P.second == SJ->getPointerOperand())
1292bba23ed672c4cedd61a302497f45bf6f53fec7b2Hal Finkel          continue;
1293bba23ed672c4cedd61a302497f45bf6f53fec7b2Hal Finkel
1294de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        // Look for <I, J>:
129500f63b1b84d059a1ffa572e76708e03750a9e523Hal Finkel        if (CandidatePairsSet.count(ValuePair(*I, *J))) {
129672465ea23d010507d3746adc126d719005981e05Hal Finkel          VPPair VP(P, ValuePair(*I, *J));
129797a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel          ConnectedPairs[VP.first].push_back(VP.second);
129872465ea23d010507d3746adc126d719005981e05Hal Finkel          PairConnectionTypes.insert(VPPairWithType(VP, PairConnectionDirect));
129972465ea23d010507d3746adc126d719005981e05Hal Finkel        }
1300de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
1301de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        // Look for <J, I>:
130200f63b1b84d059a1ffa572e76708e03750a9e523Hal Finkel        if (CandidatePairsSet.count(ValuePair(*J, *I))) {
130372465ea23d010507d3746adc126d719005981e05Hal Finkel          VPPair VP(P, ValuePair(*J, *I));
130497a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel          ConnectedPairs[VP.first].push_back(VP.second);
130572465ea23d010507d3746adc126d719005981e05Hal Finkel          PairConnectionTypes.insert(VPPairWithType(VP, PairConnectionSwap));
130672465ea23d010507d3746adc126d719005981e05Hal Finkel        }
1307de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      }
1308de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
1309bef377b7d7ce31edb40c87f8786d1b7bb6cdd6b1Hongbin Zheng      if (Config.SplatBreaksChain) continue;
1310de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      // Look for cases where just the first value in the pair is used by
1311de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      // both members of another pair (splatting).
1312de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      for (Value::use_iterator J = P.first->use_begin(); J != E; ++J) {
1313bba23ed672c4cedd61a302497f45bf6f53fec7b2Hal Finkel        if ((SJ = dyn_cast<StoreInst>(*J)) &&
1314bba23ed672c4cedd61a302497f45bf6f53fec7b2Hal Finkel            P.first == SJ->getPointerOperand())
1315bba23ed672c4cedd61a302497f45bf6f53fec7b2Hal Finkel          continue;
1316bba23ed672c4cedd61a302497f45bf6f53fec7b2Hal Finkel
131700f63b1b84d059a1ffa572e76708e03750a9e523Hal Finkel        if (CandidatePairsSet.count(ValuePair(*I, *J))) {
131872465ea23d010507d3746adc126d719005981e05Hal Finkel          VPPair VP(P, ValuePair(*I, *J));
131997a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel          ConnectedPairs[VP.first].push_back(VP.second);
132072465ea23d010507d3746adc126d719005981e05Hal Finkel          PairConnectionTypes.insert(VPPairWithType(VP, PairConnectionSplat));
132172465ea23d010507d3746adc126d719005981e05Hal Finkel        }
1322de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      }
1323de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    }
1324de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
1325bef377b7d7ce31edb40c87f8786d1b7bb6cdd6b1Hongbin Zheng    if (Config.SplatBreaksChain) return;
1326de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    // Look for cases where just the second value in the pair is used by
1327de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    // both members of another pair (splatting).
1328de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    for (Value::use_iterator I = P.second->use_begin(),
1329de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel         E = P.second->use_end(); I != E; ++I) {
1330bba23ed672c4cedd61a302497f45bf6f53fec7b2Hal Finkel      if (isa<LoadInst>(*I))
1331bba23ed672c4cedd61a302497f45bf6f53fec7b2Hal Finkel        continue;
1332bba23ed672c4cedd61a302497f45bf6f53fec7b2Hal Finkel      else if ((SI = dyn_cast<StoreInst>(*I)) &&
1333bba23ed672c4cedd61a302497f45bf6f53fec7b2Hal Finkel               P.second == SI->getPointerOperand())
1334bba23ed672c4cedd61a302497f45bf6f53fec7b2Hal Finkel        continue;
1335bba23ed672c4cedd61a302497f45bf6f53fec7b2Hal Finkel
1336de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      for (Value::use_iterator J = P.second->use_begin(); J != E; ++J) {
1337bba23ed672c4cedd61a302497f45bf6f53fec7b2Hal Finkel        if ((SJ = dyn_cast<StoreInst>(*J)) &&
1338bba23ed672c4cedd61a302497f45bf6f53fec7b2Hal Finkel            P.second == SJ->getPointerOperand())
1339bba23ed672c4cedd61a302497f45bf6f53fec7b2Hal Finkel          continue;
1340bba23ed672c4cedd61a302497f45bf6f53fec7b2Hal Finkel
134100f63b1b84d059a1ffa572e76708e03750a9e523Hal Finkel        if (CandidatePairsSet.count(ValuePair(*I, *J))) {
134272465ea23d010507d3746adc126d719005981e05Hal Finkel          VPPair VP(P, ValuePair(*I, *J));
134397a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel          ConnectedPairs[VP.first].push_back(VP.second);
134472465ea23d010507d3746adc126d719005981e05Hal Finkel          PairConnectionTypes.insert(VPPairWithType(VP, PairConnectionSplat));
134572465ea23d010507d3746adc126d719005981e05Hal Finkel        }
1346de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      }
1347de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    }
1348de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  }
1349de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
1350de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  // This function figures out which pairs are connected.  Two pairs are
1351de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  // connected if some output of the first pair forms an input to both members
1352de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  // of the second pair.
1353de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  void BBVectorize::computeConnectedPairs(
135497a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel                  DenseMap<Value *, std::vector<Value *> > &CandidatePairs,
135597a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel                  DenseSet<ValuePair> &CandidatePairsSet,
135697a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel                  std::vector<Value *> &PairableInsts,
135797a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel                  DenseMap<ValuePair, std::vector<ValuePair> > &ConnectedPairs,
135897a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel                  DenseMap<VPPair, unsigned> &PairConnectionTypes) {
1359de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    for (std::vector<Value *>::iterator PI = PairableInsts.begin(),
1360de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel         PE = PairableInsts.end(); PI != PE; ++PI) {
13616ca6d3b1eac5b8611f3a9e2c270c2e794d37e1f5Hal Finkel      DenseMap<Value *, std::vector<Value *> >::iterator PP =
13626ca6d3b1eac5b8611f3a9e2c270c2e794d37e1f5Hal Finkel        CandidatePairs.find(*PI);
13636ca6d3b1eac5b8611f3a9e2c270c2e794d37e1f5Hal Finkel      if (PP == CandidatePairs.end())
13646ca6d3b1eac5b8611f3a9e2c270c2e794d37e1f5Hal Finkel        continue;
1365de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
13666ca6d3b1eac5b8611f3a9e2c270c2e794d37e1f5Hal Finkel      for (std::vector<Value *>::iterator P = PP->second.begin(),
13676ca6d3b1eac5b8611f3a9e2c270c2e794d37e1f5Hal Finkel           E = PP->second.end(); P != E; ++P)
1368b1a82589339fed148c12b052d30861a539552f1aHal Finkel        computePairsConnectedTo(CandidatePairs, CandidatePairsSet,
1369b1a82589339fed148c12b052d30861a539552f1aHal Finkel                                PairableInsts, ConnectedPairs,
13706ca6d3b1eac5b8611f3a9e2c270c2e794d37e1f5Hal Finkel                                PairConnectionTypes, ValuePair(*PI, *P));
1371de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    }
1372de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
137397a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel    DEBUG(size_t TotalPairs = 0;
137497a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel          for (DenseMap<ValuePair, std::vector<ValuePair> >::iterator I =
137597a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel               ConnectedPairs.begin(), IE = ConnectedPairs.end(); I != IE; ++I)
137697a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel            TotalPairs += I->second.size();
137797a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel          dbgs() << "BBV: found " << TotalPairs
1378de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel                 << " pair connections.\n");
1379de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  }
1380de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
1381de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  // This function builds a set of use tuples such that <A, B> is in the set
1382f64a7a83bea5f1d2ab1e71231616c6cb0487d56eHal Finkel  // if B is in the use dag of A. If B is in the use dag of A, then B
1383de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  // depends on the output of A.
1384de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  void BBVectorize::buildDepMap(
1385de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel                      BasicBlock &BB,
13866ca6d3b1eac5b8611f3a9e2c270c2e794d37e1f5Hal Finkel                      DenseMap<Value *, std::vector<Value *> > &CandidatePairs,
1387de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel                      std::vector<Value *> &PairableInsts,
1388de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel                      DenseSet<ValuePair> &PairableInstUsers) {
1389de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    DenseSet<Value *> IsInPair;
13906ca6d3b1eac5b8611f3a9e2c270c2e794d37e1f5Hal Finkel    for (DenseMap<Value *, std::vector<Value *> >::iterator C =
13916ca6d3b1eac5b8611f3a9e2c270c2e794d37e1f5Hal Finkel         CandidatePairs.begin(), E = CandidatePairs.end(); C != E; ++C) {
1392de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      IsInPair.insert(C->first);
13936ca6d3b1eac5b8611f3a9e2c270c2e794d37e1f5Hal Finkel      IsInPair.insert(C->second.begin(), C->second.end());
1394de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    }
1395de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
13967a8aba906416b6998347b52c3c08610fdc190638Hal Finkel    // Iterate through the basic block, recording all users of each
1397de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    // pairable instruction.
1398de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
1399c951003faf4d475d221f5e839971673d2350b983Hal Finkel    BasicBlock::iterator E = BB.end(), EL =
1400c951003faf4d475d221f5e839971673d2350b983Hal Finkel      BasicBlock::iterator(cast<Instruction>(PairableInsts.back()));
1401de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    for (BasicBlock::iterator I = BB.getFirstInsertionPt(); I != E; ++I) {
1402de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      if (IsInPair.find(I) == IsInPair.end()) continue;
1403de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
1404de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      DenseSet<Value *> Users;
1405de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      AliasSetTracker WriteSet(*AA);
1406c951003faf4d475d221f5e839971673d2350b983Hal Finkel      for (BasicBlock::iterator J = llvm::next(I); J != E; ++J) {
1407de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        (void) trackUsesOfI(Users, WriteSet, I, J);
1408de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
1409c951003faf4d475d221f5e839971673d2350b983Hal Finkel        if (J == EL)
1410c951003faf4d475d221f5e839971673d2350b983Hal Finkel          break;
1411c951003faf4d475d221f5e839971673d2350b983Hal Finkel      }
1412c951003faf4d475d221f5e839971673d2350b983Hal Finkel
1413de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      for (DenseSet<Value *>::iterator U = Users.begin(), E = Users.end();
14148f3359a4b396d3f1a7b2726e02f199be74c62e4cHal Finkel           U != E; ++U) {
14158f3359a4b396d3f1a7b2726e02f199be74c62e4cHal Finkel        if (IsInPair.find(*U) == IsInPair.end()) continue;
1416de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        PairableInstUsers.insert(ValuePair(I, *U));
14178f3359a4b396d3f1a7b2726e02f199be74c62e4cHal Finkel      }
1418c951003faf4d475d221f5e839971673d2350b983Hal Finkel
1419c951003faf4d475d221f5e839971673d2350b983Hal Finkel      if (I == EL)
1420c951003faf4d475d221f5e839971673d2350b983Hal Finkel        break;
1421de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    }
1422de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  }
1423de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
1424de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  // Returns true if an input to pair P is an output of pair Q and also an
1425de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  // input of pair Q is an output of pair P. If this is the case, then these
1426de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  // two pairs cannot be simultaneously fused.
1427de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  bool BBVectorize::pairsConflict(ValuePair P, ValuePair Q,
142897a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel             DenseSet<ValuePair> &PairableInstUsers,
142997a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel             DenseMap<ValuePair, std::vector<ValuePair> > *PairableInstUserMap,
143097a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel             DenseSet<VPPair> *PairableInstUserPairSet) {
1431de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    // Two pairs are in conflict if they are mutual Users of eachother.
1432de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    bool QUsesP = PairableInstUsers.count(ValuePair(P.first,  Q.first))  ||
1433de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel                  PairableInstUsers.count(ValuePair(P.first,  Q.second)) ||
1434de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel                  PairableInstUsers.count(ValuePair(P.second, Q.first))  ||
1435de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel                  PairableInstUsers.count(ValuePair(P.second, Q.second));
1436de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    bool PUsesQ = PairableInstUsers.count(ValuePair(Q.first,  P.first))  ||
1437de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel                  PairableInstUsers.count(ValuePair(Q.first,  P.second)) ||
1438de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel                  PairableInstUsers.count(ValuePair(Q.second, P.first))  ||
1439de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel                  PairableInstUsers.count(ValuePair(Q.second, P.second));
1440de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    if (PairableInstUserMap) {
1441de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      // FIXME: The expensive part of the cycle check is not so much the cycle
1442de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      // check itself but this edge insertion procedure. This needs some
144397a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel      // profiling and probably a different data structure.
1444de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      if (PUsesQ) {
1445da20ea696d8b24d89ae157106ddad2337296ed50Hal Finkel        if (PairableInstUserPairSet->insert(VPPair(Q, P)).second)
144697a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel          (*PairableInstUserMap)[Q].push_back(P);
1447de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      }
1448de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      if (QUsesP) {
1449da20ea696d8b24d89ae157106ddad2337296ed50Hal Finkel        if (PairableInstUserPairSet->insert(VPPair(P, Q)).second)
145097a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel          (*PairableInstUserMap)[P].push_back(Q);
1451de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      }
1452de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    }
1453de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
1454de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    return (QUsesP && PUsesQ);
1455de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  }
1456de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
1457de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  // This function walks the use graph of current pairs to see if, starting
1458de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  // from P, the walk returns to P.
1459de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  bool BBVectorize::pairWillFormCycle(ValuePair P,
146097a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel             DenseMap<ValuePair, std::vector<ValuePair> > &PairableInstUserMap,
146197a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel             DenseSet<ValuePair> &CurrentPairs) {
1462de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    DEBUG(if (DebugCycleCheck)
1463de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel            dbgs() << "BBV: starting cycle check for : " << *P.first << " <-> "
1464de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel                   << *P.second << "\n");
1465de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    // A lookup table of visisted pairs is kept because the PairableInstUserMap
1466de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    // contains non-direct associations.
1467de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    DenseSet<ValuePair> Visited;
146835564dc3ae1c377abad425cb09928eaf676dcb3cHal Finkel    SmallVector<ValuePair, 32> Q;
1469de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    // General depth-first post-order traversal:
1470de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    Q.push_back(P);
147135564dc3ae1c377abad425cb09928eaf676dcb3cHal Finkel    do {
147235564dc3ae1c377abad425cb09928eaf676dcb3cHal Finkel      ValuePair QTop = Q.pop_back_val();
1473de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      Visited.insert(QTop);
1474de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
1475de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      DEBUG(if (DebugCycleCheck)
1476de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel              dbgs() << "BBV: cycle check visiting: " << *QTop.first << " <-> "
1477de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel                     << *QTop.second << "\n");
147897a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel      DenseMap<ValuePair, std::vector<ValuePair> >::iterator QQ =
147997a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel        PairableInstUserMap.find(QTop);
148097a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel      if (QQ == PairableInstUserMap.end())
148197a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel        continue;
148297a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel
148397a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel      for (std::vector<ValuePair>::iterator C = QQ->second.begin(),
148497a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel           CE = QQ->second.end(); C != CE; ++C) {
148597a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel        if (*C == P) {
1486de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel          DEBUG(dbgs()
1487de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel                 << "BBV: rejected to prevent non-trivial cycle formation: "
148897a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel                 << QTop.first << " <-> " << C->second << "\n");
1489de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel          return true;
1490de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        }
1491de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
149297a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel        if (CurrentPairs.count(*C) && !Visited.count(*C))
149397a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel          Q.push_back(*C);
1494de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      }
149535564dc3ae1c377abad425cb09928eaf676dcb3cHal Finkel    } while (!Q.empty());
1496de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
1497de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    return false;
1498de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  }
1499de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
1500f64a7a83bea5f1d2ab1e71231616c6cb0487d56eHal Finkel  // This function builds the initial dag of connected pairs with the
1501de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  // pair J at the root.
1502f64a7a83bea5f1d2ab1e71231616c6cb0487d56eHal Finkel  void BBVectorize::buildInitialDAGFor(
150397a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel                  DenseMap<Value *, std::vector<Value *> > &CandidatePairs,
150497a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel                  DenseSet<ValuePair> &CandidatePairsSet,
150597a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel                  std::vector<Value *> &PairableInsts,
150697a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel                  DenseMap<ValuePair, std::vector<ValuePair> > &ConnectedPairs,
150797a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel                  DenseSet<ValuePair> &PairableInstUsers,
150897a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel                  DenseMap<Value *, Value *> &ChosenPairs,
1509f64a7a83bea5f1d2ab1e71231616c6cb0487d56eHal Finkel                  DenseMap<ValuePair, size_t> &DAG, ValuePair J) {
1510f64a7a83bea5f1d2ab1e71231616c6cb0487d56eHal Finkel    // Each of these pairs is viewed as the root node of a DAG. The DAG
1511de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    // is then walked (depth-first). As this happens, we keep track of
1512f64a7a83bea5f1d2ab1e71231616c6cb0487d56eHal Finkel    // the pairs that compose the DAG and the maximum depth of the DAG.
151335564dc3ae1c377abad425cb09928eaf676dcb3cHal Finkel    SmallVector<ValuePairWithDepth, 32> Q;
1514de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    // General depth-first post-order traversal:
1515de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    Q.push_back(ValuePairWithDepth(J, getDepthFactor(J.first)));
151635564dc3ae1c377abad425cb09928eaf676dcb3cHal Finkel    do {
1517de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      ValuePairWithDepth QTop = Q.back();
1518de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
1519de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      // Push each child onto the queue:
1520de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      bool MoreChildren = false;
1521de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      size_t MaxChildDepth = QTop.second;
152297a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel      DenseMap<ValuePair, std::vector<ValuePair> >::iterator QQ =
152397a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel        ConnectedPairs.find(QTop.first);
152497a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel      if (QQ != ConnectedPairs.end())
152597a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel        for (std::vector<ValuePair>::iterator k = QQ->second.begin(),
152697a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel             ke = QQ->second.end(); k != ke; ++k) {
152797a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel          // Make sure that this child pair is still a candidate:
152897a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel          if (CandidatePairsSet.count(*k)) {
1529f64a7a83bea5f1d2ab1e71231616c6cb0487d56eHal Finkel            DenseMap<ValuePair, size_t>::iterator C = DAG.find(*k);
1530f64a7a83bea5f1d2ab1e71231616c6cb0487d56eHal Finkel            if (C == DAG.end()) {
153197a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel              size_t d = getDepthFactor(k->first);
153297a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel              Q.push_back(ValuePairWithDepth(*k, QTop.second+d));
153397a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel              MoreChildren = true;
153497a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel            } else {
153597a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel              MaxChildDepth = std::max(MaxChildDepth, C->second);
153697a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel            }
1537de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel          }
1538de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        }
1539de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
1540de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      if (!MoreChildren) {
1541f64a7a83bea5f1d2ab1e71231616c6cb0487d56eHal Finkel        // Record the current pair as part of the DAG:
1542f64a7a83bea5f1d2ab1e71231616c6cb0487d56eHal Finkel        DAG.insert(ValuePairWithDepth(QTop.first, MaxChildDepth));
1543de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        Q.pop_back();
1544de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      }
154535564dc3ae1c377abad425cb09928eaf676dcb3cHal Finkel    } while (!Q.empty());
1546de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  }
1547de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
1548f64a7a83bea5f1d2ab1e71231616c6cb0487d56eHal Finkel  // Given some initial dag, prune it by removing conflicting pairs (pairs
1549de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  // that cannot be simultaneously chosen for vectorization).
1550f64a7a83bea5f1d2ab1e71231616c6cb0487d56eHal Finkel  void BBVectorize::pruneDAGFor(
155197a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel              DenseMap<Value *, std::vector<Value *> > &CandidatePairs,
155297a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel              std::vector<Value *> &PairableInsts,
155397a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel              DenseMap<ValuePair, std::vector<ValuePair> > &ConnectedPairs,
155497a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel              DenseSet<ValuePair> &PairableInstUsers,
155597a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel              DenseMap<ValuePair, std::vector<ValuePair> > &PairableInstUserMap,
155697a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel              DenseSet<VPPair> &PairableInstUserPairSet,
155797a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel              DenseMap<Value *, Value *> &ChosenPairs,
1558f64a7a83bea5f1d2ab1e71231616c6cb0487d56eHal Finkel              DenseMap<ValuePair, size_t> &DAG,
1559f64a7a83bea5f1d2ab1e71231616c6cb0487d56eHal Finkel              DenseSet<ValuePair> &PrunedDAG, ValuePair J,
156097a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel              bool UseCycleCheck) {
156135564dc3ae1c377abad425cb09928eaf676dcb3cHal Finkel    SmallVector<ValuePairWithDepth, 32> Q;
1562de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    // General depth-first post-order traversal:
1563de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    Q.push_back(ValuePairWithDepth(J, getDepthFactor(J.first)));
156435564dc3ae1c377abad425cb09928eaf676dcb3cHal Finkel    do {
156535564dc3ae1c377abad425cb09928eaf676dcb3cHal Finkel      ValuePairWithDepth QTop = Q.pop_back_val();
1566f64a7a83bea5f1d2ab1e71231616c6cb0487d56eHal Finkel      PrunedDAG.insert(QTop.first);
1567de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
1568de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      // Visit each child, pruning as necessary...
156997d19ebe5b5bf27617e536a16fa232116cefe914Hal Finkel      SmallVector<ValuePairWithDepth, 8> BestChildren;
157097a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel      DenseMap<ValuePair, std::vector<ValuePair> >::iterator QQ =
157197a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel        ConnectedPairs.find(QTop.first);
157297a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel      if (QQ == ConnectedPairs.end())
157397a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel        continue;
157497a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel
157597a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel      for (std::vector<ValuePair>::iterator K = QQ->second.begin(),
157697a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel           KE = QQ->second.end(); K != KE; ++K) {
1577f64a7a83bea5f1d2ab1e71231616c6cb0487d56eHal Finkel        DenseMap<ValuePair, size_t>::iterator C = DAG.find(*K);
1578f64a7a83bea5f1d2ab1e71231616c6cb0487d56eHal Finkel        if (C == DAG.end()) continue;
1579de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
1580f64a7a83bea5f1d2ab1e71231616c6cb0487d56eHal Finkel        // This child is in the DAG, now we need to make sure it is the
1581de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        // best of any conflicting children. There could be multiple
1582de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        // conflicting children, so first, determine if we're keeping
1583de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        // this child, then delete conflicting children as necessary.
1584de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
1585de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        // It is also necessary to guard against pairing-induced
1586de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        // dependencies. Consider instructions a .. x .. y .. b
1587de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        // such that (a,b) are to be fused and (x,y) are to be fused
1588de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        // but a is an input to x and b is an output from y. This
1589de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        // means that y cannot be moved after b but x must be moved
1590de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        // after b for (a,b) to be fused. In other words, after
1591de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        // fusing (a,b) we have y .. a/b .. x where y is an input
1592de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        // to a/b and x is an output to a/b: x and y can no longer
1593de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        // be legally fused. To prevent this condition, we must
1594f64a7a83bea5f1d2ab1e71231616c6cb0487d56eHal Finkel        // make sure that a child pair added to the DAG is not
1595de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        // both an input and output of an already-selected pair.
1596de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
1597de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        // Pairing-induced dependencies can also form from more complicated
1598de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        // cycles. The pair vs. pair conflicts are easy to check, and so
1599de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        // that is done explicitly for "fast rejection", and because for
1600de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        // child vs. child conflicts, we may prefer to keep the current
1601de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        // pair in preference to the already-selected child.
1602de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        DenseSet<ValuePair> CurrentPairs;
1603de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
1604de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        bool CanAdd = true;
160597d19ebe5b5bf27617e536a16fa232116cefe914Hal Finkel        for (SmallVector<ValuePairWithDepth, 8>::iterator C2
160643ec0f4921e315dd9507be7467e633a837ad23dbSebastian Pop              = BestChildren.begin(), E2 = BestChildren.end();
1607de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel             C2 != E2; ++C2) {
1608de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel          if (C2->first.first == C->first.first ||
1609de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel              C2->first.first == C->first.second ||
1610de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel              C2->first.second == C->first.first ||
1611de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel              C2->first.second == C->first.second ||
1612de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel              pairsConflict(C2->first, C->first, PairableInstUsers,
1613da20ea696d8b24d89ae157106ddad2337296ed50Hal Finkel                            UseCycleCheck ? &PairableInstUserMap : 0,
1614da20ea696d8b24d89ae157106ddad2337296ed50Hal Finkel                            UseCycleCheck ? &PairableInstUserPairSet : 0)) {
1615de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel            if (C2->second >= C->second) {
1616de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel              CanAdd = false;
1617de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel              break;
1618de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel            }
1619de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
1620de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel            CurrentPairs.insert(C2->first);
1621de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel          }
1622de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        }
1623de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        if (!CanAdd) continue;
1624de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
1625de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        // Even worse, this child could conflict with another node already
1626f64a7a83bea5f1d2ab1e71231616c6cb0487d56eHal Finkel        // selected for the DAG. If that is the case, ignore this child.
1627f64a7a83bea5f1d2ab1e71231616c6cb0487d56eHal Finkel        for (DenseSet<ValuePair>::iterator T = PrunedDAG.begin(),
1628f64a7a83bea5f1d2ab1e71231616c6cb0487d56eHal Finkel             E2 = PrunedDAG.end(); T != E2; ++T) {
1629de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel          if (T->first == C->first.first ||
1630de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel              T->first == C->first.second ||
1631de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel              T->second == C->first.first ||
1632de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel              T->second == C->first.second ||
1633de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel              pairsConflict(*T, C->first, PairableInstUsers,
1634da20ea696d8b24d89ae157106ddad2337296ed50Hal Finkel                            UseCycleCheck ? &PairableInstUserMap : 0,
1635da20ea696d8b24d89ae157106ddad2337296ed50Hal Finkel                            UseCycleCheck ? &PairableInstUserPairSet : 0)) {
1636de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel            CanAdd = false;
1637de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel            break;
1638de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel          }
1639de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
1640de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel          CurrentPairs.insert(*T);
1641de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        }
1642de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        if (!CanAdd) continue;
1643de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
1644de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        // And check the queue too...
164535564dc3ae1c377abad425cb09928eaf676dcb3cHal Finkel        for (SmallVector<ValuePairWithDepth, 32>::iterator C2 = Q.begin(),
1646de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel             E2 = Q.end(); C2 != E2; ++C2) {
1647de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel          if (C2->first.first == C->first.first ||
1648de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel              C2->first.first == C->first.second ||
1649de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel              C2->first.second == C->first.first ||
1650de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel              C2->first.second == C->first.second ||
1651de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel              pairsConflict(C2->first, C->first, PairableInstUsers,
1652da20ea696d8b24d89ae157106ddad2337296ed50Hal Finkel                            UseCycleCheck ? &PairableInstUserMap : 0,
1653da20ea696d8b24d89ae157106ddad2337296ed50Hal Finkel                            UseCycleCheck ? &PairableInstUserPairSet : 0)) {
1654de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel            CanAdd = false;
1655de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel            break;
1656de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel          }
1657de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
1658de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel          CurrentPairs.insert(C2->first);
1659de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        }
1660de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        if (!CanAdd) continue;
1661de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
1662de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        // Last but not least, check for a conflict with any of the
1663de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        // already-chosen pairs.
1664de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        for (DenseMap<Value *, Value *>::iterator C2 =
1665de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel              ChosenPairs.begin(), E2 = ChosenPairs.end();
1666de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel             C2 != E2; ++C2) {
1667de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel          if (pairsConflict(*C2, C->first, PairableInstUsers,
1668da20ea696d8b24d89ae157106ddad2337296ed50Hal Finkel                            UseCycleCheck ? &PairableInstUserMap : 0,
1669da20ea696d8b24d89ae157106ddad2337296ed50Hal Finkel                            UseCycleCheck ? &PairableInstUserPairSet : 0)) {
1670de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel            CanAdd = false;
1671de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel            break;
1672de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel          }
1673de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
1674de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel          CurrentPairs.insert(*C2);
1675de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        }
1676de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        if (!CanAdd) continue;
1677de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
16781230ad6e8cb7977527ac64dcf5005464d7d6c20bSebastian Pop        // To check for non-trivial cycles formed by the addition of the
16791230ad6e8cb7977527ac64dcf5005464d7d6c20bSebastian Pop        // current pair we've formed a list of all relevant pairs, now use a
16801230ad6e8cb7977527ac64dcf5005464d7d6c20bSebastian Pop        // graph walk to check for a cycle. We start from the current pair and
1681f64a7a83bea5f1d2ab1e71231616c6cb0487d56eHal Finkel        // walk the use dag to see if we again reach the current pair. If we
16821230ad6e8cb7977527ac64dcf5005464d7d6c20bSebastian Pop        // do, then the current pair is rejected.
1683de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
1684de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        // FIXME: It may be more efficient to use a topological-ordering
1685de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        // algorithm to improve the cycle check. This should be investigated.
1686de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        if (UseCycleCheck &&
1687de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel            pairWillFormCycle(C->first, PairableInstUserMap, CurrentPairs))
1688de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel          continue;
1689de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
1690de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        // This child can be added, but we may have chosen it in preference
1691de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        // to an already-selected child. Check for this here, and if a
1692de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        // conflict is found, then remove the previously-selected child
1693de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        // before adding this one in its place.
169497d19ebe5b5bf27617e536a16fa232116cefe914Hal Finkel        for (SmallVector<ValuePairWithDepth, 8>::iterator C2
169543ec0f4921e315dd9507be7467e633a837ad23dbSebastian Pop              = BestChildren.begin(); C2 != BestChildren.end();) {
1696de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel          if (C2->first.first == C->first.first ||
1697de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel              C2->first.first == C->first.second ||
1698de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel              C2->first.second == C->first.first ||
1699de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel              C2->first.second == C->first.second ||
1700de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel              pairsConflict(C2->first, C->first, PairableInstUsers))
1701d7a3425f06d51ed579bd9aefeb835b7fa4ce7849Hal Finkel            C2 = BestChildren.erase(C2);
1702de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel          else
1703de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel            ++C2;
1704de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        }
1705de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
1706d7a3425f06d51ed579bd9aefeb835b7fa4ce7849Hal Finkel        BestChildren.push_back(ValuePairWithDepth(C->first, C->second));
1707de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      }
1708de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
170997d19ebe5b5bf27617e536a16fa232116cefe914Hal Finkel      for (SmallVector<ValuePairWithDepth, 8>::iterator C
171043ec0f4921e315dd9507be7467e633a837ad23dbSebastian Pop            = BestChildren.begin(), E2 = BestChildren.end();
1711de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel           C != E2; ++C) {
1712de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        size_t DepthF = getDepthFactor(C->first.first);
1713de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        Q.push_back(ValuePairWithDepth(C->first, QTop.second+DepthF));
1714de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      }
171535564dc3ae1c377abad425cb09928eaf676dcb3cHal Finkel    } while (!Q.empty());
1716de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  }
1717de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
1718f64a7a83bea5f1d2ab1e71231616c6cb0487d56eHal Finkel  // This function finds the best dag of mututally-compatible connected
1719de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  // pairs, given the choice of root pairs as an iterator range.
1720f64a7a83bea5f1d2ab1e71231616c6cb0487d56eHal Finkel  void BBVectorize::findBestDAGFor(
172197a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel              DenseMap<Value *, std::vector<Value *> > &CandidatePairs,
172297a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel              DenseSet<ValuePair> &CandidatePairsSet,
172397a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel              DenseMap<ValuePair, int> &CandidatePairCostSavings,
172497a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel              std::vector<Value *> &PairableInsts,
172597a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel              DenseSet<ValuePair> &FixedOrderPairs,
172697a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel              DenseMap<VPPair, unsigned> &PairConnectionTypes,
172797a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel              DenseMap<ValuePair, std::vector<ValuePair> > &ConnectedPairs,
172897a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel              DenseMap<ValuePair, std::vector<ValuePair> > &ConnectedPairDeps,
172997a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel              DenseSet<ValuePair> &PairableInstUsers,
173097a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel              DenseMap<ValuePair, std::vector<ValuePair> > &PairableInstUserMap,
173197a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel              DenseSet<VPPair> &PairableInstUserPairSet,
173297a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel              DenseMap<Value *, Value *> &ChosenPairs,
1733f64a7a83bea5f1d2ab1e71231616c6cb0487d56eHal Finkel              DenseSet<ValuePair> &BestDAG, size_t &BestMaxDepth,
173497a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel              int &BestEffSize, Value *II, std::vector<Value *>&JJ,
173597a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel              bool UseCycleCheck) {
17366ca6d3b1eac5b8611f3a9e2c270c2e794d37e1f5Hal Finkel    for (std::vector<Value *>::iterator J = JJ.begin(), JE = JJ.end();
17376ca6d3b1eac5b8611f3a9e2c270c2e794d37e1f5Hal Finkel         J != JE; ++J) {
17386ca6d3b1eac5b8611f3a9e2c270c2e794d37e1f5Hal Finkel      ValuePair IJ(II, *J);
17396ca6d3b1eac5b8611f3a9e2c270c2e794d37e1f5Hal Finkel      if (!CandidatePairsSet.count(IJ))
17406ca6d3b1eac5b8611f3a9e2c270c2e794d37e1f5Hal Finkel        continue;
1741de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
1742de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      // Before going any further, make sure that this pair does not
1743de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      // conflict with any already-selected pairs (see comment below
1744f64a7a83bea5f1d2ab1e71231616c6cb0487d56eHal Finkel      // near the DAG pruning for more details).
1745de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      DenseSet<ValuePair> ChosenPairSet;
1746de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      bool DoesConflict = false;
1747de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      for (DenseMap<Value *, Value *>::iterator C = ChosenPairs.begin(),
1748de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel           E = ChosenPairs.end(); C != E; ++C) {
17496ca6d3b1eac5b8611f3a9e2c270c2e794d37e1f5Hal Finkel        if (pairsConflict(*C, IJ, PairableInstUsers,
1750da20ea696d8b24d89ae157106ddad2337296ed50Hal Finkel                          UseCycleCheck ? &PairableInstUserMap : 0,
1751da20ea696d8b24d89ae157106ddad2337296ed50Hal Finkel                          UseCycleCheck ? &PairableInstUserPairSet : 0)) {
1752de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel          DoesConflict = true;
1753de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel          break;
1754de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        }
1755de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
1756de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        ChosenPairSet.insert(*C);
1757de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      }
1758de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      if (DoesConflict) continue;
1759de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
1760de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      if (UseCycleCheck &&
17616ca6d3b1eac5b8611f3a9e2c270c2e794d37e1f5Hal Finkel          pairWillFormCycle(IJ, PairableInstUserMap, ChosenPairSet))
1762de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        continue;
1763de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
1764f64a7a83bea5f1d2ab1e71231616c6cb0487d56eHal Finkel      DenseMap<ValuePair, size_t> DAG;
1765f64a7a83bea5f1d2ab1e71231616c6cb0487d56eHal Finkel      buildInitialDAGFor(CandidatePairs, CandidatePairsSet,
1766b1a82589339fed148c12b052d30861a539552f1aHal Finkel                          PairableInsts, ConnectedPairs,
1767f64a7a83bea5f1d2ab1e71231616c6cb0487d56eHal Finkel                          PairableInstUsers, ChosenPairs, DAG, IJ);
1768de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
1769de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      // Because we'll keep the child with the largest depth, the largest
1770f64a7a83bea5f1d2ab1e71231616c6cb0487d56eHal Finkel      // depth is still the same in the unpruned DAG.
1771f64a7a83bea5f1d2ab1e71231616c6cb0487d56eHal Finkel      size_t MaxDepth = DAG.lookup(IJ);
1772de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
1773f64a7a83bea5f1d2ab1e71231616c6cb0487d56eHal Finkel      DEBUG(if (DebugPairSelection) dbgs() << "BBV: found DAG for pair {"
177476a05c93b13a8debec4497a6e4e753d7531709e5Hal Finkel                   << *IJ.first << " <-> " << *IJ.second << "} of depth " <<
1775f64a7a83bea5f1d2ab1e71231616c6cb0487d56eHal Finkel                   MaxDepth << " and size " << DAG.size() << "\n");
1776de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
1777f64a7a83bea5f1d2ab1e71231616c6cb0487d56eHal Finkel      // At this point the DAG has been constructed, but, may contain
1778de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      // contradictory children (meaning that different children of
1779f64a7a83bea5f1d2ab1e71231616c6cb0487d56eHal Finkel      // some dag node may be attempting to fuse the same instruction).
1780f64a7a83bea5f1d2ab1e71231616c6cb0487d56eHal Finkel      // So now we walk the dag again, in the case of a conflict,
1781de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      // keep only the child with the largest depth. To break a tie,
1782de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      // favor the first child.
1783de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
1784f64a7a83bea5f1d2ab1e71231616c6cb0487d56eHal Finkel      DenseSet<ValuePair> PrunedDAG;
1785f64a7a83bea5f1d2ab1e71231616c6cb0487d56eHal Finkel      pruneDAGFor(CandidatePairs, PairableInsts, ConnectedPairs,
1786b1a82589339fed148c12b052d30861a539552f1aHal Finkel                   PairableInstUsers, PairableInstUserMap,
1787b1a82589339fed148c12b052d30861a539552f1aHal Finkel                   PairableInstUserPairSet,
1788f64a7a83bea5f1d2ab1e71231616c6cb0487d56eHal Finkel                   ChosenPairs, DAG, PrunedDAG, IJ, UseCycleCheck);
1789de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
179065309660fa61a837cc05323f69c618a7d8134d56Hal Finkel      int EffSize = 0;
1791abc227d9b39baed266c453f459ca14242f5c1eeaChandler Carruth      if (TTI) {
1792f64a7a83bea5f1d2ab1e71231616c6cb0487d56eHal Finkel        DenseSet<Value *> PrunedDAGInstrs;
1793f64a7a83bea5f1d2ab1e71231616c6cb0487d56eHal Finkel        for (DenseSet<ValuePair>::iterator S = PrunedDAG.begin(),
1794f64a7a83bea5f1d2ab1e71231616c6cb0487d56eHal Finkel             E = PrunedDAG.end(); S != E; ++S) {
1795f64a7a83bea5f1d2ab1e71231616c6cb0487d56eHal Finkel          PrunedDAGInstrs.insert(S->first);
1796f64a7a83bea5f1d2ab1e71231616c6cb0487d56eHal Finkel          PrunedDAGInstrs.insert(S->second);
179778fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel        }
179878fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel
179978fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel        // The set of pairs that have already contributed to the total cost.
180078fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel        DenseSet<ValuePair> IncomingPairs;
180178fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel
18024387b8c95971a512e07bfda30dea6459e8419e8fHal Finkel        // If the cost model were perfect, this might not be necessary; but we
18034387b8c95971a512e07bfda30dea6459e8419e8fHal Finkel        // need to make sure that we don't get stuck vectorizing our own
18044387b8c95971a512e07bfda30dea6459e8419e8fHal Finkel        // shuffle chains.
18054387b8c95971a512e07bfda30dea6459e8419e8fHal Finkel        bool HasNontrivialInsts = false;
18064387b8c95971a512e07bfda30dea6459e8419e8fHal Finkel
180786ccc55c82651f91fd6a312c5f6a4b511bcd1aecHal Finkel        // The node weights represent the cost savings associated with
180886ccc55c82651f91fd6a312c5f6a4b511bcd1aecHal Finkel        // fusing the pair of instructions.
1809f64a7a83bea5f1d2ab1e71231616c6cb0487d56eHal Finkel        for (DenseSet<ValuePair>::iterator S = PrunedDAG.begin(),
1810f64a7a83bea5f1d2ab1e71231616c6cb0487d56eHal Finkel             E = PrunedDAG.end(); S != E; ++S) {
18114387b8c95971a512e07bfda30dea6459e8419e8fHal Finkel          if (!isa<ShuffleVectorInst>(S->first) &&
18124387b8c95971a512e07bfda30dea6459e8419e8fHal Finkel              !isa<InsertElementInst>(S->first) &&
18134387b8c95971a512e07bfda30dea6459e8419e8fHal Finkel              !isa<ExtractElementInst>(S->first))
18144387b8c95971a512e07bfda30dea6459e8419e8fHal Finkel            HasNontrivialInsts = true;
18154387b8c95971a512e07bfda30dea6459e8419e8fHal Finkel
181678fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel          bool FlipOrder = false;
181778fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel
181878fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel          if (getDepthFactor(S->first)) {
181978fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel            int ESContrib = CandidatePairCostSavings.find(*S)->second;
182078fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel            DEBUG(if (DebugPairSelection) dbgs() << "\tweight {"
182178fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel                   << *S->first << " <-> " << *S->second << "} = " <<
182278fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel                   ESContrib << "\n");
182378fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel            EffSize += ESContrib;
182478fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel          }
182586ccc55c82651f91fd6a312c5f6a4b511bcd1aecHal Finkel
182678fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel          // The edge weights contribute in a negative sense: they represent
182778fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel          // the cost of shuffles.
182897a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel          DenseMap<ValuePair, std::vector<ValuePair> >::iterator SS =
182997a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel            ConnectedPairDeps.find(*S);
183097a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel          if (SS != ConnectedPairDeps.end()) {
183186ccc55c82651f91fd6a312c5f6a4b511bcd1aecHal Finkel            unsigned NumDepsDirect = 0, NumDepsSwap = 0;
183297a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel            for (std::vector<ValuePair>::iterator T = SS->second.begin(),
183397a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel                 TE = SS->second.end(); T != TE; ++T) {
183497a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel              VPPair Q(*S, *T);
1835f64a7a83bea5f1d2ab1e71231616c6cb0487d56eHal Finkel              if (!PrunedDAG.count(Q.second))
183678fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel                continue;
183786ccc55c82651f91fd6a312c5f6a4b511bcd1aecHal Finkel              DenseMap<VPPair, unsigned>::iterator R =
183897a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel                PairConnectionTypes.find(VPPair(Q.second, Q.first));
183986ccc55c82651f91fd6a312c5f6a4b511bcd1aecHal Finkel              assert(R != PairConnectionTypes.end() &&
184086ccc55c82651f91fd6a312c5f6a4b511bcd1aecHal Finkel                     "Cannot find pair connection type");
184186ccc55c82651f91fd6a312c5f6a4b511bcd1aecHal Finkel              if (R->second == PairConnectionDirect)
184286ccc55c82651f91fd6a312c5f6a4b511bcd1aecHal Finkel                ++NumDepsDirect;
184386ccc55c82651f91fd6a312c5f6a4b511bcd1aecHal Finkel              else if (R->second == PairConnectionSwap)
184486ccc55c82651f91fd6a312c5f6a4b511bcd1aecHal Finkel                ++NumDepsSwap;
184586ccc55c82651f91fd6a312c5f6a4b511bcd1aecHal Finkel            }
184686ccc55c82651f91fd6a312c5f6a4b511bcd1aecHal Finkel
184786ccc55c82651f91fd6a312c5f6a4b511bcd1aecHal Finkel            // If there are more swaps than direct connections, then
184886ccc55c82651f91fd6a312c5f6a4b511bcd1aecHal Finkel            // the pair order will be flipped during fusion. So the real
184986ccc55c82651f91fd6a312c5f6a4b511bcd1aecHal Finkel            // number of swaps is the minimum number.
185078fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel            FlipOrder = !FixedOrderPairs.count(*S) &&
185186ccc55c82651f91fd6a312c5f6a4b511bcd1aecHal Finkel              ((NumDepsSwap > NumDepsDirect) ||
185286ccc55c82651f91fd6a312c5f6a4b511bcd1aecHal Finkel                FixedOrderPairs.count(ValuePair(S->second, S->first)));
185386ccc55c82651f91fd6a312c5f6a4b511bcd1aecHal Finkel
185497a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel            for (std::vector<ValuePair>::iterator T = SS->second.begin(),
185597a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel                 TE = SS->second.end(); T != TE; ++T) {
185697a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel              VPPair Q(*S, *T);
1857f64a7a83bea5f1d2ab1e71231616c6cb0487d56eHal Finkel              if (!PrunedDAG.count(Q.second))
185878fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel                continue;
185986ccc55c82651f91fd6a312c5f6a4b511bcd1aecHal Finkel              DenseMap<VPPair, unsigned>::iterator R =
186097a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel                PairConnectionTypes.find(VPPair(Q.second, Q.first));
186186ccc55c82651f91fd6a312c5f6a4b511bcd1aecHal Finkel              assert(R != PairConnectionTypes.end() &&
186286ccc55c82651f91fd6a312c5f6a4b511bcd1aecHal Finkel                     "Cannot find pair connection type");
186397a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel              Type *Ty1 = Q.second.first->getType(),
186497a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel                   *Ty2 = Q.second.second->getType();
186586ccc55c82651f91fd6a312c5f6a4b511bcd1aecHal Finkel              Type *VTy = getVecTypeForPair(Ty1, Ty2);
186686ccc55c82651f91fd6a312c5f6a4b511bcd1aecHal Finkel              if ((R->second == PairConnectionDirect && FlipOrder) ||
186786ccc55c82651f91fd6a312c5f6a4b511bcd1aecHal Finkel                  (R->second == PairConnectionSwap && !FlipOrder)  ||
186878fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel                  R->second == PairConnectionSplat) {
186978fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel                int ESContrib = (int) getInstrCost(Instruction::ShuffleVector,
187078fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel                                                   VTy, VTy);
1871245b657ab636a505066ea6a81591a9a8b93604d2Hal Finkel
1872245b657ab636a505066ea6a81591a9a8b93604d2Hal Finkel                if (VTy->getVectorNumElements() == 2) {
1873245b657ab636a505066ea6a81591a9a8b93604d2Hal Finkel                  if (R->second == PairConnectionSplat)
1874245b657ab636a505066ea6a81591a9a8b93604d2Hal Finkel                    ESContrib = std::min(ESContrib, (int) TTI->getShuffleCost(
1875245b657ab636a505066ea6a81591a9a8b93604d2Hal Finkel                      TargetTransformInfo::SK_Broadcast, VTy));
1876245b657ab636a505066ea6a81591a9a8b93604d2Hal Finkel                  else
1877245b657ab636a505066ea6a81591a9a8b93604d2Hal Finkel                    ESContrib = std::min(ESContrib, (int) TTI->getShuffleCost(
1878245b657ab636a505066ea6a81591a9a8b93604d2Hal Finkel                      TargetTransformInfo::SK_Reverse, VTy));
1879245b657ab636a505066ea6a81591a9a8b93604d2Hal Finkel                }
1880245b657ab636a505066ea6a81591a9a8b93604d2Hal Finkel
188178fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel                DEBUG(if (DebugPairSelection) dbgs() << "\tcost {" <<
188297a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel                  *Q.second.first << " <-> " << *Q.second.second <<
188378fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel                    "} -> {" <<
188478fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel                  *S->first << " <-> " << *S->second << "} = " <<
188578fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel                   ESContrib << "\n");
188678fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel                EffSize -= ESContrib;
188778fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel              }
188878fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel            }
188978fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel          }
189078fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel
189178fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel          // Compute the cost of outgoing edges. We assume that edges outgoing
189278fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel          // to shuffles, inserts or extracts can be merged, and so contribute
189378fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel          // no additional cost.
189478fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel          if (!S->first->getType()->isVoidTy()) {
189578fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel            Type *Ty1 = S->first->getType(),
189678fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel                 *Ty2 = S->second->getType();
189778fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel            Type *VTy = getVecTypeForPair(Ty1, Ty2);
189878fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel
189978fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel            bool NeedsExtraction = false;
190078fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel            for (Value::use_iterator I = S->first->use_begin(),
190178fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel                 IE = S->first->use_end(); I != IE; ++I) {
190286c88c938aec8006d2ce83325ec1f31e1154620bHal Finkel              if (ShuffleVectorInst *SI = dyn_cast<ShuffleVectorInst>(*I)) {
190386c88c938aec8006d2ce83325ec1f31e1154620bHal Finkel                // Shuffle can be folded if it has no other input
190486c88c938aec8006d2ce83325ec1f31e1154620bHal Finkel                if (isa<UndefValue>(SI->getOperand(1)))
190586c88c938aec8006d2ce83325ec1f31e1154620bHal Finkel                  continue;
190686c88c938aec8006d2ce83325ec1f31e1154620bHal Finkel              }
190786c88c938aec8006d2ce83325ec1f31e1154620bHal Finkel              if (isa<ExtractElementInst>(*I))
190878fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel                continue;
1909f64a7a83bea5f1d2ab1e71231616c6cb0487d56eHal Finkel              if (PrunedDAGInstrs.count(*I))
191078fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel                continue;
191178fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel              NeedsExtraction = true;
191278fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel              break;
191378fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel            }
191478fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel
191578fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel            if (NeedsExtraction) {
191678fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel              int ESContrib;
1917245b657ab636a505066ea6a81591a9a8b93604d2Hal Finkel              if (Ty1->isVectorTy()) {
191878fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel                ESContrib = (int) getInstrCost(Instruction::ShuffleVector,
191978fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel                                               Ty1, VTy);
1920245b657ab636a505066ea6a81591a9a8b93604d2Hal Finkel                ESContrib = std::min(ESContrib, (int) TTI->getShuffleCost(
1921245b657ab636a505066ea6a81591a9a8b93604d2Hal Finkel                  TargetTransformInfo::SK_ExtractSubvector, VTy, 0, Ty1));
1922245b657ab636a505066ea6a81591a9a8b93604d2Hal Finkel              } else
1923abc227d9b39baed266c453f459ca14242f5c1eeaChandler Carruth                ESContrib = (int) TTI->getVectorInstrCost(
192478fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel                                    Instruction::ExtractElement, VTy, 0);
192578fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel
192678fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel              DEBUG(if (DebugPairSelection) dbgs() << "\tcost {" <<
192778fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel                *S->first << "} = " << ESContrib << "\n");
192878fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel              EffSize -= ESContrib;
192978fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel            }
193078fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel
193178fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel            NeedsExtraction = false;
193278fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel            for (Value::use_iterator I = S->second->use_begin(),
193378fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel                 IE = S->second->use_end(); I != IE; ++I) {
193486c88c938aec8006d2ce83325ec1f31e1154620bHal Finkel              if (ShuffleVectorInst *SI = dyn_cast<ShuffleVectorInst>(*I)) {
193586c88c938aec8006d2ce83325ec1f31e1154620bHal Finkel                // Shuffle can be folded if it has no other input
193686c88c938aec8006d2ce83325ec1f31e1154620bHal Finkel                if (isa<UndefValue>(SI->getOperand(1)))
193786c88c938aec8006d2ce83325ec1f31e1154620bHal Finkel                  continue;
193886c88c938aec8006d2ce83325ec1f31e1154620bHal Finkel              }
193986c88c938aec8006d2ce83325ec1f31e1154620bHal Finkel              if (isa<ExtractElementInst>(*I))
194078fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel                continue;
1941f64a7a83bea5f1d2ab1e71231616c6cb0487d56eHal Finkel              if (PrunedDAGInstrs.count(*I))
194278fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel                continue;
194378fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel              NeedsExtraction = true;
194478fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel              break;
194578fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel            }
194678fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel
194778fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel            if (NeedsExtraction) {
194878fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel              int ESContrib;
1949245b657ab636a505066ea6a81591a9a8b93604d2Hal Finkel              if (Ty2->isVectorTy()) {
195078fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel                ESContrib = (int) getInstrCost(Instruction::ShuffleVector,
195178fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel                                               Ty2, VTy);
1952245b657ab636a505066ea6a81591a9a8b93604d2Hal Finkel                ESContrib = std::min(ESContrib, (int) TTI->getShuffleCost(
1953245b657ab636a505066ea6a81591a9a8b93604d2Hal Finkel                  TargetTransformInfo::SK_ExtractSubvector, VTy,
1954245b657ab636a505066ea6a81591a9a8b93604d2Hal Finkel                  Ty1->isVectorTy() ? Ty1->getVectorNumElements() : 1, Ty2));
1955245b657ab636a505066ea6a81591a9a8b93604d2Hal Finkel              } else
1956abc227d9b39baed266c453f459ca14242f5c1eeaChandler Carruth                ESContrib = (int) TTI->getVectorInstrCost(
195778fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel                                    Instruction::ExtractElement, VTy, 1);
195878fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel              DEBUG(if (DebugPairSelection) dbgs() << "\tcost {" <<
195978fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel                *S->second << "} = " << ESContrib << "\n");
196078fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel              EffSize -= ESContrib;
196178fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel            }
196278fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel          }
196378fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel
196478fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel          // Compute the cost of incoming edges.
196578fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel          if (!isa<LoadInst>(S->first) && !isa<StoreInst>(S->first)) {
196678fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel            Instruction *S1 = cast<Instruction>(S->first),
196778fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel                        *S2 = cast<Instruction>(S->second);
196878fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel            for (unsigned o = 0; o < S1->getNumOperands(); ++o) {
196978fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel              Value *O1 = S1->getOperand(o), *O2 = S2->getOperand(o);
197078fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel
197178fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel              // Combining constants into vector constants (or small vector
197278fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel              // constants into larger ones are assumed free).
197378fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel              if (isa<Constant>(O1) && isa<Constant>(O2))
197478fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel                continue;
197578fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel
197678fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel              if (FlipOrder)
197778fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel                std::swap(O1, O2);
197878fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel
197978fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel              ValuePair VP  = ValuePair(O1, O2);
198078fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel              ValuePair VPR = ValuePair(O2, O1);
198178fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel
198278fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel              // Internal edges are not handled here.
1983f64a7a83bea5f1d2ab1e71231616c6cb0487d56eHal Finkel              if (PrunedDAG.count(VP) || PrunedDAG.count(VPR))
198478fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel                continue;
198578fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel
198678fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel              Type *Ty1 = O1->getType(),
198778fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel                   *Ty2 = O2->getType();
198878fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel              Type *VTy = getVecTypeForPair(Ty1, Ty2);
198978fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel
199078fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel              // Combining vector operations of the same type is also assumed
199178fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel              // folded with other operations.
199286c88c938aec8006d2ce83325ec1f31e1154620bHal Finkel              if (Ty1 == Ty2) {
199386c88c938aec8006d2ce83325ec1f31e1154620bHal Finkel                // If both are insert elements, then both can be widened.
1994b2b2469a9178f7e22cd7a69f3093e54d67d6b712Hal Finkel                InsertElementInst *IEO1 = dyn_cast<InsertElementInst>(O1),
1995b2b2469a9178f7e22cd7a69f3093e54d67d6b712Hal Finkel                                  *IEO2 = dyn_cast<InsertElementInst>(O2);
1996b2b2469a9178f7e22cd7a69f3093e54d67d6b712Hal Finkel                if (IEO1 && IEO2 && isPureIEChain(IEO1) && isPureIEChain(IEO2))
199786c88c938aec8006d2ce83325ec1f31e1154620bHal Finkel                  continue;
199886c88c938aec8006d2ce83325ec1f31e1154620bHal Finkel                // If both are extract elements, and both have the same input
199986c88c938aec8006d2ce83325ec1f31e1154620bHal Finkel                // type, then they can be replaced with a shuffle
200086c88c938aec8006d2ce83325ec1f31e1154620bHal Finkel                ExtractElementInst *EIO1 = dyn_cast<ExtractElementInst>(O1),
200186c88c938aec8006d2ce83325ec1f31e1154620bHal Finkel                                   *EIO2 = dyn_cast<ExtractElementInst>(O2);
200286c88c938aec8006d2ce83325ec1f31e1154620bHal Finkel                if (EIO1 && EIO2 &&
200386c88c938aec8006d2ce83325ec1f31e1154620bHal Finkel                    EIO1->getOperand(0)->getType() ==
200486c88c938aec8006d2ce83325ec1f31e1154620bHal Finkel                      EIO2->getOperand(0)->getType())
200586c88c938aec8006d2ce83325ec1f31e1154620bHal Finkel                  continue;
200686c88c938aec8006d2ce83325ec1f31e1154620bHal Finkel                // If both are a shuffle with equal operand types and only two
200786c88c938aec8006d2ce83325ec1f31e1154620bHal Finkel                // unqiue operands, then they can be replaced with a single
200886c88c938aec8006d2ce83325ec1f31e1154620bHal Finkel                // shuffle
200986c88c938aec8006d2ce83325ec1f31e1154620bHal Finkel                ShuffleVectorInst *SIO1 = dyn_cast<ShuffleVectorInst>(O1),
201086c88c938aec8006d2ce83325ec1f31e1154620bHal Finkel                                  *SIO2 = dyn_cast<ShuffleVectorInst>(O2);
201186c88c938aec8006d2ce83325ec1f31e1154620bHal Finkel                if (SIO1 && SIO2 &&
201286c88c938aec8006d2ce83325ec1f31e1154620bHal Finkel                    SIO1->getOperand(0)->getType() ==
201386c88c938aec8006d2ce83325ec1f31e1154620bHal Finkel                      SIO2->getOperand(0)->getType()) {
201486c88c938aec8006d2ce83325ec1f31e1154620bHal Finkel                  SmallSet<Value *, 4> SIOps;
201586c88c938aec8006d2ce83325ec1f31e1154620bHal Finkel                  SIOps.insert(SIO1->getOperand(0));
201686c88c938aec8006d2ce83325ec1f31e1154620bHal Finkel                  SIOps.insert(SIO1->getOperand(1));
201786c88c938aec8006d2ce83325ec1f31e1154620bHal Finkel                  SIOps.insert(SIO2->getOperand(0));
201886c88c938aec8006d2ce83325ec1f31e1154620bHal Finkel                  SIOps.insert(SIO2->getOperand(1));
201986c88c938aec8006d2ce83325ec1f31e1154620bHal Finkel                  if (SIOps.size() <= 2)
202086c88c938aec8006d2ce83325ec1f31e1154620bHal Finkel                    continue;
202186c88c938aec8006d2ce83325ec1f31e1154620bHal Finkel                }
202286c88c938aec8006d2ce83325ec1f31e1154620bHal Finkel              }
202378fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel
202478fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel              int ESContrib;
202578fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel              // This pair has already been formed.
202678fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel              if (IncomingPairs.count(VP)) {
202778fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel                continue;
202878fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel              } else if (IncomingPairs.count(VPR)) {
202978fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel                ESContrib = (int) getInstrCost(Instruction::ShuffleVector,
203078fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel                                               VTy, VTy);
2031245b657ab636a505066ea6a81591a9a8b93604d2Hal Finkel
2032245b657ab636a505066ea6a81591a9a8b93604d2Hal Finkel                if (VTy->getVectorNumElements() == 2)
2033245b657ab636a505066ea6a81591a9a8b93604d2Hal Finkel                  ESContrib = std::min(ESContrib, (int) TTI->getShuffleCost(
2034245b657ab636a505066ea6a81591a9a8b93604d2Hal Finkel                    TargetTransformInfo::SK_Reverse, VTy));
203578fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel              } else if (!Ty1->isVectorTy() && !Ty2->isVectorTy()) {
2036abc227d9b39baed266c453f459ca14242f5c1eeaChandler Carruth                ESContrib = (int) TTI->getVectorInstrCost(
203778fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel                                    Instruction::InsertElement, VTy, 0);
2038abc227d9b39baed266c453f459ca14242f5c1eeaChandler Carruth                ESContrib += (int) TTI->getVectorInstrCost(
203978fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel                                     Instruction::InsertElement, VTy, 1);
204078fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel              } else if (!Ty1->isVectorTy()) {
204178fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel                // O1 needs to be inserted into a vector of size O2, and then
204278fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel                // both need to be shuffled together.
2043abc227d9b39baed266c453f459ca14242f5c1eeaChandler Carruth                ESContrib = (int) TTI->getVectorInstrCost(
204478fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel                                    Instruction::InsertElement, Ty2, 0);
204578fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel                ESContrib += (int) getInstrCost(Instruction::ShuffleVector,
204678fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel                                                VTy, Ty2);
204778fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel              } else if (!Ty2->isVectorTy()) {
204878fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel                // O2 needs to be inserted into a vector of size O1, and then
204978fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel                // both need to be shuffled together.
2050abc227d9b39baed266c453f459ca14242f5c1eeaChandler Carruth                ESContrib = (int) TTI->getVectorInstrCost(
205178fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel                                    Instruction::InsertElement, Ty1, 0);
205278fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel                ESContrib += (int) getInstrCost(Instruction::ShuffleVector,
205378fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel                                                VTy, Ty1);
205478fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel              } else {
205578fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel                Type *TyBig = Ty1, *TySmall = Ty2;
205678fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel                if (Ty2->getVectorNumElements() > Ty1->getVectorNumElements())
205778fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel                  std::swap(TyBig, TySmall);
205878fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel
205978fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel                ESContrib = (int) getInstrCost(Instruction::ShuffleVector,
206078fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel                                               VTy, TyBig);
206178fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel                if (TyBig != TySmall)
206278fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel                  ESContrib += (int) getInstrCost(Instruction::ShuffleVector,
206378fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel                                                  TyBig, TySmall);
206478fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel              }
206578fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel
206678fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel              DEBUG(if (DebugPairSelection) dbgs() << "\tcost {"
206778fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel                     << *O1 << " <-> " << *O2 << "} = " <<
206878fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel                     ESContrib << "\n");
206978fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel              EffSize -= ESContrib;
207078fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel              IncomingPairs.insert(VP);
207186ccc55c82651f91fd6a312c5f6a4b511bcd1aecHal Finkel            }
207286ccc55c82651f91fd6a312c5f6a4b511bcd1aecHal Finkel          }
207365309660fa61a837cc05323f69c618a7d8134d56Hal Finkel        }
20744387b8c95971a512e07bfda30dea6459e8419e8fHal Finkel
20754387b8c95971a512e07bfda30dea6459e8419e8fHal Finkel        if (!HasNontrivialInsts) {
20764387b8c95971a512e07bfda30dea6459e8419e8fHal Finkel          DEBUG(if (DebugPairSelection) dbgs() <<
2077f64a7a83bea5f1d2ab1e71231616c6cb0487d56eHal Finkel                "\tNo non-trivial instructions in DAG;"
20784387b8c95971a512e07bfda30dea6459e8419e8fHal Finkel                " override to zero effective size\n");
20794387b8c95971a512e07bfda30dea6459e8419e8fHal Finkel          EffSize = 0;
20804387b8c95971a512e07bfda30dea6459e8419e8fHal Finkel        }
208165309660fa61a837cc05323f69c618a7d8134d56Hal Finkel      } else {
2082f64a7a83bea5f1d2ab1e71231616c6cb0487d56eHal Finkel        for (DenseSet<ValuePair>::iterator S = PrunedDAG.begin(),
2083f64a7a83bea5f1d2ab1e71231616c6cb0487d56eHal Finkel             E = PrunedDAG.end(); S != E; ++S)
208465309660fa61a837cc05323f69c618a7d8134d56Hal Finkel          EffSize += (int) getDepthFactor(S->first);
208565309660fa61a837cc05323f69c618a7d8134d56Hal Finkel      }
2086de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
2087de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      DEBUG(if (DebugPairSelection)
2088f64a7a83bea5f1d2ab1e71231616c6cb0487d56eHal Finkel             dbgs() << "BBV: found pruned DAG for pair {"
208976a05c93b13a8debec4497a6e4e753d7531709e5Hal Finkel             << *IJ.first << " <-> " << *IJ.second << "} of depth " <<
2090f64a7a83bea5f1d2ab1e71231616c6cb0487d56eHal Finkel             MaxDepth << " and size " << PrunedDAG.size() <<
2091de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel            " (effective size: " << EffSize << ")\n");
2092abc227d9b39baed266c453f459ca14242f5c1eeaChandler Carruth      if (((TTI && !UseChainDepthWithTI) ||
209378fd353d5e5daedc47ecc31b6193ca48793c249cHal Finkel            MaxDepth >= Config.ReqChainDepth) &&
209465309660fa61a837cc05323f69c618a7d8134d56Hal Finkel          EffSize > 0 && EffSize > BestEffSize) {
2095de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        BestMaxDepth = MaxDepth;
2096de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        BestEffSize = EffSize;
2097f64a7a83bea5f1d2ab1e71231616c6cb0487d56eHal Finkel        BestDAG = PrunedDAG;
2098de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      }
2099de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    }
2100de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  }
2101de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
2102de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  // Given the list of candidate pairs, this function selects those
2103de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  // that will be fused into vector instructions.
2104de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  void BBVectorize::choosePairs(
210597a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel                DenseMap<Value *, std::vector<Value *> > &CandidatePairs,
210697a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel                DenseSet<ValuePair> &CandidatePairsSet,
210797a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel                DenseMap<ValuePair, int> &CandidatePairCostSavings,
210897a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel                std::vector<Value *> &PairableInsts,
210997a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel                DenseSet<ValuePair> &FixedOrderPairs,
211097a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel                DenseMap<VPPair, unsigned> &PairConnectionTypes,
211197a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel                DenseMap<ValuePair, std::vector<ValuePair> > &ConnectedPairs,
211297a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel                DenseMap<ValuePair, std::vector<ValuePair> > &ConnectedPairDeps,
211397a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel                DenseSet<ValuePair> &PairableInstUsers,
211497a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel                DenseMap<Value *, Value *>& ChosenPairs) {
2115bef377b7d7ce31edb40c87f8786d1b7bb6cdd6b1Hongbin Zheng    bool UseCycleCheck =
21166ca6d3b1eac5b8611f3a9e2c270c2e794d37e1f5Hal Finkel     CandidatePairsSet.size() <= Config.MaxCandPairsForCycleCheck;
21176ca6d3b1eac5b8611f3a9e2c270c2e794d37e1f5Hal Finkel
21186ca6d3b1eac5b8611f3a9e2c270c2e794d37e1f5Hal Finkel    DenseMap<Value *, std::vector<Value *> > CandidatePairs2;
21196ca6d3b1eac5b8611f3a9e2c270c2e794d37e1f5Hal Finkel    for (DenseSet<ValuePair>::iterator I = CandidatePairsSet.begin(),
21206ca6d3b1eac5b8611f3a9e2c270c2e794d37e1f5Hal Finkel         E = CandidatePairsSet.end(); I != E; ++I) {
21216ca6d3b1eac5b8611f3a9e2c270c2e794d37e1f5Hal Finkel      std::vector<Value *> &JJ = CandidatePairs2[I->second];
21226ca6d3b1eac5b8611f3a9e2c270c2e794d37e1f5Hal Finkel      if (JJ.empty()) JJ.reserve(32);
21236ca6d3b1eac5b8611f3a9e2c270c2e794d37e1f5Hal Finkel      JJ.push_back(I->first);
21246ca6d3b1eac5b8611f3a9e2c270c2e794d37e1f5Hal Finkel    }
21256ca6d3b1eac5b8611f3a9e2c270c2e794d37e1f5Hal Finkel
212697a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel    DenseMap<ValuePair, std::vector<ValuePair> > PairableInstUserMap;
2127da20ea696d8b24d89ae157106ddad2337296ed50Hal Finkel    DenseSet<VPPair> PairableInstUserPairSet;
2128de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    for (std::vector<Value *>::iterator I = PairableInsts.begin(),
2129de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel         E = PairableInsts.end(); I != E; ++I) {
2130de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      // The number of possible pairings for this variable:
21316ca6d3b1eac5b8611f3a9e2c270c2e794d37e1f5Hal Finkel      size_t NumChoices = CandidatePairs.lookup(*I).size();
2132de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      if (!NumChoices) continue;
2133de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
21346ca6d3b1eac5b8611f3a9e2c270c2e794d37e1f5Hal Finkel      std::vector<Value *> &JJ = CandidatePairs[*I];
2135de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
2136f64a7a83bea5f1d2ab1e71231616c6cb0487d56eHal Finkel      // The best pair to choose and its dag:
213765309660fa61a837cc05323f69c618a7d8134d56Hal Finkel      size_t BestMaxDepth = 0;
213865309660fa61a837cc05323f69c618a7d8134d56Hal Finkel      int BestEffSize = 0;
2139f64a7a83bea5f1d2ab1e71231616c6cb0487d56eHal Finkel      DenseSet<ValuePair> BestDAG;
2140f64a7a83bea5f1d2ab1e71231616c6cb0487d56eHal Finkel      findBestDAGFor(CandidatePairs, CandidatePairsSet,
2141b1a82589339fed148c12b052d30861a539552f1aHal Finkel                      CandidatePairCostSavings,
214286ccc55c82651f91fd6a312c5f6a4b511bcd1aecHal Finkel                      PairableInsts, FixedOrderPairs, PairConnectionTypes,
214386ccc55c82651f91fd6a312c5f6a4b511bcd1aecHal Finkel                      ConnectedPairs, ConnectedPairDeps,
2144da20ea696d8b24d89ae157106ddad2337296ed50Hal Finkel                      PairableInstUsers, PairableInstUserMap,
2145da20ea696d8b24d89ae157106ddad2337296ed50Hal Finkel                      PairableInstUserPairSet, ChosenPairs,
2146f64a7a83bea5f1d2ab1e71231616c6cb0487d56eHal Finkel                      BestDAG, BestMaxDepth, BestEffSize, *I, JJ,
2147de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel                      UseCycleCheck);
2148de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
2149f64a7a83bea5f1d2ab1e71231616c6cb0487d56eHal Finkel      if (BestDAG.empty())
21506ca6d3b1eac5b8611f3a9e2c270c2e794d37e1f5Hal Finkel        continue;
21516ca6d3b1eac5b8611f3a9e2c270c2e794d37e1f5Hal Finkel
2152f64a7a83bea5f1d2ab1e71231616c6cb0487d56eHal Finkel      // A dag has been chosen (or not) at this point. If no dag was
2153de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      // chosen, then this instruction, I, cannot be paired (and is no longer
2154de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      // considered).
2155de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
2156f64a7a83bea5f1d2ab1e71231616c6cb0487d56eHal Finkel      DEBUG(dbgs() << "BBV: selected pairs in the best DAG for: "
21576ca6d3b1eac5b8611f3a9e2c270c2e794d37e1f5Hal Finkel                   << *cast<Instruction>(*I) << "\n");
2158de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
2159f64a7a83bea5f1d2ab1e71231616c6cb0487d56eHal Finkel      for (DenseSet<ValuePair>::iterator S = BestDAG.begin(),
2160f64a7a83bea5f1d2ab1e71231616c6cb0487d56eHal Finkel           SE2 = BestDAG.end(); S != SE2; ++S) {
2161f64a7a83bea5f1d2ab1e71231616c6cb0487d56eHal Finkel        // Insert the members of this dag into the list of chosen pairs.
2162de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        ChosenPairs.insert(ValuePair(S->first, S->second));
2163de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        DEBUG(dbgs() << "BBV: selected pair: " << *S->first << " <-> " <<
2164de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel               *S->second << "\n");
2165de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
2166f64a7a83bea5f1d2ab1e71231616c6cb0487d56eHal Finkel        // Remove all candidate pairs that have values in the chosen dag.
2167f79f136cc64b0625b77c7b9008ed8c5b848b6b17Hal Finkel        std::vector<Value *> &KK = CandidatePairs[S->first];
21686ca6d3b1eac5b8611f3a9e2c270c2e794d37e1f5Hal Finkel        for (std::vector<Value *>::iterator K = KK.begin(), KE = KK.end();
21696ca6d3b1eac5b8611f3a9e2c270c2e794d37e1f5Hal Finkel             K != KE; ++K) {
21706ca6d3b1eac5b8611f3a9e2c270c2e794d37e1f5Hal Finkel          if (*K == S->second)
21716ca6d3b1eac5b8611f3a9e2c270c2e794d37e1f5Hal Finkel            continue;
21726ca6d3b1eac5b8611f3a9e2c270c2e794d37e1f5Hal Finkel
21736ca6d3b1eac5b8611f3a9e2c270c2e794d37e1f5Hal Finkel          CandidatePairsSet.erase(ValuePair(S->first, *K));
21746ca6d3b1eac5b8611f3a9e2c270c2e794d37e1f5Hal Finkel        }
2175f79f136cc64b0625b77c7b9008ed8c5b848b6b17Hal Finkel
2176f79f136cc64b0625b77c7b9008ed8c5b848b6b17Hal Finkel        std::vector<Value *> &LL = CandidatePairs2[S->second];
21776ca6d3b1eac5b8611f3a9e2c270c2e794d37e1f5Hal Finkel        for (std::vector<Value *>::iterator L = LL.begin(), LE = LL.end();
21786ca6d3b1eac5b8611f3a9e2c270c2e794d37e1f5Hal Finkel             L != LE; ++L) {
21796ca6d3b1eac5b8611f3a9e2c270c2e794d37e1f5Hal Finkel          if (*L == S->first)
21806ca6d3b1eac5b8611f3a9e2c270c2e794d37e1f5Hal Finkel            continue;
21816ca6d3b1eac5b8611f3a9e2c270c2e794d37e1f5Hal Finkel
21826ca6d3b1eac5b8611f3a9e2c270c2e794d37e1f5Hal Finkel          CandidatePairsSet.erase(ValuePair(*L, S->second));
21836ca6d3b1eac5b8611f3a9e2c270c2e794d37e1f5Hal Finkel        }
2184f79f136cc64b0625b77c7b9008ed8c5b848b6b17Hal Finkel
2185f79f136cc64b0625b77c7b9008ed8c5b848b6b17Hal Finkel        std::vector<Value *> &MM = CandidatePairs[S->second];
21866ca6d3b1eac5b8611f3a9e2c270c2e794d37e1f5Hal Finkel        for (std::vector<Value *>::iterator M = MM.begin(), ME = MM.end();
21876ca6d3b1eac5b8611f3a9e2c270c2e794d37e1f5Hal Finkel             M != ME; ++M) {
21886ca6d3b1eac5b8611f3a9e2c270c2e794d37e1f5Hal Finkel          assert(*M != S->first && "Flipped pair in candidate list?");
21896ca6d3b1eac5b8611f3a9e2c270c2e794d37e1f5Hal Finkel          CandidatePairsSet.erase(ValuePair(S->second, *M));
21906ca6d3b1eac5b8611f3a9e2c270c2e794d37e1f5Hal Finkel        }
2191f79f136cc64b0625b77c7b9008ed8c5b848b6b17Hal Finkel
2192f79f136cc64b0625b77c7b9008ed8c5b848b6b17Hal Finkel        std::vector<Value *> &NN = CandidatePairs2[S->first];
21936ca6d3b1eac5b8611f3a9e2c270c2e794d37e1f5Hal Finkel        for (std::vector<Value *>::iterator N = NN.begin(), NE = NN.end();
21946ca6d3b1eac5b8611f3a9e2c270c2e794d37e1f5Hal Finkel             N != NE; ++N) {
21956ca6d3b1eac5b8611f3a9e2c270c2e794d37e1f5Hal Finkel          assert(*N != S->second && "Flipped pair in candidate list?");
21966ca6d3b1eac5b8611f3a9e2c270c2e794d37e1f5Hal Finkel          CandidatePairsSet.erase(ValuePair(*N, S->first));
2197de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        }
2198de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      }
2199de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    }
2200de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
2201de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    DEBUG(dbgs() << "BBV: selected " << ChosenPairs.size() << " pairs.\n");
2202de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  }
2203de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
2204de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  std::string getReplacementName(Instruction *I, bool IsInput, unsigned o,
2205de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel                     unsigned n = 0) {
2206de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    if (!I->hasName())
2207de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      return "";
2208de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
2209de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    return (I->getName() + (IsInput ? ".v.i" : ".v.r") + utostr(o) +
2210de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel             (n > 0 ? "." + utostr(n) : "")).str();
2211de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  }
2212de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
2213de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  // Returns the value that is to be used as the pointer input to the vector
2214de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  // instruction that fuses I with J.
2215de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  Value *BBVectorize::getReplacementPointerInput(LLVMContext& Context,
2216202d1cb8a587a9513d8bb65bf4a3d88a55132860Hal Finkel                     Instruction *I, Instruction *J, unsigned o) {
2217de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    Value *IPtr, *JPtr;
221865309660fa61a837cc05323f69c618a7d8134d56Hal Finkel    unsigned IAlignment, JAlignment, IAddressSpace, JAddressSpace;
2219de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    int64_t OffsetInElmts;
2220282969ed3641ffa426e0440d3824dd219152b2d8Hal Finkel
2221202d1cb8a587a9513d8bb65bf4a3d88a55132860Hal Finkel    // Note: the analysis might fail here, that is why the pair order has
2222282969ed3641ffa426e0440d3824dd219152b2d8Hal Finkel    // been precomputed (OffsetInElmts must be unused here).
2223de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    (void) getPairPtrInfo(I, J, IPtr, JPtr, IAlignment, JAlignment,
222465309660fa61a837cc05323f69c618a7d8134d56Hal Finkel                          IAddressSpace, JAddressSpace,
222593f6f457614299eee3d22f376ab8f42a130f1912Hal Finkel                          OffsetInElmts, false);
2226de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
2227de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    // The pointer value is taken to be the one with the lowest offset.
2228202d1cb8a587a9513d8bb65bf4a3d88a55132860Hal Finkel    Value *VPtr = IPtr;
2229de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
223064e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel    Type *ArgTypeI = cast<PointerType>(IPtr->getType())->getElementType();
223164e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel    Type *ArgTypeJ = cast<PointerType>(JPtr->getType())->getElementType();
223264e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel    Type *VArgType = getVecTypeForPair(ArgTypeI, ArgTypeJ);
2233de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    Type *VArgPtrType = PointerType::get(VArgType,
2234de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      cast<PointerType>(IPtr->getType())->getAddressSpace());
2235de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    return new BitCastInst(VPtr, VArgPtrType, getReplacementName(I, true, o),
2236202d1cb8a587a9513d8bb65bf4a3d88a55132860Hal Finkel                        /* insert before */ I);
2237de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  }
2238de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
2239de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  void BBVectorize::fillNewShuffleMask(LLVMContext& Context, Instruction *J,
224064e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel                     unsigned MaskOffset, unsigned NumInElem,
224164e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel                     unsigned NumInElem1, unsigned IdxOffset,
224264e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel                     std::vector<Constant*> &Mask) {
224364e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel    unsigned NumElem1 = cast<VectorType>(J->getType())->getNumElements();
224464e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel    for (unsigned v = 0; v < NumElem1; ++v) {
2245de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      int m = cast<ShuffleVectorInst>(J)->getMaskValue(v);
2246de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      if (m < 0) {
2247de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        Mask[v+MaskOffset] = UndefValue::get(Type::getInt32Ty(Context));
2248de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      } else {
2249de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        unsigned mm = m + (int) IdxOffset;
225064e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        if (m >= (int) NumInElem1)
2251de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel          mm += (int) NumInElem;
2252de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
2253de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        Mask[v+MaskOffset] =
2254de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel          ConstantInt::get(Type::getInt32Ty(Context), mm);
2255de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      }
2256de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    }
2257de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  }
2258de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
2259de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  // Returns the value that is to be used as the vector-shuffle mask to the
2260de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  // vector instruction that fuses I with J.
2261de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  Value *BBVectorize::getReplacementShuffleMask(LLVMContext& Context,
2262de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel                     Instruction *I, Instruction *J) {
2263de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    // This is the shuffle mask. We need to append the second
2264de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    // mask to the first, and the numbers need to be adjusted.
2265de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
226664e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel    Type *ArgTypeI = I->getType();
226764e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel    Type *ArgTypeJ = J->getType();
226864e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel    Type *VArgType = getVecTypeForPair(ArgTypeI, ArgTypeJ);
226964e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel
227064e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel    unsigned NumElemI = cast<VectorType>(ArgTypeI)->getNumElements();
2271de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
2272de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    // Get the total number of elements in the fused vector type.
2273de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    // By definition, this must equal the number of elements in
2274de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    // the final mask.
2275de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    unsigned NumElem = cast<VectorType>(VArgType)->getNumElements();
2276de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    std::vector<Constant*> Mask(NumElem);
2277de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
227864e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel    Type *OpTypeI = I->getOperand(0)->getType();
227964e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel    unsigned NumInElemI = cast<VectorType>(OpTypeI)->getNumElements();
228064e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel    Type *OpTypeJ = J->getOperand(0)->getType();
228164e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel    unsigned NumInElemJ = cast<VectorType>(OpTypeJ)->getNumElements();
228264e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel
228364e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel    // The fused vector will be:
228464e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel    // -----------------------------------------------------
228564e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel    // | NumInElemI | NumInElemJ | NumInElemI | NumInElemJ |
228664e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel    // -----------------------------------------------------
228764e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel    // from which we'll extract NumElem total elements (where the first NumElemI
228864e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel    // of them come from the mask in I and the remainder come from the mask
228964e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel    // in J.
2290de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
2291de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    // For the mask from the first pair...
229264e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel    fillNewShuffleMask(Context, I, 0,        NumInElemJ, NumInElemI,
229364e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel                       0,          Mask);
2294de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
2295de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    // For the mask from the second pair...
229664e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel    fillNewShuffleMask(Context, J, NumElemI, NumInElemI, NumInElemJ,
229764e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel                       NumInElemI, Mask);
2298de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
2299de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    return ConstantVector::get(Mask);
2300de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  }
2301de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
230264e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel  bool BBVectorize::expandIEChain(LLVMContext& Context, Instruction *I,
230364e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel                                  Instruction *J, unsigned o, Value *&LOp,
230464e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel                                  unsigned numElemL,
230564e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel                                  Type *ArgTypeL, Type *ArgTypeH,
230672465ea23d010507d3746adc126d719005981e05Hal Finkel                                  bool IBeforeJ, unsigned IdxOff) {
230764e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel    bool ExpandedIEChain = false;
230864e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel    if (InsertElementInst *LIE = dyn_cast<InsertElementInst>(LOp)) {
230964e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel      // If we have a pure insertelement chain, then this can be rewritten
231064e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel      // into a chain that directly builds the larger type.
2311b2b2469a9178f7e22cd7a69f3093e54d67d6b712Hal Finkel      if (isPureIEChain(LIE)) {
231264e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        SmallVector<Value *, 8> VectElemts(numElemL,
231364e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel          UndefValue::get(ArgTypeL->getScalarType()));
231464e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        InsertElementInst *LIENext = LIE;
231564e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        do {
231664e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel          unsigned Idx =
231764e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel            cast<ConstantInt>(LIENext->getOperand(2))->getSExtValue();
231864e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel          VectElemts[Idx] = LIENext->getOperand(1);
231964e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        } while ((LIENext =
232064e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel                   dyn_cast<InsertElementInst>(LIENext->getOperand(0))));
232164e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel
232264e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        LIENext = 0;
232364e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        Value *LIEPrev = UndefValue::get(ArgTypeH);
232464e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        for (unsigned i = 0; i < numElemL; ++i) {
232564e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel          if (isa<UndefValue>(VectElemts[i])) continue;
232664e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel          LIENext = InsertElementInst::Create(LIEPrev, VectElemts[i],
232764e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel                             ConstantInt::get(Type::getInt32Ty(Context),
232864e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel                                              i + IdxOff),
232972465ea23d010507d3746adc126d719005981e05Hal Finkel                             getReplacementName(IBeforeJ ? I : J,
233072465ea23d010507d3746adc126d719005981e05Hal Finkel                                                true, o, i+1));
233172465ea23d010507d3746adc126d719005981e05Hal Finkel          LIENext->insertBefore(IBeforeJ ? J : I);
233264e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel          LIEPrev = LIENext;
233364e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        }
233464e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel
233564e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        LOp = LIENext ? (Value*) LIENext : UndefValue::get(ArgTypeH);
233664e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        ExpandedIEChain = true;
233764e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel      }
233864e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel    }
233964e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel
234064e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel    return ExpandedIEChain;
234164e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel  }
234264e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel
2343de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  // Returns the value to be used as the specified operand of the vector
2344de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  // instruction that fuses I with J.
2345de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  Value *BBVectorize::getReplacementInput(LLVMContext& Context, Instruction *I,
234672465ea23d010507d3746adc126d719005981e05Hal Finkel                     Instruction *J, unsigned o, bool IBeforeJ) {
2347de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    Value *CV0 = ConstantInt::get(Type::getInt32Ty(Context), 0);
2348de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    Value *CV1 = ConstantInt::get(Type::getInt32Ty(Context), 1);
2349de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
235064e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel    // Compute the fused vector type for this operand
235164e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel    Type *ArgTypeI = I->getOperand(o)->getType();
235264e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel    Type *ArgTypeJ = J->getOperand(o)->getType();
235364e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel    VectorType *VArgType = getVecTypeForPair(ArgTypeI, ArgTypeJ);
2354de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
2355de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    Instruction *L = I, *H = J;
235664e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel    Type *ArgTypeL = ArgTypeI, *ArgTypeH = ArgTypeJ;
2357de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
235864e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel    unsigned numElemL;
235964e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel    if (ArgTypeL->isVectorTy())
236064e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel      numElemL = cast<VectorType>(ArgTypeL)->getNumElements();
236164e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel    else
236264e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel      numElemL = 1;
2363de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
236464e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel    unsigned numElemH;
236564e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel    if (ArgTypeH->isVectorTy())
236664e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel      numElemH = cast<VectorType>(ArgTypeH)->getNumElements();
236764e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel    else
236864e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel      numElemH = 1;
236964e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel
237064e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel    Value *LOp = L->getOperand(o);
237164e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel    Value *HOp = H->getOperand(o);
237264e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel    unsigned numElem = VArgType->getNumElements();
237364e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel
237464e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel    // First, we check if we can reuse the "original" vector outputs (if these
237564e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel    // exist). We might need a shuffle.
237664e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel    ExtractElementInst *LEE = dyn_cast<ExtractElementInst>(LOp);
237764e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel    ExtractElementInst *HEE = dyn_cast<ExtractElementInst>(HOp);
237864e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel    ShuffleVectorInst *LSV = dyn_cast<ShuffleVectorInst>(LOp);
237964e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel    ShuffleVectorInst *HSV = dyn_cast<ShuffleVectorInst>(HOp);
238064e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel
238164e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel    // FIXME: If we're fusing shuffle instructions, then we can't apply this
238264e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel    // optimization. The input vectors to the shuffle might be a different
238364e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel    // length from the shuffle outputs. Unfortunately, the replacement
238464e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel    // shuffle mask has already been formed, and the mask entries are sensitive
238564e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel    // to the sizes of the inputs.
238664e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel    bool IsSizeChangeShuffle =
238764e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel      isa<ShuffleVectorInst>(L) &&
238864e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        (LOp->getType() != L->getType() || HOp->getType() != H->getType());
238964e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel
239064e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel    if ((LEE || LSV) && (HEE || HSV) && !IsSizeChangeShuffle) {
239164e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel      // We can have at most two unique vector inputs.
239264e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel      bool CanUseInputs = true;
239364e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel      Value *I1, *I2 = 0;
239464e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel      if (LEE) {
239564e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        I1 = LEE->getOperand(0);
239664e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel      } else {
239764e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        I1 = LSV->getOperand(0);
239864e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        I2 = LSV->getOperand(1);
239964e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        if (I2 == I1 || isa<UndefValue>(I2))
240064e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel          I2 = 0;
240164e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel      }
240264e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel
240364e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel      if (HEE) {
240464e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        Value *I3 = HEE->getOperand(0);
240564e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        if (!I2 && I3 != I1)
240664e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel          I2 = I3;
240764e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        else if (I3 != I1 && I3 != I2)
240864e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel          CanUseInputs = false;
240964e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel      } else {
241064e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        Value *I3 = HSV->getOperand(0);
241164e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        if (!I2 && I3 != I1)
241264e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel          I2 = I3;
241364e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        else if (I3 != I1 && I3 != I2)
241464e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel          CanUseInputs = false;
241564e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel
241664e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        if (CanUseInputs) {
241764e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel          Value *I4 = HSV->getOperand(1);
241864e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel          if (!isa<UndefValue>(I4)) {
241964e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel            if (!I2 && I4 != I1)
242064e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel              I2 = I4;
242164e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel            else if (I4 != I1 && I4 != I2)
242264e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel              CanUseInputs = false;
242364e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel          }
242464e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        }
242564e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel      }
242664e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel
242764e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel      if (CanUseInputs) {
242864e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        unsigned LOpElem =
242964e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel          cast<VectorType>(cast<Instruction>(LOp)->getOperand(0)->getType())
243064e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel            ->getNumElements();
243164e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        unsigned HOpElem =
243264e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel          cast<VectorType>(cast<Instruction>(HOp)->getOperand(0)->getType())
243364e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel            ->getNumElements();
243464e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel
243564e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        // We have one or two input vectors. We need to map each index of the
243664e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        // operands to the index of the original vector.
243764e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        SmallVector<std::pair<int, int>, 8>  II(numElem);
243864e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        for (unsigned i = 0; i < numElemL; ++i) {
243964e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel          int Idx, INum;
244064e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel          if (LEE) {
244164e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel            Idx =
244264e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel              cast<ConstantInt>(LEE->getOperand(1))->getSExtValue();
244364e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel            INum = LEE->getOperand(0) == I1 ? 0 : 1;
244464e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel          } else {
244564e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel            Idx = LSV->getMaskValue(i);
244664e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel            if (Idx < (int) LOpElem) {
244764e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel              INum = LSV->getOperand(0) == I1 ? 0 : 1;
244864e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel            } else {
244964e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel              Idx -= LOpElem;
245064e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel              INum = LSV->getOperand(1) == I1 ? 0 : 1;
245164e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel            }
245264e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel          }
245364e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel
245464e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel          II[i] = std::pair<int, int>(Idx, INum);
245564e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        }
245664e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        for (unsigned i = 0; i < numElemH; ++i) {
245764e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel          int Idx, INum;
245864e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel          if (HEE) {
245964e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel            Idx =
246064e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel              cast<ConstantInt>(HEE->getOperand(1))->getSExtValue();
246164e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel            INum = HEE->getOperand(0) == I1 ? 0 : 1;
246264e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel          } else {
246364e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel            Idx = HSV->getMaskValue(i);
246464e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel            if (Idx < (int) HOpElem) {
246564e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel              INum = HSV->getOperand(0) == I1 ? 0 : 1;
246664e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel            } else {
246764e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel              Idx -= HOpElem;
246864e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel              INum = HSV->getOperand(1) == I1 ? 0 : 1;
246964e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel            }
247064e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel          }
247164e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel
247264e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel          II[i + numElemL] = std::pair<int, int>(Idx, INum);
247364e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        }
247464e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel
247564e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        // We now have an array which tells us from which index of which
247664e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        // input vector each element of the operand comes.
247764e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        VectorType *I1T = cast<VectorType>(I1->getType());
247864e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        unsigned I1Elem = I1T->getNumElements();
247964e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel
248064e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        if (!I2) {
248164e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel          // In this case there is only one underlying vector input. Check for
248264e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel          // the trivial case where we can use the input directly.
248364e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel          if (I1Elem == numElem) {
248464e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel            bool ElemInOrder = true;
248564e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel            for (unsigned i = 0; i < numElem; ++i) {
248664e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel              if (II[i].first != (int) i && II[i].first != -1) {
248764e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel                ElemInOrder = false;
248864e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel                break;
248964e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel              }
249064e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel            }
249164e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel
249264e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel            if (ElemInOrder)
249364e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel              return I1;
249464e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel          }
249564e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel
249664e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel          // A shuffle is needed.
249764e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel          std::vector<Constant *> Mask(numElem);
249864e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel          for (unsigned i = 0; i < numElem; ++i) {
249964e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel            int Idx = II[i].first;
250064e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel            if (Idx == -1)
250164e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel              Mask[i] = UndefValue::get(Type::getInt32Ty(Context));
250264e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel            else
250364e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel              Mask[i] = ConstantInt::get(Type::getInt32Ty(Context), Idx);
250464e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel          }
250564e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel
250664e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel          Instruction *S =
250764e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel            new ShuffleVectorInst(I1, UndefValue::get(I1T),
250864e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel                                  ConstantVector::get(Mask),
250972465ea23d010507d3746adc126d719005981e05Hal Finkel                                  getReplacementName(IBeforeJ ? I : J,
251072465ea23d010507d3746adc126d719005981e05Hal Finkel                                                     true, o));
251172465ea23d010507d3746adc126d719005981e05Hal Finkel          S->insertBefore(IBeforeJ ? J : I);
251264e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel          return S;
251364e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        }
251464e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel
251564e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        VectorType *I2T = cast<VectorType>(I2->getType());
251664e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        unsigned I2Elem = I2T->getNumElements();
251764e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel
251864e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        // This input comes from two distinct vectors. The first step is to
251964e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        // make sure that both vectors are the same length. If not, the
252064e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        // smaller one will need to grow before they can be shuffled together.
252164e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        if (I1Elem < I2Elem) {
252264e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel          std::vector<Constant *> Mask(I2Elem);
252364e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel          unsigned v = 0;
252464e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel          for (; v < I1Elem; ++v)
252564e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel            Mask[v] = ConstantInt::get(Type::getInt32Ty(Context), v);
252664e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel          for (; v < I2Elem; ++v)
252764e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel            Mask[v] = UndefValue::get(Type::getInt32Ty(Context));
252864e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel
252964e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel          Instruction *NewI1 =
253064e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel            new ShuffleVectorInst(I1, UndefValue::get(I1T),
253164e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel                                  ConstantVector::get(Mask),
253272465ea23d010507d3746adc126d719005981e05Hal Finkel                                  getReplacementName(IBeforeJ ? I : J,
253372465ea23d010507d3746adc126d719005981e05Hal Finkel                                                     true, o, 1));
253472465ea23d010507d3746adc126d719005981e05Hal Finkel          NewI1->insertBefore(IBeforeJ ? J : I);
253564e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel          I1 = NewI1;
253664e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel          I1T = I2T;
253764e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel          I1Elem = I2Elem;
253864e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        } else if (I1Elem > I2Elem) {
253964e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel          std::vector<Constant *> Mask(I1Elem);
254064e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel          unsigned v = 0;
254164e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel          for (; v < I2Elem; ++v)
254264e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel            Mask[v] = ConstantInt::get(Type::getInt32Ty(Context), v);
254364e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel          for (; v < I1Elem; ++v)
254464e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel            Mask[v] = UndefValue::get(Type::getInt32Ty(Context));
254564e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel
254664e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel          Instruction *NewI2 =
254764e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel            new ShuffleVectorInst(I2, UndefValue::get(I2T),
254864e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel                                  ConstantVector::get(Mask),
254972465ea23d010507d3746adc126d719005981e05Hal Finkel                                  getReplacementName(IBeforeJ ? I : J,
255072465ea23d010507d3746adc126d719005981e05Hal Finkel                                                     true, o, 1));
255172465ea23d010507d3746adc126d719005981e05Hal Finkel          NewI2->insertBefore(IBeforeJ ? J : I);
255264e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel          I2 = NewI2;
255364e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel          I2T = I1T;
255464e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel          I2Elem = I1Elem;
255564e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        }
255664e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel
255764e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        // Now that both I1 and I2 are the same length we can shuffle them
255864e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        // together (and use the result).
255964e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        std::vector<Constant *> Mask(numElem);
256064e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        for (unsigned v = 0; v < numElem; ++v) {
256164e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel          if (II[v].first == -1) {
256264e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel            Mask[v] = UndefValue::get(Type::getInt32Ty(Context));
256364e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel          } else {
256464e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel            int Idx = II[v].first + II[v].second * I1Elem;
256564e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel            Mask[v] = ConstantInt::get(Type::getInt32Ty(Context), Idx);
256664e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel          }
256764e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        }
256864e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel
256964e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        Instruction *NewOp =
257064e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel          new ShuffleVectorInst(I1, I2, ConstantVector::get(Mask),
257172465ea23d010507d3746adc126d719005981e05Hal Finkel                                getReplacementName(IBeforeJ ? I : J, true, o));
257272465ea23d010507d3746adc126d719005981e05Hal Finkel        NewOp->insertBefore(IBeforeJ ? J : I);
257364e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        return NewOp;
257464e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel      }
2575de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    }
2576de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
257764e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel    Type *ArgType = ArgTypeL;
257864e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel    if (numElemL < numElemH) {
257964e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel      if (numElemL == 1 && expandIEChain(Context, I, J, o, HOp, numElemH,
258072465ea23d010507d3746adc126d719005981e05Hal Finkel                                         ArgTypeL, VArgType, IBeforeJ, 1)) {
258164e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        // This is another short-circuit case: we're combining a scalar into
258264e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        // a vector that is formed by an IE chain. We've just expanded the IE
258364e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        // chain, now insert the scalar and we're done.
258464e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel
258564e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        Instruction *S = InsertElementInst::Create(HOp, LOp, CV0,
258672465ea23d010507d3746adc126d719005981e05Hal Finkel                           getReplacementName(IBeforeJ ? I : J, true, o));
258772465ea23d010507d3746adc126d719005981e05Hal Finkel        S->insertBefore(IBeforeJ ? J : I);
258864e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        return S;
258964e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel      } else if (!expandIEChain(Context, I, J, o, LOp, numElemL, ArgTypeL,
259072465ea23d010507d3746adc126d719005981e05Hal Finkel                                ArgTypeH, IBeforeJ)) {
259164e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        // The two vector inputs to the shuffle must be the same length,
259264e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        // so extend the smaller vector to be the same length as the larger one.
259364e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        Instruction *NLOp;
259464e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        if (numElemL > 1) {
259564e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel
259664e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel          std::vector<Constant *> Mask(numElemH);
259764e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel          unsigned v = 0;
259864e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel          for (; v < numElemL; ++v)
259964e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel            Mask[v] = ConstantInt::get(Type::getInt32Ty(Context), v);
260064e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel          for (; v < numElemH; ++v)
260164e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel            Mask[v] = UndefValue::get(Type::getInt32Ty(Context));
260264e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel
260364e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel          NLOp = new ShuffleVectorInst(LOp, UndefValue::get(ArgTypeL),
260464e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel                                       ConstantVector::get(Mask),
260572465ea23d010507d3746adc126d719005981e05Hal Finkel                                       getReplacementName(IBeforeJ ? I : J,
260672465ea23d010507d3746adc126d719005981e05Hal Finkel                                                          true, o, 1));
260764e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        } else {
260864e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel          NLOp = InsertElementInst::Create(UndefValue::get(ArgTypeH), LOp, CV0,
260972465ea23d010507d3746adc126d719005981e05Hal Finkel                                           getReplacementName(IBeforeJ ? I : J,
261072465ea23d010507d3746adc126d719005981e05Hal Finkel                                                              true, o, 1));
261164e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        }
261264e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel
261372465ea23d010507d3746adc126d719005981e05Hal Finkel        NLOp->insertBefore(IBeforeJ ? J : I);
261464e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        LOp = NLOp;
261564e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel      }
261664e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel
261764e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel      ArgType = ArgTypeH;
261864e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel    } else if (numElemL > numElemH) {
261964e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel      if (numElemH == 1 && expandIEChain(Context, I, J, o, LOp, numElemL,
262072465ea23d010507d3746adc126d719005981e05Hal Finkel                                         ArgTypeH, VArgType, IBeforeJ)) {
262164e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        Instruction *S =
262264e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel          InsertElementInst::Create(LOp, HOp,
262364e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel                                    ConstantInt::get(Type::getInt32Ty(Context),
262464e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel                                                     numElemL),
262572465ea23d010507d3746adc126d719005981e05Hal Finkel                                    getReplacementName(IBeforeJ ? I : J,
262672465ea23d010507d3746adc126d719005981e05Hal Finkel                                                       true, o));
262772465ea23d010507d3746adc126d719005981e05Hal Finkel        S->insertBefore(IBeforeJ ? J : I);
262864e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        return S;
262964e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel      } else if (!expandIEChain(Context, I, J, o, HOp, numElemH, ArgTypeH,
263072465ea23d010507d3746adc126d719005981e05Hal Finkel                                ArgTypeL, IBeforeJ)) {
263164e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        Instruction *NHOp;
263264e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        if (numElemH > 1) {
263364e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel          std::vector<Constant *> Mask(numElemL);
263464e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel          unsigned v = 0;
263564e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel          for (; v < numElemH; ++v)
263664e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel            Mask[v] = ConstantInt::get(Type::getInt32Ty(Context), v);
263764e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel          for (; v < numElemL; ++v)
263864e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel            Mask[v] = UndefValue::get(Type::getInt32Ty(Context));
263964e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel
264064e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel          NHOp = new ShuffleVectorInst(HOp, UndefValue::get(ArgTypeH),
264164e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel                                       ConstantVector::get(Mask),
264272465ea23d010507d3746adc126d719005981e05Hal Finkel                                       getReplacementName(IBeforeJ ? I : J,
264372465ea23d010507d3746adc126d719005981e05Hal Finkel                                                          true, o, 1));
264464e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        } else {
264564e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel          NHOp = InsertElementInst::Create(UndefValue::get(ArgTypeL), HOp, CV0,
264672465ea23d010507d3746adc126d719005981e05Hal Finkel                                           getReplacementName(IBeforeJ ? I : J,
264772465ea23d010507d3746adc126d719005981e05Hal Finkel                                                              true, o, 1));
264864e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        }
264964e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel
265072465ea23d010507d3746adc126d719005981e05Hal Finkel        NHOp->insertBefore(IBeforeJ ? J : I);
265164e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        HOp = NHOp;
2652de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      }
265364e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel    }
2654de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
265564e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel    if (ArgType->isVectorTy()) {
265664e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel      unsigned numElem = cast<VectorType>(VArgType)->getNumElements();
265764e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel      std::vector<Constant*> Mask(numElem);
265864e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel      for (unsigned v = 0; v < numElem; ++v) {
265964e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        unsigned Idx = v;
266064e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        // If the low vector was expanded, we need to skip the extra
266164e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        // undefined entries.
266264e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        if (v >= numElemL && numElemH > numElemL)
266364e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel          Idx += (numElemH - numElemL);
266464e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        Mask[v] = ConstantInt::get(Type::getInt32Ty(Context), Idx);
266564e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel      }
2666de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
266764e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel      Instruction *BV = new ShuffleVectorInst(LOp, HOp,
266872465ea23d010507d3746adc126d719005981e05Hal Finkel                          ConstantVector::get(Mask),
266972465ea23d010507d3746adc126d719005981e05Hal Finkel                          getReplacementName(IBeforeJ ? I : J, true, o));
267072465ea23d010507d3746adc126d719005981e05Hal Finkel      BV->insertBefore(IBeforeJ ? J : I);
2671de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      return BV;
2672de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    }
2673de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
2674de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    Instruction *BV1 = InsertElementInst::Create(
267564e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel                                          UndefValue::get(VArgType), LOp, CV0,
267672465ea23d010507d3746adc126d719005981e05Hal Finkel                                          getReplacementName(IBeforeJ ? I : J,
267772465ea23d010507d3746adc126d719005981e05Hal Finkel                                                             true, o, 1));
267872465ea23d010507d3746adc126d719005981e05Hal Finkel    BV1->insertBefore(IBeforeJ ? J : I);
267964e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel    Instruction *BV2 = InsertElementInst::Create(BV1, HOp, CV1,
268072465ea23d010507d3746adc126d719005981e05Hal Finkel                                          getReplacementName(IBeforeJ ? I : J,
268172465ea23d010507d3746adc126d719005981e05Hal Finkel                                                             true, o, 2));
268272465ea23d010507d3746adc126d719005981e05Hal Finkel    BV2->insertBefore(IBeforeJ ? J : I);
2683de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    return BV2;
2684de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  }
2685de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
2686de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  // This function creates an array of values that will be used as the inputs
2687de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  // to the vector instruction that fuses I with J.
2688de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  void BBVectorize::getReplacementInputsForPair(LLVMContext& Context,
2689de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel                     Instruction *I, Instruction *J,
269072465ea23d010507d3746adc126d719005981e05Hal Finkel                     SmallVector<Value *, 3> &ReplacedOperands,
269172465ea23d010507d3746adc126d719005981e05Hal Finkel                     bool IBeforeJ) {
2692de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    unsigned NumOperands = I->getNumOperands();
2693de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
2694de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    for (unsigned p = 0, o = NumOperands-1; p < NumOperands; ++p, --o) {
2695de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      // Iterate backward so that we look at the store pointer
2696de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      // first and know whether or not we need to flip the inputs.
2697de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
2698de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      if (isa<LoadInst>(I) || (o == 1 && isa<StoreInst>(I))) {
2699de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        // This is the pointer for a load/store instruction.
2700202d1cb8a587a9513d8bb65bf4a3d88a55132860Hal Finkel        ReplacedOperands[o] = getReplacementPointerInput(Context, I, J, o);
2701de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        continue;
27026173ed95daf2f209fe3883faee45967e4800ae75Hal Finkel      } else if (isa<CallInst>(I)) {
2703de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        Function *F = cast<CallInst>(I)->getCalledFunction();
2704a77728415857196035c0090f7b2749d7971811a2Hal Finkel        Intrinsic::ID IID = (Intrinsic::ID) F->getIntrinsicID();
27056173ed95daf2f209fe3883faee45967e4800ae75Hal Finkel        if (o == NumOperands-1) {
27066173ed95daf2f209fe3883faee45967e4800ae75Hal Finkel          BasicBlock &BB = *I->getParent();
2707bef377b7d7ce31edb40c87f8786d1b7bb6cdd6b1Hongbin Zheng
27086173ed95daf2f209fe3883faee45967e4800ae75Hal Finkel          Module *M = BB.getParent()->getParent();
270964e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel          Type *ArgTypeI = I->getType();
271064e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel          Type *ArgTypeJ = J->getType();
271164e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel          Type *VArgType = getVecTypeForPair(ArgTypeI, ArgTypeJ);
2712bef377b7d7ce31edb40c87f8786d1b7bb6cdd6b1Hongbin Zheng
2713a77728415857196035c0090f7b2749d7971811a2Hal Finkel          ReplacedOperands[o] = Intrinsic::getDeclaration(M, IID, VArgType);
27146173ed95daf2f209fe3883faee45967e4800ae75Hal Finkel          continue;
27156173ed95daf2f209fe3883faee45967e4800ae75Hal Finkel        } else if (IID == Intrinsic::powi && o == 1) {
27166173ed95daf2f209fe3883faee45967e4800ae75Hal Finkel          // The second argument of powi is a single integer and we've already
27176173ed95daf2f209fe3883faee45967e4800ae75Hal Finkel          // checked that both arguments are equal. As a result, we just keep
27186173ed95daf2f209fe3883faee45967e4800ae75Hal Finkel          // I's second argument.
27196173ed95daf2f209fe3883faee45967e4800ae75Hal Finkel          ReplacedOperands[o] = I->getOperand(o);
27206173ed95daf2f209fe3883faee45967e4800ae75Hal Finkel          continue;
27216173ed95daf2f209fe3883faee45967e4800ae75Hal Finkel        }
2722de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      } else if (isa<ShuffleVectorInst>(I) && o == NumOperands-1) {
2723de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        ReplacedOperands[o] = getReplacementShuffleMask(Context, I, J);
2724de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        continue;
2725de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      }
2726de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
272772465ea23d010507d3746adc126d719005981e05Hal Finkel      ReplacedOperands[o] = getReplacementInput(Context, I, J, o, IBeforeJ);
2728de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    }
2729de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  }
2730de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
2731de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  // This function creates two values that represent the outputs of the
2732de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  // original I and J instructions. These are generally vector shuffles
2733de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  // or extracts. In many cases, these will end up being unused and, thus,
2734de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  // eliminated by later passes.
2735de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  void BBVectorize::replaceOutputsOfPair(LLVMContext& Context, Instruction *I,
2736de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel                     Instruction *J, Instruction *K,
2737de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel                     Instruction *&InsertionPt,
2738202d1cb8a587a9513d8bb65bf4a3d88a55132860Hal Finkel                     Instruction *&K1, Instruction *&K2) {
2739de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    if (isa<StoreInst>(I)) {
2740de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      AA->replaceWithNewValue(I, K);
2741de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      AA->replaceWithNewValue(J, K);
2742de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    } else {
2743de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      Type *IType = I->getType();
274464e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel      Type *JType = J->getType();
274564e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel
274664e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel      VectorType *VType = getVecTypeForPair(IType, JType);
274764e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel      unsigned numElem = VType->getNumElements();
274864e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel
274964e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel      unsigned numElemI, numElemJ;
275064e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel      if (IType->isVectorTy())
275164e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        numElemI = cast<VectorType>(IType)->getNumElements();
275264e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel      else
275364e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        numElemI = 1;
275464e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel
275564e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel      if (JType->isVectorTy())
275664e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        numElemJ = cast<VectorType>(JType)->getNumElements();
275764e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel      else
275864e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        numElemJ = 1;
2759de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
2760de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      if (IType->isVectorTy()) {
276164e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        std::vector<Constant*> Mask1(numElemI), Mask2(numElemI);
276264e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        for (unsigned v = 0; v < numElemI; ++v) {
276364e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel          Mask1[v] = ConstantInt::get(Type::getInt32Ty(Context), v);
276464e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel          Mask2[v] = ConstantInt::get(Type::getInt32Ty(Context), numElemJ+v);
276564e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        }
2766de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
276764e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        K1 = new ShuffleVectorInst(K, UndefValue::get(VType),
2768202d1cb8a587a9513d8bb65bf4a3d88a55132860Hal Finkel                                   ConstantVector::get( Mask1),
276964e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel                                   getReplacementName(K, false, 1));
2770de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      } else {
277164e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        Value *CV0 = ConstantInt::get(Type::getInt32Ty(Context), 0);
2772202d1cb8a587a9513d8bb65bf4a3d88a55132860Hal Finkel        K1 = ExtractElementInst::Create(K, CV0,
2773de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel                                          getReplacementName(K, false, 1));
277464e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel      }
277564e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel
277664e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel      if (JType->isVectorTy()) {
277764e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        std::vector<Constant*> Mask1(numElemJ), Mask2(numElemJ);
277864e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        for (unsigned v = 0; v < numElemJ; ++v) {
277964e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel          Mask1[v] = ConstantInt::get(Type::getInt32Ty(Context), v);
278064e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel          Mask2[v] = ConstantInt::get(Type::getInt32Ty(Context), numElemI+v);
278164e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        }
278264e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel
278364e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        K2 = new ShuffleVectorInst(K, UndefValue::get(VType),
2784202d1cb8a587a9513d8bb65bf4a3d88a55132860Hal Finkel                                   ConstantVector::get( Mask2),
278564e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel                                   getReplacementName(K, false, 2));
278664e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel      } else {
278764e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel        Value *CV1 = ConstantInt::get(Type::getInt32Ty(Context), numElem-1);
2788202d1cb8a587a9513d8bb65bf4a3d88a55132860Hal Finkel        K2 = ExtractElementInst::Create(K, CV1,
2789de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel                                          getReplacementName(K, false, 2));
2790de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      }
2791de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
2792de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      K1->insertAfter(K);
2793de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      K2->insertAfter(K1);
2794de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      InsertionPt = K2;
2795de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    }
2796de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  }
2797de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
2798de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  // Move all uses of the function I (including pairing-induced uses) after J.
2799de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  bool BBVectorize::canMoveUsesOfIAfterJ(BasicBlock &BB,
28002f0e63cc16feb39480805bd00f53bbe5e3031d29Hal Finkel                     DenseSet<ValuePair> &LoadMoveSetPairs,
2801de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel                     Instruction *I, Instruction *J) {
2802de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    // Skip to the first instruction past I.
2803ded681d2725907c7de9db53d59cee0c51fad6fcbBenjamin Kramer    BasicBlock::iterator L = llvm::next(BasicBlock::iterator(I));
2804de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
2805de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    DenseSet<Value *> Users;
2806de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    AliasSetTracker WriteSet(*AA);
2807de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    for (; cast<Instruction>(L) != J; ++L)
28082f0e63cc16feb39480805bd00f53bbe5e3031d29Hal Finkel      (void) trackUsesOfI(Users, WriteSet, I, L, true, &LoadMoveSetPairs);
2809de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
2810de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    assert(cast<Instruction>(L) == J &&
2811de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      "Tracking has not proceeded far enough to check for dependencies");
2812de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    // If J is now in the use set of I, then trackUsesOfI will return true
2813de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    // and we have a dependency cycle (and the fusing operation must abort).
28142f0e63cc16feb39480805bd00f53bbe5e3031d29Hal Finkel    return !trackUsesOfI(Users, WriteSet, I, J, true, &LoadMoveSetPairs);
2815de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  }
2816de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
2817de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  // Move all uses of the function I (including pairing-induced uses) after J.
2818de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  void BBVectorize::moveUsesOfIAfterJ(BasicBlock &BB,
28192f0e63cc16feb39480805bd00f53bbe5e3031d29Hal Finkel                     DenseSet<ValuePair> &LoadMoveSetPairs,
2820de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel                     Instruction *&InsertionPt,
2821de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel                     Instruction *I, Instruction *J) {
2822de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    // Skip to the first instruction past I.
2823ded681d2725907c7de9db53d59cee0c51fad6fcbBenjamin Kramer    BasicBlock::iterator L = llvm::next(BasicBlock::iterator(I));
2824de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
2825de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    DenseSet<Value *> Users;
2826de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    AliasSetTracker WriteSet(*AA);
2827de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    for (; cast<Instruction>(L) != J;) {
28282f0e63cc16feb39480805bd00f53bbe5e3031d29Hal Finkel      if (trackUsesOfI(Users, WriteSet, I, L, true, &LoadMoveSetPairs)) {
2829de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        // Move this instruction
2830de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        Instruction *InstToMove = L; ++L;
2831de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
2832de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        DEBUG(dbgs() << "BBV: moving: " << *InstToMove <<
2833de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel                        " to after " << *InsertionPt << "\n");
2834de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        InstToMove->removeFromParent();
2835de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        InstToMove->insertAfter(InsertionPt);
2836de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        InsertionPt = InstToMove;
2837de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      } else {
2838de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        ++L;
2839de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      }
2840de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    }
2841de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  }
2842de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
2843de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  // Collect all load instruction that are in the move set of a given first
2844de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  // pair member.  These loads depend on the first instruction, I, and so need
2845de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  // to be moved after J (the second instruction) when the pair is fused.
2846de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  void BBVectorize::collectPairLoadMoveSet(BasicBlock &BB,
2847de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel                     DenseMap<Value *, Value *> &ChosenPairs,
284897a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel                     DenseMap<Value *, std::vector<Value *> > &LoadMoveSet,
28492f0e63cc16feb39480805bd00f53bbe5e3031d29Hal Finkel                     DenseSet<ValuePair> &LoadMoveSetPairs,
2850de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel                     Instruction *I) {
2851de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    // Skip to the first instruction past I.
2852ded681d2725907c7de9db53d59cee0c51fad6fcbBenjamin Kramer    BasicBlock::iterator L = llvm::next(BasicBlock::iterator(I));
2853de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
2854de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    DenseSet<Value *> Users;
2855de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    AliasSetTracker WriteSet(*AA);
2856de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
2857de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    // Note: We cannot end the loop when we reach J because J could be moved
2858de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    // farther down the use chain by another instruction pairing. Also, J
2859de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    // could be before I if this is an inverted input.
2860de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    for (BasicBlock::iterator E = BB.end(); cast<Instruction>(L) != E; ++L) {
2861de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      if (trackUsesOfI(Users, WriteSet, I, L)) {
28622f0e63cc16feb39480805bd00f53bbe5e3031d29Hal Finkel        if (L->mayReadFromMemory()) {
286397a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel          LoadMoveSet[L].push_back(I);
28642f0e63cc16feb39480805bd00f53bbe5e3031d29Hal Finkel          LoadMoveSetPairs.insert(ValuePair(L, I));
28652f0e63cc16feb39480805bd00f53bbe5e3031d29Hal Finkel        }
2866de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      }
2867de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    }
2868de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  }
2869de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
2870de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  // In cases where both load/stores and the computation of their pointers
2871de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  // are chosen for vectorization, we can end up in a situation where the
2872de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  // aliasing analysis starts returning different query results as the
2873de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  // process of fusing instruction pairs continues. Because the algorithm
2874f64a7a83bea5f1d2ab1e71231616c6cb0487d56eHal Finkel  // relies on finding the same use dags here as were found earlier, we'll
2875de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  // need to precompute the necessary aliasing information here and then
2876de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  // manually update it during the fusion process.
2877de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  void BBVectorize::collectLoadMoveSet(BasicBlock &BB,
2878de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel                     std::vector<Value *> &PairableInsts,
2879de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel                     DenseMap<Value *, Value *> &ChosenPairs,
288097a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel                     DenseMap<Value *, std::vector<Value *> > &LoadMoveSet,
28812f0e63cc16feb39480805bd00f53bbe5e3031d29Hal Finkel                     DenseSet<ValuePair> &LoadMoveSetPairs) {
2882de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    for (std::vector<Value *>::iterator PI = PairableInsts.begin(),
2883de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel         PIE = PairableInsts.end(); PI != PIE; ++PI) {
2884de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      DenseMap<Value *, Value *>::iterator P = ChosenPairs.find(*PI);
2885de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      if (P == ChosenPairs.end()) continue;
2886de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
2887de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      Instruction *I = cast<Instruction>(P->first);
28882f0e63cc16feb39480805bd00f53bbe5e3031d29Hal Finkel      collectPairLoadMoveSet(BB, ChosenPairs, LoadMoveSet,
28892f0e63cc16feb39480805bd00f53bbe5e3031d29Hal Finkel                             LoadMoveSetPairs, I);
2890de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    }
2891de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  }
2892de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
2893ab4684e26fe21857d8c8bc6ba7a5234c35117c83Hal Finkel  // When the first instruction in each pair is cloned, it will inherit its
2894ab4684e26fe21857d8c8bc6ba7a5234c35117c83Hal Finkel  // parent's metadata. This metadata must be combined with that of the other
2895ab4684e26fe21857d8c8bc6ba7a5234c35117c83Hal Finkel  // instruction in a safe way.
2896ab4684e26fe21857d8c8bc6ba7a5234c35117c83Hal Finkel  void BBVectorize::combineMetadata(Instruction *K, const Instruction *J) {
2897ab4684e26fe21857d8c8bc6ba7a5234c35117c83Hal Finkel    SmallVector<std::pair<unsigned, MDNode*>, 4> Metadata;
2898ab4684e26fe21857d8c8bc6ba7a5234c35117c83Hal Finkel    K->getAllMetadataOtherThanDebugLoc(Metadata);
2899ab4684e26fe21857d8c8bc6ba7a5234c35117c83Hal Finkel    for (unsigned i = 0, n = Metadata.size(); i < n; ++i) {
2900ab4684e26fe21857d8c8bc6ba7a5234c35117c83Hal Finkel      unsigned Kind = Metadata[i].first;
2901ab4684e26fe21857d8c8bc6ba7a5234c35117c83Hal Finkel      MDNode *JMD = J->getMetadata(Kind);
2902ab4684e26fe21857d8c8bc6ba7a5234c35117c83Hal Finkel      MDNode *KMD = Metadata[i].second;
2903ab4684e26fe21857d8c8bc6ba7a5234c35117c83Hal Finkel
2904ab4684e26fe21857d8c8bc6ba7a5234c35117c83Hal Finkel      switch (Kind) {
2905ab4684e26fe21857d8c8bc6ba7a5234c35117c83Hal Finkel      default:
2906ab4684e26fe21857d8c8bc6ba7a5234c35117c83Hal Finkel        K->setMetadata(Kind, 0); // Remove unknown metadata
2907ab4684e26fe21857d8c8bc6ba7a5234c35117c83Hal Finkel        break;
2908ab4684e26fe21857d8c8bc6ba7a5234c35117c83Hal Finkel      case LLVMContext::MD_tbaa:
2909ab4684e26fe21857d8c8bc6ba7a5234c35117c83Hal Finkel        K->setMetadata(Kind, MDNode::getMostGenericTBAA(JMD, KMD));
2910ab4684e26fe21857d8c8bc6ba7a5234c35117c83Hal Finkel        break;
2911ab4684e26fe21857d8c8bc6ba7a5234c35117c83Hal Finkel      case LLVMContext::MD_fpmath:
2912ab4684e26fe21857d8c8bc6ba7a5234c35117c83Hal Finkel        K->setMetadata(Kind, MDNode::getMostGenericFPMath(JMD, KMD));
2913ab4684e26fe21857d8c8bc6ba7a5234c35117c83Hal Finkel        break;
2914ab4684e26fe21857d8c8bc6ba7a5234c35117c83Hal Finkel      }
2915ab4684e26fe21857d8c8bc6ba7a5234c35117c83Hal Finkel    }
2916ab4684e26fe21857d8c8bc6ba7a5234c35117c83Hal Finkel  }
2917ab4684e26fe21857d8c8bc6ba7a5234c35117c83Hal Finkel
2918de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  // This function fuses the chosen instruction pairs into vector instructions,
2919de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  // taking care preserve any needed scalar outputs and, then, it reorders the
2920de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  // remaining instructions as needed (users of the first member of the pair
2921de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  // need to be moved to after the location of the second member of the pair
2922de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  // because the vector instruction is inserted in the location of the pair's
2923de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  // second member).
2924de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  void BBVectorize::fuseChosenPairs(BasicBlock &BB,
292597a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel             std::vector<Value *> &PairableInsts,
292697a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel             DenseMap<Value *, Value *> &ChosenPairs,
292797a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel             DenseSet<ValuePair> &FixedOrderPairs,
292897a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel             DenseMap<VPPair, unsigned> &PairConnectionTypes,
292997a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel             DenseMap<ValuePair, std::vector<ValuePair> > &ConnectedPairs,
293097a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel             DenseMap<ValuePair, std::vector<ValuePair> > &ConnectedPairDeps) {
2931de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    LLVMContext& Context = BB.getContext();
2932de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
2933de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    // During the vectorization process, the order of the pairs to be fused
2934de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    // could be flipped. So we'll add each pair, flipped, into the ChosenPairs
2935de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    // list. After a pair is fused, the flipped pair is removed from the list.
293672465ea23d010507d3746adc126d719005981e05Hal Finkel    DenseSet<ValuePair> FlippedPairs;
2937de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    for (DenseMap<Value *, Value *>::iterator P = ChosenPairs.begin(),
2938de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel         E = ChosenPairs.end(); P != E; ++P)
293972465ea23d010507d3746adc126d719005981e05Hal Finkel      FlippedPairs.insert(ValuePair(P->second, P->first));
294072465ea23d010507d3746adc126d719005981e05Hal Finkel    for (DenseSet<ValuePair>::iterator P = FlippedPairs.begin(),
2941de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel         E = FlippedPairs.end(); P != E; ++P)
2942de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      ChosenPairs.insert(*P);
2943de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
294497a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel    DenseMap<Value *, std::vector<Value *> > LoadMoveSet;
29452f0e63cc16feb39480805bd00f53bbe5e3031d29Hal Finkel    DenseSet<ValuePair> LoadMoveSetPairs;
29462f0e63cc16feb39480805bd00f53bbe5e3031d29Hal Finkel    collectLoadMoveSet(BB, PairableInsts, ChosenPairs,
29472f0e63cc16feb39480805bd00f53bbe5e3031d29Hal Finkel                       LoadMoveSet, LoadMoveSetPairs);
2948de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
2949de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    DEBUG(dbgs() << "BBV: initial: \n" << BB << "\n");
2950de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
2951de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    for (BasicBlock::iterator PI = BB.getFirstInsertionPt(); PI != BB.end();) {
2952de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      DenseMap<Value *, Value *>::iterator P = ChosenPairs.find(PI);
2953de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      if (P == ChosenPairs.end()) {
2954de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        ++PI;
2955de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        continue;
2956de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      }
2957de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
2958de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      if (getDepthFactor(P->first) == 0) {
2959de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        // These instructions are not really fused, but are tracked as though
2960de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        // they are. Any case in which it would be interesting to fuse them
2961de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        // will be taken care of by InstCombine.
2962de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        --NumFusedOps;
2963de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        ++PI;
2964de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        continue;
2965de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      }
2966de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
2967de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      Instruction *I = cast<Instruction>(P->first),
2968de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        *J = cast<Instruction>(P->second);
2969de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
2970de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      DEBUG(dbgs() << "BBV: fusing: " << *I <<
2971de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel             " <-> " << *J << "\n");
2972de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
2973de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      // Remove the pair and flipped pair from the list.
2974de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      DenseMap<Value *, Value *>::iterator FP = ChosenPairs.find(P->second);
2975de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      assert(FP != ChosenPairs.end() && "Flipped pair not found in list");
2976de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      ChosenPairs.erase(FP);
2977de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      ChosenPairs.erase(P);
2978de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
29792f0e63cc16feb39480805bd00f53bbe5e3031d29Hal Finkel      if (!canMoveUsesOfIAfterJ(BB, LoadMoveSetPairs, I, J)) {
2980de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        DEBUG(dbgs() << "BBV: fusion of: " << *I <<
2981de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel               " <-> " << *J <<
2982de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel               " aborted because of non-trivial dependency cycle\n");
2983de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        --NumFusedOps;
2984de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        ++PI;
2985de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        continue;
2986de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      }
2987de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
2988a9779bfbc9ab0cf3f157453fd0afd110b04a9fdcHal Finkel      // If the pair must have the other order, then flip it.
2989a9779bfbc9ab0cf3f157453fd0afd110b04a9fdcHal Finkel      bool FlipPairOrder = FixedOrderPairs.count(ValuePair(J, I));
299072465ea23d010507d3746adc126d719005981e05Hal Finkel      if (!FlipPairOrder && !FixedOrderPairs.count(ValuePair(I, J))) {
299172465ea23d010507d3746adc126d719005981e05Hal Finkel        // This pair does not have a fixed order, and so we might want to
299272465ea23d010507d3746adc126d719005981e05Hal Finkel        // flip it if that will yield fewer shuffles. We count the number
299372465ea23d010507d3746adc126d719005981e05Hal Finkel        // of dependencies connected via swaps, and those directly connected,
299472465ea23d010507d3746adc126d719005981e05Hal Finkel        // and flip the order if the number of swaps is greater.
299572465ea23d010507d3746adc126d719005981e05Hal Finkel        bool OrigOrder = true;
299697a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel        DenseMap<ValuePair, std::vector<ValuePair> >::iterator IJ =
299797a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel          ConnectedPairDeps.find(ValuePair(I, J));
299897a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel        if (IJ == ConnectedPairDeps.end()) {
299997a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel          IJ = ConnectedPairDeps.find(ValuePair(J, I));
300072465ea23d010507d3746adc126d719005981e05Hal Finkel          OrigOrder = false;
300172465ea23d010507d3746adc126d719005981e05Hal Finkel        }
300272465ea23d010507d3746adc126d719005981e05Hal Finkel
300397a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel        if (IJ != ConnectedPairDeps.end()) {
300472465ea23d010507d3746adc126d719005981e05Hal Finkel          unsigned NumDepsDirect = 0, NumDepsSwap = 0;
300597a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel          for (std::vector<ValuePair>::iterator T = IJ->second.begin(),
300697a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel               TE = IJ->second.end(); T != TE; ++T) {
300797a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel            VPPair Q(IJ->first, *T);
300872465ea23d010507d3746adc126d719005981e05Hal Finkel            DenseMap<VPPair, unsigned>::iterator R =
300997a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel              PairConnectionTypes.find(VPPair(Q.second, Q.first));
301072465ea23d010507d3746adc126d719005981e05Hal Finkel            assert(R != PairConnectionTypes.end() &&
301172465ea23d010507d3746adc126d719005981e05Hal Finkel                   "Cannot find pair connection type");
301272465ea23d010507d3746adc126d719005981e05Hal Finkel            if (R->second == PairConnectionDirect)
301372465ea23d010507d3746adc126d719005981e05Hal Finkel              ++NumDepsDirect;
301472465ea23d010507d3746adc126d719005981e05Hal Finkel            else if (R->second == PairConnectionSwap)
301572465ea23d010507d3746adc126d719005981e05Hal Finkel              ++NumDepsSwap;
301672465ea23d010507d3746adc126d719005981e05Hal Finkel          }
301772465ea23d010507d3746adc126d719005981e05Hal Finkel
301872465ea23d010507d3746adc126d719005981e05Hal Finkel          if (!OrigOrder)
301972465ea23d010507d3746adc126d719005981e05Hal Finkel            std::swap(NumDepsDirect, NumDepsSwap);
302072465ea23d010507d3746adc126d719005981e05Hal Finkel
302172465ea23d010507d3746adc126d719005981e05Hal Finkel          if (NumDepsSwap > NumDepsDirect) {
302272465ea23d010507d3746adc126d719005981e05Hal Finkel            FlipPairOrder = true;
302372465ea23d010507d3746adc126d719005981e05Hal Finkel            DEBUG(dbgs() << "BBV: reordering pair: " << *I <<
302472465ea23d010507d3746adc126d719005981e05Hal Finkel                            " <-> " << *J << "\n");
302572465ea23d010507d3746adc126d719005981e05Hal Finkel          }
302672465ea23d010507d3746adc126d719005981e05Hal Finkel        }
302772465ea23d010507d3746adc126d719005981e05Hal Finkel      }
3028282969ed3641ffa426e0440d3824dd219152b2d8Hal Finkel
3029202d1cb8a587a9513d8bb65bf4a3d88a55132860Hal Finkel      Instruction *L = I, *H = J;
3030a9779bfbc9ab0cf3f157453fd0afd110b04a9fdcHal Finkel      if (FlipPairOrder)
3031202d1cb8a587a9513d8bb65bf4a3d88a55132860Hal Finkel        std::swap(H, L);
3032202d1cb8a587a9513d8bb65bf4a3d88a55132860Hal Finkel
303372465ea23d010507d3746adc126d719005981e05Hal Finkel      // If the pair being fused uses the opposite order from that in the pair
303472465ea23d010507d3746adc126d719005981e05Hal Finkel      // connection map, then we need to flip the types.
303597a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel      DenseMap<ValuePair, std::vector<ValuePair> >::iterator HL =
303697a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel        ConnectedPairs.find(ValuePair(H, L));
303797a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel      if (HL != ConnectedPairs.end())
303897a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel        for (std::vector<ValuePair>::iterator T = HL->second.begin(),
303997a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel             TE = HL->second.end(); T != TE; ++T) {
304097a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel          VPPair Q(HL->first, *T);
304197a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel          DenseMap<VPPair, unsigned>::iterator R = PairConnectionTypes.find(Q);
304297a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel          assert(R != PairConnectionTypes.end() &&
304397a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel                 "Cannot find pair connection type");
304497a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel          if (R->second == PairConnectionDirect)
304597a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel            R->second = PairConnectionSwap;
304697a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel          else if (R->second == PairConnectionSwap)
304797a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel            R->second = PairConnectionDirect;
304897a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel        }
304972465ea23d010507d3746adc126d719005981e05Hal Finkel
305072465ea23d010507d3746adc126d719005981e05Hal Finkel      bool LBeforeH = !FlipPairOrder;
3051de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      unsigned NumOperands = I->getNumOperands();
3052de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      SmallVector<Value *, 3> ReplacedOperands(NumOperands);
305372465ea23d010507d3746adc126d719005981e05Hal Finkel      getReplacementInputsForPair(Context, L, H, ReplacedOperands,
305472465ea23d010507d3746adc126d719005981e05Hal Finkel                                  LBeforeH);
3055de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
3056de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      // Make a copy of the original operation, change its type to the vector
3057de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      // type and replace its operands with the vector operands.
305872465ea23d010507d3746adc126d719005981e05Hal Finkel      Instruction *K = L->clone();
305972465ea23d010507d3746adc126d719005981e05Hal Finkel      if (L->hasName())
306072465ea23d010507d3746adc126d719005981e05Hal Finkel        K->takeName(L);
306172465ea23d010507d3746adc126d719005981e05Hal Finkel      else if (H->hasName())
306272465ea23d010507d3746adc126d719005981e05Hal Finkel        K->takeName(H);
3063de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
3064de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      if (!isa<StoreInst>(K))
3065202d1cb8a587a9513d8bb65bf4a3d88a55132860Hal Finkel        K->mutateType(getVecTypeForPair(L->getType(), H->getType()));
3066de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
306772465ea23d010507d3746adc126d719005981e05Hal Finkel      combineMetadata(K, H);
3068430b9079c614cd3f45015a6516590d33742cc802Hal Finkel      K->intersectOptionalDataWith(H);
3069ab4684e26fe21857d8c8bc6ba7a5234c35117c83Hal Finkel
3070de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      for (unsigned o = 0; o < NumOperands; ++o)
3071de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        K->setOperand(o, ReplacedOperands[o]);
3072de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
3073de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      K->insertAfter(J);
3074de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
3075de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      // Instruction insertion point:
3076de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      Instruction *InsertionPt = K;
3077de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      Instruction *K1 = 0, *K2 = 0;
3078202d1cb8a587a9513d8bb65bf4a3d88a55132860Hal Finkel      replaceOutputsOfPair(Context, L, H, K, InsertionPt, K1, K2);
3079de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
3080f64a7a83bea5f1d2ab1e71231616c6cb0487d56eHal Finkel      // The use dag of the first original instruction must be moved to after
3081f64a7a83bea5f1d2ab1e71231616c6cb0487d56eHal Finkel      // the location of the second instruction. The entire use dag of the
3082f64a7a83bea5f1d2ab1e71231616c6cb0487d56eHal Finkel      // first instruction is disjoint from the input dag of the second
3083de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      // (by definition), and so commutes with it.
3084de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
30852f0e63cc16feb39480805bd00f53bbe5e3031d29Hal Finkel      moveUsesOfIAfterJ(BB, LoadMoveSetPairs, InsertionPt, I, J);
3086de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
3087de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      if (!isa<StoreInst>(I)) {
308872465ea23d010507d3746adc126d719005981e05Hal Finkel        L->replaceAllUsesWith(K1);
308972465ea23d010507d3746adc126d719005981e05Hal Finkel        H->replaceAllUsesWith(K2);
309072465ea23d010507d3746adc126d719005981e05Hal Finkel        AA->replaceWithNewValue(L, K1);
309172465ea23d010507d3746adc126d719005981e05Hal Finkel        AA->replaceWithNewValue(H, K2);
3092de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      }
3093de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
3094de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      // Instructions that may read from memory may be in the load move set.
3095de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      // Once an instruction is fused, we no longer need its move set, and so
3096de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      // the values of the map never need to be updated. However, when a load
3097de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      // is fused, we need to merge the entries from both instructions in the
3098de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      // pair in case those instructions were in the move set of some other
3099de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      // yet-to-be-fused pair. The loads in question are the keys of the map.
3100de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      if (I->mayReadFromMemory()) {
3101de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        std::vector<ValuePair> NewSetMembers;
310297a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel        DenseMap<Value *, std::vector<Value *> >::iterator II =
310397a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel          LoadMoveSet.find(I);
310497a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel        if (II != LoadMoveSet.end())
310597a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel          for (std::vector<Value *>::iterator N = II->second.begin(),
310697a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel               NE = II->second.end(); N != NE; ++N)
310797a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel            NewSetMembers.push_back(ValuePair(K, *N));
310897a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel        DenseMap<Value *, std::vector<Value *> >::iterator JJ =
310997a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel          LoadMoveSet.find(J);
311097a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel        if (JJ != LoadMoveSet.end())
311197a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel          for (std::vector<Value *>::iterator N = JJ->second.begin(),
311297a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel               NE = JJ->second.end(); N != NE; ++N)
311397a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel            NewSetMembers.push_back(ValuePair(K, *N));
3114de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        for (std::vector<ValuePair>::iterator A = NewSetMembers.begin(),
31152f0e63cc16feb39480805bd00f53bbe5e3031d29Hal Finkel             AE = NewSetMembers.end(); A != AE; ++A) {
311697a241b173a1413df5a93fdd891ddfac36dabad9Hal Finkel          LoadMoveSet[A->first].push_back(A->second);
31172f0e63cc16feb39480805bd00f53bbe5e3031d29Hal Finkel          LoadMoveSetPairs.insert(*A);
31182f0e63cc16feb39480805bd00f53bbe5e3031d29Hal Finkel        }
3119de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      }
3120de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
3121de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      // Before removing I, set the iterator to the next instruction.
3122de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      PI = llvm::next(BasicBlock::iterator(I));
3123de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      if (cast<Instruction>(PI) == J)
3124de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel        ++PI;
3125de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
3126de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      SE->forgetValue(I);
3127de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      SE->forgetValue(J);
3128de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      I->eraseFromParent();
3129de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel      J->eraseFromParent();
313072465ea23d010507d3746adc126d719005981e05Hal Finkel
313172465ea23d010507d3746adc126d719005981e05Hal Finkel      DEBUG(if (PrintAfterEveryPair) dbgs() << "BBV: block is now: \n" <<
313272465ea23d010507d3746adc126d719005981e05Hal Finkel                                               BB << "\n");
3133de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    }
3134de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
3135de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel    DEBUG(dbgs() << "BBV: final: \n" << BB << "\n");
3136de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel  }
3137de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel}
3138de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
3139de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkelchar BBVectorize::ID = 0;
3140de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkelstatic const char bb_vectorize_name[] = "Basic-Block Vectorization";
3141de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal FinkelINITIALIZE_PASS_BEGIN(BBVectorize, BBV_NAME, bb_vectorize_name, false, false)
3142de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal FinkelINITIALIZE_AG_DEPENDENCY(AliasAnalysis)
31438bd6c52396ab6e7955fdcc1bce099b7cba29a308Chandler CarruthINITIALIZE_AG_DEPENDENCY(TargetTransformInfo)
3144e29c19091cca58db668407dfc5dd86c70e8b3d49Hal FinkelINITIALIZE_PASS_DEPENDENCY(DominatorTree)
3145de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal FinkelINITIALIZE_PASS_DEPENDENCY(ScalarEvolution)
3146de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal FinkelINITIALIZE_PASS_END(BBVectorize, BBV_NAME, bb_vectorize_name, false, false)
3147de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
3148bef377b7d7ce31edb40c87f8786d1b7bb6cdd6b1Hongbin ZhengBasicBlockPass *llvm::createBBVectorizePass(const VectorizeConfig &C) {
3149bef377b7d7ce31edb40c87f8786d1b7bb6cdd6b1Hongbin Zheng  return new BBVectorize(C);
3150de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel}
3151de5e5ec3045a73a06b1054417f9ac6c02929e9ceHal Finkel
3152bef377b7d7ce31edb40c87f8786d1b7bb6cdd6b1Hongbin Zhengbool
3153bef377b7d7ce31edb40c87f8786d1b7bb6cdd6b1Hongbin Zhengllvm::vectorizeBasicBlock(Pass *P, BasicBlock &BB, const VectorizeConfig &C) {
3154bef377b7d7ce31edb40c87f8786d1b7bb6cdd6b1Hongbin Zheng  BBVectorize BBVectorizer(P, C);
315587825e7970a361ce5a8bab19bc880ff7f6242ca2Hongbin Zheng  return BBVectorizer.vectorizeBB(BB);
315687825e7970a361ce5a8bab19bc880ff7f6242ca2Hongbin Zheng}
3157bef377b7d7ce31edb40c87f8786d1b7bb6cdd6b1Hongbin Zheng
3158bef377b7d7ce31edb40c87f8786d1b7bb6cdd6b1Hongbin Zheng//===----------------------------------------------------------------------===//
3159bef377b7d7ce31edb40c87f8786d1b7bb6cdd6b1Hongbin ZhengVectorizeConfig::VectorizeConfig() {
3160bef377b7d7ce31edb40c87f8786d1b7bb6cdd6b1Hongbin Zheng  VectorBits = ::VectorBits;
3161768edf3cd037aab10391abc279f71470df8e3156Hal Finkel  VectorizeBools = !::NoBools;
316286312cc15f29ce2bbd9647b94862e068045280c3Hongbin Zheng  VectorizeInts = !::NoInts;
316386312cc15f29ce2bbd9647b94862e068045280c3Hongbin Zheng  VectorizeFloats = !::NoFloats;
3164f3f5a1e6f77a842ccb24cc81766437da5197d712Hal Finkel  VectorizePointers = !::NoPointers;
316586312cc15f29ce2bbd9647b94862e068045280c3Hongbin Zheng  VectorizeCasts = !::NoCasts;
316686312cc15f29ce2bbd9647b94862e068045280c3Hongbin Zheng  VectorizeMath = !::NoMath;
316786312cc15f29ce2bbd9647b94862e068045280c3Hongbin Zheng  VectorizeFMA = !::NoFMA;
3168fc3665c87519850f629c9565535e3be447e10addHal Finkel  VectorizeSelect = !::NoSelect;
3169e415f96b6a43ac8861148a11a4258bc38c247e8fHal Finkel  VectorizeCmp = !::NoCmp;
3170f3f5a1e6f77a842ccb24cc81766437da5197d712Hal Finkel  VectorizeGEP = !::NoGEP;
317186312cc15f29ce2bbd9647b94862e068045280c3Hongbin Zheng  VectorizeMemOps = !::NoMemOps;
3172bef377b7d7ce31edb40c87f8786d1b7bb6cdd6b1Hongbin Zheng  AlignedOnly = ::AlignedOnly;
3173bef377b7d7ce31edb40c87f8786d1b7bb6cdd6b1Hongbin Zheng  ReqChainDepth= ::ReqChainDepth;
3174bef377b7d7ce31edb40c87f8786d1b7bb6cdd6b1Hongbin Zheng  SearchLimit = ::SearchLimit;
3175bef377b7d7ce31edb40c87f8786d1b7bb6cdd6b1Hongbin Zheng  MaxCandPairsForCycleCheck = ::MaxCandPairsForCycleCheck;
3176bef377b7d7ce31edb40c87f8786d1b7bb6cdd6b1Hongbin Zheng  SplatBreaksChain = ::SplatBreaksChain;
3177bef377b7d7ce31edb40c87f8786d1b7bb6cdd6b1Hongbin Zheng  MaxInsts = ::MaxInsts;
3178ab90084bca42b74a5b5edad9b416bd81e105dad0Hal Finkel  MaxPairs = ::MaxPairs;
3179bef377b7d7ce31edb40c87f8786d1b7bb6cdd6b1Hongbin Zheng  MaxIter = ::MaxIter;
318064e1b28643d87e70734deb5f3d2d298e859c2fd2Hal Finkel  Pow2LenOnly = ::Pow2LenOnly;
3181bef377b7d7ce31edb40c87f8786d1b7bb6cdd6b1Hongbin Zheng  NoMemOpBoost = ::NoMemOpBoost;
3182bef377b7d7ce31edb40c87f8786d1b7bb6cdd6b1Hongbin Zheng  FastDep = ::FastDep;
3183bef377b7d7ce31edb40c87f8786d1b7bb6cdd6b1Hongbin Zheng}
3184