KmlScadss(SPD/HPD)Init?I
初始化数据结构,并将用户提供的系数矩阵关联到求解系统中,初始化集群版的LLT\LLH方法时使用上三角的CSR格式。
接口定义
C Interface:
int KmlScadssSpdInitSI(KmlScasolverTask **pHandle, int n, float *a, int *ja, int *ia, MPI_Comm comm);
int KmlScadssSpdInitDI(KmlScasolverTask **pHandle, int n, double *a, int *ja, int *ia, MPI_Comm comm);
int KmlScadssHpdInitCI(KmlScasolverTask **pHandle, int n, kml_complex_float *a, int *ja, int *ia, MPI_Comm comm);
int KmlScadssHpdInitZI(KmlScasolverTask **pHandle, int n, kml_complex_double *a, int *ja, int *ia, MPI_Comm comm);
参数
参数名 |
类型 |
描述 |
输入/输出 |
---|---|---|---|
pHandle |
KmlScasolverTask ** |
求解器句柄,pHandle不可为空指针。 |
输出 |
n |
int |
n为矩阵A的维度。 |
输入 |
a |
|
a为维度为非零元素的个数的数组,a[i]为第i个非零元的值。该数组的生命周期与phandle相同。 |
输入 |
ja |
int |
ja为维度为非零元素的个数的数组,ja[i]为第i个非零元的列号。该数组的生命周期与phandle相同。 |
输入 |
ia |
int |
ia为n+1维数组,ia[i]为第i行第一个非零元在所有非零元中的索引。该数组的生命周期与phandle相同。 |
输入 |
comm |
MPI_Comm |
为MPI进程通信域。 |
输入 |
返回值
返回值 |
类型 |
描述 |
---|---|---|
KMLSS_NO_ERROR |
int |
正常执行。 |
KMLSS_NULL_ARGUMENT |
int |
pHandle,a,ja,ia中有空指针。 |
KMLSS_BAD_N |
int |
矩阵维数输入值小于等于0。 |
KMLSS_NO_MEMORY |
int |
内存开辟失败。 |
依赖
#include "kml_scadss.h"
示例
MPI_Init(NULL, NULL); int size, rank; MPI_Comm_size(MPI_COMM_WORLD, &size);//获取总进程数 MPI_Comm_rank(MPI_COMM_WORLD, &rank);//获取当前进程标识 int n = 8; int ia[9] = {0, 3, 7, 9, 11, 13, 15, 16, 17}; int ja[17] = {0, 3, 4, 1, 2, 3, 5, 2, 7, 3, 6, 4, 5, 5, 7, 6, 7}; double a[17] = {1.0, 1.0, 2.0, 9.0, 2.0, 1.0, -3.0, 3.0, 2.0, 9.0, -5.0, 6.0, 1.0, 4.0, 1.0, 7.0, 2.0}; KmlScasolverTask* matrix_handle = NULL; int ierr = KmlScadssSpdInitDI(&matrix_handle, n, a, ja, ia, MPI_COMM_WORLD); if (ierr != 0) { printf("\nERROR in KmlScadssSpdInitDI: %d", ierr); }