[mc] Enable the custom mm malloc only in MC

[simgrid.git] / src / mc / mc_memory.c

diff --git a/src/mc/mc_memory.c b/src/mc/mc_memory.c
index d1c804b..6d9700e 100644 (file)
--- a/src/mc/mc_memory.c
+++ b/src/mc/mc_memory.c
@@ -8,6 +8,8 @@
 #include <fcntl.h>
 
 #include "xbt/log.h"
+#include "xbt/dynar.h"
+#include "xbt/virtu.h"
 
 #include "mc/mc.h"
 #include "mc_object_info.h"
@@ -24,6 +26,9 @@ xbt_mheap_t mc_heap = NULL;           /* memory persistent over the MC rollbacks
 /* It creates the two heap regions: std_heap and mc_heap */
 void MC_memory_init()
 {
+  mmalloc_ensure_using_mm(
+    xbt_dynar_length(xbt_cmdline), xbt_dynar_get_ptr(xbt_cmdline, 0));
+
   /* Create the first region HEAP_OFFSET bytes after the heap break address */
   std_heap = mmalloc_get_default_md();
   xbt_assert(std_heap != NULL);