Searched refs:wPerm (Results 1 – 2 of 2) sorted by relevance
| /petsc/src/mat/impls/aij/seq/seqhipsparse/ |
| H A D | aijhipsparse.hip.cxx | 4186 auto wPerm = new THRUSTINTARRAY32(Annz + Bnnz); in MatSeqAIJHIPSPARSEMergeMats() local
4191 …::make_tuple(Ccoo->begin(), Ccsr->column_indices->begin(), Ccsr->values->begin(), wPerm->begin())); in MatSeqAIJHIPSPARSEMergeMats()
4199 …PetscCallThrust(thrust::partition_copy(thrust::device, cci, cce, wPerm->begin(), p1, p2, thrust::i… in MatSeqAIJHIPSPARSEMergeMats()
4202 PetscCallThrust(thrust::copy_if(thrust::device, cci, cce, wPerm->begin(), p1, pred)); in MatSeqAIJHIPSPARSEMergeMats()
4203 PetscCallThrust(thrust::remove_copy_if(thrust::device, cci, cce, wPerm->begin(), p2, pred)); in MatSeqAIJHIPSPARSEMergeMats()
4207 delete wPerm; in MatSeqAIJHIPSPARSEMergeMats()
|
| /petsc/src/mat/impls/aij/seq/seqcusparse/ |
| H A D | aijcusparse.cu | 4842 auto wPerm = new THRUSTINTARRAY32(Annz + Bnnz); in MatSeqAIJCUSPARSEMergeMats() local
4847 …::make_tuple(Ccoo->begin(), Ccsr->column_indices->begin(), Ccsr->values->begin(), wPerm->begin())); in MatSeqAIJCUSPARSEMergeMats()
4862 …PetscCallThrust(thrust::partition_copy(thrust::device,cci,cce,wPerm->begin(),p1,p2,thrust::identit… in MatSeqAIJCUSPARSEMergeMats()
4869 PetscCallThrust(thrust::copy_if(thrust::device, cci, cce, wPerm->begin(), p1, pred)); in MatSeqAIJCUSPARSEMergeMats()
4870 PetscCallThrust(thrust::remove_copy_if(thrust::device, cci, cce, wPerm->begin(), p2, pred)); in MatSeqAIJCUSPARSEMergeMats()
4875 delete wPerm; in MatSeqAIJCUSPARSEMergeMats()
|