KmlScaissCgInitStripes?I

初始化数据结构，并将用户提供的系数矩阵关联到求解系统中，初始化集群版的CG方法时，每个进程使用局部的CSR格式矩阵。

接口定义

C Interface：

int KmlScaissCgInitStripesSI(KmlScasolverTask **handle, int n, int nstripes, const int *stripeWidth,const int *stripeRow, const float * const *a, const int * const *ja, const int * const *ia, MPI_Comm comm);

int KmlScaissCgInitStripesDI(KmlScasolverTask **handle, int n, int nstripes, const int *stripeWidth, const int *stripeRow, const double * const *a, const int * const *ja, const int * const *ia, MPI_Comm comm);

参数

参数名	类型	描述	输入/输出
handle	KmlScasolverTask **	求解器句柄，handle不可为空指针。	输入/输出
n	int	n为矩阵A的维度。	输入
nstripes	int	nstripes为存储在当前MPI进程中的矩阵数，目前只能取1。	输入
stripeWidth	int *	stripeWidth为当前MPI进程中的矩阵行数。该变量的生命周期与handle相同。	输入
stripeRow	int *	stripeRow为当前MPI进程中的矩阵的全局行索引，即首行在全局矩阵中对应行数。该数组的生命周期与handle相同。	输入
a	在KmlScaissCgInitStripesSI中为float 。在KmlScaissCgInitStripesDI中为double 。	a为维度为非零元素的个数的数组，a[i]为第i个非零元的值。该数组的生命周期与handle相同。	输入
ja	int *	ja为维度为非零元素的个数的数组，ja[i]为第i个非零元的列号。该数组的生命周期与handle相同。	输入
ia	int *	ia为stripeWidth+1维数组，ia[i]为第i行第一个非零元在所有非零元中的索引，ia[0]=0。该数组的生命周期与handle相同。	输入
comm	MPI_Comm	为MPI进程通信域。	输入

返回值

返回值	类型	描述
KMLSS_NO_ERROR	int	正常执行。
KMLSS_NULL_ARGUMENT	int	handle，stripeWidth，stripeRow，a，ja，ia中存在空参数。
KMLSS_BAD_N	int	参数n不是正整数。
KMLSS_NOT_IMPLEMENTED	int	参数nstripes不等于1。
KMLSS_NO_MEMORY	int	内存不足。

依赖

#include "kml_scaiss.h"

示例

C Interface：

MPI_Init(NULL, NULL);
int size, rank;
MPI_Comm_size(MPI_COMM_WORLD, &size);//获取总进程数
MPI_Comm_rank(MPI_COMM_WORLD, &rank);//获取当前进程标识

int mat_size = 8;
int n = mat_size / size;
int n_beg = n * rank;
if (n * size != mat_size && rank == (size - 1)) {
    n = mat_size - n * rank;
}
int ja[26] = { 0, 3, 4, 1, 2, 3, 5, 1, 2, 7, 0, 1, 3, 6, 0, 4, 5, 1, 4, 5, 7, 3, 6, 2, 5, 7 };
double a[26] = { 1.0, 1.0, 2.0, 9.0, 2.0, 1.0, -3.0, 2.0, 3.0, 2.0, 1.0, 1.0, 9.0, -5.0, 2.0, 6.0, 1.0, -3.0, 1.0, 4.0, 1.0, -5.0, 7.0, 2.0, 1.0, 2.0 };
int ia[9] = { 0, 3, 7, 10, 14, 17, 21, 23, 26 };
int a_beg = ia[n_beg];
for (int i = n_beg; i < (n_beg + n + 1); i++) {
    ia[i] -= a_beg;
}

/* Internal KML_SCAISS structure */
KmlScasolverTask *handle;

/* KML_SCAISS control parameters */
int error;            /* Output error handle */

/* Create data structures */
const double *a_holder = &a[a_beg];
const int *ja_holder = &ja[a_beg];
const int *ia_holder = &ia[n_beg];
error = KmlScaissCgInitStripesDI(&handle, mat_size, 1, &n, &n_beg, &a_holder, &ja_holder, &ia_holder, MPI_COMM_WORLD);

父主题： SCAISS-CG