Lines Matching refs:loTriFactor
266 …Mat_SeqAIJHIPSPARSETriFactorStruct *loTriFactor = (Mat_SeqAIJHIPSPARSETriFactorStruct *)hi… in MatSeqAIJHIPSPARSEBuildILULowerTriMatrix() local
278 if (!loTriFactor) { in MatSeqAIJHIPSPARSEBuildILULowerTriMatrix()
312 PetscCall(PetscNew(&loTriFactor)); in MatSeqAIJHIPSPARSEBuildILULowerTriMatrix()
313 loTriFactor->solvePolicy = HIPSPARSE_SOLVE_POLICY_USE_LEVEL; in MatSeqAIJHIPSPARSEBuildILULowerTriMatrix()
315 PetscCallHIPSPARSE(hipsparseCreateMatDescr(&loTriFactor->descr)); in MatSeqAIJHIPSPARSEBuildILULowerTriMatrix()
316 PetscCallHIPSPARSE(hipsparseSetMatIndexBase(loTriFactor->descr, HIPSPARSE_INDEX_BASE_ZERO)); in MatSeqAIJHIPSPARSEBuildILULowerTriMatrix()
317 PetscCallHIPSPARSE(hipsparseSetMatType(loTriFactor->descr, HIPSPARSE_MATRIX_TYPE_GENERAL)); in MatSeqAIJHIPSPARSEBuildILULowerTriMatrix()
318 PetscCallHIPSPARSE(hipsparseSetMatFillMode(loTriFactor->descr, HIPSPARSE_FILL_MODE_LOWER)); in MatSeqAIJHIPSPARSEBuildILULowerTriMatrix()
319 PetscCallHIPSPARSE(hipsparseSetMatDiagType(loTriFactor->descr, HIPSPARSE_DIAG_TYPE_UNIT)); in MatSeqAIJHIPSPARSEBuildILULowerTriMatrix()
322 loTriFactor->solveOp = HIPSPARSE_OPERATION_NON_TRANSPOSE; in MatSeqAIJHIPSPARSEBuildILULowerTriMatrix()
325 loTriFactor->csrMat = new CsrMatrix; in MatSeqAIJHIPSPARSEBuildILULowerTriMatrix()
326 loTriFactor->csrMat->num_rows = n; in MatSeqAIJHIPSPARSEBuildILULowerTriMatrix()
327 loTriFactor->csrMat->num_cols = n; in MatSeqAIJHIPSPARSEBuildILULowerTriMatrix()
328 loTriFactor->csrMat->num_entries = nzLower; in MatSeqAIJHIPSPARSEBuildILULowerTriMatrix()
329 loTriFactor->csrMat->row_offsets = new THRUSTINTARRAY32(n + 1); in MatSeqAIJHIPSPARSEBuildILULowerTriMatrix()
330 loTriFactor->csrMat->column_indices = new THRUSTINTARRAY32(nzLower); in MatSeqAIJHIPSPARSEBuildILULowerTriMatrix()
331 loTriFactor->csrMat->values = new THRUSTARRAY(nzLower); in MatSeqAIJHIPSPARSEBuildILULowerTriMatrix()
333 loTriFactor->csrMat->row_offsets->assign(AiLo, AiLo + n + 1); in MatSeqAIJHIPSPARSEBuildILULowerTriMatrix()
334 loTriFactor->csrMat->column_indices->assign(AjLo, AjLo + nzLower); in MatSeqAIJHIPSPARSEBuildILULowerTriMatrix()
335 loTriFactor->csrMat->values->assign(AALo, AALo + nzLower); in MatSeqAIJHIPSPARSEBuildILULowerTriMatrix()
339 PetscCallHIPSPARSE(hipsparseCreateCsrsvInfo(&loTriFactor->solveInfo)); in MatSeqAIJHIPSPARSEBuildILULowerTriMatrix()
340 …Factors->handle, loTriFactor->solveOp, loTriFactor->csrMat->num_rows, loTriFactor->csrMat->num_ent… in MatSeqAIJHIPSPARSEBuildILULowerTriMatrix()
341 …loTriFactor->csrMat->row_offsets->data().get(), loTriFactor->csrMat->column_indices->data().get(),… in MatSeqAIJHIPSPARSEBuildILULowerTriMatrix()
342 PetscCallHIP(hipMalloc(&loTriFactor->solveBuffer, loTriFactor->solveBufferSize)); in MatSeqAIJHIPSPARSEBuildILULowerTriMatrix()
345 …Factors->handle, loTriFactor->solveOp, loTriFactor->csrMat->num_rows, loTriFactor->csrMat->num_ent… in MatSeqAIJHIPSPARSEBuildILULowerTriMatrix()
346 …loTriFactor->csrMat->row_offsets->data().get(), loTriFactor->csrMat->column_indices->data().get(),… in MatSeqAIJHIPSPARSEBuildILULowerTriMatrix()
352 ((Mat_SeqAIJHIPSPARSETriFactors *)A->spptr)->loTriFactorPtr = loTriFactor; in MatSeqAIJHIPSPARSEBuildILULowerTriMatrix()
353 loTriFactor->AA_h = AALo; in MatSeqAIJHIPSPARSEBuildILULowerTriMatrix()
358 …if (!loTriFactor->AA_h) PetscCallHIP(hipHostMalloc((void **)&loTriFactor->AA_h, nzLower * sizeof(P… in MatSeqAIJHIPSPARSEBuildILULowerTriMatrix()
360 loTriFactor->AA_h[0] = 1.0; in MatSeqAIJHIPSPARSEBuildILULowerTriMatrix()
366 PetscCall(PetscArraycpy(&loTriFactor->AA_h[offset], v, nz)); in MatSeqAIJHIPSPARSEBuildILULowerTriMatrix()
368 loTriFactor->AA_h[offset] = 1.0; in MatSeqAIJHIPSPARSEBuildILULowerTriMatrix()
372 loTriFactor->csrMat->values->assign(loTriFactor->AA_h, loTriFactor->AA_h + nzLower); in MatSeqAIJHIPSPARSEBuildILULowerTriMatrix()
542 …Mat_SeqAIJHIPSPARSETriFactorStruct *loTriFactor = (Mat_SeqAIJHIPSPARSETriFactorStruct *)hi… in MatSeqAIJHIPSPARSEBuildICCTriMatrices() local
558 if (!upTriFactor && !loTriFactor) { in MatSeqAIJHIPSPARSEBuildICCTriMatrices()
635 PetscCall(PetscNew(&loTriFactor)); in MatSeqAIJHIPSPARSEBuildICCTriMatrices()
636 loTriFactor->solvePolicy = HIPSPARSE_SOLVE_POLICY_USE_LEVEL; in MatSeqAIJHIPSPARSEBuildICCTriMatrices()
639 PetscCallHIPSPARSE(hipsparseCreateMatDescr(&loTriFactor->descr)); in MatSeqAIJHIPSPARSEBuildICCTriMatrices()
640 PetscCallHIPSPARSE(hipsparseSetMatIndexBase(loTriFactor->descr, HIPSPARSE_INDEX_BASE_ZERO)); in MatSeqAIJHIPSPARSEBuildICCTriMatrices()
641 PetscCallHIPSPARSE(hipsparseSetMatType(loTriFactor->descr, HIPSPARSE_MATRIX_TYPE_GENERAL)); in MatSeqAIJHIPSPARSEBuildICCTriMatrices()
642 PetscCallHIPSPARSE(hipsparseSetMatFillMode(loTriFactor->descr, HIPSPARSE_FILL_MODE_UPPER)); in MatSeqAIJHIPSPARSEBuildICCTriMatrices()
643 … PetscCallHIPSPARSE(hipsparseSetMatDiagType(loTriFactor->descr, HIPSPARSE_DIAG_TYPE_NON_UNIT)); in MatSeqAIJHIPSPARSEBuildICCTriMatrices()
646 loTriFactor->solveOp = HIPSPARSE_OPERATION_TRANSPOSE; in MatSeqAIJHIPSPARSEBuildICCTriMatrices()
649 loTriFactor->csrMat = new CsrMatrix; in MatSeqAIJHIPSPARSEBuildICCTriMatrices()
650 loTriFactor->csrMat->num_rows = A->rmap->n; in MatSeqAIJHIPSPARSEBuildICCTriMatrices()
651 loTriFactor->csrMat->num_cols = A->cmap->n; in MatSeqAIJHIPSPARSEBuildICCTriMatrices()
652 loTriFactor->csrMat->num_entries = a->nz; in MatSeqAIJHIPSPARSEBuildICCTriMatrices()
653 loTriFactor->csrMat->row_offsets = new THRUSTINTARRAY32(A->rmap->n + 1); in MatSeqAIJHIPSPARSEBuildICCTriMatrices()
654 loTriFactor->csrMat->column_indices = new THRUSTINTARRAY32(a->nz); in MatSeqAIJHIPSPARSEBuildICCTriMatrices()
655 loTriFactor->csrMat->values = new THRUSTARRAY(a->nz); in MatSeqAIJHIPSPARSEBuildICCTriMatrices()
656 loTriFactor->csrMat->row_offsets->assign(AiUp, AiUp + A->rmap->n + 1); in MatSeqAIJHIPSPARSEBuildICCTriMatrices()
657 loTriFactor->csrMat->column_indices->assign(AjUp, AjUp + a->nz); in MatSeqAIJHIPSPARSEBuildICCTriMatrices()
658 loTriFactor->csrMat->values->assign(AALo, AALo + a->nz); in MatSeqAIJHIPSPARSEBuildICCTriMatrices()
662 PetscCallHIPSPARSE(hipsparseCreateCsrsvInfo(&loTriFactor->solveInfo)); in MatSeqAIJHIPSPARSEBuildICCTriMatrices()
663 …Factors->handle, loTriFactor->solveOp, loTriFactor->csrMat->num_rows, loTriFactor->csrMat->num_ent… in MatSeqAIJHIPSPARSEBuildICCTriMatrices()
664 …loTriFactor->csrMat->row_offsets->data().get(), loTriFactor->csrMat->column_indices->data().get(),… in MatSeqAIJHIPSPARSEBuildICCTriMatrices()
665 PetscCallHIP(hipMalloc(&loTriFactor->solveBuffer, loTriFactor->solveBufferSize)); in MatSeqAIJHIPSPARSEBuildICCTriMatrices()
668 …Factors->handle, loTriFactor->solveOp, loTriFactor->csrMat->num_rows, loTriFactor->csrMat->num_ent… in MatSeqAIJHIPSPARSEBuildICCTriMatrices()
669 …loTriFactor->csrMat->row_offsets->data().get(), loTriFactor->csrMat->column_indices->data().get(),… in MatSeqAIJHIPSPARSEBuildICCTriMatrices()
675 ((Mat_SeqAIJHIPSPARSETriFactors *)A->spptr)->loTriFactorPtr = loTriFactor; in MatSeqAIJHIPSPARSEBuildICCTriMatrices()
703 PetscCheck(loTriFactor, PETSC_COMM_SELF, PETSC_ERR_COR, "Missing hipsparseTriFactors"); in MatSeqAIJHIPSPARSEBuildICCTriMatrices()
705 loTriFactor->csrMat->values->assign(AALo, AALo + a->nz); in MatSeqAIJHIPSPARSEBuildICCTriMatrices()
785 …Mat_SeqAIJHIPSPARSETriFactorStruct *loTriFactor = (Mat_SeqAIJHIPSPARSETriFactorStruct *)hi… in MatSeqAIJHIPSPARSEAnalyzeTransposeForSolve() local
800 matrixType = hipsparseGetMatType(loTriFactor->descr); in MatSeqAIJHIPSPARSEAnalyzeTransposeForSolve()
801 indexBase = hipsparseGetMatIndexBase(loTriFactor->descr); in MatSeqAIJHIPSPARSEAnalyzeTransposeForSolve()
802 …fillMode = hipsparseGetMatFillMode(loTriFactor->descr) == HIPSPARSE_FILL_MODE_UPPER ? HIPSPARSE_… in MatSeqAIJHIPSPARSEAnalyzeTransposeForSolve()
803 diagType = hipsparseGetMatDiagType(loTriFactor->descr); in MatSeqAIJHIPSPARSEAnalyzeTransposeForSolve()
817 loTriFactorT->csrMat->num_rows = loTriFactor->csrMat->num_cols; in MatSeqAIJHIPSPARSEAnalyzeTransposeForSolve()
818 loTriFactorT->csrMat->num_cols = loTriFactor->csrMat->num_rows; in MatSeqAIJHIPSPARSEAnalyzeTransposeForSolve()
819 loTriFactorT->csrMat->num_entries = loTriFactor->csrMat->num_entries; in MatSeqAIJHIPSPARSEAnalyzeTransposeForSolve()
835 … loTriFactor->csrMat->num_rows, loTriFactor->csrMat->num_cols, loTriFactor->csrMat->num_entries, l… in MatSeqAIJHIPSPARSEAnalyzeTransposeForSolve()
836 … loTriFactor->csrMat->column_indices->data().get(), loTriFactorT->csrMat->values->data().get(), in MatSeqAIJHIPSPARSEAnalyzeTransposeForSolve()
839 …alartype, HIPSPARSE_ACTION_NUMERIC, indexBase, HIPSPARSE_CSR2CSC_ALG1, loTriFactor->csr2cscBuffer)… in MatSeqAIJHIPSPARSEAnalyzeTransposeForSolve()
1182 …Mat_SeqAIJHIPSPARSETriFactorStruct *loTriFactor = (Mat_SeqAIJHIPSPARSETriFactorStruct *)… in MatSolve_SeqAIJHIPSPARSE() local
1198 …ndle, loTriFactor->solveOp, loTriFactor->csrMat->num_rows, loTriFactor->csrMat->num_entries, &PETS… in MatSolve_SeqAIJHIPSPARSE()
1199 …loTriFactor->csrMat->row_offsets->data().get(), loTriFactor->csrMat->column_indices->data().get(),… in MatSolve_SeqAIJHIPSPARSE()
1220 …Mat_SeqAIJHIPSPARSETriFactorStruct *loTriFactor = (Mat_SeqAIJHIPSPARSETriFactorStruct *)hi… in MatSolve_SeqAIJHIPSPARSE_NaturalOrdering() local
1231 …ndle, loTriFactor->solveOp, loTriFactor->csrMat->num_rows, loTriFactor->csrMat->num_entries, &PETS… in MatSolve_SeqAIJHIPSPARSE_NaturalOrdering()
1232 …loTriFactor->csrMat->row_offsets->data().get(), loTriFactor->csrMat->column_indices->data().get(),… in MatSolve_SeqAIJHIPSPARSE_NaturalOrdering()