|
@@ -281,8 +281,10 @@ static size_t do_free_mem_chunk(struct _starpu_mem_chunk *mc, unsigned node)
|
|
|
size_t size;
|
|
size_t size;
|
|
|
starpu_data_handle_t handle = mc->data;
|
|
starpu_data_handle_t handle = mc->data;
|
|
|
|
|
|
|
|
- if (handle)
|
|
|
|
|
|
|
+ if (handle) {
|
|
|
_starpu_spin_checklocked(&handle->header_lock);
|
|
_starpu_spin_checklocked(&handle->header_lock);
|
|
|
|
|
+ mc->size = _starpu_data_get_size(handle);
|
|
|
|
|
+ }
|
|
|
|
|
|
|
|
mc->replicate->mc=NULL;
|
|
mc->replicate->mc=NULL;
|
|
|
|
|
|