Lines Matching refs:Bcsr
2178 CsrMatrix *Bcsr; member
2198 delete mmdata->Bcsr; in MatProductCtxDestroy_MatMatHipsparse()
2414 CsrMatrix *Acsr, *Bcsr, *Ccsr; in MatProductNumeric_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE() local
2485 …Bcsr = mmdata->Bcsr ? mmdata->Bcsr : (CsrMatrix *)Bmat->mat; /* B may be in compressed row storage… in MatProductNumeric_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE()
2488 PetscCheck(Bcsr, PetscObjectComm((PetscObject)C), PETSC_ERR_GPU, "Missing B CSR struct"); in MatProductNumeric_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE()
2492 …BmatSpDescr = mmdata->Bcsr ? mmdata->matSpBDescr : Bmat->matDescr; /* B may be in compressed row s… in MatProductNumeric_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE()
2501 …PetscCallHIPSPARSE(hipsparse_csr_spgemm(Ccusp->handle, opA, opB, Acsr->num_rows, Bcsr->num_cols, A… in MatProductNumeric_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE()
2502 …Bcsr->num_entries, Bcsr->values->data().get(), Bcsr->row_offsets->data().get(), Bcsr->column_indic… in MatProductNumeric_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE()
2529 CsrMatrix *Acsr, *Bcsr, *Ccsr; in MatProductSymbolic_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE() local
2660 Bcsr = (CsrMatrix *)Bmat->mat; in MatProductSymbolic_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE()
2664 Bcsr = new CsrMatrix; in MatProductSymbolic_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE()
2665 Bcsr->num_rows = B->rmap->n; in MatProductSymbolic_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE()
2666 Bcsr->num_cols = cBcsr->num_cols; in MatProductSymbolic_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE()
2667 Bcsr->num_entries = cBcsr->num_entries; in MatProductSymbolic_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE()
2668 Bcsr->column_indices = cBcsr->column_indices; in MatProductSymbolic_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE()
2669 Bcsr->values = cBcsr->values; in MatProductSymbolic_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE()
2675 Bcsr->row_offsets = Bcusp->rowoffsets_gpu; in MatProductSymbolic_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE()
2676 mmdata->Bcsr = Bcsr; in MatProductSymbolic_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE()
2677 if (Bcsr->num_rows && Bcsr->num_cols) { in MatProductSymbolic_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE()
2678 …>matSpBDescr, Bcsr->num_rows, Bcsr->num_cols, Bcsr->num_entries, Bcsr->row_offsets->data().get(), … in MatProductSymbolic_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE()
2683 PetscCheck(Bcsr, PetscObjectComm((PetscObject)C), PETSC_ERR_GPU, "Missing B CSR struct"); in MatProductSymbolic_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE()
2787 … Bcsr->num_cols, Acsr->num_cols, Amat->descr, Acsr->num_entries, Acsr->row_offsets->data().get(), … in MatProductSymbolic_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE()
2788 …Bcsr->row_offsets->data().get(), Bcsr->column_indices->data().get(), Cmat->descr, Ccsr->row_offset… in MatProductSymbolic_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE()
2799 …PetscCallHIPSPARSE(hipsparse_csr_spgemm(Ccusp->handle, opA, opB, Acsr->num_rows, Bcsr->num_cols, A… in MatProductSymbolic_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE()
2800 …Bcsr->num_entries, Bcsr->values->data().get(), Bcsr->row_offsets->data().get(), Bcsr->column_indic… in MatProductSymbolic_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE()
4097 CsrMatrix *Acsr, *Bcsr, *Ccsr; in MatSeqAIJHIPSPARSEMergeMats() local
4147 Bcsr = (CsrMatrix *)Bcusp->mat->mat; in MatSeqAIJHIPSPARSEMergeMats()
4149 Bnnz = (PetscInt)Bcsr->column_indices->size(); in MatSeqAIJHIPSPARSEMergeMats()
4177 } else Broff = Bcsr->row_offsets; in MatSeqAIJHIPSPARSEMergeMats()
4184 … auto Bcib = thrust::make_transform_iterator(Bcsr->column_indices->begin(), Shift(A->cmap->n)); in MatSeqAIJHIPSPARSEMergeMats()
4185 auto Bcie = thrust::make_transform_iterator(Bcsr->column_indices->end(), Shift(A->cmap->n)); in MatSeqAIJHIPSPARSEMergeMats()
4189 …auto Bzb = thrust::make_zip_iterator(thrust::make_tuple(Bcoo->begin(), Bcib, Bcsr->values->begin… in MatSeqAIJHIPSPARSEMergeMats()
4190 …auto Bze = thrust::make_zip_iterator(thrust::make_tuple(Bcoo->end(), Bcie, Bcsr->values->end(), … in MatSeqAIJHIPSPARSEMergeMats()
4314 Bcsr = (CsrMatrix *)Bcusp->mat->mat; in MatSeqAIJHIPSPARSEMergeMats()
4317 …Bcsr->num_entries == (PetscInt)Bcsr->values->size(), PETSC_COMM_SELF, PETSC_ERR_COR, "B nnz %" Pet… in MatSeqAIJHIPSPARSEMergeMats()
4319 …Bcsr->num_entries, PETSC_COMM_SELF, PETSC_ERR_COR, "C nnz %" PetscInt_FMT " != %" PetscInt_FMT " +… in MatSeqAIJHIPSPARSEMergeMats()
4327 …auto zibbit = thrust::make_zip_iterator(thrust::make_tuple(Bcsr->values->begin(), thrust::make_per… in MatSeqAIJHIPSPARSEMergeMats()
4328 …auto ziebit = thrust::make_zip_iterator(thrust::make_tuple(Bcsr->values->end(), thrust::make_permu… in MatSeqAIJHIPSPARSEMergeMats()