Lines Matching refs:column_indices
385 loTriFactor->csrMat->column_indices = new THRUSTINTARRAY32(nzLower); in MatSeqAIJCUSPARSEBuildILULowerTriMatrix()
386 loTriFactor->csrMat->column_indices->assign(AjLo, AjLo + nzLower); in MatSeqAIJCUSPARSEBuildILULowerTriMatrix()
396 …loTriFactor->csrMat->row_offsets->data().get(), loTriFactor->csrMat->column_indices->data().get(),… in MatSeqAIJCUSPARSEBuildILULowerTriMatrix()
402 …loTriFactor->csrMat->row_offsets->data().get(), loTriFactor->csrMat->column_indices->data().get(),… in MatSeqAIJCUSPARSEBuildILULowerTriMatrix()
514 upTriFactor->csrMat->column_indices = new THRUSTINTARRAY32(nzUpper); in MatSeqAIJCUSPARSEBuildILUUpperTriMatrix()
515 upTriFactor->csrMat->column_indices->assign(AjUp, AjUp + nzUpper); in MatSeqAIJCUSPARSEBuildILUUpperTriMatrix()
525 …upTriFactor->csrMat->row_offsets->data().get(), upTriFactor->csrMat->column_indices->data().get(),… in MatSeqAIJCUSPARSEBuildILUUpperTriMatrix()
531 …upTriFactor->csrMat->row_offsets->data().get(), upTriFactor->csrMat->column_indices->data().get(),… in MatSeqAIJCUSPARSEBuildILUUpperTriMatrix()
850 upTriFactor->csrMat->column_indices = new THRUSTINTARRAY32(a->nz); in MatSeqAIJCUSPARSEBuildICCTriMatrices()
851 upTriFactor->csrMat->column_indices->assign(AjUp, AjUp + a->nz); in MatSeqAIJCUSPARSEBuildICCTriMatrices()
864 …upTriFactor->csrMat->row_offsets->data().get(), upTriFactor->csrMat->column_indices->data().get(),… in MatSeqAIJCUSPARSEBuildICCTriMatrices()
870 …upTriFactor->csrMat->row_offsets->data().get(), upTriFactor->csrMat->column_indices->data().get(),… in MatSeqAIJCUSPARSEBuildICCTriMatrices()
905 loTriFactor->csrMat->column_indices = new THRUSTINTARRAY32(a->nz); in MatSeqAIJCUSPARSEBuildICCTriMatrices()
906 loTriFactor->csrMat->column_indices->assign(AjUp, AjUp + a->nz); in MatSeqAIJCUSPARSEBuildICCTriMatrices()
916 …loTriFactor->csrMat->row_offsets->data().get(), loTriFactor->csrMat->column_indices->data().get(),… in MatSeqAIJCUSPARSEBuildICCTriMatrices()
922 …loTriFactor->csrMat->row_offsets->data().get(), loTriFactor->csrMat->column_indices->data().get(),… in MatSeqAIJCUSPARSEBuildICCTriMatrices()
1086 loTriFactorT->csrMat->column_indices = new THRUSTINTARRAY32(loTriFactorT->csrMat->num_entries); in MatSeqAIJCUSPARSEAnalyzeTransposeForSolve()
1092 …loTriFactor->csrMat->row_offsets->data().get(), loTriFactor->csrMat->column_indices->data().get(),… in MatSeqAIJCUSPARSEAnalyzeTransposeForSolve()
1093 …loTriFactorT->csrMat->column_indices->data().get(), cusparse_scalartype, CUSPARSE_ACTION_NUMERIC, … in MatSeqAIJCUSPARSEAnalyzeTransposeForSolve()
1101 … loTriFactor->csrMat->column_indices->data().get(), loTriFactorT->csrMat->values->data().get(), in MatSeqAIJCUSPARSEAnalyzeTransposeForSolve()
1103 …TriFactorT->csrMat->row_offsets->data().get(), loTriFactorT->csrMat->column_indices->data().get(),… in MatSeqAIJCUSPARSEAnalyzeTransposeForSolve()
1105 …loTriFactorT->csrMat->column_indices->data().get(), loTriFactorT->csrMat->row_offsets->data().get(… in MatSeqAIJCUSPARSEAnalyzeTransposeForSolve()
1118 …TriFactorT->csrMat->row_offsets->data().get(), loTriFactorT->csrMat->column_indices->data().get(),… in MatSeqAIJCUSPARSEAnalyzeTransposeForSolve()
1124 …TriFactorT->csrMat->row_offsets->data().get(), loTriFactorT->csrMat->column_indices->data().get(),… in MatSeqAIJCUSPARSEAnalyzeTransposeForSolve()
1162 upTriFactorT->csrMat->column_indices = new THRUSTINTARRAY32(upTriFactorT->csrMat->num_entries); in MatSeqAIJCUSPARSEAnalyzeTransposeForSolve()
1168 …upTriFactor->csrMat->row_offsets->data().get(), upTriFactor->csrMat->column_indices->data().get(),… in MatSeqAIJCUSPARSEAnalyzeTransposeForSolve()
1169 …upTriFactorT->csrMat->column_indices->data().get(), cusparse_scalartype, CUSPARSE_ACTION_NUMERIC, … in MatSeqAIJCUSPARSEAnalyzeTransposeForSolve()
1177 … upTriFactor->csrMat->column_indices->data().get(), upTriFactorT->csrMat->values->data().get(), in MatSeqAIJCUSPARSEAnalyzeTransposeForSolve()
1179 …TriFactorT->csrMat->row_offsets->data().get(), upTriFactorT->csrMat->column_indices->data().get(),… in MatSeqAIJCUSPARSEAnalyzeTransposeForSolve()
1181 …upTriFactorT->csrMat->column_indices->data().get(), upTriFactorT->csrMat->row_offsets->data().get(… in MatSeqAIJCUSPARSEAnalyzeTransposeForSolve()
1194 …TriFactorT->csrMat->row_offsets->data().get(), upTriFactorT->csrMat->column_indices->data().get(),… in MatSeqAIJCUSPARSEAnalyzeTransposeForSolve()
1201 …TriFactorT->csrMat->row_offsets->data().get(), upTriFactorT->csrMat->column_indices->data().get(),… in MatSeqAIJCUSPARSEAnalyzeTransposeForSolve()
1256 matrixT->column_indices = new THRUSTINTARRAY32(a->nz); in MatSeqAIJCUSPARSEFormExplicitTranspose()
1264 …, matrixT->num_entries, matrixT->row_offsets->data().get(), matrixT->column_indices->data().get(),… in MatSeqAIJCUSPARSEFormExplicitTranspose()
1276 …, matrixT->num_entries, matrixT->row_offsets->data().get(), matrixT->column_indices->data().get(),… in MatSeqAIJCUSPARSEFormExplicitTranspose()
1296 temp->column_indices = new THRUSTINTARRAY32(a->nz); in MatSeqAIJCUSPARSEFormExplicitTranspose()
1299 …, temp->values->data().get(), temp->row_offsets->data().get(), temp->column_indices->data().get()); in MatSeqAIJCUSPARSEFormExplicitTranspose()
1307 tempT->column_indices = new THRUSTINTARRAY32(a->nz); in MatSeqAIJCUSPARSEFormExplicitTranspose()
1310 …, temp->values->data().get(), temp->row_offsets->data().get(), temp->column_indices->data().get(),… in MatSeqAIJCUSPARSEFormExplicitTranspose()
1311 …tempT->column_indices->data().get(), tempT->row_offsets->data().get(), CUSPARSE_ACTION_NUMERIC, in… in MatSeqAIJCUSPARSEFormExplicitTranspose()
1318 …empT->values->data().get(), tempT->row_offsets->data().get(), tempT->column_indices->data().get(),… in MatSeqAIJCUSPARSEFormExplicitTranspose()
1327 if (tempT->column_indices) delete (THRUSTINTARRAY32 *)tempT->column_indices; in MatSeqAIJCUSPARSEFormExplicitTranspose()
1333 if (temp->column_indices) delete (THRUSTINTARRAY32 *)temp->column_indices; in MatSeqAIJCUSPARSEFormExplicitTranspose()
1345 PetscCheck(matrix->column_indices, PETSC_COMM_SELF, PETSC_ERR_GPU, "Missing CsrMatrix cols"); in MatSeqAIJCUSPARSEFormExplicitTranspose()
1349 PetscCheck(matrixT->column_indices, PETSC_COMM_SELF, PETSC_ERR_GPU, "Missing CsrMatrixT cols"); in MatSeqAIJCUSPARSEFormExplicitTranspose()
1364 …>data().get(), cusparsestruct->rowoffsets_gpu->data().get(), matrix->column_indices->data().get(),… in MatSeqAIJCUSPARSEFormExplicitTranspose()
1365 …matrixT->row_offsets->data().get(), matrixT->column_indices->data().get(), cusparse_scalartype, CU… in MatSeqAIJCUSPARSEFormExplicitTranspose()
1378 ….data().get(), cusparsestruct->rowoffsets_gpu->data().get(), matrix->column_indices->data().get(),… in MatSeqAIJCUSPARSEFormExplicitTranspose()
1380 …matrixT->row_offsets->data().get(), matrixT->column_indices->data().get(), cusparse_scalartype, CU… in MatSeqAIJCUSPARSEFormExplicitTranspose()
1383 …matrixT->column_indices->data().get(), matrixT->row_offsets->data().get(), CUSPARSE_ACTION_NUMERIC… in MatSeqAIJCUSPARSEFormExplicitTranspose()
1566 …TriFactorT->csrMat->row_offsets->data().get(), upTriFactorT->csrMat->column_indices->data().get(),… in MatSolveTranspose_SeqAIJCUSPARSE()
1570 …TriFactorT->csrMat->row_offsets->data().get(), loTriFactorT->csrMat->column_indices->data().get(),… in MatSolveTranspose_SeqAIJCUSPARSE()
1610 …TriFactorT->csrMat->row_offsets->data().get(), upTriFactorT->csrMat->column_indices->data().get(),… in MatSolveTranspose_SeqAIJCUSPARSE_NaturalOrdering()
1614 …TriFactorT->csrMat->row_offsets->data().get(), loTriFactorT->csrMat->column_indices->data().get(),… in MatSolveTranspose_SeqAIJCUSPARSE_NaturalOrdering()
1648 …loTriFactor->csrMat->row_offsets->data().get(), loTriFactor->csrMat->column_indices->data().get(),… in MatSolve_SeqAIJCUSPARSE()
1652 …upTriFactor->csrMat->row_offsets->data().get(), upTriFactor->csrMat->column_indices->data().get(),… in MatSolve_SeqAIJCUSPARSE()
1681 …loTriFactor->csrMat->row_offsets->data().get(), loTriFactor->csrMat->column_indices->data().get(),… in MatSolve_SeqAIJCUSPARSE_NaturalOrdering()
1685 …upTriFactor->csrMat->row_offsets->data().get(), upTriFactor->csrMat->column_indices->data().get(),… in MatSolve_SeqAIJCUSPARSE_NaturalOrdering()
2413 *j = matrix->column_indices->data().get(); in MatSeqAIJGetCSRAndMemType_SeqAIJCUSPARSE()
2496 PetscCallCXX(mat->column_indices = new THRUSTINTARRAY32(nnz)); in MatSeqAIJCUSPARSECopyToGPU()
2497 mat->column_indices->assign(a->j, a->j + nnz); in MatSeqAIJCUSPARSECopyToGPU()
2506 …at->num_cols, mat->num_entries, mat->row_offsets->data().get(), mat->column_indices->data().get(),… in MatSeqAIJCUSPARSECopyToGPU()
2522 PetscCallCXX(mat->column_indices = new THRUSTINTARRAY32(nnz)); in MatSeqAIJCUSPARSECopyToGPU()
2523 mat->column_indices->assign(a->j, a->j + nnz); in MatSeqAIJCUSPARSECopyToGPU()
2531 …scr, mat->values->data().get(), mat->row_offsets->data().get(), mat->column_indices->data().get(),… in MatSeqAIJCUSPARSECopyToGPU()
2538 if (mat->column_indices) delete (THRUSTINTARRAY32 *)mat->column_indices; in MatSeqAIJCUSPARSECopyToGPU()
2759 …ols, csrmat->num_entries, csrmat->row_offsets->data().get(), csrmat->column_indices->data().get(),… in MatProductNumeric_SeqAIJCUSPARSE_SeqDENSECUDA()
2803 …at->values->data().get(), csrmat->row_offsets->data().get(), csrmat->column_indices->data().get(),… in MatProductNumeric_SeqAIJCUSPARSE_SeqDENSECUDA()
3003 …, Acsr->values->data().get(), Acsr->row_offsets->data().get(), Acsr->column_indices->data().get(),… in MatProductNumeric_SeqAIJCUSPARSE_SeqAIJCUSPARSE()
3004 ….get(), Bcsr->column_indices->data().get(), Cmat->descr, Ccsr->values->data().get(), Ccsr->row_off… in MatProductNumeric_SeqAIJCUSPARSE_SeqAIJCUSPARSE()
3154 Ccsr->column_indices = new THRUSTINTARRAY32(c->nz); in MatProductSymbolic_SeqAIJCUSPARSE_SeqAIJCUSPARSE()
3173 Bcsr->column_indices = cBcsr->column_indices; in MatProductSymbolic_SeqAIJCUSPARSE_SeqAIJCUSPARSE()
3184 …>num_cols, Bcsr->num_entries, Bcsr->row_offsets->data().get(), Bcsr->column_indices->data().get(),… in MatProductSymbolic_SeqAIJCUSPARSE_SeqAIJCUSPARSE()
3258 Ccsr->column_indices = new THRUSTINTARRAY32(c->nz); in MatProductSymbolic_SeqAIJCUSPARSE_SeqAIJCUSPARSE()
3263 …srSetPointers(Cmat->matDescr, Ccsr->row_offsets->data().get(), Ccsr->column_indices->data().get(),… in MatProductSymbolic_SeqAIJCUSPARSE_SeqAIJCUSPARSE()
3302 Ccsr->column_indices = new THRUSTINTARRAY32(c->nz); in MatProductSymbolic_SeqAIJCUSPARSE_SeqAIJCUSPARSE()
3306 …srSetPointers(Cmat->matDescr, Ccsr->row_offsets->data().get(), Ccsr->column_indices->data().get(),… in MatProductSymbolic_SeqAIJCUSPARSE_SeqAIJCUSPARSE()
3313 …at->descr, Acsr->num_entries, Acsr->row_offsets->data().get(), Acsr->column_indices->data().get(),… in MatProductSymbolic_SeqAIJCUSPARSE_SeqAIJCUSPARSE()
3314 …Bcsr->row_offsets->data().get(), Bcsr->column_indices->data().get(), Cmat->descr, Ccsr->row_offset… in MatProductSymbolic_SeqAIJCUSPARSE_SeqAIJCUSPARSE()
3317 Ccsr->column_indices = new THRUSTINTARRAY32(c->nz); in MatProductSymbolic_SeqAIJCUSPARSE_SeqAIJCUSPARSE()
3326 …, Acsr->values->data().get(), Acsr->row_offsets->data().get(), Acsr->column_indices->data().get(),… in MatProductSymbolic_SeqAIJCUSPARSE_SeqAIJCUSPARSE()
3327 ….get(), Bcsr->column_indices->data().get(), Cmat->descr, Ccsr->values->data().get(), Ccsr->row_off… in MatProductSymbolic_SeqAIJCUSPARSE_SeqAIJCUSPARSE()
3340 THRUSTINTARRAY jj(Ccsr->column_indices->size()); in MatProductSymbolic_SeqAIJCUSPARSE_SeqAIJCUSPARSE()
3342 jj = *Ccsr->column_indices; in MatProductSymbolic_SeqAIJCUSPARSE_SeqAIJCUSPARSE()
3345 …PetscCallCUDA(cudaMemcpy(c->j, jj.data().get(), Ccsr->column_indices->size() * sizeof(PetscInt), c… in MatProductSymbolic_SeqAIJCUSPARSE_SeqAIJCUSPARSE()
3350 …PetscCallCUDA(cudaMemcpy(c->j, Ccsr->column_indices->data().get(), Ccsr->column_indices->size() * … in MatProductSymbolic_SeqAIJCUSPARSE_SeqAIJCUSPARSE()
3362 …PetscCall(PetscLogGpuToCpu((Ccsr->column_indices->size() + Ccsr->row_offsets->size()) * sizeof(Pet… in MatProductSymbolic_SeqAIJCUSPARSE_SeqAIJCUSPARSE()
3655 …at->num_cols, mat->num_entries, mat->row_offsets->data().get(), mat->column_indices->data().get(),… in MatMultAddKernel_SeqAIJCUSPARSE()
3679 …scr, mat->values->data().get(), mat->row_offsets->data().get(), mat->column_indices->data().get(),… in MatMultAddKernel_SeqAIJCUSPARSE()
3769 …6, 0, PetscDefaultCudaStream>>>(mat->row_offsets->data().get(), mat->column_indices->data().get(),… in MatGetDiagonal_SeqAIJCUSPARSE()
3893 … = thrust::equal(thrust::device, csry->column_indices->begin(), csry->column_indices->end(), csrx-… in MatAXPY_SeqAIJCUSPARSE()
3910 …a, cx->mat->descr, x->nz, ax, csrx->row_offsets->data().get(), csrx->column_indices->data().get(),… in MatAXPY_SeqAIJCUSPARSE()
3911 …csry->column_indices->data().get(), cy->mat->descr, ay, csry->row_offsets->data().get(), csry->col… in MatAXPY_SeqAIJCUSPARSE()
3914 …a, cx->mat->descr, x->nz, ax, csrx->row_offsets->data().get(), csrx->column_indices->data().get(),… in MatAXPY_SeqAIJCUSPARSE()
3915 …csry->column_indices->data().get(), cy->mat->descr, ay, csry->row_offsets->data().get(), csry->col… in MatAXPY_SeqAIJCUSPARSE()
3921 …a, cx->mat->descr, x->nz, ax, csrx->row_offsets->data().get(), csrx->column_indices->data().get(),… in MatAXPY_SeqAIJCUSPARSE()
3922 …csry->column_indices->data().get(), cy->mat->descr, ay, csry->row_offsets->data().get(), csry->col… in MatAXPY_SeqAIJCUSPARSE()
4197 delete (*mat)->column_indices; in CsrMatrix_Destroy()
4508 if (j) *j = csr->column_indices->data().get(); in MatSeqAIJCUSPARSEGetIJ()
4795 Annz = (PetscInt)Acsr->column_indices->size(); in MatSeqAIJCUSPARSEMergeMats()
4796 Bnnz = (PetscInt)Bcsr->column_indices->size(); in MatSeqAIJCUSPARSEMergeMats()
4799 Ccsr->column_indices = new THRUSTINTARRAY32(c->nz); in MatSeqAIJCUSPARSEMergeMats()
4834 auto Bcib = thrust::make_transform_iterator(Bcsr->column_indices->begin(), Shift(A->cmap->n)); in MatSeqAIJCUSPARSEMergeMats()
4835 auto Bcie = thrust::make_transform_iterator(Bcsr->column_indices->end(), Shift(A->cmap->n)); in MatSeqAIJCUSPARSEMergeMats()
4838 auto Bcib = Bcsr->column_indices->begin(); in MatSeqAIJCUSPARSEMergeMats()
4839 auto Bcie = Bcsr->column_indices->end(); in MatSeqAIJCUSPARSEMergeMats()
4843 …auto Azb = thrust::make_zip_iterator(thrust::make_tuple(Acoo->begin(), Acsr->column_indices->beg… in MatSeqAIJCUSPARSEMergeMats()
4844 …auto Aze = thrust::make_zip_iterator(thrust::make_tuple(Acoo->end(), Acsr->column_indices->end()… in MatSeqAIJCUSPARSEMergeMats()
4847 …auto Czb = thrust::make_zip_iterator(thrust::make_tuple(Ccoo->begin(), Ccsr->column_indices->beg… in MatSeqAIJCUSPARSEMergeMats()
4880 …>num_cols, Ccsr->num_entries, Ccsr->row_offsets->data().get(), Ccsr->column_indices->data().get(),… in MatSeqAIJCUSPARSEMergeMats()
4902 CcsrT->column_indices = new THRUSTINTARRAY32(c->nz); in MatSeqAIJCUSPARSEMergeMats()
4920 auto cT = CcsrT->column_indices->begin(); in MatSeqAIJCUSPARSEMergeMats()
4921 if (AT) cT = thrust::copy(AcsrT->column_indices->begin(), AcsrT->column_indices->end(), cT); in MatSeqAIJCUSPARSEMergeMats()
4922 if (BT) thrust::copy(BcsrT->column_indices->begin(), BcsrT->column_indices->end(), cT); in MatSeqAIJCUSPARSEMergeMats()
4938 …m_cols, CcsrT->num_entries, CcsrT->row_offsets->data().get(), CcsrT->column_indices->data().get(),… in MatSeqAIJCUSPARSEMergeMats()
4951 THRUSTINTARRAY jj(Ccsr->column_indices->size()); in MatSeqAIJCUSPARSEMergeMats()
4953 jj = *Ccsr->column_indices; in MatSeqAIJCUSPARSEMergeMats()
4955 …PetscCallCUDA(cudaMemcpy(c->j, jj.data().get(), Ccsr->column_indices->size() * sizeof(PetscInt), c… in MatSeqAIJCUSPARSEMergeMats()
4958 …PetscCallCUDA(cudaMemcpy(c->j, Ccsr->column_indices->data().get(), Ccsr->column_indices->size() * … in MatSeqAIJCUSPARSEMergeMats()
4960 …PetscCall(PetscLogGpuToCpu((Ccsr->column_indices->size() + Ccsr->row_offsets->size()) * sizeof(Pet… in MatSeqAIJCUSPARSEMergeMats()