Home
last modified time | relevance | path

Searched refs:colidx (Results 1 – 9 of 9) sorted by relevance

/petsc/src/ksp/ksp/tests/
H A Dex2.c24 PetscInt N = 2, rowidx, colidx; in main() local
43 colidx = 0; in main()
45 PetscCall(MatSetValues(C, 1, &rowidx, 1, &colidx, &v, INSERT_VALUES)); in main()
47 colidx = 1; in main()
49 PetscCall(MatSetValues(C, 1, &rowidx, 1, &colidx, &v, INSERT_VALUES)); in main()
52 colidx = 0; in main()
54 PetscCall(MatSetValues(C, 1, &rowidx, 1, &colidx, &v, INSERT_VALUES)); in main()
56 colidx = 1; in main()
58 PetscCall(MatSetValues(C, 1, &rowidx, 1, &colidx, &v, INSERT_VALUES)); in main()
/petsc/src/mat/impls/sell/seq/
H A Dsell.h34 PetscInt *colidx; /* column index */ \
78 static inline PetscErrorCode MatSeqXSELLFreeSELL(Mat AA, MatScalar **val, PetscInt **colidx) in MatSeqXSELLFreeSELL() argument
82 PetscCall(PetscFree2(*val, *colidx)); in MatSeqXSELLFreeSELL()
85 if (A->free_colidx) PetscCall(PetscFree(*colidx)); in MatSeqXSELLFreeSELL()
113 PetscCall(MatSeqXSELLFreeSELL(A, &Ain->val, &Ain->colidx)); \
115 Ain->colidx = new_colidx; \
159 PetscCall(PetscArraycpy(new_colidx, a->colidx, a->sliidx[row / a->sliceheight + 1])); \
161 …ow / a->sliceheight + 1] + a->sliceheight, PetscSafePointerPlusOffset(a->colidx, a->sliidx[row / a…
166 PetscCall(MatSeqXSELLFreeSELL(A, &a->val, &a->colidx)); \
168 a->colidx = new_colidx; \
[all …]
H A Dsell.c177 PetscCall(MatSeqXSELLFreeSELL(B, &b->val, &b->colidx)); in MatSeqSELLSetPreallocation_SeqSELL()
179 PetscCall(PetscMalloc2(b->sliidx[totalslices], &b->val, b->sliidx[totalslices], &b->colidx)); in MatSeqSELLSetPreallocation_SeqSELL()
216 for (j = 0; j < a->rlen[row]; j++) a->getrowcols[j] = a->colidx[shift + a->sliceheight * j]; in MatGetRow_SeqSELL()
327 const PetscInt *acolidx = a->colidx; in MatMult_SeqSELL()
542 const PetscInt *acolidx = a->colidx; in MatMultAdd_SeqSELL()
719 const PetscInt *acolidx = a->colidx; in MatMultTransposeAdd_SeqSELL()
790 if (a->colidx[shift + a->sliceheight * j] == i) { in MatGetDiagonalMarkers_SeqSELL()
812 if (a->colidx[shift + a->sliceheight * j] == i) { in MatGetDiagonalMarkers_SeqSELL()
893 PetscCall(MatSeqXSELLFreeSELL(A, &a->val, &a->colidx)); in MatDestroy_SeqSELL()
986 if (a->colidx[shift + a->sliceheight * j] == i) { in MatGetDiagonal_SeqSELL()
[all …]
H A Dfdsell.c32 if (isnonzero) collengths[a->colidx[j]]++; in MatGetColumnIJ_SeqSELL_Color()
44 col = a->colidx[j]; in MatGetColumnIJ_SeqSELL_Color()
/petsc/src/mat/impls/sell/seq/seqhip/
H A Dsellhip.hip.cxx13 PetscInt *colidx; /* column index array, device pointer */ member
29 if ((*hipstruct)->colidx) PetscCallHIP(hipFree((*hipstruct)->colidx)); in MatSeqSELLHIP_Destroy()
51 if (hipstruct->colidx) PetscCallHIP(hipFree(hipstruct->colidx)); in MatSeqSELLHIPCopyToGPU()
59 … PetscCallHIP(hipMalloc((void **)&hipstruct->colidx, a->maxallocmat * sizeof(*hipstruct->colidx))); in MatSeqSELLHIPCopyToGPU()
62 …PetscCallHIP(hipMemcpy(hipstruct->colidx, a->colidx, a->sliidx[a->totalslices] * sizeof(*a->colidx in MatSeqSELLHIPCopyToGPU()
594 acolidx = hipstruct->colidx; in MatMult_SeqSELLHIP()
716 PetscInt *acolidx = hipstruct->colidx; in MatMultAdd_SeqSELLHIP()
/petsc/src/mat/impls/sell/seq/seqcuda/
H A Dsellcuda.cu10 PetscInt *colidx; /* column index array, device pointer */ member
26 if ((*cudastruct)->colidx) PetscCallCUDA(cudaFree((*cudastruct)->colidx)); in MatSeqSELLCUDA_Destroy()
48 if (cudastruct->colidx) PetscCallCUDA(cudaFree(cudastruct->colidx)); in MatSeqSELLCUDACopyToGPU()
56 …PetscCallCUDA(cudaMalloc((void **)&cudastruct->colidx, a->maxallocmat * sizeof(*cudastruct->colidx in MatSeqSELLCUDACopyToGPU()
59 …PetscCallCUDA(cudaMemcpy(cudastruct->colidx, a->colidx, a->sliidx[a->totalslices] * sizeof(*a->col… in MatSeqSELLCUDACopyToGPU()
589 acolidx = cudastruct->colidx; in MatMult_SeqSELLCUDA()
716 PetscInt *acolidx = cudastruct->colidx; in MatMultAdd_SeqSELLCUDA()
/petsc/src/mat/impls/sell/mpi/
H A Dmmsell.c60 …Call(MatSetValue(Bnew, Bsell->sliceheight * i + row, sell->garray[Bsell->colidx[j]], Bsell->val[j]… in MatDisAssemble_MPISELL()
77 PetscInt i, j, *bcolidx = B->colidx, ec = 0, *garray, totalslices; in MatSetUpMultiply_MPISELL()
H A Dmpisell.c92 …am, 1, nrow1, a->sliidx, a->sliceheight, row / sliceheight, row, col, a->colidx, a->val, cp1, vp1,…
136 …bm, 1, nrow2, b->sliidx, b->sliceheight, row / sliceheight, row, col, b->colidx, b->val, cp2, vp2,…
175 cp1 = PetscSafePointerPlusOffset(a->colidx, shift1); in MatSetValues_MPISELL()
182 cp2 = PetscSafePointerPlusOffset(b->colidx, shift2); in MatSetValues_MPISELL()
217 cp2 = b->colidx + shift2; in MatSetValues_MPISELL()
620 acolidx = Aloc->colidx; in MatView_MPISELL_ASCIIorDraworSocket()
637 acolidx = Aloc->colidx; in MatView_MPISELL_ASCIIorDraworSocket()
/petsc/src/mat/impls/aij/mpi/mumps/
H A Dmumps.c834 for (i = 0; i < nz; i++) PetscCall(PetscMUMPSIntCast(a->colidx[i] + shift, &col[i])); in MatConvertToTriples_seqsell_seqaij()