Subzero: Add memory usage to "-szstats" output.

src/IceGlobalContext.cpp

 //===- subzero/src/IceGlobalContext.cpp - Global context defs -------------===//
 //
 //                        The Subzero Code Generator
 //
 // This file is distributed under the University of Illinois Open Source
 // License. See LICENSE.TXT for details.
 //
 //===----------------------------------------------------------------------===//
 //
 // This file defines aspects of the compilation that persist across
 // multiple functions.
 //
 //===----------------------------------------------------------------------===//
 
 #include <ctype.h> // isdigit(), isupper()
 #include <locale>  // locale
 #include <unordered_map>
 
+#include "llvm/Support/Timer.h"
+
 #include "IceCfg.h"
 #include "IceClFlags.h"
 #include "IceDefs.h"
 #include "IceGlobalContext.h"
 #include "IceGlobalInits.h"
 #include "IceOperand.h"
 #include "IceTargetLowering.h"
 #include "IceTimerTree.h"
 #include "IceTypes.h"
 
@@ skipping 69 matching lines @@
   TypePool<IceType_f64, double, ConstantDouble> Doubles;
   TypePool<IceType_i1, int8_t, ConstantInteger32> Integers1;
   TypePool<IceType_i8, int8_t, ConstantInteger32> Integers8;
   TypePool<IceType_i16, int16_t, ConstantInteger32> Integers16;
   TypePool<IceType_i32, int32_t, ConstantInteger32> Integers32;
   TypePool<IceType_i64, int64_t, ConstantInteger64> Integers64;
   TypePool<IceType_i32, RelocatableTuple, ConstantRelocatable> Relocatables;
   UndefPool Undefs;
 };
 
+void CodeStats::dump(const IceString &Name, Ostream &Str) {
+  if (!ALLOW_DUMP)
+    return;
+  Str << "|" << Name << "|Inst Count  |" << InstructionsEmitted << "\n";
+  Str << "|" << Name << "|Regs Saved  |" << RegistersSaved << "\n";
+  Str << "|" << Name << "|Frame Bytes |" << FrameBytes << "\n";
+  Str << "|" << Name << "|Spills      |" << Spills << "\n";
+  Str << "|" << Name << "|Fills       |" << Fills << "\n";
+  Str << "|" << Name << "|Spills+Fills|" << Spills + Fills << "\n";
+  Str << "|" << Name << "|Memory Usage|";
+  if (ssize_t MemUsed = llvm::TimeRecord::getCurrentTime(false).getMemUsed())
+    Str << MemUsed;
+  else
+    Str << "(requires '-track-memory')";
+  Str << "\n";
+}
+
 GlobalContext::GlobalContext(Ostream *OsDump, Ostream *OsEmit,
                              ELFStreamer *ELFStr, VerboseMask Mask,
                              TargetArch Arch, OptLevel Opt,
                              IceString TestPrefix, const ClFlags &Flags)
     : StrDump(OsDump), StrEmit(OsEmit), VMask(Mask),
       ConstPool(new ConstantPool()), Arch(Arch), Opt(Opt),
       TestPrefix(TestPrefix), Flags(Flags), HasEmittedFirstMethod(false),
       RNG(""), ObjectWriter() {
   // Pre-register built-in stack names.
   if (ALLOW_DUMP) {
@@ skipping 373 matching lines @@
 TimerMarker::TimerMarker(TimerIdT ID, const Cfg *Func)
     : ID(ID), Ctx(Func->getContext()), Active(false) {
   if (ALLOW_DUMP) {
     Active = Func->getFocusedTiming() || Ctx->getFlags().SubzeroTimingEnabled;
     if (Active)
       Ctx->pushTimer(ID);
   }
 }
 
 } // end of namespace Ice