2 files changed, 7 insertions, 2 deletions
diff --git a/include/llvm/CodeGen/DwarfWriter.h b/include/llvm/CodeGen/DwarfWriter.h
index f199dd25fd..aec1077eed 100644
--- a/include/llvm/CodeGen/DwarfWriter.h
+++ b/include/llvm/CodeGen/DwarfWriter.h
@@ -32,6 +32,7 @@ namespace llvm {
   // Forward declarations.
   //
   class AsmPrinter;
+  class CompileUnitWrapper;
   class DIE;
   class DwarfWriter; 
   class DWContext;
@@ -645,10 +646,10 @@ public:
                            unsigned Size, unsigned Align);
 
 private:
+
     /// NewCompileUnit - Create new compile unit information.
     ///
-    DIE *NewCompileUnit(const std::string &Directory,
-                        const std::string &SourceName);
+    DIE *DwarfWriter::NewCompileUnit(const CompileUnitWrapper &CompileUnit);
 
     /// EmitInitial - Emit initial Dwarf declarations.
     ///
diff --git a/include/llvm/CodeGen/MachineDebugInfo.h b/include/llvm/CodeGen/MachineDebugInfo.h
index 7bbd11db7a..fcd75f4749 100644
--- a/include/llvm/CodeGen/MachineDebugInfo.h
+++ b/include/llvm/CodeGen/MachineDebugInfo.h
@@ -271,6 +271,10 @@ public:
   ///
   void MachineDebugInfo::SetupCompileUnits(Module &M);
 
+  /// getCompileUnits - Return a vector of debug compile units.
+  ///
+  const UniqueVector<CompileUnitWrapper> getCompileUnits() const;
+
   /// getGlobalVariables - Return a vector of debug global variables.
   ///
   static std::vector<GlobalWrapper> getGlobalVariables(Module &M);