Home
last modified time | relevance | path

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

/petsc/src/ksp/ksp/impls/fcg/pipefcg/
H A Dpipefcg.c35 PetscCall(KSPCreateVecs(ksp, nnewvecs, &pipefcg->pQvecs[pipefcg->nchunks], 0, NULL)); in KSPAllocateVectors_PIPEFCG()
36 PetscCall(KSPCreateVecs(ksp, nnewvecs, &pipefcg->pZETAvecs[pipefcg->nchunks], 0, NULL)); in KSPAllocateVectors_PIPEFCG()
37 PetscCall(KSPCreateVecs(ksp, nnewvecs, &pipefcg->pPvecs[pipefcg->nchunks], 0, NULL)); in KSPAllocateVectors_PIPEFCG()
38 PetscCall(KSPCreateVecs(ksp, nnewvecs, &pipefcg->pSvecs[pipefcg->nchunks], 0, NULL)); in KSPAllocateVectors_PIPEFCG()
41 pipefcg->Qvecs[nvecsprev + i] = pipefcg->pQvecs[pipefcg->nchunks][i]; in KSPAllocateVectors_PIPEFCG()
42 pipefcg->ZETAvecs[nvecsprev + i] = pipefcg->pZETAvecs[pipefcg->nchunks][i]; in KSPAllocateVectors_PIPEFCG()
43 pipefcg->Pvecs[nvecsprev + i] = pipefcg->pPvecs[pipefcg->nchunks][i]; in KSPAllocateVectors_PIPEFCG()
44 pipefcg->Svecs[nvecsprev + i] = pipefcg->pSvecs[pipefcg->nchunks][i]; in KSPAllocateVectors_PIPEFCG()
46 pipefcg->chunksizes[pipefcg->nchunks] = nnewvecs; in KSPAllocateVectors_PIPEFCG()
47 ++pipefcg->nchunks; in KSPAllocateVectors_PIPEFCG()
[all …]
H A Dpipefcgimpl.h13 PetscInt nchunks; /* Number of chunks */ member
/petsc/src/ksp/ksp/impls/gcr/pipegcr/
H A Dpipegcr.c39 PetscCall(KSPCreateVecs(ksp, nnewvecs, &pipegcr->ppvecs[pipegcr->nchunks], 0, NULL)); in KSPAllocateVectors_PIPEGCR()
40 PetscCall(KSPCreateVecs(ksp, nnewvecs, &pipegcr->psvecs[pipegcr->nchunks], 0, NULL)); in KSPAllocateVectors_PIPEGCR()
41 PetscCall(KSPCreateVecs(ksp, nnewvecs, &pipegcr->pqvecs[pipegcr->nchunks], 0, NULL)); in KSPAllocateVectors_PIPEGCR()
42 …if (pipegcr->unroll_w) PetscCall(KSPCreateVecs(ksp, nnewvecs, &pipegcr->ptvecs[pipegcr->nchunks], … in KSPAllocateVectors_PIPEGCR()
45 pipegcr->qvecs[nvecsprev + i] = pipegcr->pqvecs[pipegcr->nchunks][i]; in KSPAllocateVectors_PIPEGCR()
46 pipegcr->pvecs[nvecsprev + i] = pipegcr->ppvecs[pipegcr->nchunks][i]; in KSPAllocateVectors_PIPEGCR()
47 pipegcr->svecs[nvecsprev + i] = pipegcr->psvecs[pipegcr->nchunks][i]; in KSPAllocateVectors_PIPEGCR()
48 if (pipegcr->unroll_w) pipegcr->tvecs[nvecsprev + i] = pipegcr->ptvecs[pipegcr->nchunks][i]; in KSPAllocateVectors_PIPEGCR()
50 pipegcr->chunksizes[pipegcr->nchunks] = nnewvecs; in KSPAllocateVectors_PIPEGCR()
51 pipegcr->nchunks++; in KSPAllocateVectors_PIPEGCR()
[all …]
H A Dpipegcrimpl.h12 PetscInt nchunks; /* Number of chunks */ member
/petsc/src/ksp/ksp/impls/fcg/
H A Dfcg.c25 PetscCall(KSPCreateVecs(ksp, nnewvecs, &fcg->pCvecs[fcg->nchunks], 0, NULL)); in KSPAllocateVectors_FCG()
26 PetscCall(KSPCreateVecs(ksp, nnewvecs, &fcg->pPvecs[fcg->nchunks], 0, NULL)); in KSPAllocateVectors_FCG()
29 fcg->Cvecs[nvecsprev + i] = fcg->pCvecs[fcg->nchunks][i]; in KSPAllocateVectors_FCG()
30 fcg->Pvecs[nvecsprev + i] = fcg->pPvecs[fcg->nchunks][i]; in KSPAllocateVectors_FCG()
32 fcg->chunksizes[fcg->nchunks] = nnewvecs; in KSPAllocateVectors_FCG()
33 ++fcg->nchunks; in KSPAllocateVectors_FCG()
278 for (i = 0; i < fcg->nchunks; ++i) { in KSPReset_FCG()
282 fcg->nchunks = fcg->nvecs = 0; in KSPReset_FCG()
542 fcg->nchunks = 0; in KSPCreate_FCG()
H A Dfcgimpl.h17 PetscInt nchunks; /* Number of chunks */ member
/petsc/src/mat/impls/sell/seq/seqcuda/
H A Dsellcuda.cu579 PetscInt chunksperblock, nchunks, *chunk_slice_map; in MatMult_SeqSELLCUDA() local
661 nchunks = cudastruct->totalchunks; in MatMult_SeqSELLCUDA()
663 nblocks = 1 + (nchunks - 1) / chunksperblock; in MatMult_SeqSELLCUDA()
666 …kernel8<2><<<nblocks, dim3(32, 2)>>>(nrows, sliceheight, chunksperblock, nchunks, chunk_slice_map,… in MatMult_SeqSELLCUDA()
668 …kernel8<4><<<nblocks, dim3(32, 4)>>>(nrows, sliceheight, chunksperblock, nchunks, chunk_slice_map,… in MatMult_SeqSELLCUDA()
670 …kernel8<8><<<nblocks, dim3(32, 8)>>>(nrows, sliceheight, chunksperblock, nchunks, chunk_slice_map,… in MatMult_SeqSELLCUDA()
672 …rnel8<16><<<nblocks, dim3(32, 16)>>>(nrows, sliceheight, chunksperblock, nchunks, chunk_slice_map,… in MatMult_SeqSELLCUDA()
674 …rnel8<32><<<nblocks, dim3(32, 32)>>>(nrows, sliceheight, chunksperblock, nchunks, chunk_slice_map,… in MatMult_SeqSELLCUDA()
676 …kernel8<2><<<nblocks, dim3(32, 2)>>>(nrows, sliceheight, chunksperblock, nchunks, chunk_slice_map,… in MatMult_SeqSELLCUDA()
720 PetscInt chunksperblock, nchunks, *chunk_slice_map; in MatMultAdd_SeqSELLCUDA() local
[all …]
/petsc/src/mat/impls/sell/seq/seqhip/
H A Dsellhip.hip.cxx584 PetscInt chunksperblock, nchunks, *chunk_slice_map; in MatMult_SeqSELLHIP() local
663 nchunks = hipstruct->totalchunks; in MatMult_SeqSELLHIP()
665 nblocks = 1 + (nchunks - 1) / chunksperblock; in MatMult_SeqSELLHIP()
668 …<2><<<nblocks, dim3(WARP_SIZE, 2)>>>(nrows, sliceheight, chunksperblock, nchunks, chunk_slice_map,… in MatMult_SeqSELLHIP()
670 …<4><<<nblocks, dim3(WARP_SIZE, 4)>>>(nrows, sliceheight, chunksperblock, nchunks, chunk_slice_map,… in MatMult_SeqSELLHIP()
672 …<8><<<nblocks, dim3(WARP_SIZE, 8)>>>(nrows, sliceheight, chunksperblock, nchunks, chunk_slice_map,… in MatMult_SeqSELLHIP()
674 …6><<<nblocks, dim3(WARP_SIZE, 16)>>>(nrows, sliceheight, chunksperblock, nchunks, chunk_slice_map,… in MatMult_SeqSELLHIP()
676 …<2><<<nblocks, dim3(WARP_SIZE, 2)>>>(nrows, sliceheight, chunksperblock, nchunks, chunk_slice_map,… in MatMult_SeqSELLHIP()
720 PetscInt chunksperblock, nchunks, *chunk_slice_map; in MatMultAdd_SeqSELLHIP() local
754 nchunks = hipstruct->totalchunks; in MatMultAdd_SeqSELLHIP()
[all …]