Lines Matching refs:Cmat
2413 Mat_SeqAIJHIPSPARSEMultStruct *Amat, *Bmat, *Cmat; in MatProductNumeric_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE() local
2433 Cmat = Ccusp->mat; in MatProductNumeric_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE()
2434 …PetscCheck(Cmat, PetscObjectComm((PetscObject)C), PETSC_ERR_GPU, "Missing C mult struct for produc… in MatProductNumeric_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE()
2435 Ccsr = (CsrMatrix *)Cmat->mat; in MatProductNumeric_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE()
2480 Cmat = Ccusp->mat; in MatProductNumeric_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE()
2483 …PetscCheck(Cmat, PetscObjectComm((PetscObject)C), PETSC_ERR_GPU, "Missing C mult struct for produc… in MatProductNumeric_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE()
2486 Ccsr = (CsrMatrix *)Cmat->mat; in MatProductNumeric_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE()
2495 …GEMMreuse_compute(Ccusp->handle, opA, opB, Cmat->alpha_one, Amat->matDescr, BmatSpDescr, Cmat->bet… in MatProductNumeric_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE()
2497 …rseSpGEMM_compute(Ccusp->handle, opA, opB, Cmat->alpha_one, Amat->matDescr, BmatSpDescr, Cmat->bet… in MatProductNumeric_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE()
2498 …sparseSpGEMM_copy(Ccusp->handle, opA, opB, Cmat->alpha_one, Amat->matDescr, BmatSpDescr, Cmat->bet… in MatProductNumeric_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE()
2502 …t(), Bcsr->row_offsets->data().get(), Bcsr->column_indices->data().get(), Cmat->descr, Ccsr->value… in MatProductNumeric_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE()
2528 Mat_SeqAIJHIPSPARSEMultStruct *Amat, *Bmat, *Cmat; in MatProductSymbolic_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE() local
2615 Cmat = new Mat_SeqAIJHIPSPARSEMultStruct; in MatProductSymbolic_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE()
2624 Cmat->cprowIndices = new THRUSTINTARRAY(c->compressedrow.nrows); in MatProductSymbolic_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE()
2625 …Cmat->cprowIndices->assign(c->compressedrow.rindex, c->compressedrow.rindex + c->compressedrow.nro… in MatProductSymbolic_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE()
2631 Cmat->cprowIndices = NULL; in MatProductSymbolic_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE()
2634 Ccusp->mat = Cmat; in MatProductSymbolic_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE()
2639 PetscCallHIPSPARSE(hipsparseCreateMatDescr(&Cmat->descr)); in MatProductSymbolic_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE()
2640 PetscCallHIPSPARSE(hipsparseSetMatIndexBase(Cmat->descr, HIPSPARSE_INDEX_BASE_ZERO)); in MatProductSymbolic_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE()
2641 PetscCallHIPSPARSE(hipsparseSetMatType(Cmat->descr, HIPSPARSE_MATRIX_TYPE_GENERAL)); in MatProductSymbolic_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE()
2642 PetscCallHIP(hipMalloc((void **)&Cmat->alpha_one, sizeof(PetscScalar))); in MatProductSymbolic_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE()
2643 PetscCallHIP(hipMalloc((void **)&Cmat->beta_zero, sizeof(PetscScalar))); in MatProductSymbolic_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE()
2644 PetscCallHIP(hipMalloc((void **)&Cmat->beta_one, sizeof(PetscScalar))); in MatProductSymbolic_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE()
2645 …PetscCallHIP(hipMemcpy(Cmat->alpha_one, &PETSC_HIPSPARSE_ONE, sizeof(PetscScalar), hipMemcpyHostTo… in MatProductSymbolic_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE()
2646 …PetscCallHIP(hipMemcpy(Cmat->beta_zero, &PETSC_HIPSPARSE_ZERO, sizeof(PetscScalar), hipMemcpyHostT… in MatProductSymbolic_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE()
2647 …PetscCallHIP(hipMemcpy(Cmat->beta_one, &PETSC_HIPSPARSE_ONE, sizeof(PetscScalar), hipMemcpyHostToD… in MatProductSymbolic_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE()
2706 …PetscCallHIPSPARSE(hipsparseCreateCsr(&Cmat->matDescr, Ccsr->num_rows, Ccsr->num_cols, 0, Ccsr->ro… in MatProductSymbolic_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE()
2724 …euse_workEstimation(Ccusp->handle, opA, opB, Amat->matDescr, BmatSpDescr, Cmat->matDescr, HIPSPARS… in MatProductSymbolic_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE()
2727 …euse_workEstimation(Ccusp->handle, opA, opB, Amat->matDescr, BmatSpDescr, Cmat->matDescr, HIPSPARS… in MatProductSymbolic_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE()
2729 …arseSpGEMMreuse_nnz(Ccusp->handle, opA, opB, Amat->matDescr, BmatSpDescr, Cmat->matDescr, HIPSPARS… in MatProductSymbolic_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE()
2733 …arseSpGEMMreuse_nnz(Ccusp->handle, opA, opB, Amat->matDescr, BmatSpDescr, Cmat->matDescr, HIPSPARS… in MatProductSymbolic_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE()
2738 PetscCallHIPSPARSE(hipsparseSpMatGetSize(Cmat->matDescr, &C_num_rows1, &C_num_cols1, &C_nnz1)); in MatProductSymbolic_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE()
2747 …PetscCallHIPSPARSE(hipsparseCsrSetPointers(Cmat->matDescr, Ccsr->row_offsets->data().get(), Ccsr->… in MatProductSymbolic_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE()
2749 …rseSpGEMMreuse_copy(Ccusp->handle, opA, opB, Amat->matDescr, BmatSpDescr, Cmat->matDescr, HIPSPARS… in MatProductSymbolic_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE()
2751 …rseSpGEMMreuse_copy(Ccusp->handle, opA, opB, Amat->matDescr, BmatSpDescr, Cmat->matDescr, HIPSPARS… in MatProductSymbolic_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE()
2753 …GEMMreuse_compute(Ccusp->handle, opA, opB, Cmat->alpha_one, Amat->matDescr, BmatSpDescr, Cmat->bet… in MatProductSymbolic_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE()
2760 …MM_workEstimation(Ccusp->handle, opA, opB, Cmat->alpha_one, Amat->matDescr, BmatSpDescr, Cmat->bet… in MatProductSymbolic_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE()
2763 …MM_workEstimation(Ccusp->handle, opA, opB, Cmat->alpha_one, Amat->matDescr, BmatSpDescr, Cmat->bet… in MatProductSymbolic_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE()
2765 …rseSpGEMM_compute(Ccusp->handle, opA, opB, Cmat->alpha_one, Amat->matDescr, BmatSpDescr, Cmat->bet… in MatProductSymbolic_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE()
2772 …rseSpGEMM_compute(Ccusp->handle, opA, opB, Cmat->alpha_one, Amat->matDescr, BmatSpDescr, Cmat->bet… in MatProductSymbolic_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE()
2774 PetscCallHIPSPARSE(hipsparseSpMatGetSize(Cmat->matDescr, &C_num_rows1, &C_num_cols1, &C_nnz1)); in MatProductSymbolic_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE()
2782 …PetscCallHIPSPARSE(hipsparseCsrSetPointers(Cmat->matDescr, Ccsr->row_offsets->data().get(), Ccsr->… in MatProductSymbolic_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE()
2783 …sparseSpGEMM_copy(Ccusp->handle, opA, opB, Cmat->alpha_one, Amat->matDescr, BmatSpDescr, Cmat->bet… in MatProductSymbolic_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE()
2788 …Bcsr->row_offsets->data().get(), Bcsr->column_indices->data().get(), Cmat->descr, Ccsr->row_offset… in MatProductSymbolic_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE()
2800 …t(), Bcsr->row_offsets->data().get(), Bcsr->column_indices->data().get(), Cmat->descr, Ccsr->value… in MatProductSymbolic_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE()
4096 Mat_SeqAIJHIPSPARSEMultStruct *Cmat; in MatSeqAIJHIPSPARSEMergeMats() local
4119 Cmat = new Mat_SeqAIJHIPSPARSEMultStruct; in MatSeqAIJHIPSPARSEMergeMats()
4121 Cmat->cprowIndices = NULL; in MatSeqAIJHIPSPARSEMergeMats()
4128 Ccusp->mat = Cmat; in MatSeqAIJHIPSPARSEMergeMats()
4132 PetscCallHIPSPARSE(hipsparseCreateMatDescr(&Cmat->descr)); in MatSeqAIJHIPSPARSEMergeMats()
4133 PetscCallHIPSPARSE(hipsparseSetMatIndexBase(Cmat->descr, HIPSPARSE_INDEX_BASE_ZERO)); in MatSeqAIJHIPSPARSEMergeMats()
4134 PetscCallHIPSPARSE(hipsparseSetMatType(Cmat->descr, HIPSPARSE_MATRIX_TYPE_GENERAL)); in MatSeqAIJHIPSPARSEMergeMats()
4135 PetscCallHIP(hipMalloc((void **)&Cmat->alpha_one, sizeof(PetscScalar))); in MatSeqAIJHIPSPARSEMergeMats()
4136 PetscCallHIP(hipMalloc((void **)&Cmat->beta_zero, sizeof(PetscScalar))); in MatSeqAIJHIPSPARSEMergeMats()
4137 PetscCallHIP(hipMalloc((void **)&Cmat->beta_one, sizeof(PetscScalar))); in MatSeqAIJHIPSPARSEMergeMats()
4138 …PetscCallHIP(hipMemcpy(Cmat->alpha_one, &PETSC_HIPSPARSE_ONE, sizeof(PetscScalar), hipMemcpyHostTo… in MatSeqAIJHIPSPARSEMergeMats()
4139 …PetscCallHIP(hipMemcpy(Cmat->beta_zero, &PETSC_HIPSPARSE_ZERO, sizeof(PetscScalar), hipMemcpyHostT… in MatSeqAIJHIPSPARSEMergeMats()
4140 …PetscCallHIP(hipMemcpy(Cmat->beta_one, &PETSC_HIPSPARSE_ONE, sizeof(PetscScalar), hipMemcpyHostToD… in MatSeqAIJHIPSPARSEMergeMats()
4211 …PetscCallHIPSPARSE(hipsparseCreateCsr(&Cmat->matDescr, Ccsr->num_rows, Ccsr->num_cols, Ccsr->num_e… in MatSeqAIJHIPSPARSEMergeMats()