LAMMP 4.1.0
Lamina High-Precision Arithmetic Library
载入中...
搜索中...
未找到
宏定义 | 函数
mat22_mul.c 文件参考
#include "../../../include/lammp/impl/signed.h"
#include "../../../include/lammp/impl/tmp_alloc.h"
#include "../../../include/lammp/lmmpn.h"
#include "../../../include/lammp/impl/mparam.h"
#include "../../../include/lammp/matrix.h"
+ mat22_mul.c 的引用(Include)关系图:

浏览源代码.

宏定义

#define A11   (matA->a00)
 
#define A11   (mat->a00)
 
#define A11n   (matA->n00)
 
#define A11n   (mat->n00)
 
#define A12   (matA->a01)
 
#define A12   (mat->a01)
 
#define A12n   (matA->n01)
 
#define A12n   (mat->n01)
 
#define A21   (matA->a10)
 
#define A21   (mat->a10)
 
#define A21n   (matA->n10)
 
#define A21n   (mat->n10)
 
#define A22   (matA->a11)
 
#define A22   (mat->a11)
 
#define A22n   (matA->n11)
 
#define A22n   (mat->n11)
 
#define B11   (matB->a00)
 
#define B11n   (matB->n00)
 
#define B12   (matB->a01)
 
#define B12n   (matB->n01)
 
#define B21   (matB->a10)
 
#define B21n   (matB->n10)
 
#define B22   (matB->a11)
 
#define B22n   (matB->n11)
 
#define C11   (dst->a00)
 
#define C11   (dst->a00)
 
#define C11n   (dst->n00)
 
#define C11n   (dst->n00)
 
#define C12   (dst->a01)
 
#define C12   (dst->a01)
 
#define C12n   (dst->n01)
 
#define C12n   (dst->n01)
 
#define C21   (dst->a10)
 
#define C21   (dst->a10)
 
#define C21n   (dst->n10)
 
#define C21n   (dst->n10)
 
#define C22   (dst->a11)
 
#define C22   (dst->a11)
 
#define C22n   (dst->n11)
 
#define C22n   (dst->n11)
 
#define p1   tp
 
#define p1   tp
 
#define p1   (tp)
 
#define p1   (tp)
 
#define p1n   n1
 
#define p1n   n1
 
#define p2   tp + tn
 
#define p2   tp + tn
 
#define p2   (tp + tn)
 
#define p2   (tp + tn)
 
#define p2n   n5
 
#define p2n   n5
 
#define p3   (tp + 2 * tn)
 
#define p3   (tp + 2 * tn)
 
#define p3n   n2
 
#define p3n   n2
 
#define p4   (tp + 3 * tn)
 
#define p4   (tp + 3 * tn)
 
#define p4n   n7
 
#define p4n   n7
 
#define p5   (tp + 4 * tn)
 
#define p5   (tp + 4 * tn)
 
#define p5n   n6
 
#define p5n   n6
 
#define p6   (tp + 5 * tn)
 
#define p6   (tp + 5 * tn)
 
#define p6n   n3
 
#define p6n   n3
 
#define p7   (tp + 6 * tn)
 
#define p7   (tp + 6 * tn)
 
#define p7n   n4
 
#define p7n   n4
 
#define s1   (dst->a00)
 
#define s1   (dst->a00)
 
#define s2   (dst->a01)
 
#define s2   (dst->a01)
 
#define s3   (dst->a10)
 
#define s3   (dst->a10)
 
#define s4   (dst->a11)
 
#define s4   (dst->a11)
 
#define t1   (dst->a00 + maxa)
 
#define t2   (dst->a01 + maxa)
 
#define t3   (dst->a10 + maxa)
 
#define t4   (dst->a11 + maxa)
 
#define U1   p5
 
#define U1   p5
 
#define U1n   p5n
 
#define U1n   p5n
 
#define U2   p1
 
#define U2   p1
 
#define U2n   p1n
 
#define U2n   p1n
 
#define U3   U1
 
#define U3   U1
 
#define U3n   n8
 
#define U3n   n8
 

函数

void lmmp_mat22_mul_basecase_ (lmmp_mat22_t *dst, const lmmp_mat22_t *matA, const lmmp_mat22_t *matB, mp_ptr tp, mp_size_t tn)
 计算2x2矩阵和2x2矩阵的乘积
 
int lmmp_mat22_mul_size_ (lmmp_mat22_t *dst, const lmmp_mat22_t *matA, const lmmp_mat22_t *matB, mp_size_t *tn, mp_size_t *maxa)
 计算2x2矩阵和2x2矩阵的乘积需要分配的内存
 
void lmmp_mat22_mul_strassen_ (lmmp_mat22_t *dst, const lmmp_mat22_t *matA, const lmmp_mat22_t *matB, mp_ptr tp, mp_size_t tn, mp_size_t maxa)
 计算（稠密）2x2矩阵和（稠密）2x2矩阵的乘积（STRASSEN算法）
 
