Home
last modified time | relevance | path

Searched refs:totalslices (Results 1 – 8 of 8) sorted by relevance

/petsc/src/mat/impls/sell/seq/
H A Dsell.c90 PetscInt i, j, totalslices; in MatSeqSELLSetPreallocation_SeqSELL() local
127 totalslices = PetscCeilInt(B->rmap->n, b->sliceheight); in MatSeqSELLSetPreallocation_SeqSELL()
128 b->totalslices = totalslices; in MatSeqSELLSetPreallocation_SeqSELL()
133 PetscCall(PetscMalloc1(totalslices + 1, &b->sliidx)); in MatSeqSELLSetPreallocation_SeqSELL()
143 for (i = 0; i <= totalslices; i++) b->sliidx[i] = b->sliceheight * i * maxallocrow; in MatSeqSELLSetPreallocation_SeqSELL()
150 for (i = 1; i < totalslices; i++) { in MatSeqSELLSetPreallocation_SeqSELL()
163 b->sliidx[totalslices] = 0; in MatSeqSELLSetPreallocation_SeqSELL()
164 …for (j = b->sliceheight * (totalslices - 1); j < B->rmap->n; j++) b->sliidx[totalslices] = PetscMa… in MatSeqSELLSetPreallocation_SeqSELL()
168 b->sliidx[totalslices] = ((b->sliidx[totalslices] - 1) / mul + 1) * mul; in MatSeqSELLSetPreallocation_SeqSELL()
171 maxallocrow = PetscMax(b->sliidx[totalslices], maxallocrow); in MatSeqSELLSetPreallocation_SeqSELL()
[all …]
H A Dfdsell.c13 PetscInt i, j, *collengths, *cia, *cja, n = A->cmap->n, totalslices; in MatGetColumnIJ_SeqSELL_Color() local
28 totalslices = PetscCeilInt(A->rmap->n, 8); in MatGetColumnIJ_SeqSELL_Color()
29 for (i = 0; i < totalslices; i++) { /* loop over slices */ in MatGetColumnIJ_SeqSELL_Color()
40 for (i = 0; i < totalslices; i++) { /* loop over slices */ in MatGetColumnIJ_SeqSELL_Color()
H A Dsell.h46 PetscInt totalslices; /* total number of slices */ \
105 …cpy(new_val + SIDX[SID + 1] + SH * MUL, VAL + SIDX[SID + 1], SIDX[Ain->totalslices] - SIDX[SID + 1…
106 …w_colidx + SIDX[SID + 1] + SH * MUL, COLIDX + SIDX[SID + 1], SIDX[Ain->totalslices] - SIDX[SID + 1…
108 for (ii = SID + 1; ii <= Ain->totalslices; ii++) SIDX[ii] += SH * MUL; \
150 if (a->maxallocmat < a->sliidx[a->totalslices] + a->sliceheight) { \
160 …rPlusOffset(a->val, a->sliidx[row / a->sliceheight + 1]), a->sliidx[a->totalslices] - a->sliidx[ro…
161 …usOffset(a->colidx, a->sliidx[row / a->sliceheight + 1]), a->sliidx[a->totalslices] - a->sliidx[ro…
174 …liceheight, a->val + a->sliidx[row / a->sliceheight + 1], a->sliidx[a->totalslices] - a->sliidx[ro…
175 …eheight, a->colidx + a->sliidx[row / a->sliceheight + 1], a->sliidx[a->totalslices] - a->sliidx[ro…
178 … for (ii = row / a->sliceheight + 1; ii <= a->totalslices; ii++) a->sliidx[ii] += a->sliceheight; \
/petsc/src/mat/impls/sell/mpi/
H A Dmmsell.c21 PetscInt i, j, totalslices, N = A->cmap->N, row; in MatDisAssemble_MPISELL() local
56 totalslices = PetscCeilInt(B->rmap->n, Bsell->sliceheight); in MatDisAssemble_MPISELL()
57 for (i = 0; i < totalslices; i++) { /* loop over slices */ in MatDisAssemble_MPISELL()
77 PetscInt i, j, *bcolidx = B->colidx, ec = 0, *garray, totalslices; in MatSetUpMultiply_MPISELL() local
90 totalslices = PetscCeilInt(sell->B->rmap->n, B->sliceheight); in MatSetUpMultiply_MPISELL()
96 for (i = 0; i < totalslices; i++) { /* loop over slices */ in MatSetUpMultiply_MPISELL()
125 for (i = 0; i < totalslices; i++) { /* loop over slices */ in MatSetUpMultiply_MPISELL()
143 for (i = 0; i < totalslices; i++) { /* loop over slices */ in MatSetUpMultiply_MPISELL()
164 for (i = 0; i < totalslices; i++) { /* loop over slices */ in MatSetUpMultiply_MPISELL()
H A Dmpisell.c622 for (i = 0; i < Aloc->totalslices; i++) { /* loop over slices */ in MatView_MPISELL_ASCIIorDraworSocket()
639 for (i = 0; i < Aloc->totalslices; i++) { in MatView_MPISELL_ASCIIorDraworSocket()
/petsc/src/mat/impls/sell/seq/seqhip/
H A Dsellhip.hip.cxx15 PetscInt totalslices; member
48 …PetscCallHIP(hipMemcpy(hipstruct->val, a->val, a->sliidx[a->totalslices] * sizeof(MatScalar), hipM… in MatSeqSELLHIPCopyToGPU()
49 PetscCall(PetscLogCpuToGpu(a->sliidx[a->totalslices] * (sizeof(MatScalar)))); in MatSeqSELLHIPCopyToGPU()
56 hipstruct->totalentries = a->sliidx[a->totalslices]; in MatSeqSELLHIPCopyToGPU()
57 hipstruct->totalslices = a->totalslices; in MatSeqSELLHIPCopyToGPU()
62 …PetscCallHIP(hipMemcpy(hipstruct->colidx, a->colidx, a->sliidx[a->totalslices] * sizeof(*a->colidx… in MatSeqSELLHIPCopyToGPU()
63 …PetscCallHIP(hipMemcpy(hipstruct->val, a->val, a->sliidx[a->totalslices] * sizeof(*a->val), hipMem… in MatSeqSELLHIPCopyToGPU()
65 …PetscCallHIP(hipMalloc((void **)&hipstruct->sliidx, (a->totalslices + 1) * sizeof(*hipstruct->slii… in MatSeqSELLHIPCopyToGPU()
66 …PetscCallHIP(hipMemcpy(hipstruct->sliidx, a->sliidx, (a->totalslices + 1) * sizeof(*a->sliidx), hi… in MatSeqSELLHIPCopyToGPU()
69 …PetscCall(PetscLogCpuToGpu(a->sliidx[a->totalslices] * (sizeof(MatScalar) + sizeof(PetscInt)) + (a… in MatSeqSELLHIPCopyToGPU()
[all …]
/petsc/src/mat/impls/sell/seq/seqcuda/
H A Dsellcuda.cu12 PetscInt totalslices; member
45 …PetscCallCUDA(cudaMemcpy(cudastruct->val, a->val, a->sliidx[a->totalslices] * sizeof(MatScalar), c… in MatSeqSELLCUDACopyToGPU()
46 PetscCall(PetscLogCpuToGpu(a->sliidx[a->totalslices] * (sizeof(MatScalar)))); in MatSeqSELLCUDACopyToGPU()
53 cudastruct->totalentries = a->sliidx[a->totalslices]; in MatSeqSELLCUDACopyToGPU()
54 cudastruct->totalslices = a->totalslices; in MatSeqSELLCUDACopyToGPU()
59 …PetscCallCUDA(cudaMemcpy(cudastruct->colidx, a->colidx, a->sliidx[a->totalslices] * sizeof(*a->col… in MatSeqSELLCUDACopyToGPU()
60 …PetscCallCUDA(cudaMemcpy(cudastruct->val, a->val, a->sliidx[a->totalslices] * sizeof(*a->val), cud… in MatSeqSELLCUDACopyToGPU()
62 …PetscCallCUDA(cudaMalloc((void **)&cudastruct->sliidx, (a->totalslices + 1) * sizeof(*cudastruct->… in MatSeqSELLCUDACopyToGPU()
63 …PetscCallCUDA(cudaMemcpy(cudastruct->sliidx, a->sliidx, (a->totalslices + 1) * sizeof(*a->sliidx),… in MatSeqSELLCUDACopyToGPU()
66 …PetscCall(PetscLogCpuToGpu(a->sliidx[a->totalslices] * (sizeof(MatScalar) + sizeof(PetscInt)) + (a… in MatSeqSELLCUDACopyToGPU()
[all …]
/petsc/src/mat/impls/aij/mpi/mumps/
H A Dmumps.c828 nz = a->sliidx[a->totalslices]; in MatConvertToTriples_seqsell_seqaij()
831 for (i = k = 0; i < a->totalslices; i++) { in MatConvertToTriples_seqsell_seqaij()