Searched refs:widx (Results 1 – 3 of 3) sorted by relevance
| /petsc/src/vec/is/ao/impls/memscalable/ |
| H A D | aomemscalable.c | 101 PetscMPIInt source, widx; in AOMap_MemoryScalable_private() local
196 …PetscCallMPI(MPI_Waitany(nreceives, recv_waits, &widx, &recv_status)); /* idx: index of handle for… in AOMap_MemoryScalable_private()
198 rbuf = rindices + nmax * widx; /* global index */ in AOMap_MemoryScalable_private()
206 PetscCallMPI(MPIU_Isend(sbuf, nindices, MPIU_INT, source, tag2, comm, send_waits2 + widx)); in AOMap_MemoryScalable_private()
214 PetscCallMPI(MPI_Waitany(nsends, recv_waits2, &widx, &recv_status)); in AOMap_MemoryScalable_private()
277 PetscMPIInt widx; in AOCreateMemoryScalable_private() local
364 PetscCallMPI(MPI_Waitany(nreceives_i, recv_waits, &widx, &recv_status)); in AOCreateMemoryScalable_private()
366 rbuf = rindices + nmax * widx; /* global index */ in AOCreateMemoryScalable_private()
|
| /petsc/src/mat/impls/aij/seq/seqhipsparse/ |
| H A D | aijhipsparse.hip.cxx | 4361 THRUSTINTARRAY widx(n); in MatSeqAIJCopySubArray_SeqAIJHIPSPARSE() local
4362 widx.assign(idx, idx + n); in MatSeqAIJCopySubArray_SeqAIJHIPSPARSE()
4374 …t::make_zip_iterator(thrust::make_tuple(thrust::make_permutation_iterator(dav, widx.begin()), dv)); in MatSeqAIJCopySubArray_SeqAIJHIPSPARSE()
4375 …:make_zip_iterator(thrust::make_tuple(thrust::make_permutation_iterator(dav, widx.end()), dv + n)); in MatSeqAIJCopySubArray_SeqAIJHIPSPARSE()
|
| /petsc/src/mat/impls/aij/seq/seqcusparse/ |
| H A D | aijcusparse.cu | 5042 THRUSTINTARRAY widx(n); in MatSeqAIJCopySubArray_SeqAIJCUSPARSE() local
5043 widx.assign(idx, idx + n); in MatSeqAIJCopySubArray_SeqAIJCUSPARSE()
5056 …t::make_zip_iterator(thrust::make_tuple(thrust::make_permutation_iterator(dav, widx.begin()), dv)); in MatSeqAIJCopySubArray_SeqAIJCUSPARSE()
5057 …:make_zip_iterator(thrust::make_tuple(thrust::make_permutation_iterator(dav, widx.end()), dv + n)); in MatSeqAIJCopySubArray_SeqAIJCUSPARSE()
|