KmlDss(Spd/Hpd)Solve?I

求解LL^Tx=b或者LL^Hx=b。

接口定义

C Interface：

int KmlDssSpdSolveSI(KmlSolverTask **pHandle, int nb, float *x, int ldx, float *b, int ldb);

int KmlDssSpdSolveDI(KmlSolverTask **pHandle, int nb, double *x, int ldx, double *b, int ldb);

int KmlDssHpdSolveCI(KmlSolverTask **pHandle, int nb, kml_complex_float *x, int ldx, kml_complex_float *b, int ldb);

int KmlDssHpdSolveZI(KmlSolverTask **pHandle, int nb, kml_complex_double *x, int ldx, kml_complex_double *b, int ldb);

参数

参数名	类型	描述	输入/输出
pHandle	KmlSolverTask **	求解器句柄，传入之前步骤的变量。	输入
nb	int	右端项矢量的个数。	输入
x	在KmlDssSpdSolveSI中为float 。在KmlDssSpdSolveDI中为double 。在KmlDssHpdSolveCI中为kml_complex_float 。在KmlDssHpdSolveZI中为kml_complex_double 。	稠密向量存储求解结果，其元素个数为ldxnb，第k个矢量的第i个元素位置为x[i + k ldx]。	输出
ldx	int	x的leading dimension大小，要求ldx ≥ max(1, n)。	输入
b	在KmDssSpdSolveSI中为float 。在KmlDssSpdSolveDI中为double 。在KmlDssHpdSolveCI中为kml_complex_float 。在KmlDssHpdSolveZI中为kml_complex_double 。	稠密向量存储右端项，其元素个数为ldbnb，第k个矢量的第i个元素位置为b[i + k ldb]。	输入
ldb	int	b的leading dimension大小，要求ldb ≥ max(1, n)。	输入

返回值

返回值	类型	描述
KMLSS_NO_ERROR	int	正常执行。
KMLSS_NULL_ARGUMENT	int	pHandle，x，b为空。
KMLSS_BAD_NB	int	输入参数nb小于等于0。
KMLSS_BAD_HANDLE	int	pHandle格式存在问题。
KMLSS_BAD_LDX	int	ldx取值不在有效范围内。
KMLSS_BAD_LDB	int	ldb取值不在有效范围内。
KMLSS_INVALID_CALL_ORDER	int	函数执行顺序错误，尚未执行分解阶段就提前执行求解函数。
KMLSS_NO_MEMORY	int	开辟内存失败。

依赖

#include "kml_dss.h"或#include "kml_solver.h"

示例

C Interface：

int n = 2;
int ia[3] = {0,2,4};
int ja[4] = {0,1,0,1};
double a[4] = {2,1,1,2};
double b[2] = {3,1};
double x[2];

int nrhs = 1;
int ldx = n, ldb = n;

KmlSolverTask *handle;
int ierr = KmlDssSpdInitDI(&handle, n, a, ja, ia);
if (ierr != 0) {
        printf("\nERROR in KmlDssSpdInitDI: %d", ierr);
}
ierr = KmlDssSpdAnalyzeDI(&handle);
if (ierr != 0) {
        printf("\nERROR in KmlDssSpdAnalyzeDI: %d", ierr);
}
ierr = KmlDssSpdFactorizeDI(&handle);
if (ierr != 0) {
        printf("\nERROR in KmlDssSpdFactorizeDI: %d", ierr);
}
ierr = KmlDssSpdSolveDI(&handle, nrhs, x, ldx, b, ldb);
if (ierr != 0) {
        printf("\nERROR in KmlDssSpdSolveDI: %d", ierr);
}
for (int i = 0; i < n; i++) {
        printf("%lf ", x[i]);
}
printf("\n");

运行结果：

1.666667 -0.333333

父主题： LLT/LLH