X-Git-Url: http://plrg.eecs.uci.edu/git/?a=blobdiff_plain;f=lib%2FTransforms%2FUtils%2FValueMapper.cpp;h=5d5ca1309eec1b4f320d95c23865122a9f95e6bd;hb=4ee451de366474b9c228b4e5fa573795a715216d;hp=f1724db1d4773f36ac034f62333ab370cd55a9ea;hpb=3805dea3fe742002f32a945abb19b74e2bb576d9;p=oota-llvm.git diff --git a/lib/Transforms/Utils/ValueMapper.cpp b/lib/Transforms/Utils/ValueMapper.cpp index f1724db1d47..5d5ca1309ee 100644 --- a/lib/Transforms/Utils/ValueMapper.cpp +++ b/lib/Transforms/Utils/ValueMapper.cpp @@ -2,8 +2,8 @@ // // The LLVM Compiler Infrastructure // -// This file was developed by the LLVM research group and is distributed under -// the University of Illinois Open Source License. See LICENSE.TXT for details. +// This file is distributed under the University of Illinois Open Source +// License. See LICENSE.TXT for details. // //===----------------------------------------------------------------------===// // @@ -12,15 +12,18 @@ // //===----------------------------------------------------------------------===// -#include "ValueMapper.h" +#include "llvm/Transforms/Utils/ValueMapper.h" #include "llvm/Constants.h" #include "llvm/GlobalValue.h" #include "llvm/Instruction.h" using namespace llvm; -Value *llvm::MapValue(const Value *V, std::map &VM) { +Value *llvm::MapValue(const Value *V, ValueMapTy &VM) { Value *&VMSlot = VM[V]; if (VMSlot) return VMSlot; // Does it exist in the map yet? + + // NOTE: VMSlot can be invalidated by any reference to VM, which can grow the + // DenseMap. This includes any recursive calls to MapValue. // Global values do not need to be seeded into the ValueMap if they are using // the identity mapping. @@ -28,9 +31,9 @@ Value *llvm::MapValue(const Value *V, std::map &VM) { return VMSlot = const_cast(V); if (Constant *C = const_cast(dyn_cast(V))) { - if (isa(C) || isa(C) || + if (isa(C) || isa(C) || isa(C) || isa(C) || - isa(C) || isa(V)) + isa(C)) return VMSlot = C; // Primitive constants map directly else if (ConstantArray *CA = dyn_cast(C)) { for (unsigned i = 0, e = CA->getNumOperands(); i != e; ++i) { @@ -46,10 +49,10 @@ Value *llvm::MapValue(const Value *V, std::map &VM) { Values.push_back(cast(MV)); for (++i; i != e; ++i) Values.push_back(cast(MapValue(CA->getOperand(i), VM))); - return VMSlot = ConstantArray::get(CA->getType(), Values); + return VM[V] = ConstantArray::get(CA->getType(), Values); } } - return VMSlot = C; + return VM[V] = C; } else if (ConstantStruct *CS = dyn_cast(C)) { for (unsigned i = 0, e = CS->getNumOperands(); i != e; ++i) { @@ -65,39 +68,22 @@ Value *llvm::MapValue(const Value *V, std::map &VM) { Values.push_back(cast(MV)); for (++i; i != e; ++i) Values.push_back(cast(MapValue(CS->getOperand(i), VM))); - return VMSlot = ConstantStruct::get(CS->getType(), Values); + return VM[V] = ConstantStruct::get(CS->getType(), Values); } } - return VMSlot = C; + return VM[V] = C; } else if (ConstantExpr *CE = dyn_cast(C)) { - if (CE->getOpcode() == Instruction::Cast) { - Constant *MV = cast(MapValue(CE->getOperand(0), VM)); - return VMSlot = ConstantExpr::getCast(MV, CE->getType()); - } else if (CE->getOpcode() == Instruction::GetElementPtr) { - std::vector Idx; - Constant *MV = cast(MapValue(CE->getOperand(0), VM)); - for (unsigned i = 1, e = CE->getNumOperands(); i != e; ++i) - Idx.push_back(cast(MapValue(CE->getOperand(i), VM))); - return VMSlot = ConstantExpr::getGetElementPtr(MV, Idx); - } else if (CE->getOpcode() == Instruction::Select) { - Constant *MV1 = cast(MapValue(CE->getOperand(0), VM)); - Constant *MV2 = cast(MapValue(CE->getOperand(1), VM)); - Constant *MV3 = cast(MapValue(CE->getOperand(2), VM)); - return VMSlot = ConstantExpr::getSelect(MV1, MV2, MV3); - } else { - assert(CE->getNumOperands() == 2 && "Must be binary operator?"); - Constant *MV1 = cast(MapValue(CE->getOperand(0), VM)); - Constant *MV2 = cast(MapValue(CE->getOperand(1), VM)); - return VMSlot = ConstantExpr::get(CE->getOpcode(), MV1, MV2); - } - - } else if (ConstantPacked *CP = dyn_cast(C)) { + std::vector Ops; + for (unsigned i = 0, e = CE->getNumOperands(); i != e; ++i) + Ops.push_back(cast(MapValue(CE->getOperand(i), VM))); + return VM[V] = CE->getWithOperands(Ops); + } else if (ConstantVector *CP = dyn_cast(C)) { for (unsigned i = 0, e = CP->getNumOperands(); i != e; ++i) { Value *MV = MapValue(CP->getOperand(i), VM); if (MV != CP->getOperand(i)) { - // This packed value must contain a reference to a global, make a new - // packed constant and return it. + // This vector value must contain a reference to a global, make a new + // vector constant and return it. // std::vector Values; Values.reserve(CP->getNumOperands()); @@ -106,26 +92,23 @@ Value *llvm::MapValue(const Value *V, std::map &VM) { Values.push_back(cast(MV)); for (++i; i != e; ++i) Values.push_back(cast(MapValue(CP->getOperand(i), VM))); - return VMSlot = ConstantPacked::get(Values); + return VM[V] = ConstantVector::get(Values); } } - return VMSlot = C; + return VM[V] = C; } else { assert(0 && "Unknown type of constant!"); } } - V->dump(); - assert(0 && "Unknown value type: why didn't it get resolved?!"); return 0; } /// RemapInstruction - Convert the instruction operands from referencing the /// current values into those specified by ValueMap. /// -void llvm::RemapInstruction(Instruction *I, - std::map &ValueMap) { +void llvm::RemapInstruction(Instruction *I, ValueMapTy &ValueMap) { for (unsigned op = 0, E = I->getNumOperands(); op != E; ++op) { const Value *Op = I->getOperand(op); Value *V = MapValue(Op, ValueMap);