Lines Matching refs:upTriFactor
387 …Mat_SeqAIJHIPSPARSETriFactorStruct *upTriFactor = (Mat_SeqAIJHIPSPARSETriFactorStruct *)hi… in MatSeqAIJHIPSPARSEBuildILUUpperTriMatrix() local
400 if (!upTriFactor) { in MatSeqAIJHIPSPARSEBuildILUUpperTriMatrix()
428 PetscCall(PetscNew(&upTriFactor)); in MatSeqAIJHIPSPARSEBuildILUUpperTriMatrix()
429 upTriFactor->solvePolicy = HIPSPARSE_SOLVE_POLICY_USE_LEVEL; in MatSeqAIJHIPSPARSEBuildILUUpperTriMatrix()
432 PetscCallHIPSPARSE(hipsparseCreateMatDescr(&upTriFactor->descr)); in MatSeqAIJHIPSPARSEBuildILUUpperTriMatrix()
433 PetscCallHIPSPARSE(hipsparseSetMatIndexBase(upTriFactor->descr, HIPSPARSE_INDEX_BASE_ZERO)); in MatSeqAIJHIPSPARSEBuildILUUpperTriMatrix()
434 PetscCallHIPSPARSE(hipsparseSetMatType(upTriFactor->descr, HIPSPARSE_MATRIX_TYPE_GENERAL)); in MatSeqAIJHIPSPARSEBuildILUUpperTriMatrix()
435 PetscCallHIPSPARSE(hipsparseSetMatFillMode(upTriFactor->descr, HIPSPARSE_FILL_MODE_UPPER)); in MatSeqAIJHIPSPARSEBuildILUUpperTriMatrix()
436 … PetscCallHIPSPARSE(hipsparseSetMatDiagType(upTriFactor->descr, HIPSPARSE_DIAG_TYPE_NON_UNIT)); in MatSeqAIJHIPSPARSEBuildILUUpperTriMatrix()
439 upTriFactor->solveOp = HIPSPARSE_OPERATION_NON_TRANSPOSE; in MatSeqAIJHIPSPARSEBuildILUUpperTriMatrix()
442 upTriFactor->csrMat = new CsrMatrix; in MatSeqAIJHIPSPARSEBuildILUUpperTriMatrix()
443 upTriFactor->csrMat->num_rows = n; in MatSeqAIJHIPSPARSEBuildILUUpperTriMatrix()
444 upTriFactor->csrMat->num_cols = n; in MatSeqAIJHIPSPARSEBuildILUUpperTriMatrix()
445 upTriFactor->csrMat->num_entries = nzUpper; in MatSeqAIJHIPSPARSEBuildILUUpperTriMatrix()
446 upTriFactor->csrMat->row_offsets = new THRUSTINTARRAY32(n + 1); in MatSeqAIJHIPSPARSEBuildILUUpperTriMatrix()
447 upTriFactor->csrMat->column_indices = new THRUSTINTARRAY32(nzUpper); in MatSeqAIJHIPSPARSEBuildILUUpperTriMatrix()
448 upTriFactor->csrMat->values = new THRUSTARRAY(nzUpper); in MatSeqAIJHIPSPARSEBuildILUUpperTriMatrix()
449 upTriFactor->csrMat->row_offsets->assign(AiUp, AiUp + n + 1); in MatSeqAIJHIPSPARSEBuildILUUpperTriMatrix()
450 upTriFactor->csrMat->column_indices->assign(AjUp, AjUp + nzUpper); in MatSeqAIJHIPSPARSEBuildILUUpperTriMatrix()
451 upTriFactor->csrMat->values->assign(AAUp, AAUp + nzUpper); in MatSeqAIJHIPSPARSEBuildILUUpperTriMatrix()
455 PetscCallHIPSPARSE(hipsparseCreateCsrsvInfo(&upTriFactor->solveInfo)); in MatSeqAIJHIPSPARSEBuildILUUpperTriMatrix()
456 …Factors->handle, upTriFactor->solveOp, upTriFactor->csrMat->num_rows, upTriFactor->csrMat->num_ent… in MatSeqAIJHIPSPARSEBuildILUUpperTriMatrix()
457 …upTriFactor->csrMat->row_offsets->data().get(), upTriFactor->csrMat->column_indices->data().get(),… in MatSeqAIJHIPSPARSEBuildILUUpperTriMatrix()
458 PetscCallHIP(hipMalloc(&upTriFactor->solveBuffer, upTriFactor->solveBufferSize)); in MatSeqAIJHIPSPARSEBuildILUUpperTriMatrix()
461 …Factors->handle, upTriFactor->solveOp, upTriFactor->csrMat->num_rows, upTriFactor->csrMat->num_ent… in MatSeqAIJHIPSPARSEBuildILUUpperTriMatrix()
462 …upTriFactor->csrMat->row_offsets->data().get(), upTriFactor->csrMat->column_indices->data().get(),… in MatSeqAIJHIPSPARSEBuildILUUpperTriMatrix()
468 ((Mat_SeqAIJHIPSPARSETriFactors *)A->spptr)->upTriFactorPtr = upTriFactor; in MatSeqAIJHIPSPARSEBuildILUUpperTriMatrix()
469 upTriFactor->AA_h = AAUp; in MatSeqAIJHIPSPARSEBuildILUUpperTriMatrix()
474 …if (!upTriFactor->AA_h) PetscCallHIP(hipHostMalloc((void **)&upTriFactor->AA_h, nzUpper * sizeof(P… in MatSeqAIJHIPSPARSEBuildILUUpperTriMatrix()
483 upTriFactor->AA_h[offset] = 1. / v[nz]; in MatSeqAIJHIPSPARSEBuildILUUpperTriMatrix()
484 PetscCall(PetscArraycpy(&upTriFactor->AA_h[offset + 1], v, nz)); in MatSeqAIJHIPSPARSEBuildILUUpperTriMatrix()
486 upTriFactor->csrMat->values->assign(upTriFactor->AA_h, upTriFactor->AA_h + nzUpper); in MatSeqAIJHIPSPARSEBuildILUUpperTriMatrix()
543 …Mat_SeqAIJHIPSPARSETriFactorStruct *upTriFactor = (Mat_SeqAIJHIPSPARSETriFactorStruct *)hi… in MatSeqAIJHIPSPARSEBuildICCTriMatrices() local
558 if (!upTriFactor && !loTriFactor) { in MatSeqAIJHIPSPARSEBuildICCTriMatrices()
592 PetscCall(PetscNew(&upTriFactor)); in MatSeqAIJHIPSPARSEBuildICCTriMatrices()
593 upTriFactor->solvePolicy = HIPSPARSE_SOLVE_POLICY_USE_LEVEL; in MatSeqAIJHIPSPARSEBuildICCTriMatrices()
596 PetscCallHIPSPARSE(hipsparseCreateMatDescr(&upTriFactor->descr)); in MatSeqAIJHIPSPARSEBuildICCTriMatrices()
597 PetscCallHIPSPARSE(hipsparseSetMatIndexBase(upTriFactor->descr, HIPSPARSE_INDEX_BASE_ZERO)); in MatSeqAIJHIPSPARSEBuildICCTriMatrices()
598 PetscCallHIPSPARSE(hipsparseSetMatType(upTriFactor->descr, HIPSPARSE_MATRIX_TYPE_GENERAL)); in MatSeqAIJHIPSPARSEBuildICCTriMatrices()
599 PetscCallHIPSPARSE(hipsparseSetMatFillMode(upTriFactor->descr, HIPSPARSE_FILL_MODE_UPPER)); in MatSeqAIJHIPSPARSEBuildICCTriMatrices()
600 PetscCallHIPSPARSE(hipsparseSetMatDiagType(upTriFactor->descr, HIPSPARSE_DIAG_TYPE_UNIT)); in MatSeqAIJHIPSPARSEBuildICCTriMatrices()
603 upTriFactor->csrMat = new CsrMatrix; in MatSeqAIJHIPSPARSEBuildICCTriMatrices()
604 upTriFactor->csrMat->num_rows = A->rmap->n; in MatSeqAIJHIPSPARSEBuildICCTriMatrices()
605 upTriFactor->csrMat->num_cols = A->cmap->n; in MatSeqAIJHIPSPARSEBuildICCTriMatrices()
606 upTriFactor->csrMat->num_entries = a->nz; in MatSeqAIJHIPSPARSEBuildICCTriMatrices()
607 upTriFactor->csrMat->row_offsets = new THRUSTINTARRAY32(A->rmap->n + 1); in MatSeqAIJHIPSPARSEBuildICCTriMatrices()
608 upTriFactor->csrMat->column_indices = new THRUSTINTARRAY32(a->nz); in MatSeqAIJHIPSPARSEBuildICCTriMatrices()
609 upTriFactor->csrMat->values = new THRUSTARRAY(a->nz); in MatSeqAIJHIPSPARSEBuildICCTriMatrices()
610 upTriFactor->csrMat->row_offsets->assign(AiUp, AiUp + A->rmap->n + 1); in MatSeqAIJHIPSPARSEBuildICCTriMatrices()
611 upTriFactor->csrMat->column_indices->assign(AjUp, AjUp + a->nz); in MatSeqAIJHIPSPARSEBuildICCTriMatrices()
612 upTriFactor->csrMat->values->assign(AAUp, AAUp + a->nz); in MatSeqAIJHIPSPARSEBuildICCTriMatrices()
615 upTriFactor->solveOp = HIPSPARSE_OPERATION_NON_TRANSPOSE; in MatSeqAIJHIPSPARSEBuildICCTriMatrices()
619 PetscCallHIPSPARSE(hipsparseCreateCsrsvInfo(&upTriFactor->solveInfo)); in MatSeqAIJHIPSPARSEBuildICCTriMatrices()
620 …Factors->handle, upTriFactor->solveOp, upTriFactor->csrMat->num_rows, upTriFactor->csrMat->num_ent… in MatSeqAIJHIPSPARSEBuildICCTriMatrices()
621 …upTriFactor->csrMat->row_offsets->data().get(), upTriFactor->csrMat->column_indices->data().get(),… in MatSeqAIJHIPSPARSEBuildICCTriMatrices()
622 PetscCallHIP(hipMalloc(&upTriFactor->solveBuffer, upTriFactor->solveBufferSize)); in MatSeqAIJHIPSPARSEBuildICCTriMatrices()
625 …Factors->handle, upTriFactor->solveOp, upTriFactor->csrMat->num_rows, upTriFactor->csrMat->num_ent… in MatSeqAIJHIPSPARSEBuildICCTriMatrices()
626 …upTriFactor->csrMat->row_offsets->data().get(), upTriFactor->csrMat->column_indices->data().get(),… in MatSeqAIJHIPSPARSEBuildICCTriMatrices()
632 ((Mat_SeqAIJHIPSPARSETriFactors *)A->spptr)->upTriFactorPtr = upTriFactor; in MatSeqAIJHIPSPARSEBuildICCTriMatrices()
702 PetscCheck(upTriFactor, PETSC_COMM_SELF, PETSC_ERR_COR, "Missing hipsparseTriFactors"); in MatSeqAIJHIPSPARSEBuildICCTriMatrices()
704 upTriFactor->csrMat->values->assign(AAUp, AAUp + a->nz); in MatSeqAIJHIPSPARSEBuildICCTriMatrices()
786 …Mat_SeqAIJHIPSPARSETriFactorStruct *upTriFactor = (Mat_SeqAIJHIPSPARSETriFactorStruct *)hi… in MatSeqAIJHIPSPARSEAnalyzeTransposeForSolve() local
873 matrixType = hipsparseGetMatType(upTriFactor->descr); in MatSeqAIJHIPSPARSEAnalyzeTransposeForSolve()
874 indexBase = hipsparseGetMatIndexBase(upTriFactor->descr); in MatSeqAIJHIPSPARSEAnalyzeTransposeForSolve()
875 …fillMode = hipsparseGetMatFillMode(upTriFactor->descr) == HIPSPARSE_FILL_MODE_UPPER ? HIPSPARSE_… in MatSeqAIJHIPSPARSEAnalyzeTransposeForSolve()
876 diagType = hipsparseGetMatDiagType(upTriFactor->descr); in MatSeqAIJHIPSPARSEAnalyzeTransposeForSolve()
890 upTriFactorT->csrMat->num_rows = upTriFactor->csrMat->num_cols; in MatSeqAIJHIPSPARSEAnalyzeTransposeForSolve()
891 upTriFactorT->csrMat->num_cols = upTriFactor->csrMat->num_rows; in MatSeqAIJHIPSPARSEAnalyzeTransposeForSolve()
892 upTriFactorT->csrMat->num_entries = upTriFactor->csrMat->num_entries; in MatSeqAIJHIPSPARSEAnalyzeTransposeForSolve()
907 … upTriFactor->csrMat->num_rows, upTriFactor->csrMat->num_cols, upTriFactor->csrMat->num_entries, u… in MatSeqAIJHIPSPARSEAnalyzeTransposeForSolve()
908 … upTriFactor->csrMat->column_indices->data().get(), upTriFactorT->csrMat->values->data().get(), in MatSeqAIJHIPSPARSEAnalyzeTransposeForSolve()
911 …alartype, HIPSPARSE_ACTION_NUMERIC, indexBase, HIPSPARSE_CSR2CSC_ALG1, upTriFactor->csr2cscBuffer)… in MatSeqAIJHIPSPARSEAnalyzeTransposeForSolve()
1183 …Mat_SeqAIJHIPSPARSETriFactorStruct *upTriFactor = (Mat_SeqAIJHIPSPARSETriFactorStruct *)… in MatSolve_SeqAIJHIPSPARSE() local
1202 …ndle, upTriFactor->solveOp, upTriFactor->csrMat->num_rows, upTriFactor->csrMat->num_entries, &PETS… in MatSolve_SeqAIJHIPSPARSE()
1203 …upTriFactor->csrMat->row_offsets->data().get(), upTriFactor->csrMat->column_indices->data().get(),… in MatSolve_SeqAIJHIPSPARSE()
1221 …Mat_SeqAIJHIPSPARSETriFactorStruct *upTriFactor = (Mat_SeqAIJHIPSPARSETriFactorStruct *)hi… in MatSolve_SeqAIJHIPSPARSE_NaturalOrdering() local
1235 …ndle, upTriFactor->solveOp, upTriFactor->csrMat->num_rows, upTriFactor->csrMat->num_entries, &PETS… in MatSolve_SeqAIJHIPSPARSE_NaturalOrdering()
1236 …upTriFactor->csrMat->row_offsets->data().get(), upTriFactor->csrMat->column_indices->data().get(),… in MatSolve_SeqAIJHIPSPARSE_NaturalOrdering()