Searched refs:mmBuffer (Results 1 – 2 of 2) sorted by relevance
| /petsc/src/mat/impls/aij/seq/seqhipsparse/ |
| H A D | aijhipsparse.hip.cxx | 2188 void *mmBuffer, *mmBuffer2; /* SpGEMM WorkEstimation buffer */ member
2207 if (mmdata->mmBuffer) PetscCallHIP(hipFree(mmdata->mmBuffer)); in MatProductCtxDestroy_MatMatHipsparse()
2312 if ((mmdata->mmBuffer && mmdata->mmBufferSize < mmBufferSize) || !mmdata->mmBuffer) { in MatProductNumeric_SeqAIJHIPSPARSE_SeqDENSEHIP()
2313 PetscCallHIP(hipFree(mmdata->mmBuffer)); in MatProductNumeric_SeqAIJHIPSPARSE_SeqDENSEHIP()
2314 PetscCallHIP(hipMalloc(&mmdata->mmBuffer, mmBufferSize)); in MatProductNumeric_SeqAIJHIPSPARSE_SeqDENSEHIP()
2326 …BDescr, mat->beta_zero, mmdata->matCDescr, hipsparse_scalartype, cusp->spmmAlg, mmdata->mmBuffer)); in MatProductNumeric_SeqAIJHIPSPARSE_SeqDENSEHIP()
2497 …calartype, HIPSPARSE_SPGEMM_DEFAULT, mmdata->spgemmDesc, &mmdata->mmBufferSize, mmdata->mmBuffer)); in MatProductNumeric_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE()
2770 PetscCallHIP(hipMalloc((void **)&mmdata->mmBuffer, mmdata->mmBufferSize)); in MatProductSymbolic_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE()
2772 …calartype, HIPSPARSE_SPGEMM_DEFAULT, mmdata->spgemmDesc, &mmdata->mmBufferSize, mmdata->mmBuffer)); in MatProductSymbolic_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE()
|
| /petsc/src/mat/impls/aij/seq/seqcusparse/ |
| H A D | aijcusparse.cu | 2615 void *mmBuffer; member
2637 if (mmdata->mmBuffer) PetscCallCUDA(cudaFree(mmdata->mmBuffer)); in MatProductCtxDestroy_MatMatCusparse()
2766 if ((mmdata->mmBuffer && mmdata->mmBufferSize < mmBufferSize) || !mmdata->mmBuffer) { in MatProductNumeric_SeqAIJCUSPARSE_SeqDENSECUDA()
2767 PetscCallCUDA(cudaFree(mmdata->mmBuffer)); in MatProductNumeric_SeqAIJCUSPARSE_SeqDENSECUDA()
2768 PetscCallCUDA(cudaMalloc(&mmdata->mmBuffer, mmBufferSize)); in MatProductNumeric_SeqAIJCUSPARSE_SeqDENSECUDA()
2773 …tBDescr, mat->beta_zero, mmdata->matCDescr, cusparse_scalartype, cusp->spmmAlg, mmdata->mmBuffer)); in MatProductNumeric_SeqAIJCUSPARSE_SeqDENSECUDA()
2785 …tBDescr, mat->beta_zero, mmdata->matCDescr, cusparse_scalartype, cusp->spmmAlg, mmdata->mmBuffer)); in MatProductNumeric_SeqAIJCUSPARSE_SeqDENSECUDA()
2997 …_scalartype, CUSPARSE_SPGEMM_DEFAULT, mmdata->spgemmDesc, &mmdata->mmBufferSize, mmdata->mmBuffer); in MatProductNumeric_SeqAIJCUSPARSE_SeqAIJCUSPARSE()
3293 PetscCallCUDA(cudaMalloc((void **)&mmdata->mmBuffer, mmdata->mmBufferSize)); in MatProductSymbolic_SeqAIJCUSPARSE_SeqAIJCUSPARSE()
3295 …_scalartype, CUSPARSE_SPGEMM_DEFAULT, mmdata->spgemmDesc, &mmdata->mmBufferSize, mmdata->mmBuffer); in MatProductSymbolic_SeqAIJCUSPARSE_SeqAIJCUSPARSE()
|