void lmmp_mat22_sqr_basecase_ (lmmp_mat22_t *dst, const lmmp_mat22_t *matA, mp_ptr tp, mp_size_t tn)
 计算2x2矩阵平方
 
void lmmp_mat22_sqr_strassen_ (lmmp_mat22_t *dst, const lmmp_mat22_t *mat, mp_ptr tp, mp_size_t tn)
 计算（稠密）2x2矩阵平方（STRASSEN算法）
 

宏定义说明

◆ A11 [1/2]

#define A11   (matA->a00)

◆ A11 [2/2]

#define A11   (mat->a00)

◆ A11n [1/2]

#define A11n   (matA->n00)

◆ A11n [2/2]

#define A11n   (mat->n00)

◆ A12 [1/2]

#define A12   (matA->a01)

◆ A12 [2/2]

#define A12   (mat->a01)

◆ A12n [1/2]

#define A12n   (matA->n01)

◆ A12n [2/2]

#define A12n   (mat->n01)

◆ A21 [1/2]

#define A21   (matA->a10)

◆ A21 [2/2]

#define A21   (mat->a10)

◆ A21n [1/2]

#define A21n   (matA->n10)

◆ A21n [2/2]

#define A21n   (mat->n10)

◆ A22 [1/2]

#define A22   (matA->a11)

◆ A22 [2/2]

#define A22   (mat->a11)

◆ A22n [1/2]

#define A22n   (matA->n11)

◆ A22n [2/2]

#define A22n   (mat->n11)

◆ B11

#define B11   (matB->a00)

◆ B11n

#define B11n   (matB->n00)

◆ B12

#define B12   (matB->a01)

◆ B12n

#define B12n   (matB->n01)

◆ B21

#define B21   (matB->a10)

◆ B21n

#define B21n   (matB->n10)

◆ B22

#define B22   (matB->a11)

◆ B22n

#define B22n   (matB->n11)

◆ C11 [1/2]

#define C11   (dst->a00)

◆ C11 [2/2]

#define C11   (dst->a00)

◆ C11n [1/2]

#define C11n   (dst->n00)

◆ C11n [2/2]

#define C11n   (dst->n00)

◆ C12 [1/2]

#define C12   (dst->a01)

◆ C12 [2/2]

#define C12   (dst->a01)

◆ C12n [1/2]

#define C12n   (dst->n01)

◆ C12n [2/2]

#define C12n   (dst->n01)

◆ C21 [1/2]

#define C21   (dst->a10)

◆ C21 [2/2]

#define C21   (dst->a10)

◆ C21n [1/2]

#define C21n   (dst->n10)

◆ C21n [2/2]

#define C21n   (dst->n10)

◆ C22 [1/2]

#define C22   (dst->a11)

◆ C22 [2/2]

#define C22   (dst->a11)

◆ C22n [1/2]

#define C22n   (dst->n11)

◆ C22n [2/2]

#define C22n   (dst->n11)

◆ p1 [1/4]

#define p1   tp

◆ p1 [2/4]

#define p1   tp

◆ p1 [3/4]

#define p1   (tp)

◆ p1 [4/4]

#define p1   (tp)

◆ p1n [1/2]

#define p1n   n1

◆ p1n [2/2]

#define p1n   n1

◆ p2 [1/4]

#define p2   tp + tn

◆ p2 [2/4]

#define p2   tp + tn

◆ p2 [3/4]

#define p2   (tp + tn)

◆ p2 [4/4]

#define p2   (tp + tn)

◆ p2n [1/2]

#define p2n   n5

◆ p2n [2/2]

#define p2n   n5

◆ p3 [1/2]

#define p3   (tp + 2 * tn)

◆ p3 [2/2]

#define p3   (tp + 2 * tn)

◆ p3n [1/2]

#define p3n   n2

◆ p3n [2/2]

#define p3n   n2

◆ p4 [1/2]

#define p4   (tp + 3 * tn)

◆ p4 [2/2]

#define p4   (tp + 3 * tn)

◆ p4n [1/2]

#define p4n   n7

◆ p4n [2/2]

#define p4n   n7

◆ p5 [1/2]

#define p5   (tp + 4 * tn)

◆ p5 [2/2]

#define p5   (tp + 4 * tn)

◆ p5n [1/2]

#define p5n   n6

◆ p5n [2/2]

#define p5n   n6

◆ p6 [1/2]

#define p6   (tp + 5 * tn)

◆ p6 [2/2]

#define p6   (tp + 5 * tn)

◆ p6n [1/2]

#define p6n   n3

◆ p6n [2/2]

#define p6n   n3

◆ p7 [1/2]

#define p7   (tp + 6 * tn)

◆ p7 [2/2]

#define p7   (tp + 6 * tn)

◆ p7n [1/2]

#define p7n   n4

◆ p7n [2/2]

#define p7n   n4

◆ s1 [1/2]

#define s1   (dst->a00)

◆ s1 [2/2]

#define s1   (dst->a00)

◆ s2 [1/2]

#define s2   (dst->a01)

◆ s2 [2/2]

#define s2   (dst->a01)

