Searched refs:bufferSize5 (Results 1 – 2 of 2) sorted by relevance
| /petsc/src/mat/impls/aij/seq/seqhipsparse/ |
| H A D | aijhipsparse.hip.cxx | 2721 size_t bufferSize5 = 0; in MatProductSymbolic_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE() local
2749 …r, BmatSpDescr, Cmat->matDescr, HIPSPARSE_SPGEMM_DEFAULT, mmdata->spgemmDesc, &bufferSize5, NULL)); in MatProductSymbolic_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE()
2750 PetscCallHIP(hipMalloc((void **)&mmdata->dBuffer5, bufferSize5)); in MatProductSymbolic_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE()
2751 …pDescr, Cmat->matDescr, HIPSPARSE_SPGEMM_DEFAULT, mmdata->spgemmDesc, &bufferSize5, mmdata->dBuffe… in MatProductSymbolic_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE()
2755 …n", MatProductTypes[ptype], m, n, k, a->nz, b->nz, c->nz, bufferSize4 / 1024, bufferSize5 / 1024)); in MatProductSymbolic_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE()
|
| /petsc/src/mat/impls/aij/seq/seqcusparse/ |
| H A D | aijcusparse.cu | 3234 size_t bufferSize5 = 0; in MatProductSymbolic_SeqAIJCUSPARSE_SeqAIJCUSPARSE() local
3266 …scr, BmatSpDescr, Cmat->matDescr, CUSPARSE_SPGEMM_DEFAULT, mmdata->spgemmDesc, &bufferSize5, NULL); in MatProductSymbolic_SeqAIJCUSPARSE_SeqAIJCUSPARSE()
3268 PetscCallCUDA(cudaMalloc((void **)&mmdata->dBuffer5, bufferSize5)); in MatProductSymbolic_SeqAIJCUSPARSE_SeqAIJCUSPARSE()
3269 …SpDescr, Cmat->matDescr, CUSPARSE_SPGEMM_DEFAULT, mmdata->spgemmDesc, &bufferSize5, mmdata->dBuffe… in MatProductSymbolic_SeqAIJCUSPARSE_SeqAIJCUSPARSE()
3274 …n", MatProductTypes[ptype], m, n, k, a->nz, b->nz, c->nz, bufferSize4 / 1024, bufferSize5 / 1024)); in MatProductSymbolic_SeqAIJCUSPARSE_SeqAIJCUSPARSE()
|