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
|
/* Should be imported before Python.h */
#include <Python.h>
#define NO_IMPORT_ARRAY
#define PY_ARRAY_UNIQUE_SYMBOL _scipy_sparse_superlu_ARRAY_API
#include "_superluobject.h"
#include "numpy/npy_3kcompat.h"
#include <setjmp.h>
jmp_buf _superlu_py_jmpbuf;
PyObject *_superlumodule_memory_dict = NULL;
/* Abort to be used inside the superlu module so that memory allocation
errors don't exit Python and memory allocated internal to SuperLU is freed.
Calling program should deallocate (using SUPERLU_FREE) all memory that could have
been allocated. (It's ok to FREE unallocated memory)---will be ignored.
*/
void superlu_python_module_abort(char *msg)
{
PyErr_SetString(PyExc_RuntimeError, msg);
longjmp(_superlu_py_jmpbuf, -1);
}
void *superlu_python_module_malloc(size_t size)
{
PyObject *key = NULL;
void *mem_ptr;
if (_superlumodule_memory_dict == NULL) {
_superlumodule_memory_dict = PyDict_New();
}
mem_ptr = malloc(size);
if (mem_ptr == NULL)
return NULL;
key = PyLong_FromVoidPtr(mem_ptr);
if (key == NULL)
goto fail;
if (PyDict_SetItem(_superlumodule_memory_dict, key, Py_None))
goto fail;
Py_DECREF(key);
return mem_ptr;
fail:
Py_XDECREF(key);
free(mem_ptr);
superlu_python_module_abort
("superlu_malloc: Cannot set dictionary key value in malloc.");
return NULL;
}
void superlu_python_module_free(void *ptr)
{
PyObject *key;
PyObject *ptype, *pvalue, *ptraceback;
if (ptr == NULL)
return;
PyErr_Fetch(&ptype, &pvalue, &ptraceback);
key = PyLong_FromVoidPtr(ptr);
/* This will only free the pointer if it could find it in the dictionary
* of already allocated pointers --- thus after abort, the module can free all
* the memory that "might" have been allocated to avoid memory leaks on abort
* calls.
*/
if (_superlumodule_memory_dict &&
!(PyDict_DelItem(_superlumodule_memory_dict, key))) {
free(ptr);
}
Py_DECREF(key);
PyErr_Restore(ptype, pvalue, ptraceback);
return;
}
/*
* Stubs for Harwell Subroutine Library functions that SuperLU tries to call.
*/
void mc64id_(int *a)
{
superlu_python_module_abort("chosen functionality not available");
}
void mc64ad_(int *a, int *b, int *c, int d[], int e[], double f[],
int *g, int h[], int *i, int j[], int *k, double l[],
int m[], int n[])
{
superlu_python_module_abort("chosen functionality not available");
}
|
