1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
|
/*
* Copyright (C) 2019-2020 Paul Cercueil <paul@crapouillou.net>
*
* This library is free software; you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public
* License as published by the Free Software Foundation; either
* version 2.1 of the License, or (at your option) any later version.
*
* This library is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* Lesser General Public License for more details.
*/
#include "config.h"
#include "lightrec-private.h"
#include "memmanager.h"
#include <stdlib.h>
#if ENABLE_TINYMM
#include <tinymm.h>
#endif
#ifdef ENABLE_THREADED_COMPILER
#include <stdatomic.h>
static atomic_uint lightrec_bytes[MEM_TYPE_END];
void lightrec_register(enum mem_type type, unsigned int len)
{
atomic_fetch_add(&lightrec_bytes[type], len);
}
void lightrec_unregister(enum mem_type type, unsigned int len)
{
atomic_fetch_sub(&lightrec_bytes[type], len);
}
unsigned int lightrec_get_mem_usage(enum mem_type type)
{
return atomic_load(&lightrec_bytes[type]);
}
#else /* ENABLE_THREADED_COMPILER */
static unsigned int lightrec_bytes[MEM_TYPE_END];
void lightrec_register(enum mem_type type, unsigned int len)
{
lightrec_bytes[type] += len;
}
void lightrec_unregister(enum mem_type type, unsigned int len)
{
lightrec_bytes[type] -= len;
}
unsigned int lightrec_get_mem_usage(enum mem_type type)
{
return lightrec_bytes[type];
}
#endif /* ENABLE_THREADED_COMPILER */
unsigned int lightrec_get_total_mem_usage(void)
{
unsigned int i, count;
for (i = 0, count = 0; i < MEM_TYPE_END; i++)
count += lightrec_get_mem_usage((enum mem_type)i);
return count;
}
void * lightrec_malloc(struct lightrec_state *state,
enum mem_type type, unsigned int len)
{
void *ptr;
#if ENABLE_TINYMM
if (type == MEM_FOR_IR)
ptr = tinymm_malloc(state->tinymm, len);
else
#endif
ptr = malloc(len);
if (!ptr)
return NULL;
lightrec_register(type, len);
return ptr;
}
void * lightrec_calloc(struct lightrec_state *state,
enum mem_type type, unsigned int len)
{
void *ptr;
#if ENABLE_TINYMM
if (type == MEM_FOR_IR)
ptr = tinymm_zalloc(state->tinymm, len);
else
#endif
ptr = calloc(1, len);
if (!ptr)
return NULL;
lightrec_register(type, len);
return ptr;
}
void lightrec_free(struct lightrec_state *state,
enum mem_type type, unsigned int len, void *ptr)
{
lightrec_unregister(type, len);
#if ENABLE_TINYMM
if (type == MEM_FOR_IR)
tinymm_free(state->tinymm, ptr);
else
#endif
free(ptr);
}
float lightrec_get_average_ipi(void)
{
unsigned int code_mem = lightrec_get_mem_usage(MEM_FOR_CODE);
unsigned int native_mem = lightrec_get_mem_usage(MEM_FOR_MIPS_CODE);
return native_mem ? (float)code_mem / (float)native_mem : 0.0f;
}
|