◆ s3 [1/2]

#define s3   (dst->a10)

◆ s3 [2/2]

#define s3   (dst->a10)

◆ s4 [1/2]

#define s4   (dst->a11)

◆ s4 [2/2]

#define s4   (dst->a11)

◆ t1

#define t1   (dst->a00 + maxa)

◆ t2

#define t2   (dst->a01 + maxa)

◆ t3

#define t3   (dst->a10 + maxa)

◆ t4

#define t4   (dst->a11 + maxa)

◆ U1 [1/2]

#define U1   p5

◆ U1 [2/2]

#define U1   p5

◆ U1n [1/2]

#define U1n   p5n

◆ U1n [2/2]

#define U1n   p5n

◆ U2 [1/2]

#define U2   p1

◆ U2 [2/2]

#define U2   p1

◆ U2n [1/2]

#define U2n   p1n

◆ U2n [2/2]

#define U2n   p1n

◆ U3 [1/2]

#define U3   U1

◆ U3 [2/2]

#define U3   U1

◆ U3n [1/2]

#define U3n   n8

◆ U3n [2/2]

#define U3n   n8

函数说明

◆ lmmp_mat22_mul_basecase_()

void lmmp_mat22_mul_basecase_ ( lmmp_mat22_t *  dst,
const lmmp_mat22_t *  matA,
const lmmp_mat22_t *  matB,
mp_ptr  tp,
mp_size_t  tn 
)

计算2x2矩阵和2x2矩阵的乘积

参数
dst结果矩阵。
matA矩阵A
matB矩阵B
tp临时缓冲区，用于存储中间结果，需要分配2*tn个limb，若为NULL，则会自动分配。
tn缓冲区的limb长度
警告
dst!=NULL, nonull(dst), [matA|matB]!=NULL, nonull([matA|matB]), sep(dst,[matA|matB]), tn>0

在文件 mat22_mul.c 第 83 行定义.

89 {
90 lmmp_param_assert(matA != NULL && matB != NULL && dst != NULL);
91 lmmp_param_assert(tn > 0);
92 if (matA == matB) {
93 lmmp_mat22_sqr_basecase_(dst, matA, tp, tn);
94 return;
95 }
96 TEMP_DECL;
97 if (tp == NULL)
98 tp = TALLOC_TYPE(tn * 2, mp_limb_t);
99#define p1 tp
100#define p2 tp + tn
101 mp_ssize_t pn1, pn2;
102 pn1 = lmmp_mul_signed_(p1, matA->a00, matA->n00, matB->a00, matB->n00);
103 pn2 = lmmp_mul_signed_(p2, matA->a01, matA->n01, matB->a10, matB->n10);
104 dst->n00 = lmmp_add_signed_(dst->a00, p1, pn1, p2, pn2);
105 pn1 = lmmp_mul_signed_(p1, matA->a00, matA->n00, matB->a01, matB->n01);
106 pn2 = lmmp_mul_signed_(p2, matA->a01, matA->n01, matB->a11, matB->n11);
107 dst->n01 = lmmp_add_signed_(dst->a01, p1, pn1, p2, pn2);
108 pn1 = lmmp_mul_signed_(p1, matA->a10, matA->n10, matB->a00, matB->n00);
109 pn2 = lmmp_mul_signed_(p2, matA->a11, matA->n11, matB->a10, matB->n10);
110 dst->n10 = lmmp_add_signed_(dst->a10, p1, pn1, p2, pn2);
111 pn1 = lmmp_mul_signed_(p1, matA->a10, matA->n10, matB->a01, matB->n01);
112 pn2 = lmmp_mul_signed_(p2, matA->a11, matA->n11, matB->a11, matB->n11);
113 dst->n11 = lmmp_add_signed_(dst->a11, p1, pn1, p2, pn2);
114#undef p1
115#undef p2
116 TEMP_FREE;
117}
mp_ssize_t
int64_t mp_ssize_t
Definition lmmp.h:214
mp_limb_t
uint64_t mp_limb_t
Definition lmmp.h:211
lmmp_param_assert
#define lmmp_param_assert(x)
Definition lmmp.h:398
p2
#define p2
lmmp_mat22_sqr_basecase_
void lmmp_mat22_sqr_basecase_(lmmp_mat22_t *dst, const lmmp_mat22_t *matA, mp_ptr tp, mp_size_t tn)
计算2x2矩阵平方
Definition mat22_mul.c:119
p1
#define p1
lmmp_mat22_t::a01
mp_ptr a01
Definition matrix.h:60
lmmp_mat22_t::n10
mp_ssize_t n10
Definition matrix.h:65
lmmp_mat22_t::a11
mp_ptr a11
Definition matrix.h:62
lmmp_mat22_t::n11
mp_ssize_t n11
Definition matrix.h:66
lmmp_mat22_t::n01
mp_ssize_t n01
Definition matrix.h:64
lmmp_mat22_t::n00
mp_ssize_t n00
Definition matrix.h:63
lmmp_mat22_t::a00
mp_ptr a00
Definition matrix.h:59
lmmp_mat22_t::a10
mp_ptr a10
Definition matrix.h:61
tp
#define tp
lmmp_add_signed_
static mp_ssize_t lmmp_add_signed_(mp_ptr dst, mp_srcptr numa, mp_ssize_t na, mp_srcptr numb, mp_ssize_t nb)
计算带符号数的加法
Definition signed.h:38
lmmp_mul_signed_
static mp_ssize_t lmmp_mul_signed_(mp_ptr dst, mp_srcptr numa, mp_ssize_t na, mp_srcptr numb, mp_ssize_t nb)
计算带符号数的乘法
Definition signed.h:146
TEMP_DECL
#define TEMP_DECL
Definition tmp_alloc.h:72
TEMP_FREE
#define TEMP_FREE
Definition tmp_alloc.h:93
TALLOC_TYPE
#define TALLOC_TYPE(n, type)
Definition tmp_alloc.h:91

