KmlScaissBicgstabInitStripes?I
初始化数据结构,并将用户提供的系数矩阵关联到求解系统中,初始化集群版的BICGSTAB方法时,每个进程使用局部的CSR格式矩阵。
接口定义
C Interface:
int KmlScaissBicgstabInitStripesSI(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 KmlScaissBicgstabInitStripesDI(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 |
|
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"
示例
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 = KmlScaissBicgstabInitStripesDI(&handle, mat_size, 1, &n, &n_beg, &a_holder, &ja_holder, &ia_holder, MPI_COMM_WORLD);