| /petsc/src/mat/tutorials/ |
| H A D | ex18.c | 61 for (PetscInt vj = 0; vj < 3; vj++) { in CreateMatrix() local
63 ooc[cnt++] = fe->vertices[3 * e + vj]; in CreateMatrix()
86 for (PetscInt vj = 0; vj < 3; vj++) s[3 * vi + vj] = vi + 2 * vj; in FillMatrixCPU() local
109 for (PetscInt vj = 0; vj < 3; vj++) s[3 * vi + vj] = vi + 2 * vj; in FillMatrixCPUCOO() local
130 for (PetscInt vj = 0; vj < 3; vj++) s[e][vi][vj] = vi + 2 * vj; in FillMatrixCPUCOO3d() local
|
| H A D | ex18k.kokkos.cxx | 17 for (PetscInt vj = 0; vj < 3; vj++) s[vi * 3 + vj] = vi + 2 * vj; in FillMatrixKokkosCOO()
|
| H A D | ex18cu.cu | 11 for (PetscInt vj = 0; vj < 3; vj++) s[vi * 3 + vj] = vi + 2 * vj; in FillValues() local
|
| /petsc/src/mat/impls/sbaij/seq/ |
| H A D | relax.h | 63 MatScalar vj; in MatMult_SeqSBAIJ_1_ushort() local
98 vj = v[j]; in MatMult_SeqSBAIJ_1_ushort()
99 z[ibt] += PetscConj(vj) * x1; /* (strict lower triangular part of A)*x */ in MatMult_SeqSBAIJ_1_ushort()
100 sum += vj * x[ibt]; /* (strict upper triangular part of A)*x */ in MatMult_SeqSBAIJ_1_ushort()
105 vj = v[j]; in MatMult_SeqSBAIJ_1_ushort()
106 z[ibt] += vj * x1; /* (strict lower triangular part of A)*x */ in MatMult_SeqSBAIJ_1_ushort()
107 sum += vj * x[ibt]; /* (strict upper triangular part of A)*x */ in MatMult_SeqSBAIJ_1_ushort()
135 const unsigned short *aj = a->jshort, *vj, *vj1; in MatSOR_SeqSBAIJ_ushort() local
137 const PetscInt *aj = a->j, *vj, *vj1; in MatSOR_SeqSBAIJ_ushort() local
170 vj = aj + ai[i] + 1; in MatSOR_SeqSBAIJ_ushort()
[all …]
|
| H A D | sbaijfact2.c | 15 PetscInt nz, *vj, k, idx, k1; in MatSolve_SeqSBAIJ_N_inplace() local
40 vj = aj + ai[k]; in MatSolve_SeqSBAIJ_N_inplace()
41 xj = t + (*vj) * bs; /* *vj-th block of x, *vj>k */ in MatSolve_SeqSBAIJ_N_inplace()
45 vj++; in MatSolve_SeqSBAIJ_N_inplace()
46 xj = t + (*vj) * bs; in MatSolve_SeqSBAIJ_N_inplace()
59 vj = aj + ai[k]; in MatSolve_SeqSBAIJ_N_inplace()
60 xj = t + (*vj) * bs; in MatSolve_SeqSBAIJ_N_inplace()
64 vj++; in MatSolve_SeqSBAIJ_N_inplace()
66 xj = t + (*vj) * bs; in MatSolve_SeqSBAIJ_N_inplace()
95 const PetscInt *vj, bs2 = bs * bs; in MatForwardSolve_SeqSBAIJ_N_NaturalOrdering() local
[all …]
|
| H A D | sbaijfact4.c | 11 PetscInt *ai, *aj, k, k1, jmin, jmax, *jl, *il, vj, nexti, ili; in MatCholeskyFactorNumeric_SeqSBAIJ_3_NaturalOrdering() local
38 vj = aj[j]; /* block col. index */ in MatCholeskyFactorNumeric_SeqSBAIJ_3_NaturalOrdering()
39 rtmp_ptr = rtmp + vj * 9; in MatCholeskyFactorNumeric_SeqSBAIJ_3_NaturalOrdering()
131 vj = bj[j]; /* block col. index of U */ in MatCholeskyFactorNumeric_SeqSBAIJ_3_NaturalOrdering()
133 rtmp_ptr = rtmp + vj * 9; in MatCholeskyFactorNumeric_SeqSBAIJ_3_NaturalOrdering()
|
| H A D | sbaijfact.c | 45 PetscInt *jl, *q, jmin, jmax, juidx, nzk, qm, *iu, *ju, k, j, vj, umax, maxadd; in MatCholeskyFactorSymbolic_SeqSBAIJ_MSR() local
91 vj = rip[aj[j]]; /* col. value */ in MatCholeskyFactorSymbolic_SeqSBAIJ_MSR()
92 if (vj > k) { in MatCholeskyFactorSymbolic_SeqSBAIJ_MSR()
97 } while (qm < vj); in MatCholeskyFactorSymbolic_SeqSBAIJ_MSR()
98 PetscCheck(qm != vj, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Duplicate entry in A"); in MatCholeskyFactorSymbolic_SeqSBAIJ_MSR()
100 q[m] = vj; in MatCholeskyFactorSymbolic_SeqSBAIJ_MSR()
101 q[vj] = qm; in MatCholeskyFactorSymbolic_SeqSBAIJ_MSR()
116 vj = ju[j]; in MatCholeskyFactorSymbolic_SeqSBAIJ_MSR()
120 } while (qm < vj); in MatCholeskyFactorSymbolic_SeqSBAIJ_MSR()
121 if (qm != vj) { in MatCholeskyFactorSymbolic_SeqSBAIJ_MSR()
[all …]
|
| H A D | sbaijfact3.c | 10 PetscInt *a2anew, i, j, k, k1, jmin, jmax, *jl, *il, vj, nexti, ili; in MatCholeskyFactorNumeric_SeqSBAIJ_3() local
76 vj = perm_ptr[aj[j]]; /* block col. index */ in MatCholeskyFactorNumeric_SeqSBAIJ_3()
77 rtmp_ptr = rtmp + vj * 9; in MatCholeskyFactorNumeric_SeqSBAIJ_3()
169 vj = bj[j]; /* block col. index of U */ in MatCholeskyFactorNumeric_SeqSBAIJ_3()
171 rtmp_ptr = rtmp + vj * 9; in MatCholeskyFactorNumeric_SeqSBAIJ_3()
|
| H A D | sbaijfact5.c | 11 PetscInt *ai, *aj, k, k1, jmin, jmax, *jl, *il, vj, nexti, ili; in MatCholeskyFactorNumeric_SeqSBAIJ_4_NaturalOrdering() local
39 vj = aj[j]; /* block col. index */ in MatCholeskyFactorNumeric_SeqSBAIJ_4_NaturalOrdering()
40 rtmp_ptr = rtmp + vj * 16; in MatCholeskyFactorNumeric_SeqSBAIJ_4_NaturalOrdering()
160 vj = bj[j]; /* block col. index of U */ in MatCholeskyFactorNumeric_SeqSBAIJ_4_NaturalOrdering()
162 rtmp_ptr = rtmp + vj * 16; in MatCholeskyFactorNumeric_SeqSBAIJ_4_NaturalOrdering()
|
| H A D | sbaijfact6.c | 10 PetscInt i, j, *a2anew, k, k1, jmin, jmax, *jl, *il, vj, nexti, ili; in MatCholeskyFactorNumeric_SeqSBAIJ_4() local
77 vj = perm_ptr[aj[j]]; /* block col. index */ in MatCholeskyFactorNumeric_SeqSBAIJ_4()
78 rtmp_ptr = rtmp + vj * 16; in MatCholeskyFactorNumeric_SeqSBAIJ_4()
199 vj = bj[j]; /* block col. index of U */ in MatCholeskyFactorNumeric_SeqSBAIJ_4()
201 rtmp_ptr = rtmp + vj * 16; in MatCholeskyFactorNumeric_SeqSBAIJ_4()
|
| H A D | sbaijfact8.c | 11 PetscInt *ai, *aj, k, k1, jmin, jmax, *jl, *il, vj, nexti, ili, ipvt[5]; in MatCholeskyFactorNumeric_SeqSBAIJ_5_NaturalOrdering() local
38 vj = aj[j]; /* block col. index */ in MatCholeskyFactorNumeric_SeqSBAIJ_5_NaturalOrdering()
39 rtmp_ptr = rtmp + vj * 25; in MatCholeskyFactorNumeric_SeqSBAIJ_5_NaturalOrdering()
185 vj = bj[j]; /* block col. index of U */ in MatCholeskyFactorNumeric_SeqSBAIJ_5_NaturalOrdering()
187 rtmp_ptr = rtmp + vj * 25; in MatCholeskyFactorNumeric_SeqSBAIJ_5_NaturalOrdering()
|
| H A D | sbaijfact7.c | 10 PetscInt i, j, *a2anew, k, k1, jmin, jmax, *jl, *il, vj, nexti, ili, ipvt[5]; in MatCholeskyFactorNumeric_SeqSBAIJ_5() local
76 vj = perm_ptr[aj[j]]; /* block col. index */ in MatCholeskyFactorNumeric_SeqSBAIJ_5()
77 rtmp_ptr = rtmp + vj * 25; in MatCholeskyFactorNumeric_SeqSBAIJ_5()
223 vj = bj[j]; /* block col. index of U */ in MatCholeskyFactorNumeric_SeqSBAIJ_5()
225 rtmp_ptr = rtmp + vj * 25; in MatCholeskyFactorNumeric_SeqSBAIJ_5()
|
| H A D | sbaijfact9.c | 10 PetscInt i, j, *a2anew, k, k1, jmin, jmax, *jl, *il, vj, nexti, ili; in MatCholeskyFactorNumeric_SeqSBAIJ_6() local
78 vj = perm_ptr[aj[j]]; /* block col. index */ in MatCholeskyFactorNumeric_SeqSBAIJ_6()
79 wp = w + vj * 36; in MatCholeskyFactorNumeric_SeqSBAIJ_6()
336 vj = bj[j]; /* block col. index of U */ in MatCholeskyFactorNumeric_SeqSBAIJ_6()
338 wp = w + vj * 36; in MatCholeskyFactorNumeric_SeqSBAIJ_6()
|
| H A D | sbaijfact10.c | 11 PetscInt *ai, *aj, k, k1, jmin, jmax, *jl, *il, vj, nexti, ili; in MatCholeskyFactorNumeric_SeqSBAIJ_6_NaturalOrdering() local
46 vj = aj[j]; /* block col. index */ in MatCholeskyFactorNumeric_SeqSBAIJ_6_NaturalOrdering()
47 wp = w + vj * 36; in MatCholeskyFactorNumeric_SeqSBAIJ_6_NaturalOrdering()
341 vj = bj[j]; /* block col. index of U */ in MatCholeskyFactorNumeric_SeqSBAIJ_6_NaturalOrdering()
343 wp = w + vj * 36; in MatCholeskyFactorNumeric_SeqSBAIJ_6_NaturalOrdering()
|
| H A D | sbaijfact12.c | 11 PetscInt *ai, *aj, k, k1, jmin, jmax, *jl, *il, vj, nexti, ili; in MatCholeskyFactorNumeric_SeqSBAIJ_7_NaturalOrdering() local
41 vj = aj[j]; /* block col. index */ in MatCholeskyFactorNumeric_SeqSBAIJ_7_NaturalOrdering()
42 wp = w + vj * 49; in MatCholeskyFactorNumeric_SeqSBAIJ_7_NaturalOrdering()
368 vj = bj[j]; /* block col. index of U */ in MatCholeskyFactorNumeric_SeqSBAIJ_7_NaturalOrdering()
370 wp = w + vj * 49; in MatCholeskyFactorNumeric_SeqSBAIJ_7_NaturalOrdering()
|
| H A D | sbaijfact11.c | 10 PetscInt *a2anew, k, k1, jmin, jmax, *jl, *il, vj, nexti, ili, i, j, mbs = a->mbs; in MatCholeskyFactorNumeric_SeqSBAIJ_7() local
79 vj = perm_ptr[aj[j]]; /* block col. index */ in MatCholeskyFactorNumeric_SeqSBAIJ_7()
80 wp = w + vj * 49; in MatCholeskyFactorNumeric_SeqSBAIJ_7()
405 vj = bj[j]; /* block col. index of U */ in MatCholeskyFactorNumeric_SeqSBAIJ_7()
407 wp = w + vj * 49; in MatCholeskyFactorNumeric_SeqSBAIJ_7()
|
| /petsc/src/ksp/pc/impls/sor/ |
| H A D | sor.tex | 47 vj = aj + 1;
52 t[vj[j]] += tmp*v[j];
55 vj += nz + 1;
62 vj = aj + ai[m-1] + 1;
68 PETSC_Prefetch(vj-nz2-1,0,PETSC_PREFETCH_HINT_NTA);
69 PetscSparseDensePlusDot(sum,x,v,vj,nz);
74 vj -= nz + 1;
|
| /petsc/src/ts/utils/dmplexlandau/kokkos/ |
| H A D | landau.kokkos.cxx | 607 …const PetscReal vj[3] = {d_x[jpidx_glb], d_y[jpidx_glb], d_z ? d_z[jpidx_glb] : … in LandauKokkosJacobian() local
616 …etscReal Ud[2][2], Uk[2][2], mask = (PetscAbs(vj[0] - x) < 100 * PETSC_SQRT_MACHINE_EPSILON && Pet… in LandauKokkosJacobian()
617 LandauTensor2D(vj, x, y, Ud, Uk, mask); in LandauKokkosJacobian()
619 …x], mask = (PetscAbs(vj[0]-x) < 100*PETSC_SQRT_MACHINE_EPSILON && PetscAbs(vj[1]-y) < 100*PETSC_SQ… in LandauKokkosJacobian()
620 LandauTensor3D(vj, x, y, z, U, mask); in LandauKokkosJacobian()
|
| /petsc/src/ts/utils/dmplexlandau/ |
| H A D | plexland.c | 341 …const PetscReal vj[3] = {xx[jpidx_glb], yy[jpidx_glb], zz ? zz[jpidx_glb] : 0}, wj = ww[jpi… in LandauFormJacobian_Internal() local
356 …etscReal Ud[2][2], Uk[2][2], mask = (PetscAbs(vj[0] - x) < 100 * PETSC_SQRT_MACHINE_EPSILON && Pet… in LandauFormJacobian_Internal()
357 LandauTensor2D(vj, x, y, Ud, Uk, mask); in LandauFormJacobian_Internal()
359 …mask = (PetscAbs(vj[0] - x) < 100 * PETSC_SQRT_MACHINE_EPSILON && PetscAbs(vj[1] - y) < 100 * PETS… in LandauFormJacobian_Internal()
361 LandauTensor3DRelativistic(vj, x, y, z, U, mask, C_0(ctx->v_0)); in LandauFormJacobian_Internal()
363 LandauTensor3D(vj, x, y, z, U, mask); in LandauFormJacobian_Internal()
1755 …PetscReal vj[LANDAU_MAX_NQND * LANDAU_DIM], detJj[LANDAU_MAX_NQND], Jdummy[LANDAU_MAX_NQND * LA… in CreateStaticData() local
1757 …PetscCall(DMPlexComputeCellGeometryFEM(ctx->plex[grid], ej + cStart, quad, vj, Jdummy, invJe, detJ… in CreateStaticData()
1764 if (dim == 2) ww[gidx] *= vj[qj * dim + 0]; /* cylindrical coordinate, w/o 2pi */ in CreateStaticData()
1781 xx[gidx] = vj[qj * dim + 0]; /* coordinate */ in CreateStaticData()
[all …]
|
| /petsc/src/ksp/ksp/utils/dm/ |
| H A D | dmproject.c | 1106 for (PetscInt vj = PetscMax(vs[1] - 1, 0); vj < PetscMin(vs[1] + 3, vend[1]); ++vj) { in DMSwarmRemap_Colella_Internal() local
1108 const PetscInt rv = vi * vend[1] + vj; in DMSwarmRemap_Colella_Internal()
1111 …if (vcdim > 1) PetscCall(W_3_Interpolation_Private((vmin[1] + (vj + 0.5) * hv[1] - v[p * dim + 1])… in DMSwarmRemap_Colella_Internal()
1114 …PetscInt_FMT ", %" PetscInt_FMT ") w = %g (%g, %g, %g, %g)\n", xi, xj, vi, vj, (double)(wp * Wx[0]… in DMSwarmRemap_Colella_Internal()
|
| /petsc/src/mat/impls/aij/seq/seqhipsparse/ |
| H A D | aijhipsparse.hip.cxx | 549 const PetscInt *ai = b->i, *aj = b->j, *vj; in MatSeqAIJHIPSPARSEBuildICCTriMatrices() local
570 vj = aj + ai[i]; in MatSeqAIJHIPSPARSEBuildICCTriMatrices()
581 PetscCall(PetscArraycpy(&AjUp[offset], vj, nz)); in MatSeqAIJHIPSPARSEBuildICCTriMatrices()
|
| /petsc/src/mat/impls/aij/seq/seqcusparse/ |
| H A D | aijcusparse.cu | 784 const PetscInt *ai = b->i, *aj = b->j, *vj; in MatSeqAIJCUSPARSEBuildICCTriMatrices() local
805 vj = aj + ai[i]; in MatSeqAIJCUSPARSEBuildICCTriMatrices()
816 PetscCall(PetscArraycpy(&AjUp[offset], vj, nz)); in MatSeqAIJCUSPARSEBuildICCTriMatrices()
|