1 files changed, 6 insertions, 2 deletions
diff --git a/bench.erl b/bench.erl
index 9fa0651..bc5102d 100755
--- a/bench.erl
+++ b/bench.erl
@@ -2,6 +2,8 @@
 %% -*- erlang -*-
 %%! -pa test/ebin -pa ../lager/ebin -pa ../lager/deps/goldrush/ebin
 
+-mode(compile).
+
 add_entries(Entries) ->
     lists:foreach(fun (Entry) ->
                           ht:add(Entry)
@@ -17,9 +19,11 @@ timeprint(Time) ->
     io_lib:format("~.2fs", [Time/1000000]).
 
 printheapsize() ->
-    case false of
+    case true of
         true ->
-            io:format("Heap size: local ~s ht ~s~n", [heapsize(self()), heapsize(ht)]);
+            Memory = erlang:memory(),
+            Total = proplists:get_value(total, Memory),
+            io:format("Heap size: local ~s ht ~s total ~.2f M~n", [heapsize(self()), heapsize(ht), Total/1024/1024]);
         false ->
             none
     end.