引用了 lmmp_mat22_t::a00, lmmp_mat22_t::a01, lmmp_mat22_t::a10, lmmp_mat22_t::a11, lmmp_add_signed_(), lmmp_mat22_sqr_basecase_(), lmmp_mul_signed_(), lmmp_param_assert, lmmp_mat22_t::n00, lmmp_mat22_t::n01, lmmp_mat22_t::n10, lmmp_mat22_t::n11, p1, p2, TALLOC_TYPE, TEMP_DECL, TEMP_FREE , 以及 tp.

被这些函数引用 lmmp_mat22_mul_().

+ 函数调用图:
+ 这是这个函数的调用关系图:

◆ lmmp_mat22_mul_size_()

int lmmp_mat22_mul_size_ ( lmmp_mat22_t *  dst,
const lmmp_mat22_t *  matA,
const lmmp_mat22_t *  matB,
mp_size_t *  tn,
mp_size_t *  maxa 
)

计算2x2矩阵和2x2矩阵的乘积需要分配的内存

参数
dst结果矩阵，dst中的n将会被覆盖为对应位置需要的limb长度，此函数不分配内存。
matA矩阵A
matB矩阵B
tn输出参数，将会被覆盖为缓冲区需要的limb长度，正数
maxa如果被覆盖，即matA中最大的元素的limb长度+1，此参数只有当确认使用STRASSEN算法时才需要
警告
dst!=NULL, [matA|matB]!=NULL, nonull([matA|matB]), sep(dst,[matA|matB]), tn!=NULL, maxa!=NULL
注解
如果你可以确认一定不使用STRASSEN算法，则不需要maxa参数，其可以为NULL。
返回
0表示选择basecase算法，1表示选择STRASSEN算法。

在文件 mat22_mul.c 第 13 行定义.

19 {
20 lmmp_param_assert(matA!= NULL && matB!= NULL && dst!= NULL);
21 lmmp_param_assert(tn != NULL);
22 if (matA == matB) {
23 mp_ssize_t A00 = LMMP_ABS(matA->n00);
24 mp_ssize_t A01 = LMMP_ABS(matA->n01);
25 mp_ssize_t A10 = LMMP_ABS(matA->n10);
26 mp_ssize_t A11 = LMMP_ABS(matA->n11);
27 if (A00 < MAT22_SQR_STRASSEN_THRESHOLD || A01 < MAT22_SQR_STRASSEN_THRESHOLD ||
28 A10 < MAT22_SQR_STRASSEN_THRESHOLD || A11 < MAT22_SQR_STRASSEN_THRESHOLD ) {
29 dst->n00 = LMMP_MAX((A00 + A00), (A01 + A10));
30 dst->n01 = LMMP_MAX((A00 + A01), (A01 + A11));
31 dst->n10 = LMMP_MAX((A10 + A00), (A11 + A10));
32 dst->n11 = LMMP_MAX((A10 + A01), (A11 + A11));
33 *tn = LMMP_MAX(LMMP_MAX(LMMP_MAX(dst->n00, dst->n01), dst->n10), dst->n11) + 1;
34 ++(dst->n00);
35 ++(dst->n01);
36 ++(dst->n10);
37 ++(dst->n11);
38 return 0;
39 } else {
40 *maxa = LMMP_MAX(LMMP_MAX(LMMP_MAX(A00, A01), A10), A11) + 1;
41 *tn = (*maxa << 1) + 1;
42 dst->n00 = *tn;
43 dst->n01 = *tn;
44 dst->n10 = *tn;
45 dst->n11 = *tn;
46 return 1;
47 }
48 } else {
49 mp_ssize_t A00 = LMMP_ABS(matA->n00);
50 mp_ssize_t A01 = LMMP_ABS(matA->n01);
51 mp_ssize_t A10 = LMMP_ABS(matA->n10);
52 mp_ssize_t A11 = LMMP_ABS(matA->n11);
53 mp_ssize_t B00 = LMMP_ABS(matB->n00);
54 mp_ssize_t B01 = LMMP_ABS(matB->n01);
55 mp_ssize_t B10 = LMMP_ABS(matB->n10);
56 mp_ssize_t B11 = LMMP_ABS(matB->n11);
57 if (A00 < MAT22_MUL_STRASSEN_THRESHOLD || A01 < MAT22_MUL_STRASSEN_THRESHOLD ||
58 A10 < MAT22_MUL_STRASSEN_THRESHOLD || A11 < MAT22_MUL_STRASSEN_THRESHOLD ||
59 B00 < MAT22_MUL_STRASSEN_THRESHOLD || B01 < MAT22_MUL_STRASSEN_THRESHOLD ||
60 B10 < MAT22_MUL_STRASSEN_THRESHOLD || B11 < MAT22_MUL_STRASSEN_THRESHOLD) {
61 dst->n00 = LMMP_MAX((A00 + B00), (A01 + B10));
62 dst->n01 = LMMP_MAX((A00 + B01), (A01 + B11));
63 dst->n10 = LMMP_MAX((A10 + B00), (A11 + B10));
64 dst->n11 = LMMP_MAX((A10 + B01), (A11 + B11));
65 *tn = LMMP_MAX(LMMP_MAX(LMMP_MAX(dst->n00, dst->n01), dst->n10), dst->n11);
66 ++(dst->n00);
67 ++(dst->n01);
68 ++(dst->n10);
69 ++(dst->n11);
70 return 0;
71 } else {
72 *maxa = LMMP_MAX(LMMP_MAX(LMMP_MAX(A00, A01), A10), A11) + 1;
73 *tn = *maxa + LMMP_MAX(LMMP_MAX(LMMP_MAX(B00, B01), B10), B11) + 1;
74 dst->n00 = *tn;
75 dst->n01 = *tn;
76 dst->n10 = *tn;
77 dst->n11 = *tn;
78 return 1;
79 }
80 }
81}
LMMP_MAX
#define LMMP_MAX(h, i)
Definition lmmp.h:350
LMMP_ABS
#define LMMP_ABS(x)
Definition lmmp.h:346
A11
#define A11
B11
#define B11
MAT22_SQR_STRASSEN_THRESHOLD
#define MAT22_SQR_STRASSEN_THRESHOLD
Definition mparam.h:101
MAT22_MUL_STRASSEN_THRESHOLD
#define MAT22_MUL_STRASSEN_THRESHOLD
Definition mparam.h:98

