@@ -152,17 +152,20 @@ class AllocService
152
152
Channel* chn,
153
153
cali::Node* label_node,
154
154
const Variant& v_size,
155
- const Variant& v_uid) {
155
+ const Variant& v_uid,
156
+ const Variant& v_addr) {
156
157
cali_id_t attr[] = {
157
158
alloc_total_size_attr.id (),
158
- alloc_uid_attr.id ()
159
+ alloc_uid_attr.id (),
160
+ alloc_addr_attr.id ()
159
161
};
160
162
Variant data[] = {
161
163
v_size,
162
- v_uid
164
+ v_uid,
165
+ v_addr
163
166
};
164
167
165
- SnapshotRecord trigger_info (1 , &label_node, 2 , attr, data);
168
+ SnapshotRecord trigger_info (1 , &label_node, 3 , attr, data);
166
169
c->push_snapshot (chn, &trigger_info);
167
170
}
168
171
@@ -199,7 +202,10 @@ class AllocService
199
202
}
200
203
201
204
if (g_track_allocations)
202
- track_mem_snapshot (c, chn, info.alloc_label_node , Variant (static_cast <int >(total_size)), info.v_uid );
205
+ track_mem_snapshot (c, chn, info.alloc_label_node ,
206
+ Variant (static_cast <int >(total_size)),
207
+ info.v_uid ,
208
+ Variant (CALI_TYPE_ADDR, &ptr, sizeof (void *)));
203
209
204
210
{
205
211
std::lock_guard<std::mutex>
@@ -242,7 +248,10 @@ class AllocService
242
248
}
243
249
244
250
if (g_track_allocations)
245
- track_mem_snapshot (c, chn, info.free_label_node , Variant (-static_cast <int >(info.total_size )), info.v_uid );
251
+ track_mem_snapshot (c, chn, info.free_label_node ,
252
+ Variant (-static_cast <int >(info.total_size )),
253
+ info.v_uid ,
254
+ Variant (CALI_TYPE_ADDR, &ptr, sizeof (void *)));
246
255
247
256
{
248
257
std::lock_guard<std::mutex>
0 commit comments