add more classes to snapshotting region
[model-checker.git] / model.h
diff --git a/model.h b/model.h
index e8a8c5eb9e868dc2d4dca77e647d2d97484fed50..9d2fd5a17f402babf0aeae888c00c6d2c0b75d46 100644 (file)
--- a/model.h
+++ b/model.h
@@ -3,10 +3,13 @@
 
 #include <list>
 #include <map>
+#include <vector>
 #include <cstddef>
 #include <ucontext.h>
 
 #include "schedule.h"
+#include "mymemory.h"
+#include <utility>
 #include "libthreads.h"
 #include "libatomic.h"
 #include "threads.h"
@@ -39,6 +42,8 @@ public:
        int switch_to_master(ModelAction *act);
 
        bool next_execution();
+
+       MEMALLOC
 private:
        int next_thread_id;
        int used_sequence_numbers;