MKL 函数 pdagemv 用于实现分布式矩阵向量乘法,即计算矩阵 A 和向量 x 的乘积 Ax。它的使用方法如下:

  1. 函数声明:
void pdagemv(char *trans, MKL_INT *m, MKL_INT *n, double *alpha, double *a, MKL_INT *ia, MKL_INT *ja, MKL_INT *desca, double *x, MKL_INT *ix, MKL_INT *jx, MKL_INT *descx, double *beta, double *y, MKL_INT *iy, MKL_INT *jy, MKL_INT *descy);
  1. 参数说明:
  • trans:矩阵 A 的转置类型,取值有'N'(不转置)、'T'(转置)和'C'(共轭转置)。
  • m:矩阵 A 的行数。
  • n:矩阵 A 的列数。
  • alpha:矩阵 A 与向量 x 的乘积中矩阵 A 的系数。
  • a:矩阵 A 的元素值。
  • ia:矩阵 A 的行偏移数组。
  • ja:矩阵 A 的列偏移数组。
  • desca:矩阵 A 的分布描述符。
  • x:向量 x 的元素值。
  • ix:向量 x 的行偏移数组。
  • jx:向量 x 的列偏移数组。
  • descx:向量 x 的分布描述符。
  • beta:矩阵 A 与向量 x 的乘积中向量 y 的系数。
  • y:向量 y 的元素值。
  • iy:向量 y 的行偏移数组。
  • jy:向量 y 的列偏移数组。
  • descy:向量 y 的分布描述符。
  1. 返回值说明:

该函数没有返回值。函数执行后,向量 y 中存储了矩阵 A 与向量 x 的乘积 Ax 的结果。

  1. 注意事项:
  • 在调用该函数之前,必须先调用 mkl_blacs_init 函数进行初始化。
  • 该函数只能用于分布式矩阵向量乘法,要想进行串行矩阵向量乘法,可以使用 mkl_dgemv 函数。
  • 调用该函数时,要保证矩阵 A 和向量 x 的分布描述符 desca 和 descx 是一致的,向量 y 的分布描述符 descy 可以与它们不同。
  • 矩阵 A 和向量 x 的偏移数组 ia、ja、ix 和 jx 的值必须按照 Morton 序排列。
MKL 函数 pdagemv 用法:分布式矩阵向量乘法

原文地址: https://www.cveoy.top/t/topic/onON 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录