Lines Matching refs:Bcsr

2602   CsrMatrix     *Bcsr;  member
2627 delete mmdata->Bcsr; in MatProductCtxDestroy_MatMatCusparse()
2910 CsrMatrix *Acsr, *Bcsr, *Ccsr; in MatProductNumeric_SeqAIJCUSPARSE_SeqAIJCUSPARSE() local
2984Bcsr = mmdata->Bcsr ? mmdata->Bcsr : (CsrMatrix *)Bmat->mat; /* B may be in compressed row storage… in MatProductNumeric_SeqAIJCUSPARSE_SeqAIJCUSPARSE()
2987 PetscCheck(Bcsr, PetscObjectComm((PetscObject)C), PETSC_ERR_GPU, "Missing B CSR struct"); in MatProductNumeric_SeqAIJCUSPARSE_SeqAIJCUSPARSE()
2991 …BmatSpDescr = mmdata->Bcsr ? mmdata->matSpBDescr : Bmat->matDescr; /* B may be in compressed row s… in MatProductNumeric_SeqAIJCUSPARSE_SeqAIJCUSPARSE()
3003Bcsr->num_cols, Acsr->num_cols, Amat->descr, Acsr->num_entries, Acsr->values->data().get(), Acsr->… in MatProductNumeric_SeqAIJCUSPARSE_SeqAIJCUSPARSE()
3004Bcsr->values->data().get(), Bcsr->row_offsets->data().get(), Bcsr->column_indices->data().get(), C… in MatProductNumeric_SeqAIJCUSPARSE_SeqAIJCUSPARSE()
3031 CsrMatrix *Acsr, *Bcsr, *Ccsr; in MatProductSymbolic_SeqAIJCUSPARSE_SeqAIJCUSPARSE() local
3163 Bcsr = (CsrMatrix *)Bmat->mat; in MatProductSymbolic_SeqAIJCUSPARSE_SeqAIJCUSPARSE()
3169 Bcsr = new CsrMatrix; in MatProductSymbolic_SeqAIJCUSPARSE_SeqAIJCUSPARSE()
3170 Bcsr->num_rows = B->rmap->n; in MatProductSymbolic_SeqAIJCUSPARSE_SeqAIJCUSPARSE()
3171 Bcsr->num_cols = cBcsr->num_cols; in MatProductSymbolic_SeqAIJCUSPARSE_SeqAIJCUSPARSE()
3172 Bcsr->num_entries = cBcsr->num_entries; in MatProductSymbolic_SeqAIJCUSPARSE_SeqAIJCUSPARSE()
3173 Bcsr->column_indices = cBcsr->column_indices; in MatProductSymbolic_SeqAIJCUSPARSE_SeqAIJCUSPARSE()
3174 Bcsr->values = cBcsr->values; in MatProductSymbolic_SeqAIJCUSPARSE_SeqAIJCUSPARSE()
3180 Bcsr->row_offsets = Bcusp->rowoffsets_gpu; in MatProductSymbolic_SeqAIJCUSPARSE_SeqAIJCUSPARSE()
3181 mmdata->Bcsr = Bcsr; in MatProductSymbolic_SeqAIJCUSPARSE_SeqAIJCUSPARSE()
3183 if (Bcsr->num_rows && Bcsr->num_cols) { in MatProductSymbolic_SeqAIJCUSPARSE_SeqAIJCUSPARSE()
3184 …>matSpBDescr, Bcsr->num_rows, Bcsr->num_cols, Bcsr->num_entries, Bcsr->row_offsets->data().get(), in MatProductSymbolic_SeqAIJCUSPARSE_SeqAIJCUSPARSE()
3191 PetscCheck(Bcsr, PetscObjectComm((PetscObject)C), PETSC_ERR_GPU, "Missing B CSR struct"); in MatProductSymbolic_SeqAIJCUSPARSE_SeqAIJCUSPARSE()
3313Bcsr->num_cols, Acsr->num_cols, Amat->descr, Acsr->num_entries, Acsr->row_offsets->data().get(), … in MatProductSymbolic_SeqAIJCUSPARSE_SeqAIJCUSPARSE()
3314Bcsr->row_offsets->data().get(), Bcsr->column_indices->data().get(), Cmat->descr, Ccsr->row_offset… in MatProductSymbolic_SeqAIJCUSPARSE_SeqAIJCUSPARSE()
3326Bcsr->num_cols, Acsr->num_cols, Amat->descr, Acsr->num_entries, Acsr->values->data().get(), Acsr->… in MatProductSymbolic_SeqAIJCUSPARSE_SeqAIJCUSPARSE()
3327Bcsr->values->data().get(), Bcsr->row_offsets->data().get(), Bcsr->column_indices->data().get(), C… in MatProductSymbolic_SeqAIJCUSPARSE_SeqAIJCUSPARSE()
4743 CsrMatrix *Acsr, *Bcsr, *Ccsr; in MatSeqAIJCUSPARSEMergeMats() local
4794 Bcsr = (CsrMatrix *)Bcusp->mat->mat; in MatSeqAIJCUSPARSEMergeMats()
4796 Bnnz = (PetscInt)Bcsr->column_indices->size(); in MatSeqAIJCUSPARSEMergeMats()
4824 } else Broff = Bcsr->row_offsets; in MatSeqAIJCUSPARSEMergeMats()
4834 auto Bcib = thrust::make_transform_iterator(Bcsr->column_indices->begin(), Shift(A->cmap->n)); in MatSeqAIJCUSPARSEMergeMats()
4835 auto Bcie = thrust::make_transform_iterator(Bcsr->column_indices->end(), Shift(A->cmap->n)); in MatSeqAIJCUSPARSEMergeMats()
4838 auto Bcib = Bcsr->column_indices->begin(); in MatSeqAIJCUSPARSEMergeMats()
4839 auto Bcie = Bcsr->column_indices->end(); in MatSeqAIJCUSPARSEMergeMats()
4845 …auto Bzb = thrust::make_zip_iterator(thrust::make_tuple(Bcoo->begin(), Bcib, Bcsr->values->begin… in MatSeqAIJCUSPARSEMergeMats()
4846 …auto Bze = thrust::make_zip_iterator(thrust::make_tuple(Bcoo->end(), Bcie, Bcsr->values->end(), … in MatSeqAIJCUSPARSEMergeMats()
4991 Bcsr = (CsrMatrix *)Bcusp->mat->mat; in MatSeqAIJCUSPARSEMergeMats()
4994Bcsr->num_entries == (PetscInt)Bcsr->values->size(), PETSC_COMM_SELF, PETSC_ERR_COR, "B nnz %" Pet… in MatSeqAIJCUSPARSEMergeMats()
4996Bcsr->num_entries, PETSC_COMM_SELF, PETSC_ERR_COR, "C nnz %" PetscInt_FMT " != %" PetscInt_FMT " +… in MatSeqAIJCUSPARSEMergeMats()
5008 …auto zibbit = thrust::make_zip_iterator(thrust::make_tuple(Bcsr->values->begin(), thrust::make_per… in MatSeqAIJCUSPARSEMergeMats()
5009 …auto ziebit = thrust::make_zip_iterator(thrust::make_tuple(Bcsr->values->end(), thrust::make_permu… in MatSeqAIJCUSPARSEMergeMats()