Refactor lookup_or_add to contain _MUCH_ less duplicated code. Add support for

numbering first class aggregate instructions while we're at it.


git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@84547 91177308-0d34-0410-b5e6-96231b3b80d8
This commit is contained in:
Owen Anderson 2009-10-19 22:14:22 +00:00
parent e306d8d6cc
commit d41ed4e2c4

View File

@ -78,13 +78,10 @@ namespace {
SHUFFLE, SELECT, TRUNC, ZEXT, SEXT, FPTOUI, SHUFFLE, SELECT, TRUNC, ZEXT, SEXT, FPTOUI,
FPTOSI, UITOFP, SITOFP, FPTRUNC, FPEXT, FPTOSI, UITOFP, SITOFP, FPTRUNC, FPEXT,
PTRTOINT, INTTOPTR, BITCAST, GEP, CALL, CONSTANT, PTRTOINT, INTTOPTR, BITCAST, GEP, CALL, CONSTANT,
EMPTY, TOMBSTONE }; INSERTVALUE, EXTRACTVALUE, EMPTY, TOMBSTONE };
ExpressionOpcode opcode; ExpressionOpcode opcode;
const Type* type; const Type* type;
uint32_t firstVN;
uint32_t secondVN;
uint32_t thirdVN;
SmallVector<uint32_t, 4> varargs; SmallVector<uint32_t, 4> varargs;
Value *function; Value *function;
@ -100,12 +97,6 @@ namespace {
return false; return false;
else if (function != other.function) else if (function != other.function)
return false; return false;
else if (firstVN != other.firstVN)
return false;
else if (secondVN != other.secondVN)
return false;
else if (thirdVN != other.thirdVN)
return false;
else { else {
if (varargs.size() != other.varargs.size()) if (varargs.size() != other.varargs.size())
return false; return false;
@ -146,6 +137,10 @@ namespace {
Expression create_expression(GetElementPtrInst* G); Expression create_expression(GetElementPtrInst* G);
Expression create_expression(CallInst* C); Expression create_expression(CallInst* C);
Expression create_expression(Constant* C); Expression create_expression(Constant* C);
Expression create_expression(ExtractValueInst* C);
Expression create_expression(InsertValueInst* C);
uint32_t lookup_or_add_call(CallInst* C);
public: public:
ValueTable() : nextValueNumber(1) { } ValueTable() : nextValueNumber(1) { }
uint32_t lookup_or_add(Value *V); uint32_t lookup_or_add(Value *V);
@ -176,13 +171,8 @@ template <> struct DenseMapInfo<Expression> {
static unsigned getHashValue(const Expression e) { static unsigned getHashValue(const Expression e) {
unsigned hash = e.opcode; unsigned hash = e.opcode;
hash = e.firstVN + hash * 37;
hash = e.secondVN + hash * 37;
hash = e.thirdVN + hash * 37;
hash = ((unsigned)((uintptr_t)e.type >> 4) ^ hash = ((unsigned)((uintptr_t)e.type >> 4) ^
(unsigned)((uintptr_t)e.type >> 9)) + (unsigned)((uintptr_t)e.type >> 9));
hash * 37;
for (SmallVector<uint32_t, 4>::const_iterator I = e.varargs.begin(), for (SmallVector<uint32_t, 4>::const_iterator I = e.varargs.begin(),
E = e.varargs.end(); I != E; ++I) E = e.varargs.end(); I != E; ++I)
@ -290,9 +280,6 @@ Expression ValueTable::create_expression(CallInst* C) {
Expression e; Expression e;
e.type = C->getType(); e.type = C->getType();
e.firstVN = 0;
e.secondVN = 0;
e.thirdVN = 0;
e.function = C->getCalledFunction(); e.function = C->getCalledFunction();
e.opcode = Expression::CALL; e.opcode = Expression::CALL;
@ -305,10 +292,8 @@ Expression ValueTable::create_expression(CallInst* C) {
Expression ValueTable::create_expression(BinaryOperator* BO) { Expression ValueTable::create_expression(BinaryOperator* BO) {
Expression e; Expression e;
e.varargs.push_back(lookup_or_add(BO->getOperand(0)));
e.firstVN = lookup_or_add(BO->getOperand(0)); e.varargs.push_back(lookup_or_add(BO->getOperand(1)));
e.secondVN = lookup_or_add(BO->getOperand(1));
e.thirdVN = 0;
e.function = 0; e.function = 0;
e.type = BO->getType(); e.type = BO->getType();
e.opcode = getOpcode(BO); e.opcode = getOpcode(BO);
@ -319,9 +304,8 @@ Expression ValueTable::create_expression(BinaryOperator* BO) {
Expression ValueTable::create_expression(CmpInst* C) { Expression ValueTable::create_expression(CmpInst* C) {
Expression e; Expression e;
e.firstVN = lookup_or_add(C->getOperand(0)); e.varargs.push_back(lookup_or_add(C->getOperand(0)));
e.secondVN = lookup_or_add(C->getOperand(1)); e.varargs.push_back(lookup_or_add(C->getOperand(1)));
e.thirdVN = 0;
e.function = 0; e.function = 0;
e.type = C->getType(); e.type = C->getType();
e.opcode = getOpcode(C); e.opcode = getOpcode(C);
@ -332,9 +316,7 @@ Expression ValueTable::create_expression(CmpInst* C) {
Expression ValueTable::create_expression(CastInst* C) { Expression ValueTable::create_expression(CastInst* C) {
Expression e; Expression e;
e.firstVN = lookup_or_add(C->getOperand(0)); e.varargs.push_back(lookup_or_add(C->getOperand(0)));
e.secondVN = 0;
e.thirdVN = 0;
e.function = 0; e.function = 0;
e.type = C->getType(); e.type = C->getType();
e.opcode = getOpcode(C); e.opcode = getOpcode(C);
@ -345,9 +327,9 @@ Expression ValueTable::create_expression(CastInst* C) {
Expression ValueTable::create_expression(ShuffleVectorInst* S) { Expression ValueTable::create_expression(ShuffleVectorInst* S) {
Expression e; Expression e;
e.firstVN = lookup_or_add(S->getOperand(0)); e.varargs.push_back(lookup_or_add(S->getOperand(0)));
e.secondVN = lookup_or_add(S->getOperand(1)); e.varargs.push_back(lookup_or_add(S->getOperand(1)));
e.thirdVN = lookup_or_add(S->getOperand(2)); e.varargs.push_back(lookup_or_add(S->getOperand(2)));
e.function = 0; e.function = 0;
e.type = S->getType(); e.type = S->getType();
e.opcode = Expression::SHUFFLE; e.opcode = Expression::SHUFFLE;
@ -358,9 +340,8 @@ Expression ValueTable::create_expression(ShuffleVectorInst* S) {
Expression ValueTable::create_expression(ExtractElementInst* E) { Expression ValueTable::create_expression(ExtractElementInst* E) {
Expression e; Expression e;
e.firstVN = lookup_or_add(E->getOperand(0)); e.varargs.push_back(lookup_or_add(E->getOperand(0)));
e.secondVN = lookup_or_add(E->getOperand(1)); e.varargs.push_back(lookup_or_add(E->getOperand(1)));
e.thirdVN = 0;
e.function = 0; e.function = 0;
e.type = E->getType(); e.type = E->getType();
e.opcode = Expression::EXTRACT; e.opcode = Expression::EXTRACT;
@ -371,9 +352,9 @@ Expression ValueTable::create_expression(ExtractElementInst* E) {
Expression ValueTable::create_expression(InsertElementInst* I) { Expression ValueTable::create_expression(InsertElementInst* I) {
Expression e; Expression e;
e.firstVN = lookup_or_add(I->getOperand(0)); e.varargs.push_back(lookup_or_add(I->getOperand(0)));
e.secondVN = lookup_or_add(I->getOperand(1)); e.varargs.push_back(lookup_or_add(I->getOperand(1)));
e.thirdVN = lookup_or_add(I->getOperand(2)); e.varargs.push_back(lookup_or_add(I->getOperand(2)));
e.function = 0; e.function = 0;
e.type = I->getType(); e.type = I->getType();
e.opcode = Expression::INSERT; e.opcode = Expression::INSERT;
@ -384,9 +365,9 @@ Expression ValueTable::create_expression(InsertElementInst* I) {
Expression ValueTable::create_expression(SelectInst* I) { Expression ValueTable::create_expression(SelectInst* I) {
Expression e; Expression e;
e.firstVN = lookup_or_add(I->getCondition()); e.varargs.push_back(lookup_or_add(I->getCondition()));
e.secondVN = lookup_or_add(I->getTrueValue()); e.varargs.push_back(lookup_or_add(I->getTrueValue()));
e.thirdVN = lookup_or_add(I->getFalseValue()); e.varargs.push_back(lookup_or_add(I->getFalseValue()));
e.function = 0; e.function = 0;
e.type = I->getType(); e.type = I->getType();
e.opcode = Expression::SELECT; e.opcode = Expression::SELECT;
@ -397,9 +378,7 @@ Expression ValueTable::create_expression(SelectInst* I) {
Expression ValueTable::create_expression(GetElementPtrInst* G) { Expression ValueTable::create_expression(GetElementPtrInst* G) {
Expression e; Expression e;
e.firstVN = lookup_or_add(G->getPointerOperand()); e.varargs.push_back(lookup_or_add(G->getPointerOperand()));
e.secondVN = 0;
e.thirdVN = 0;
e.function = 0; e.function = 0;
e.type = G->getType(); e.type = G->getType();
e.opcode = Expression::GEP; e.opcode = Expression::GEP;
@ -411,6 +390,35 @@ Expression ValueTable::create_expression(GetElementPtrInst* G) {
return e; return e;
} }
Expression ValueTable::create_expression(ExtractValueInst* E) {
Expression e;
e.varargs.push_back(lookup_or_add(E->getAggregateOperand()));
for (ExtractValueInst::idx_iterator II = E->idx_begin(), IE = E->idx_end();
II != IE; ++II)
e.varargs.push_back(*II);
e.function = 0;
e.type = E->getType();
e.opcode = Expression::EXTRACTVALUE;
return e;
}
Expression ValueTable::create_expression(InsertValueInst* E) {
Expression e;
e.varargs.push_back(lookup_or_add(E->getAggregateOperand()));
e.varargs.push_back(lookup_or_add(E->getInsertedValueOperand()));
for (InsertValueInst::idx_iterator II = E->idx_begin(), IE = E->idx_end();
II != IE; ++II)
e.varargs.push_back(*II);
e.function = 0;
e.type = E->getType();
e.opcode = Expression::INSERTVALUE;
return e;
}
//===----------------------------------------------------------------------===// //===----------------------------------------------------------------------===//
// ValueTable External Functions // ValueTable External Functions
//===----------------------------------------------------------------------===// //===----------------------------------------------------------------------===//
@ -420,33 +428,26 @@ void ValueTable::add(Value *V, uint32_t num) {
valueNumbering.insert(std::make_pair(V, num)); valueNumbering.insert(std::make_pair(V, num));
} }
/// lookup_or_add - Returns the value number for the specified value, assigning uint32_t ValueTable::lookup_or_add_call(CallInst* C) {
/// it a new number if it did not have one before.
uint32_t ValueTable::lookup_or_add(Value *V) {
DenseMap<Value*, uint32_t>::iterator VI = valueNumbering.find(V);
if (VI != valueNumbering.end())
return VI->second;
if (CallInst* C = dyn_cast<CallInst>(V)) {
if (AA->doesNotAccessMemory(C)) { if (AA->doesNotAccessMemory(C)) {
Expression exp = create_expression(C); Expression exp = create_expression(C);
uint32_t& e = expressionNumbering[exp]; uint32_t& e = expressionNumbering[exp];
if (!e) e = nextValueNumber++; if (!e) e = nextValueNumber++;
valueNumbering[V] = e; valueNumbering[C] = e;
return e; return e;
} else if (AA->onlyReadsMemory(C)) { } else if (AA->onlyReadsMemory(C)) {
Expression exp = create_expression(C); Expression exp = create_expression(C);
uint32_t& e = expressionNumbering[exp]; uint32_t& e = expressionNumbering[exp];
if (!e) { if (!e) {
e = nextValueNumber++; e = nextValueNumber++;
valueNumbering[V] = e; valueNumbering[C] = e;
return e; return e;
} }
MemDepResult local_dep = MD->getDependency(C); MemDepResult local_dep = MD->getDependency(C);
if (!local_dep.isDef() && !local_dep.isNonLocal()) { if (!local_dep.isDef() && !local_dep.isNonLocal()) {
valueNumbering[V] = nextValueNumber; valueNumbering[C] = nextValueNumber;
return nextValueNumber++; return nextValueNumber++;
} }
@ -454,7 +455,7 @@ uint32_t ValueTable::lookup_or_add(Value *V) {
CallInst* local_cdep = cast<CallInst>(local_dep.getInst()); CallInst* local_cdep = cast<CallInst>(local_dep.getInst());
if (local_cdep->getNumOperands() != C->getNumOperands()) { if (local_cdep->getNumOperands() != C->getNumOperands()) {
valueNumbering[V] = nextValueNumber; valueNumbering[C] = nextValueNumber;
return nextValueNumber++; return nextValueNumber++;
} }
@ -462,13 +463,13 @@ uint32_t ValueTable::lookup_or_add(Value *V) {
uint32_t c_vn = lookup_or_add(C->getOperand(i)); uint32_t c_vn = lookup_or_add(C->getOperand(i));
uint32_t cd_vn = lookup_or_add(local_cdep->getOperand(i)); uint32_t cd_vn = lookup_or_add(local_cdep->getOperand(i));
if (c_vn != cd_vn) { if (c_vn != cd_vn) {
valueNumbering[V] = nextValueNumber; valueNumbering[C] = nextValueNumber;
return nextValueNumber++; return nextValueNumber++;
} }
} }
uint32_t v = lookup_or_add(local_cdep); uint32_t v = lookup_or_add(local_cdep);
valueNumbering[V] = v; valueNumbering[C] = v;
return v; return v;
} }
@ -506,83 +507,118 @@ uint32_t ValueTable::lookup_or_add(Value *V) {
} }
if (!cdep) { if (!cdep) {
valueNumbering[V] = nextValueNumber; valueNumbering[C] = nextValueNumber;
return nextValueNumber++; return nextValueNumber++;
} }
if (cdep->getNumOperands() != C->getNumOperands()) { if (cdep->getNumOperands() != C->getNumOperands()) {
valueNumbering[V] = nextValueNumber; valueNumbering[C] = nextValueNumber;
return nextValueNumber++; return nextValueNumber++;
} }
for (unsigned i = 1; i < C->getNumOperands(); ++i) { for (unsigned i = 1; i < C->getNumOperands(); ++i) {
uint32_t c_vn = lookup_or_add(C->getOperand(i)); uint32_t c_vn = lookup_or_add(C->getOperand(i));
uint32_t cd_vn = lookup_or_add(cdep->getOperand(i)); uint32_t cd_vn = lookup_or_add(cdep->getOperand(i));
if (c_vn != cd_vn) { if (c_vn != cd_vn) {
valueNumbering[V] = nextValueNumber; valueNumbering[C] = nextValueNumber;
return nextValueNumber++; return nextValueNumber++;
} }
} }
uint32_t v = lookup_or_add(cdep); uint32_t v = lookup_or_add(cdep);
valueNumbering[V] = v; valueNumbering[C] = v;
return v; return v;
} else { } else {
valueNumbering[C] = nextValueNumber;
return nextValueNumber++;
}
}
/// lookup_or_add - Returns the value number for the specified value, assigning
/// it a new number if it did not have one before.
uint32_t ValueTable::lookup_or_add(Value *V) {
DenseMap<Value*, uint32_t>::iterator VI = valueNumbering.find(V);
if (VI != valueNumbering.end())
return VI->second;
if (!isa<Instruction>(V)) {
valueNumbering[V] = nextValueNumber; valueNumbering[V] = nextValueNumber;
return nextValueNumber++; return nextValueNumber++;
} }
} else if (BinaryOperator* BO = dyn_cast<BinaryOperator>(V)) {
Expression exp = create_expression(BO); Instruction* I = cast<Instruction>(V);
uint32_t& e = expressionNumbering[exp]; Expression exp;
if (!e) e = nextValueNumber++; switch (I->getOpcode()) {
valueNumbering[V] = e; case Instruction::Call:
return e; return lookup_or_add_call(cast<CallInst>(I));
} else if (CmpInst* C = dyn_cast<CmpInst>(V)) { case Instruction::Add:
Expression exp = create_expression(C); case Instruction::FAdd:
uint32_t& e = expressionNumbering[exp]; case Instruction::Sub:
if (!e) e = nextValueNumber++; case Instruction::FSub:
valueNumbering[V] = e; case Instruction::Mul:
return e; case Instruction::FMul:
} else if (ShuffleVectorInst* U = dyn_cast<ShuffleVectorInst>(V)) { case Instruction::UDiv:
Expression exp = create_expression(U); case Instruction::SDiv:
uint32_t& e = expressionNumbering[exp]; case Instruction::FDiv:
if (!e) e = nextValueNumber++; case Instruction::URem:
valueNumbering[V] = e; case Instruction::SRem:
return e; case Instruction::FRem:
} else if (ExtractElementInst* U = dyn_cast<ExtractElementInst>(V)) { case Instruction::Shl:
Expression exp = create_expression(U); case Instruction::LShr:
uint32_t& e = expressionNumbering[exp]; case Instruction::AShr:
if (!e) e = nextValueNumber++; case Instruction::And:
valueNumbering[V] = e; case Instruction::Or :
return e; case Instruction::Xor:
} else if (InsertElementInst* U = dyn_cast<InsertElementInst>(V)) { exp = create_expression(cast<BinaryOperator>(I));
Expression exp = create_expression(U); break;
uint32_t& e = expressionNumbering[exp]; case Instruction::ICmp:
if (!e) e = nextValueNumber++; case Instruction::FCmp:
valueNumbering[V] = e; exp = create_expression(cast<CmpInst>(I));
return e; break;
} else if (SelectInst* U = dyn_cast<SelectInst>(V)) { case Instruction::Trunc:
Expression exp = create_expression(U); case Instruction::ZExt:
uint32_t& e = expressionNumbering[exp]; case Instruction::SExt:
if (!e) e = nextValueNumber++; case Instruction::FPToUI:
valueNumbering[V] = e; case Instruction::FPToSI:
return e; case Instruction::UIToFP:
} else if (CastInst* U = dyn_cast<CastInst>(V)) { case Instruction::SIToFP:
Expression exp = create_expression(U); case Instruction::FPTrunc:
uint32_t& e = expressionNumbering[exp]; case Instruction::FPExt:
if (!e) e = nextValueNumber++; case Instruction::PtrToInt:
valueNumbering[V] = e; case Instruction::IntToPtr:
return e; case Instruction::BitCast:
} else if (GetElementPtrInst* U = dyn_cast<GetElementPtrInst>(V)) { exp = create_expression(cast<CastInst>(I));
Expression exp = create_expression(U); break;
uint32_t& e = expressionNumbering[exp]; case Instruction::Select:
if (!e) e = nextValueNumber++; exp = create_expression(cast<SelectInst>(I));
valueNumbering[V] = e; break;
return e; case Instruction::ExtractElement:
} else { exp = create_expression(cast<ExtractElementInst>(I));
break;
case Instruction::InsertElement:
exp = create_expression(cast<InsertElementInst>(I));
break;
case Instruction::ShuffleVector:
exp = create_expression(cast<ShuffleVectorInst>(I));
break;
case Instruction::ExtractValue:
exp = create_expression(cast<ExtractValueInst>(I));
break;
case Instruction::InsertValue:
exp = create_expression(cast<InsertValueInst>(I));
break;
case Instruction::GetElementPtr:
exp = create_expression(cast<GetElementPtrInst>(I));
break;
default:
valueNumbering[V] = nextValueNumber; valueNumbering[V] = nextValueNumber;
return nextValueNumber++; return nextValueNumber++;
} }
uint32_t& e = expressionNumbering[exp];
if (!e) e = nextValueNumber++;
valueNumbering[V] = e;
return e;
} }
/// lookup - Returns the value number of the specified value. Fails if /// lookup - Returns the value number of the specified value. Fails if