Lines Matching refs:nzUpper
446 PetscInt i, nz, nzUpper, offset; in MatSeqAIJCUSPARSEBuildILUUpperTriMatrix() local
454 nzUpper = adiag[0] - adiag[n]; in MatSeqAIJCUSPARSEBuildILUUpperTriMatrix()
458 PetscCallCUDA(cudaMallocHost((void **)&AAUp, nzUpper * sizeof(PetscScalar))); in MatSeqAIJCUSPARSEBuildILUUpperTriMatrix()
462 PetscCallCUDA(cudaMallocHost((void **)&AjUp, nzUpper * sizeof(PetscInt))); in MatSeqAIJCUSPARSEBuildILUUpperTriMatrix()
466 AiUp[n] = nzUpper; in MatSeqAIJCUSPARSEBuildILUUpperTriMatrix()
467 offset = nzUpper; in MatSeqAIJCUSPARSEBuildILUUpperTriMatrix()
509 upTriFactor->csrMat->num_entries = nzUpper; in MatSeqAIJCUSPARSEBuildILUUpperTriMatrix()
514 upTriFactor->csrMat->column_indices = new THRUSTINTARRAY32(nzUpper); in MatSeqAIJCUSPARSEBuildILUUpperTriMatrix()
515 upTriFactor->csrMat->column_indices->assign(AjUp, AjUp + nzUpper); in MatSeqAIJCUSPARSEBuildILUUpperTriMatrix()
517 upTriFactor->csrMat->values = new THRUSTARRAY(nzUpper); in MatSeqAIJCUSPARSEBuildILUUpperTriMatrix()
518 upTriFactor->csrMat->values->assign(AAUp, AAUp + nzUpper); in MatSeqAIJCUSPARSEBuildILUUpperTriMatrix()
541 … PetscCall(PetscLogCpuToGpu((n + 1 + nzUpper) * sizeof(int) + nzUpper * sizeof(PetscScalar))); in MatSeqAIJCUSPARSEBuildILUUpperTriMatrix()
543 …if (!upTriFactor->AA_h) PetscCallCUDA(cudaMallocHost((void **)&upTriFactor->AA_h, nzUpper * sizeof… in MatSeqAIJCUSPARSEBuildILUUpperTriMatrix()
545 offset = nzUpper; in MatSeqAIJCUSPARSEBuildILUUpperTriMatrix()
559 upTriFactor->csrMat->values->assign(upTriFactor->AA_h, upTriFactor->AA_h + nzUpper); in MatSeqAIJCUSPARSEBuildILUUpperTriMatrix()
560 PetscCall(PetscLogCpuToGpu(nzUpper * sizeof(PetscScalar))); in MatSeqAIJCUSPARSEBuildILUUpperTriMatrix()
782 PetscInt nzUpper = a->nz, n = A->rmap->n, i, offset, nz, j; in MatSeqAIJCUSPARSEBuildICCTriMatrices() local
791 PetscCallCUDA(cudaMallocHost((void **)&AAUp, nzUpper * sizeof(PetscScalar))); in MatSeqAIJCUSPARSEBuildICCTriMatrices()
792 PetscCallCUDA(cudaMallocHost((void **)&AALo, nzUpper * sizeof(PetscScalar))); in MatSeqAIJCUSPARSEBuildICCTriMatrices()
796 PetscCallCUDA(cudaMallocHost((void **)&AjUp, nzUpper * sizeof(PetscInt))); in MatSeqAIJCUSPARSEBuildICCTriMatrices()
800 AiUp[n] = nzUpper; in MatSeqAIJCUSPARSEBuildICCTriMatrices()