From e181deab1eeaf5a3ee34ad28f4604a8cc5cf4e38 Mon Sep 17 00:00:00 2001
From: augustin64 <me.git@augustin64.fr>
Date: Sat, 18 Feb 2023 14:10:52 +0100
Subject: [PATCH] Fix gree

---
 src/memory_management.c  | 2 +-
 src/memory_management.cu | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/memory_management.c b/src/memory_management.c
index 511b7b8..3bbca9c 100644
--- a/src/memory_management.c
+++ b/src/memory_management.c
@@ -124,7 +124,7 @@ extern "C"
 void gree(void* ptr) {
     #if defined(USE_CUDA) || defined(TEST_MEMORY_MANAGEMENT)
         pthread_mutex_lock(&memory_lock);
-        free_memory(ptr, memory);
+        memory = free_memory(ptr, memory);
         pthread_mutex_unlock(&memory_lock);
     #else
         free(ptr);
diff --git a/src/memory_management.cu b/src/memory_management.cu
index 511b7b8..3bbca9c 100644
--- a/src/memory_management.cu
+++ b/src/memory_management.cu
@@ -124,7 +124,7 @@ extern "C"
 void gree(void* ptr) {
     #if defined(USE_CUDA) || defined(TEST_MEMORY_MANAGEMENT)
         pthread_mutex_lock(&memory_lock);
-        free_memory(ptr, memory);
+        memory = free_memory(ptr, memory);
         pthread_mutex_unlock(&memory_lock);
     #else
         free(ptr);