c py_allocator update

hat-open · Aug 20, 2024 · 9332db2 · 9332db2
1 parent a1af4de
commit 9332db2
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 1 deletion.
diff --git a/src_c/hat/py_allocator.c b/src_c/hat/py_allocator.c
@@ -1,4 +1,4 @@
-#include "allocator.h"
+#include "py_allocator.h"
 #include <Python.h>
 
 
@@ -10,4 +10,14 @@ static void *py_realloc(hat_allocator_t *a, size_t size, void *old) {
 }
 
 
+static void *py_raw_realloc(hat_allocator_t *a, size_t size, void *old) {
+    if (size)
+        return PyMem_RawRealloc(old, size);
+    PyMem_RawFree(old);
+    return NULL;
+}
+
+
 hat_allocator_t hat_py_allocator = {.realloc = py_realloc};
+
+hat_allocator_t hat_py_raw_allocator = {.realloc = py_raw_realloc};
diff --git a/src_c/hat/py_allocator.h b/src_c/hat/py_allocator.h
@@ -14,6 +14,9 @@ extern "C" {
 /** \brief allocator instance */
 extern hat_allocator_t hat_py_allocator;
 
+/** \brief raw allocator instance */
+extern hat_allocator_t hat_py_raw_allocator;
+
 #ifdef __cplusplus
 }
 #endif