Lines Matching refs:zz
236 PetscErrorCode MatMult_SeqBAIJ_1(Mat A, Vec xx, Vec zz) in MatMult_SeqBAIJ_1() argument
248 PetscCall(VecGetArrayWrite(zz, &z)); in MatMult_SeqBAIJ_1()
276 PetscCall(VecRestoreArrayWrite(zz, &z)); in MatMult_SeqBAIJ_1()
281 PetscErrorCode MatMult_SeqBAIJ_2(Mat A, Vec xx, Vec zz) in MatMult_SeqBAIJ_2() argument
293 PetscCall(VecGetArrayWrite(zz, &zarray)); in MatMult_SeqBAIJ_2()
329 PetscCall(VecRestoreArrayWrite(zz, &zarray)); in MatMult_SeqBAIJ_2()
334 PetscErrorCode MatMult_SeqBAIJ_3(Mat A, Vec xx, Vec zz) in MatMult_SeqBAIJ_3() argument
349 PetscCall(VecGetArrayWrite(zz, &zarray)); in MatMult_SeqBAIJ_3()
390 PetscCall(VecRestoreArrayWrite(zz, &zarray)); in MatMult_SeqBAIJ_3()
395 PetscErrorCode MatMult_SeqBAIJ_4(Mat A, Vec xx, Vec zz) in MatMult_SeqBAIJ_4() argument
406 PetscCall(VecGetArrayWrite(zz, &zarray)); in MatMult_SeqBAIJ_4()
451 PetscCall(VecRestoreArrayWrite(zz, &zarray)); in MatMult_SeqBAIJ_4()
456 PetscErrorCode MatMult_SeqBAIJ_5(Mat A, Vec xx, Vec zz) in MatMult_SeqBAIJ_5() argument
468 PetscCall(VecGetArrayWrite(zz, &zarray)); in MatMult_SeqBAIJ_5()
516 PetscCall(VecRestoreArrayWrite(zz, &zarray)); in MatMult_SeqBAIJ_5()
521 PetscErrorCode MatMult_SeqBAIJ_6(Mat A, Vec xx, Vec zz) in MatMult_SeqBAIJ_6() argument
533 PetscCall(VecGetArrayWrite(zz, &zarray)); in MatMult_SeqBAIJ_6()
587 PetscCall(VecRestoreArrayWrite(zz, &zarray)); in MatMult_SeqBAIJ_6()
592 PetscErrorCode MatMult_SeqBAIJ_7(Mat A, Vec xx, Vec zz) in MatMult_SeqBAIJ_7() argument
604 PetscCall(VecGetArrayWrite(zz, &zarray)); in MatMult_SeqBAIJ_7()
662 PetscCall(VecRestoreArrayWrite(zz, &zarray)); in MatMult_SeqBAIJ_7()
668 PetscErrorCode MatMult_SeqBAIJ_9_AVX2(Mat A, Vec xx, Vec zz) in MatMult_SeqBAIJ_9_AVX2() argument
686 PetscCall(VecGetArrayWrite(zz, &zarray)); in MatMult_SeqBAIJ_9_AVX2()
813 PetscCall(VecRestoreArrayWrite(zz, &zarray)); in MatMult_SeqBAIJ_9_AVX2()
819 PetscErrorCode MatMult_SeqBAIJ_11(Mat A, Vec xx, Vec zz) in MatMult_SeqBAIJ_11() argument
832 PetscCall(VecGetArrayWrite(zz, &zarray)); in MatMult_SeqBAIJ_11()
897 PetscCall(VecRestoreArrayWrite(zz, &zarray)); in MatMult_SeqBAIJ_11()
903 PetscErrorCode MatMult_SeqBAIJ_12_ver1(Mat A, Vec xx, Vec zz) in MatMult_SeqBAIJ_12_ver1() argument
916 PetscCall(VecGetArrayWrite(zz, &zarray)); in MatMult_SeqBAIJ_12_ver1()
982 PetscCall(VecRestoreArrayWrite(zz, &zarray)); in MatMult_SeqBAIJ_12_ver1()
987 PetscErrorCode MatMultAdd_SeqBAIJ_12_ver1(Mat A, Vec xx, Vec yy, Vec zz) in MatMultAdd_SeqBAIJ_12_ver1() argument
1000 PetscCall(VecGetArrayPair(yy, zz, &yarray, &zarray)); in MatMultAdd_SeqBAIJ_12_ver1()
1004 if (zz != yy) PetscCall(PetscArraycpy(zarray, yarray, 12 * mbs)); in MatMultAdd_SeqBAIJ_12_ver1()
1074 PetscCall(VecRestoreArrayPair(yy, zz, &yarray, &zarray)); in MatMultAdd_SeqBAIJ_12_ver1()
1080 PetscErrorCode MatMult_SeqBAIJ_12_ver2(Mat A, Vec xx, Vec zz) in MatMult_SeqBAIJ_12_ver2() argument
1093 PetscCall(VecGetArrayWrite(zz, &zarray)); in MatMult_SeqBAIJ_12_ver2()
1186 PetscCall(VecRestoreArrayWrite(zz, &zarray)); in MatMult_SeqBAIJ_12_ver2()
1192 PetscErrorCode MatMultAdd_SeqBAIJ_12_ver2(Mat A, Vec xx, Vec yy, Vec zz) in MatMultAdd_SeqBAIJ_12_ver2() argument
1205 PetscCall(VecGetArrayPair(yy, zz, &yarray, &zarray)); in MatMultAdd_SeqBAIJ_12_ver2()
1209 if (zz != yy) PetscCall(PetscArraycpy(zarray, yarray, 12 * mbs)); in MatMultAdd_SeqBAIJ_12_ver2()
1316 PetscCall(VecRestoreArrayPair(yy, zz, &yarray, &zarray)); in MatMultAdd_SeqBAIJ_12_ver2()
1322 PetscErrorCode MatMult_SeqBAIJ_12_AVX2(Mat A, Vec xx, Vec zz) in MatMult_SeqBAIJ_12_AVX2() argument
1339 PetscCall(VecGetArrayWrite(zz, &zarray)); in MatMult_SeqBAIJ_12_AVX2()
1479 PetscCall(VecRestoreArrayWrite(zz, &zarray)); in MatMult_SeqBAIJ_12_AVX2()
1487 PetscErrorCode MatMult_SeqBAIJ_15_ver1(Mat A, Vec xx, Vec zz) in MatMult_SeqBAIJ_15_ver1() argument
1500 PetscCall(VecGetArrayWrite(zz, &zarray)); in MatMult_SeqBAIJ_15_ver1()
1577 PetscCall(VecRestoreArrayWrite(zz, &zarray)); in MatMult_SeqBAIJ_15_ver1()
1583 PetscErrorCode MatMult_SeqBAIJ_15_ver2(Mat A, Vec xx, Vec zz) in MatMult_SeqBAIJ_15_ver2() argument
1596 PetscCall(VecGetArrayWrite(zz, &zarray)); in MatMult_SeqBAIJ_15_ver2()
1738 PetscCall(VecRestoreArrayWrite(zz, &zarray)); in MatMult_SeqBAIJ_15_ver2()
1744 PetscErrorCode MatMult_SeqBAIJ_15_ver3(Mat A, Vec xx, Vec zz) in MatMult_SeqBAIJ_15_ver3() argument
1757 PetscCall(VecGetArrayWrite(zz, &zarray)); in MatMult_SeqBAIJ_15_ver3()
1864 PetscCall(VecRestoreArrayWrite(zz, &zarray)); in MatMult_SeqBAIJ_15_ver3()
1870 PetscErrorCode MatMult_SeqBAIJ_15_ver4(Mat A, Vec xx, Vec zz) in MatMult_SeqBAIJ_15_ver4() argument
1883 PetscCall(VecGetArrayWrite(zz, &zarray)); in MatMult_SeqBAIJ_15_ver4()
1972 PetscCall(VecRestoreArrayWrite(zz, &zarray)); in MatMult_SeqBAIJ_15_ver4()
1980 PetscErrorCode MatMult_SeqBAIJ_N(Mat A, Vec xx, Vec zz) in MatMult_SeqBAIJ_N() argument
1993 PetscCall(VecGetArrayWrite(zz, &zarray)); in MatMult_SeqBAIJ_N()
2029 PetscCall(VecRestoreArrayWrite(zz, &zarray)); in MatMult_SeqBAIJ_N()
2034 PetscErrorCode MatMultAdd_SeqBAIJ_1(Mat A, Vec xx, Vec yy, Vec zz) in MatMultAdd_SeqBAIJ_1() argument
2046 PetscCall(VecGetArrayPair(yy, zz, &y, &z)); in MatMultAdd_SeqBAIJ_1()
2051 if (zz != yy) PetscCall(PetscArraycpy(z, y, mbs)); in MatMultAdd_SeqBAIJ_1()
2079 PetscCall(VecRestoreArrayPair(yy, zz, &y, &z)); in MatMultAdd_SeqBAIJ_1()
2084 PetscErrorCode MatMultAdd_SeqBAIJ_2(Mat A, Vec xx, Vec yy, Vec zz) in MatMultAdd_SeqBAIJ_2() argument
2097 PetscCall(VecGetArrayPair(yy, zz, &yarray, &zarray)); in MatMultAdd_SeqBAIJ_2()
2102 if (zz != yy) PetscCall(PetscArraycpy(zarray, yarray, 2 * mbs)); in MatMultAdd_SeqBAIJ_2()
2140 PetscCall(VecRestoreArrayPair(yy, zz, &yarray, &zarray)); in MatMultAdd_SeqBAIJ_2()
2145 PetscErrorCode MatMultAdd_SeqBAIJ_3(Mat A, Vec xx, Vec yy, Vec zz) in MatMultAdd_SeqBAIJ_3() argument
2157 PetscCall(VecGetArrayPair(yy, zz, &yarray, &zarray)); in MatMultAdd_SeqBAIJ_3()
2162 if (zz != yy) PetscCall(PetscArraycpy(zarray, yarray, 3 * mbs)); in MatMultAdd_SeqBAIJ_3()
2203 PetscCall(VecRestoreArrayPair(yy, zz, &yarray, &zarray)); in MatMultAdd_SeqBAIJ_3()
2208 PetscErrorCode MatMultAdd_SeqBAIJ_4(Mat A, Vec xx, Vec yy, Vec zz) in MatMultAdd_SeqBAIJ_4() argument
2220 PetscCall(VecGetArrayPair(yy, zz, &yarray, &zarray)); in MatMultAdd_SeqBAIJ_4()
2225 if (zz != yy) PetscCall(PetscArraycpy(zarray, yarray, 4 * mbs)); in MatMultAdd_SeqBAIJ_4()
2270 PetscCall(VecRestoreArrayPair(yy, zz, &yarray, &zarray)); in MatMultAdd_SeqBAIJ_4()
2275 PetscErrorCode MatMultAdd_SeqBAIJ_5(Mat A, Vec xx, Vec yy, Vec zz) in MatMultAdd_SeqBAIJ_5() argument
2288 PetscCall(VecGetArrayPair(yy, zz, &yarray, &zarray)); in MatMultAdd_SeqBAIJ_5()
2293 if (zz != yy) PetscCall(PetscArraycpy(zarray, yarray, 5 * mbs)); in MatMultAdd_SeqBAIJ_5()
2342 PetscCall(VecRestoreArrayPair(yy, zz, &yarray, &zarray)); in MatMultAdd_SeqBAIJ_5()
2347 PetscErrorCode MatMultAdd_SeqBAIJ_6(Mat A, Vec xx, Vec yy, Vec zz) in MatMultAdd_SeqBAIJ_6() argument
2360 PetscCall(VecGetArrayPair(yy, zz, &yarray, &zarray)); in MatMultAdd_SeqBAIJ_6()
2365 if (zz != yy) PetscCall(PetscArraycpy(zarray, yarray, 6 * mbs)); in MatMultAdd_SeqBAIJ_6()
2418 PetscCall(VecRestoreArrayPair(yy, zz, &yarray, &zarray)); in MatMultAdd_SeqBAIJ_6()
2423 PetscErrorCode MatMultAdd_SeqBAIJ_7(Mat A, Vec xx, Vec yy, Vec zz) in MatMultAdd_SeqBAIJ_7() argument
2436 PetscCall(VecGetArrayPair(yy, zz, &yarray, &zarray)); in MatMultAdd_SeqBAIJ_7()
2441 if (zz != yy) PetscCall(PetscArraycpy(zarray, yarray, 7 * mbs)); in MatMultAdd_SeqBAIJ_7()
2498 PetscCall(VecRestoreArrayPair(yy, zz, &yarray, &zarray)); in MatMultAdd_SeqBAIJ_7()
2504 PetscErrorCode MatMultAdd_SeqBAIJ_9_AVX2(Mat A, Vec xx, Vec yy, Vec zz) in MatMultAdd_SeqBAIJ_9_AVX2() argument
2521 PetscCall(VecCopy(yy, zz)); in MatMultAdd_SeqBAIJ_9_AVX2()
2523 PetscCall(VecGetArray(zz, &zarray)); in MatMultAdd_SeqBAIJ_9_AVX2()
2649 PetscCall(VecRestoreArray(zz, &zarray)); in MatMultAdd_SeqBAIJ_9_AVX2()
2655 PetscErrorCode MatMultAdd_SeqBAIJ_11(Mat A, Vec xx, Vec yy, Vec zz) in MatMultAdd_SeqBAIJ_11() argument
2668 PetscCall(VecGetArrayPair(yy, zz, &yarray, &zarray)); in MatMultAdd_SeqBAIJ_11()
2673 if (zz != yy) PetscCall(PetscArraycpy(zarray, yarray, 7 * mbs)); in MatMultAdd_SeqBAIJ_11()
2746 PetscCall(VecRestoreArrayPair(yy, zz, &yarray, &zarray)); in MatMultAdd_SeqBAIJ_11()
2751 PetscErrorCode MatMultAdd_SeqBAIJ_N(Mat A, Vec xx, Vec yy, Vec zz) in MatMultAdd_SeqBAIJ_N() argument
2763 PetscCall(VecCopy(yy, zz)); in MatMultAdd_SeqBAIJ_N()
2765 PetscCall(VecGetArray(zz, &zarray)); in MatMultAdd_SeqBAIJ_N()
2800 PetscCall(VecRestoreArray(zz, &zarray)); in MatMultAdd_SeqBAIJ_N()
2805 PetscErrorCode MatMultHermitianTranspose_SeqBAIJ(Mat A, Vec xx, Vec zz) in MatMultHermitianTranspose_SeqBAIJ() argument
2810 PetscCall(VecSet(zz, zero)); in MatMultHermitianTranspose_SeqBAIJ()
2811 PetscCall(MatMultHermitianTransposeAdd_SeqBAIJ(A, xx, zz, zz)); in MatMultHermitianTranspose_SeqBAIJ()
2815 PetscErrorCode MatMultTranspose_SeqBAIJ(Mat A, Vec xx, Vec zz) in MatMultTranspose_SeqBAIJ() argument
2820 PetscCall(VecSet(zz, zero)); in MatMultTranspose_SeqBAIJ()
2821 PetscCall(MatMultTransposeAdd_SeqBAIJ(A, xx, zz, zz)); in MatMultTranspose_SeqBAIJ()
2825 PetscErrorCode MatMultHermitianTransposeAdd_SeqBAIJ(Mat A, Vec xx, Vec yy, Vec zz) in MatMultHermitianTransposeAdd_SeqBAIJ() argument
2837 if (yy != zz) PetscCall(VecCopy(yy, zz)); in MatMultHermitianTransposeAdd_SeqBAIJ()
2839 PetscCall(VecGetArray(zz, &z)); in MatMultHermitianTransposeAdd_SeqBAIJ()
2981 PetscCall(VecRestoreArray(zz, &z)); in MatMultHermitianTransposeAdd_SeqBAIJ()
2986 PetscErrorCode MatMultTransposeAdd_SeqBAIJ(Mat A, Vec xx, Vec yy, Vec zz) in MatMultTransposeAdd_SeqBAIJ() argument
2998 if (yy != zz) PetscCall(VecCopy(yy, zz)); in MatMultTransposeAdd_SeqBAIJ()
3000 PetscCall(VecGetArray(zz, &z)); in MatMultTransposeAdd_SeqBAIJ()
3139 PetscCall(VecRestoreArray(zz, &z)); in MatMultTransposeAdd_SeqBAIJ()