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
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
|
/* -*-c-*- */
/*
* _superlu object
*
* Python object representing SuperLU factorization + some utility functions.
*/
#ifndef __SUPERLU_OBJECT
#define __SUPERLU_OBJECT
#include "Python.h"
#include "SuperLU/SRC/slu_zdefs.h"
#include "numpy/arrayobject.h"
#include "SuperLU/SRC/slu_util.h"
#include "SuperLU/SRC/slu_dcomplex.h"
#include "SuperLU/SRC/slu_scomplex.h"
#define _CHECK_INTEGER(x) (PyArray_ISINTEGER(x) && (x)->descr->elsize == sizeof(int))
/*
* SuperLUObject definition
*/
typedef struct {
PyObject_HEAD
npy_intp m,n;
SuperMatrix L;
SuperMatrix U;
int *perm_r;
int *perm_c;
int type;
} SciPyLUObject;
extern PyTypeObject SciPySuperLUType;
int DenseSuper_from_Numeric(SuperMatrix *, PyObject *);
int NRFormat_from_spMatrix(SuperMatrix *, int, int, int, PyArrayObject *,
PyArrayObject *, PyArrayObject *, int);
int NCFormat_from_spMatrix(SuperMatrix *, int, int, int, PyArrayObject *,
PyArrayObject *, PyArrayObject *, int);
colperm_t superlu_module_getpermc(int);
PyObject *newSciPyLUObject(SuperMatrix *, PyObject*, int, int);
int set_superlu_options_from_dict(superlu_options_t *options,
int ilu, PyObject *option_dict,
int *panel_size, int *relax);
/*
* Definitions for other SuperLU data types than Z,
* and type-generic definitions.
*/
#define CHECK_SLU_TYPE(type) \
(type == NPY_FLOAT || type == NPY_DOUBLE || type == NPY_CFLOAT || type == NPY_CDOUBLE)
#define TYPE_GENERIC_FUNC(name, returntype) \
returntype s##name(name##_ARGS); \
returntype d##name(name##_ARGS); \
returntype c##name(name##_ARGS); \
static returntype name(int type, name##_ARGS) \
{ \
switch(type) { \
case NPY_FLOAT: s##name(name##_ARGS_REF); break; \
case NPY_DOUBLE: d##name(name##_ARGS_REF); break; \
case NPY_CFLOAT: c##name(name##_ARGS_REF); break; \
case NPY_CDOUBLE: z##name(name##_ARGS_REF); break; \
default: return; \
} \
}
#define SLU_TYPECODE_TO_NPY(s) \
( ((s) == SLU_S) ? NPY_FLOAT : \
((s) == SLU_D) ? NPY_DOUBLE : \
((s) == SLU_C) ? NPY_CFLOAT : \
((s) == SLU_Z) ? NPY_CDOUBLE : -1)
#define NPY_TYPECODE_TO_SLU(s) \
( ((s) == NPY_FLOAT) ? SLU_S : \
((s) == NPY_DOUBLE) ? SLU_D : \
((s) == NPY_CFLOAT) ? SLU_C : \
((s) == NPY_CDOUBLE) ? SLU_Z : -1)
#define gstrf_ARGS \
superlu_options_t *a, SuperMatrix *b, \
int c, int d, int *e, void *f, int g, \
int *h, int *i, SuperMatrix *j, SuperMatrix *k, \
SuperLUStat_t *l, int *m
#define gstrf_ARGS_REF a,b,c,d,e,f,g,h,i,j,k,l,m
#define gsitrf_ARGS gstrf_ARGS
#define gsitrf_ARGS_REF gstrf_ARGS_REF
#define gstrs_ARGS \
trans_t a, SuperMatrix *b, SuperMatrix *c, \
int *d, int *e, SuperMatrix *f, \
SuperLUStat_t *g, int *h
#define gstrs_ARGS_REF a,b,c,d,e,f,g,h
#define gssv_ARGS \
superlu_options_t *a, SuperMatrix *b, int *c, int *d, \
SuperMatrix *e, SuperMatrix *f, SuperMatrix *g, \
SuperLUStat_t *h, int *i
#define gssv_ARGS_REF a,b,c,d,e,f,g,h,i
#define Create_Dense_Matrix_ARGS \
SuperMatrix *a, int b, int c, void *d, int e, \
Stype_t f, Dtype_t g, Mtype_t h
#define Create_Dense_Matrix_ARGS_REF a,b,c,d,e,f,g,h
#define Create_CompRow_Matrix_ARGS \
SuperMatrix *a, int b, int c, int d, \
void *e, int *f, int *g, \
Stype_t h, Dtype_t i, Mtype_t j
#define Create_CompRow_Matrix_ARGS_REF a,b,c,d,e,f,g,h,i,j
#define Create_CompCol_Matrix_ARGS Create_CompRow_Matrix_ARGS
#define Create_CompCol_Matrix_ARGS_REF Create_CompRow_Matrix_ARGS_REF
TYPE_GENERIC_FUNC(gstrf, void);
TYPE_GENERIC_FUNC(gsitrf, void);
TYPE_GENERIC_FUNC(gstrs, void);
TYPE_GENERIC_FUNC(gssv, void);
TYPE_GENERIC_FUNC(Create_Dense_Matrix, void);
TYPE_GENERIC_FUNC(Create_CompRow_Matrix, void);
TYPE_GENERIC_FUNC(Create_CompCol_Matrix, void);
#endif /* __SUPERLU_OBJECT */
|
