38 lines
1.4 KiB
C++
38 lines
1.4 KiB
C++
/*
|
|
Copyright (c) 2013 Microsoft Corporation. All rights reserved.
|
|
Released under Apache 2.0 license as described in the file LICENSE.
|
|
|
|
Author: Leonardo de Moura
|
|
*/
|
|
#include <iostream>
|
|
#include "util/test.h"
|
|
#include "util/memory.cpp"
|
|
|
|
static void tst1() {
|
|
std::cout << "Initial: " << lean::get_allocated_memory() << "\n";
|
|
size_t old_mem = lean::get_allocated_memory();
|
|
int N = 5;
|
|
int * a = static_cast<int*>(lean::malloc(N * sizeof(N)));
|
|
lean_assert(lean::get_allocated_memory() >= old_mem + N * sizeof(N));
|
|
for (int i = 0; i < N; i++) {
|
|
a[i] = i;
|
|
}
|
|
a = static_cast<int*>(lean::realloc(a, N * 2 * sizeof(N)));
|
|
lean_assert(lean::get_allocated_memory() >= old_mem + N * 2 * sizeof(N));
|
|
std::cout << "Total: " << static_cast<size_t>(lean::get_allocated_memory()) << "\n";
|
|
std::cout << "Thread: " << static_cast<size_t>(lean::get_thread_allocated_memory()) << "\n";
|
|
#if !defined(HAS_TCMALLOC)
|
|
// When TCMALLOC is used, there is a problem during initialization, and the value of get_thread_allocated_memory is off.
|
|
lean_assert_eq(lean::get_allocated_memory(), static_cast<size_t>(lean::get_thread_allocated_memory()));
|
|
#endif
|
|
for (int i = 0; i < N; i++) {
|
|
lean_assert(a[i] == i);
|
|
}
|
|
lean::free(a);
|
|
lean_assert_eq(old_mem, lean::get_allocated_memory());
|
|
}
|
|
|
|
int main() {
|
|
tst1();
|
|
return lean::has_violations() ? 1 : 0;
|
|
}
|