Make sure that the cloned module retains the type symbol table entries!

llvm-svn: 5894

1 files changed, 9 insertions, 0 deletions

diff --git a/llvm/lib/Transforms/Utils/CloneModule.cpp b/llvm/lib/Transforms/Utils/CloneModule.cpp
index 992431619c9..93bed758397 100644
--- a/llvm/lib/Transforms/Utils/CloneModule.cpp
+++ b/llvm/lib/Transforms/Utils/CloneModule.cpp
@@ -8,6 +8,7 @@
 #include "llvm/Transforms/Utils/Cloning.h"
 #include "llvm/Module.h"
 #include "llvm/DerivedTypes.h"
+#include "llvm/SymbolTable.h"
 #include "llvm/Constant.h"
 #include "ValueMapper.h"
 
@@ -22,6 +23,14 @@ Module *CloneModule(const Module *M) {
   New->setEndianness(M->getEndianness());
   New->setPointerSize(M->getPointerSize());
 
+  // Copy all of the type symbol table entries over...
+  const SymbolTable &SymTab = M->getSymbolTable();
+  SymbolTable::const_iterator TypeI = SymTab.find(Type::TypeTy);
+  if (TypeI != SymTab.end())
+    for (SymbolTable::VarMap::const_iterator I = TypeI->second.begin(),
+           E = TypeI->second.end(); I != E; ++I)
+      New->addTypeName(I->first, cast<Type>(I->second));
+
   // Create the value map that maps things from the old module over to the new
   // module.
   std::map<const Value*, Value*> ValueMap;