Lines Matching refs:Mnz
207 PetscInt *Mi, *Mj, Mnz; in MatSeqAIJCUSPARSEBuildFactoredMatrix_LU() local
215 …Mnz = (Ai[m] - Ai[0]) + (adiag[0] - adiag[m]); // Lnz (without the unit diagonal) + Unz (with the … in MatSeqAIJCUSPARSEBuildFactoredMatrix_LU()
217 PetscCall(PetscMalloc1(Mnz, &Mj)); // Mj is temp in MatSeqAIJCUSPARSEBuildFactoredMatrix_LU()
218 PetscCall(PetscMalloc1(Mnz, &Ma)); in MatSeqAIJCUSPARSEBuildFactoredMatrix_LU()
230 PetscCallCUDA(cudaMalloc(&fs->csrColIdx, sizeof(*fs->csrColIdx) * Mnz)); in MatSeqAIJCUSPARSEBuildFactoredMatrix_LU()
231 PetscCallCUDA(cudaMalloc(&fs->csrVal, sizeof(*fs->csrVal) * Mnz)); in MatSeqAIJCUSPARSEBuildFactoredMatrix_LU()
233 …PetscCallCUDA(cudaMemcpy(fs->csrColIdx, Mj, sizeof(*fs->csrColIdx) * Mnz, cudaMemcpyHostToDevice)); in MatSeqAIJCUSPARSEBuildFactoredMatrix_LU()
244 …PetscCallCUSPARSE(cusparseCreateCsr(&fs->spMatDescr_L, m, m, Mnz, fs->csrRowPtr, fs->csrColIdx, fs… in MatSeqAIJCUSPARSEBuildFactoredMatrix_LU()
250 …PetscCallCUSPARSE(cusparseCreateCsr(&fs->spMatDescr_U, m, m, Mnz, fs->csrRowPtr, fs->csrColIdx, fs… in MatSeqAIJCUSPARSEBuildFactoredMatrix_LU()
277 Mnz = Mi[m]; in MatSeqAIJCUSPARSEBuildFactoredMatrix_LU()
285 PetscCallCUDA(cudaMemcpy(fs->csrVal, Ma, sizeof(*Ma) * Mnz, cudaMemcpyHostToDevice)); in MatSeqAIJCUSPARSEBuildFactoredMatrix_LU()
625 PetscInt *Mj, Mnz; in MatSeqAIJCUSPARSEBuildFactoredMatrix_Cholesky() local
634 Mnz = Ai[m]; // Unz (with the unit diagonal) in MatSeqAIJCUSPARSEBuildFactoredMatrix_Cholesky()
635 PetscCall(PetscMalloc1(Mnz, &Ma)); in MatSeqAIJCUSPARSEBuildFactoredMatrix_Cholesky()
636 PetscCall(PetscMalloc1(Mnz, &Mj)); // Mj[] is temp in MatSeqAIJCUSPARSEBuildFactoredMatrix_Cholesky()
645 PetscCallCUDA(cudaMalloc(&fs->csrColIdx, sizeof(*fs->csrColIdx) * Mnz)); in MatSeqAIJCUSPARSEBuildFactoredMatrix_Cholesky()
646 PetscCallCUDA(cudaMalloc(&fs->csrVal, sizeof(*fs->csrVal) * Mnz)); in MatSeqAIJCUSPARSEBuildFactoredMatrix_Cholesky()
649 PetscCallCUDA(cudaMemcpy(fs->csrColIdx, Mj, sizeof(*Mj) * Mnz, cudaMemcpyHostToDevice)); in MatSeqAIJCUSPARSEBuildFactoredMatrix_Cholesky()
660 …PetscCallCUSPARSE(cusparseCreateCsr(&fs->spMatDescr_U, m, m, Mnz, fs->csrRowPtr, fs->csrColIdx, fs… in MatSeqAIJCUSPARSEBuildFactoredMatrix_Cholesky()
688 Mnz = Ai[m]; in MatSeqAIJCUSPARSEBuildFactoredMatrix_Cholesky()
694 PetscCallCUDA(cudaMemcpy(fs->csrVal, Ma, sizeof(*Ma) * Mnz, cudaMemcpyHostToDevice)); in MatSeqAIJCUSPARSEBuildFactoredMatrix_Cholesky()