3 files changed, 3 insertions, 1 deletions
diff --git a/benchmarks/alex_rq_bench.cpp b/benchmarks/alex_rq_bench.cpp
index 00e0002..54df024 100644
--- a/benchmarks/alex_rq_bench.cpp
+++ b/benchmarks/alex_rq_bench.cpp
@@ -203,7 +203,7 @@ int main(int argc, char **argv)
     size_t insert_cnt = record_count - warmup_cnt;
 
     alex_rq_insert(alex, datafile, insert_cnt, delete_prop, to_delete, true);
-    size_t memory_usage = alex.model_size();
+    size_t memory_usage = alex.model_size() + alex.data_size();
     fprintf(stdout, "%ld\t", memory_usage);
 
     alex_rq_bench(alex, queries);
diff --git a/benchmarks/btree_irs_bench.cpp b/benchmarks/btree_irs_bench.cpp
index c64df8a..862fc6b 100644
--- a/benchmarks/btree_irs_bench.cpp
+++ b/benchmarks/btree_irs_bench.cpp
@@ -77,6 +77,7 @@ int main(int argc, char **argv)
 
     insert_tput_bench<TreeMap, btree_record>(btree, datafile, insert_cnt, delete_prop, to_delete, true);
     size_t memory_usage = btree.get_stats().inner_nodes * tlx::btree_default_traits<key_type, btree_record>::inner_slots * (sizeof(key_type) + sizeof(void*));
+    memory_usage += btree.get_stats().leaves * tlx::btree_default_traits<key_type, btree_record>::leaf_slots * sizeof(btree_record);
     fprintf(stdout, "%ld\t", memory_usage);
 
     btree_sample_bench(btree, queries);
diff --git a/benchmarks/btree_rq_bench.cpp b/benchmarks/btree_rq_bench.cpp
index 818e6f4..d92b45d 100644
--- a/benchmarks/btree_rq_bench.cpp
+++ b/benchmarks/btree_rq_bench.cpp
@@ -76,6 +76,7 @@ int main(int argc, char **argv)
 
     insert_tput_bench<TreeMap, btree_record>(btree, datafile, insert_cnt, delete_prop, to_delete, true);
     size_t memory_usage = btree.get_stats().inner_nodes * tlx::btree_default_traits<key_type, btree_record>::inner_slots * (sizeof(key_type) + sizeof(void*));
+    memory_usage += btree.get_stats().leaves * tlx::btree_default_traits<key_type, btree_record>::leaf_slots * sizeof(btree_record);
     fprintf(stdout, "%ld\t", memory_usage);
 
     btree_rq_bench(btree, queries);