Statistical memory profiling of blocks allocated in the major heap.

1 files changed, 34 insertions, 2 deletions

diff --git a/stdlib/gc.ml b/stdlib/gc.ml
index 41a8f8c8fa..28beaf66c5 100644
--- a/stdlib/gc.ml
+++ b/stdlib/gc.ml
@@ -3,9 +3,10 @@
 (*                                 OCaml                                  *)
 (*                                                                        *)
 (*             Damien Doligez, projet Para, INRIA Rocquencourt            *)
+(*             Jacques-Henri Joudan, projet Gallium, INRIA Paris          *)
 (*                                                                        *)
-(*   Copyright 1996 Institut National de Recherche en Informatique et     *)
-(*     en Automatique.                                                    *)
+(*   Copyright 1996-2016 Institut National de Recherche en Informatique   *)
+(*     et en Automatique.                                                 *)
 (*                                                                        *)
 (*   All rights reserved.  This file is distributed under the terms of    *)
 (*   the GNU Lesser General Public License version 2.1, with the          *)
@@ -117,3 +118,34 @@ let create_alarm f =
 
 
 let delete_alarm a = a := false
+
+module Memprof =
+  struct
+    type alloc_kind =
+      | Minor
+      | Major
+      | Serialized
+
+    type sample_info = {
+        n_samples: int; kind: alloc_kind; tag: int;
+        size: int; callstack: Printexc.raw_backtrace;
+    }
+
+    type 'a callback = sample_info -> (Obj.t, 'a) Ephemeron.K1.t option
+
+    type 'a ctrl = {
+        sampling_rate : float;
+        callstack_size : int;
+        callback : 'a callback
+    }
+
+    let stopped_ctrl = {
+        sampling_rate = 0.; callstack_size = 0;
+        callback = fun _ -> assert false
+    }
+
+    external set_ctrl : 'a ctrl -> unit = "caml_memprof_set"
+
+    let start = set_ctrl
+    let stop () = set_ctrl stopped_ctrl
+  end