DwarfCompileUnit: Add type safety to CompileUnit::getNode by returning DICompileUnit...

[oota-llvm.git] / lib / CodeGen / AsmPrinter / DwarfCompileUnit.h

diff --git a/lib/CodeGen/AsmPrinter/DwarfCompileUnit.h b/lib/CodeGen/AsmPrinter/DwarfCompileUnit.h
index 5b6d33f5ce6e86fa0f0049f79594209b14e009ea..059f2be2104c62c79569c06bae1fc33ad186501c 100644 (file)
--- a/lib/CodeGen/AsmPrinter/DwarfCompileUnit.h
+++ b/lib/CodeGen/AsmPrinter/DwarfCompileUnit.h
@@ -101,7 +101,7 @@ public:
   // Accessors.
   unsigned getUniqueID() const { return UniqueID; }
   uint16_t getLanguage() const { return Node.getLanguage(); }
-  const MDNode *getNode() const { return Node; }
+  DICompileUnit getNode() const { return Node; }
   DIE *getCUDie() const { return CUDie.get(); }
   const StringMap<DIE *> &getGlobalNames() const { return GlobalNames; }
   const StringMap<DIE *> &getGlobalTypes() const { return GlobalTypes; }