引用了 A11, B11, LMMP_ABS, LMMP_MAX, lmmp_param_assert, MAT22_MUL_STRASSEN_THRESHOLD, MAT22_SQR_STRASSEN_THRESHOLD, lmmp_mat22_t::n00, lmmp_mat22_t::n01, lmmp_mat22_t::n10 , 以及 lmmp_mat22_t::n11.

◆ lmmp_mat22_mul_strassen_()

void lmmp_mat22_mul_strassen_ ( lmmp_mat22_t *  dst,
const lmmp_mat22_t *  matA,
const lmmp_mat22_t *  matB,
mp_ptr  tp,
mp_size_t  tn,
mp_size_t  maxa 
)

计算（稠密）2x2矩阵和（稠密）2x2矩阵的乘积（STRASSEN算法）

参数
dst结果矩阵。
matA矩阵A
matB矩阵B
tp临时缓冲区，用于存储中间结果，需要分配7*(tn+1)个limb，若为NULL，则会自动分配。
tn缓冲区的limb长度
maxamatA中最大的元素的limb长度+1，建议由lmmp_mat22_mul_size_确定
警告
dst!=NULL, nonull(dst), [matA|matB]!=NULL, nonull([matA|matB]), sep(dst,[matA|matB]), tn>0

在文件 mat22_mul.c 第 194 行定义.

