Home
last modified time | relevance | path

Searched refs:mmBufferSize (Results 1 – 2 of 2) sorted by relevance

/petsc/src/mat/impls/aij/seq/seqhipsparse/
H A Daijhipsparse.hip.cxx2187 size_t mmBufferSize; member
2290 size_t mmBufferSize; in MatProductNumeric_SeqAIJHIPSPARSE_SeqDENSEHIP() local
2311 …matBDescr, mat->beta_zero, mmdata->matCDescr, hipsparse_scalartype, cusp->spmmAlg, &mmBufferSize)); in MatProductNumeric_SeqAIJHIPSPARSE_SeqDENSEHIP()
2312 if ((mmdata->mmBuffer && mmdata->mmBufferSize < mmBufferSize) || !mmdata->mmBuffer) { in MatProductNumeric_SeqAIJHIPSPARSE_SeqDENSEHIP()
2314 PetscCallHIP(hipMalloc(&mmdata->mmBuffer, mmBufferSize)); in MatProductNumeric_SeqAIJHIPSPARSE_SeqDENSEHIP()
2315 mmdata->mmBufferSize = mmBufferSize; in MatProductNumeric_SeqAIJHIPSPARSE_SeqDENSEHIP()
2497 …se_scalartype, HIPSPARSE_SPGEMM_DEFAULT, mmdata->spgemmDesc, &mmdata->mmBufferSize, mmdata->mmBuff… in MatProductNumeric_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE()
2765 … hipsparse_scalartype, HIPSPARSE_SPGEMM_DEFAULT, mmdata->spgemmDesc, &mmdata->mmBufferSize, NULL)); in MatProductSymbolic_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE()
2770 PetscCallHIP(hipMalloc((void **)&mmdata->mmBuffer, mmdata->mmBufferSize)); in MatProductSymbolic_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE()
2772 …se_scalartype, HIPSPARSE_SPGEMM_DEFAULT, mmdata->spgemmDesc, &mmdata->mmBufferSize, mmdata->mmBuff… in MatProductSymbolic_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE()
[all …]
/petsc/src/mat/impls/aij/seq/seqcusparse/
H A Daijcusparse.cu2614 size_t mmBufferSize; member
2732 size_t mmBufferSize; in MatProductNumeric_SeqAIJCUSPARSE_SeqDENSECUDA() local
2764 …>matBDescr, mat->beta_zero, mmdata->matCDescr, cusparse_scalartype, cusp->spmmAlg, &mmBufferSize)); in MatProductNumeric_SeqAIJCUSPARSE_SeqDENSECUDA()
2766 if ((mmdata->mmBuffer && mmdata->mmBufferSize < mmBufferSize) || !mmdata->mmBuffer) { in MatProductNumeric_SeqAIJCUSPARSE_SeqDENSECUDA()
2768 PetscCallCUDA(cudaMalloc(&mmdata->mmBuffer, mmBufferSize)); in MatProductNumeric_SeqAIJCUSPARSE_SeqDENSECUDA()
2769 mmdata->mmBufferSize = mmBufferSize; in MatProductNumeric_SeqAIJCUSPARSE_SeqDENSECUDA()
2997 …rse_scalartype, CUSPARSE_SPGEMM_DEFAULT, mmdata->spgemmDesc, &mmdata->mmBufferSize, mmdata->mmBuff… in MatProductNumeric_SeqAIJCUSPARSE_SeqAIJCUSPARSE()
3286 …cr, cusparse_scalartype, CUSPARSE_SPGEMM_DEFAULT, mmdata->spgemmDesc, &mmdata->mmBufferSize, NULL); in MatProductSymbolic_SeqAIJCUSPARSE_SeqAIJCUSPARSE()
3293 PetscCallCUDA(cudaMalloc((void **)&mmdata->mmBuffer, mmdata->mmBufferSize)); in MatProductSymbolic_SeqAIJCUSPARSE_SeqAIJCUSPARSE()
3295 …rse_scalartype, CUSPARSE_SPGEMM_DEFAULT, mmdata->spgemmDesc, &mmdata->mmBufferSize, mmdata->mmBuff… in MatProductSymbolic_SeqAIJCUSPARSE_SeqAIJCUSPARSE()
[all …]