201 {
202 lmmp_param_assert(matA != NULL && matB != NULL && dst != NULL);
203 lmmp_param_assert(tn > 0 && maxa > 0);
204 if (matA == matB) {
205 lmmp_mat22_sqr_strassen_(dst, matA, tp, tn);
206 return;
207 }
208 TEMP_B_DECL;
209 ++tn;
210 if (tp == NULL)
211 tp = BALLOC_TYPE(tn * 7, mp_limb_t);
212
213#define A11 (matA->a00)
214#define A12 (matA->a01)
215#define A21 (matA->a10)
216#define A22 (matA->a11)
217#define B11 (matB->a00)
218#define B12 (matB->a01)
219#define B21 (matB->a10)
220#define B22 (matB->a11)
221#define A11n (matA->n00)
222#define A12n (matA->n01)
223#define A21n (matA->n10)
224#define A22n (matA->n11)
225#define B11n (matB->n00)
226#define B12n (matB->n01)
227#define B21n (matB->n10)
228#define B22n (matB->n11)
229
230#define s1 (dst->a00)
231#define s2 (dst->a01)
232#define s3 (dst->a10)
233#define s4 (dst->a11)
234#define t1 (dst->a00 + maxa)
235#define t2 (dst->a01 + maxa)
236#define t3 (dst->a10 + maxa)
237#define t4 (dst->a11 + maxa)
238#define p1 (tp)
239#define p2 (tp + tn)
240#define p3 (tp + 2 * tn)
241#define p4 (tp + 3 * tn)
242#define p5 (tp + 4 * tn)
243#define p6 (tp + 5 * tn)
244#define p7 (tp + 6 * tn)
245 mp_ssize_t n1, n2, n3, n4, n5, n6, n7, n8;
246 n1 = lmmp_add_signed_(s1, A22, A22n, A12, A12n);
247 n2 = lmmp_add_signed_(s2, A22, A22n, A21, -A21n);
248 n3 = lmmp_add_signed_(s3, s2, n2, A12, A12n);
249 n4 = lmmp_add_signed_(s4, s3, n3, A11, -A11n);
250 n5 = lmmp_add_signed_(t1, B22, B22n, B12, B12n);
251 n6 = lmmp_add_signed_(t2, B22, B22n, B21, -B21n);
252 n7 = lmmp_add_signed_(t3, t2, n6, B12, B12n);
253 n8 = lmmp_add_signed_(t4, t3, n7, B11, -B11n);
254
255 n1 = lmmp_mul_signed_(p1, s1, n1, t1, n5);
256 n5 = lmmp_mul_signed_(p2, s2, n2, t2, n6);
257 n2 = lmmp_mul_signed_(p3, s3, n3, t3, n7);
258 n7 = lmmp_mul_signed_(p4, A11, A11n, B11, B11n);
259 n6 = lmmp_mul_signed_(p5, A12, A12n, B21, B21n);
260 n3 = lmmp_mul_signed_(p6, s4, n4, B12, B12n);
261 n4 = lmmp_mul_signed_(p7, A21, A21n, t4, n8);
262
263#undef s1
264#undef s2
265#undef s3
266#undef s4
267#undef t1
268#undef t2
269#undef t3
270#undef t4
271
272#define p1n n1
273#define p2n n5
274#define p3n n2
275#define p4n n7
276#define p5n n6
277#define p6n n3
278#define p7n n4
279
280#undef A11
281#undef A12
282#undef A21
283#undef A22
284#undef B11
285#undef B12
286#undef B21
287#undef B22
288#undef A11n
289#undef A12n
290#undef A21n
291#undef A22n
292#undef B11n
293#undef B12n
294#undef B21n
295#undef B22n
296
297#define C11 (dst->a00)
298#define C12 (dst->a01)
299#define C21 (dst->a10)
300#define C22 (dst->a11)
301#define C11n (dst->n00)
302#define C12n (dst->n01)
303#define C21n (dst->n10)
304#define C22n (dst->n11)
305
306 C11n = lmmp_add_signed_(C11, p4, p4n, p5, p5n);
307#define U1 p5 // U1 = p3 + p5
308#define U2 p1 // U2 = p1 - U1
309#define U3 U1 // U3 = U1 - p2
310#define U1n p5n
311#define U2n p1n
312#define U3n n8
313 U1n = lmmp_add_signed_(U1, p3, p3n, p5, p5n);
314 U2n = lmmp_add_signed_(U2, p1, p1n, U1, -U1n);
315 U3n = lmmp_add_signed_(U3, U1, U1n, p2, -p2n);
316
317 C12n = lmmp_add_signed_(C12, U3, U3n, p6, -p6n);
318 C21n = lmmp_add_signed_(C21, U2, U2n, p7, -p7n);
319 C22n = lmmp_add_signed_(C22, p2, p2n, U2, U2n);
320 TEMP_B_FREE;
321
322#undef C11
323#undef C12
324#undef C21
325#undef C22
326#undef C11n
327#undef C12n
328#undef C21n
329#undef C22n
330#undef U1
331#undef U2
332#undef U3
333#undef U1n
334#undef U2n
335#undef U3n
336
337#undef p1
338#undef p2
339#undef p3
340#undef p4
341#undef p5
342#undef p6
343#undef p7
344}
s1
#define s1
C11n
#define C11n
s4
#define s4
C22n
#define C22n
U2n
#define U2n
s3
#define s3
p6n
#define p6n
s2
#define s2
C12n
#define C12n
t4
#define t4
p6
#define p6
A22n
#define A22n
B12n
#define B12n
B21
#define B21
t1
#define t1
B21n
#define B21n
A22
#define A22
B12
#define B12
t3
#define t3
lmmp_mat22_sqr_strassen_
void lmmp_mat22_sqr_strassen_(lmmp_mat22_t *dst, const lmmp_mat22_t *mat, mp_ptr tp, mp_size_t tn)
计算（稠密）2x2矩阵平方（STRASSEN算法）
Definition mat22_mul.c:346
U3n
#define U3n
A11n
#define A11n
A12
#define A12
C12
#define C12
U3
#define U3
p1n
#define p1n
B22n
#define B22n
A21n
#define A21n
p4n
#define p4n
A21
#define A21
t2
#define t2
U1
#define U1
C21n
#define C21n
C22
#define C22
A12n
#define A12n
B22
#define B22
p3
#define p3
p3n
#define p3n
C21
#define C21
C11
#define C11
p5n
#define p5n
p7
#define p7
U1n
#define U1n
p7n
#define p7n
U2
#define U2
p2n
#define p2n
p5
#define p5
p4
#define p4
B11n
#define B11n
TEMP_B_DECL
#define TEMP_B_DECL
Definition tmp_alloc.h:75
BALLOC_TYPE
#define BALLOC_TYPE(n, type)
Definition tmp_alloc.h:89
TEMP_B_FREE
#define TEMP_B_FREE
Definition tmp_alloc.h:100

引用了 A11, A11n, A12, A12n, A21, A21n, A22, A22n, B11, B11n, B12, B12n, B21, B21n, B22, B22n, BALLOC_TYPE, C11, C11n, C12, C12n, C21, C21n, C22, C22n, lmmp_add_signed_(), lmmp_mat22_sqr_strassen_(), lmmp_mul_signed_(), lmmp_param_assert, p1, p1n, p2, p2n, p3, p3n, p4, p4n, p5, p5n, p6, p6n, p7, p7n, s1, s2, s3, s4, t1, t2, t3, t4, TEMP_B_DECL, TEMP_B_FREE, tp, U1, U1n, U2, U2n, U3 , 以及 U3n.

被这些函数引用 lmmp_mat22_mul_().

+ 函数调用图:
+ 这是这个函数的调用关系图:

◆ lmmp_mat22_sqr_basecase_()

void lmmp_mat22_sqr_basecase_ ( lmmp_mat22_t *  dst,
const lmmp_mat22_t *  matA,
mp_ptr  tp,
mp_size_t  tn 
)

计算2x2矩阵平方

参数
dst结果矩阵。
matA矩阵A
tp临时缓冲区，用于存储中间结果，需要分配2*tn个limb，若为NULL，则会自动分配。
tn缓冲区的limb长度
maxamatA中最大的元素的limb长度+1，建议由lmmp_mat22_mul_size_确定
警告
dst!=NULL, nonull(dst), matA!=NULL, nonull(matA), sep(dst,matA), tn>0

在文件 mat22_mul.c 第 119 行定义.

124 {
125 TEMP_DECL;
126 if (tp == NULL)
127 tp = TALLOC_TYPE(tn * 2, mp_limb_t);
128#define p1 tp
129#define p2 tp + tn
130 mp_ssize_t pn1, pn2;
131 pn1 = lmmp_sqr_signed_(p1, matA->a00, matA->n00);
132 pn2 = lmmp_mul_signed_(p2, matA->a01, matA->n01, matA->a10, matA->n10);
133 dst->n00 = lmmp_add_signed_(dst->a00, p1, pn1, p2, pn2);
134 pn1 = lmmp_mul_signed_(p1, matA->a00, matA->n00, matA->a01, matA->n01);
135 pn2 = lmmp_mul_signed_(p2, matA->a01, matA->n01, matA->a11, matA->n11);
136 dst->n01 = lmmp_add_signed_(dst->a01, p1, pn1, p2, pn2);
137 pn1 = lmmp_mul_signed_(p1, matA->a10, matA->n10, matA->a00, matA->n00);
138 pn2 = lmmp_mul_signed_(p2, matA->a11, matA->n11, matA->a10, matA->n10);
139 dst->n10 = lmmp_add_signed_(dst->a10, p1, pn1, p2, pn2);
140 pn1 = lmmp_mul_signed_(p1, matA->a10, matA->n10, matA->a01, matA->n01);
141 pn2 = lmmp_sqr_signed_(p2, matA->a11, matA->n11);
142 dst->n11 = lmmp_add_signed_(dst->a11, p1, pn1, p2, pn2);
143#undef p1
144#undef p2
145 TEMP_FREE;
146}
lmmp_sqr_signed_
static mp_ssize_t lmmp_sqr_signed_(mp_ptr dst, mp_srcptr numa, mp_ssize_t na)
计算带符号数的平方
Definition signed.h:171

引用了 lmmp_mat22_t::a00, lmmp_mat22_t::a01, lmmp_mat22_t::a10, lmmp_mat22_t::a11, lmmp_add_signed_(), lmmp_mul_signed_(), lmmp_sqr_signed_(), lmmp_mat22_t::n00, lmmp_mat22_t::n01, lmmp_mat22_t::n10, lmmp_mat22_t::n11, p1, p2, TALLOC_TYPE, TEMP_DECL, TEMP_FREE , 以及 tp.

被这些函数引用 lmmp_mat22_mul_basecase_() , 以及 lmmp_mat22_sqr_().

+ 函数调用图:
+ 这是这个函数的调用关系图:

◆ lmmp_mat22_sqr_strassen_()

void lmmp_mat22_sqr_strassen_ ( lmmp_mat22_t *  dst,
const lmmp_mat22_t *  matA,
mp_ptr  tp,
mp_size_t  tn 
)

计算（稠密）2x2矩阵平方（STRASSEN算法）

参数
dst结果矩阵。
matA矩阵A
tp临时缓冲区，用于存储中间结果，需要分配7*(tn+1)个limb，若为NULL，则会自动分配。
tn缓冲区的limb长度
警告
dst!=NULL, nonull(dst), matA!=NULL, nonull(matA), sep(dst,matA), tn>0

在文件 mat22_mul.c 第 346 行定义.

346 {
347 lmmp_param_assert(mat != NULL && dst != NULL);
348 TEMP_B_DECL;
349 ++tn;
350 if (tp == NULL)
351 tp = BALLOC_TYPE(tn * 7, mp_limb_t);
352
353#define A11 (mat->a00)
354#define A12 (mat->a01)
355#define A21 (mat->a10)
356#define A22 (mat->a11)
357#define A11n (mat->n00)
358#define A12n (mat->n01)
359#define A21n (mat->n10)
360#define A22n (mat->n11)
361
362#define s1 (dst->a00)
363#define s2 (dst->a01)
364#define s3 (dst->a10)
365#define s4 (dst->a11)
366#define p1 (tp)
367#define p2 (tp + tn)
368#define p3 (tp + 2 * tn)
369#define p4 (tp + 3 * tn)
370#define p5 (tp + 4 * tn)
371#define p6 (tp + 5 * tn)
372#define p7 (tp + 6 * tn)
373 mp_ssize_t n1, n2, n3, n4, n5, n6, n7, n8;
374 n1 = lmmp_add_signed_(s1, A22, A22n, A12, A12n);
375 n2 = lmmp_add_signed_(s2, A22, A22n, A21, -A21n);
376 n3 = lmmp_add_signed_(s3, s2, n2, A12, A12n);
377 n4 = lmmp_add_signed_(s4, s3, n3, A11, -A11n);
378
379 n1 = lmmp_sqr_signed_(p1, s1, n1);
380 n5 = lmmp_sqr_signed_(p2, s2, n2);
381 n2 = lmmp_sqr_signed_(p3, s3, n3);
382 n7 = lmmp_sqr_signed_(p4, A11, A11n);
383 n6 = lmmp_mul_signed_(p5, A12, A12n, A21, A21n);
384 n3 = lmmp_mul_signed_(p6, s4, n4, A12, A12n);
385 n4 = lmmp_mul_signed_(p7, A21, A21n, s4, n4);
386
387#undef s1
388#undef s2
389#undef s3
390#undef s4
391
392#define p1n n1
393#define p2n n5
394#define p3n n2
395#define p4n n7
396#define p5n n6
397#define p6n n3
398#define p7n n4
399
400#undef A11
401#undef A12
402#undef A21
403#undef A22
404#undef A11n
405#undef A12n
406#undef A21n
407#undef A22n
408
409#define C11 (dst->a00)
410#define C12 (dst->a01)
411#define C21 (dst->a10)
412#define C22 (dst->a11)
413#define C11n (dst->n00)
414#define C12n (dst->n01)
415#define C21n (dst->n10)
416#define C22n (dst->n11)
417
418 C11n = lmmp_add_signed_(C11, p4, p4n, p5, p5n);
419#define U1 p5 // U1 = p3 + p5
420#define U2 p1 // U2 = p1 - U1
421#define U3 U1 // U3 = U1 - p2
422#define U1n p5n
423#define U2n p1n
424#define U3n n8
425 U1n = lmmp_add_signed_(U1, p3, p3n, p5, p5n);
426 U2n = lmmp_add_signed_(U2, p1, p1n, U1, -U1n);
427 U3n = lmmp_add_signed_(U3, U1, U1n, p2, -p2n);
428
429 C12n = lmmp_add_signed_(C12, U3, U3n, p6, -p6n);
430 C21n = lmmp_add_signed_(C21, U2, U2n, p7, -p7n);
431 C22n = lmmp_add_signed_(C22, p2, p2n, U2, U2n);
432 TEMP_B_FREE;
433
434#undef C11
435#undef C12
436#undef C21
437#undef C22
438#undef C11n
439#undef C12n
440#undef C21n
441#undef C22n
442#undef U1
443#undef U2
444#undef U3
445#undef U1n
446#undef U2n
447#undef U3n
448
449#undef p1
450#undef p2
451#undef p3
452#undef p4
453#undef p5
454#undef p6
455#undef p7
456}

引用了 A11, A11n, A12, A12n, A21, A21n, A22, A22n, BALLOC_TYPE, C11, C11n, C12, C12n, C21, C21n, C22, C22n, lmmp_add_signed_(), lmmp_mul_signed_(), lmmp_param_assert, lmmp_sqr_signed_(), p1, p1n, p2, p2n, p3, p3n, p4, p4n, p5, p5n, p6, p6n, p7, p7n, s1, s2, s3, s4, TEMP_B_DECL, TEMP_B_FREE, tp, U1, U1n, U2, U2n, U3 , 以及 U3n.

被这些函数引用 lmmp_mat22_mul_strassen_() , 以及 lmmp_mat22_sqr_().

+ 函数调用图:
+ 这是这个函数的调用关系图: