Searched refs:bufferSize4 (Results 1 – 2 of 2) sorted by relevance
| /petsc/src/mat/impls/aij/seq/seqhipsparse/ |
| H A D | aijhipsparse.hip.cxx | 2720 size_t bufferSize4 = 0; in MatProductSymbolic_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE() local
2729 …E_SPGEMM_DEFAULT, mmdata->spgemmDesc, &bufferSize2, NULL, &bufferSize3, NULL, &bufferSize4, NULL)); in MatProductSymbolic_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE()
2732 PetscCallHIP(hipMalloc((void **)&mmdata->dBuffer4, bufferSize4)); in MatProductSymbolic_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE()
2733 …, mmdata->spgemmDesc, &bufferSize2, dBuffer2, &bufferSize3, dBuffer3, &bufferSize4, mmdata->dBuffe… in MatProductSymbolic_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE()
2755 … %ldKB %ldKB\n", MatProductTypes[ptype], m, n, k, a->nz, b->nz, c->nz, bufferSize4 / 1024, bufferS… in MatProductSymbolic_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE()
|
| /petsc/src/mat/impls/aij/seq/seqcusparse/ |
| H A D | aijcusparse.cu | 3233 size_t bufferSize4 = 0; in MatProductSymbolic_SeqAIJCUSPARSE_SeqAIJCUSPARSE() local
3244 …SE_SPGEMM_DEFAULT, mmdata->spgemmDesc, &bufferSize2, NULL, &bufferSize3, NULL, &bufferSize4, NULL); in MatProductSymbolic_SeqAIJCUSPARSE_SeqAIJCUSPARSE()
3248 PetscCallCUDA(cudaMalloc((void **)&mmdata->dBuffer4, bufferSize4)); in MatProductSymbolic_SeqAIJCUSPARSE_SeqAIJCUSPARSE()
3249 …, mmdata->spgemmDesc, &bufferSize2, dBuffer2, &bufferSize3, dBuffer3, &bufferSize4, mmdata->dBuffe… in MatProductSymbolic_SeqAIJCUSPARSE_SeqAIJCUSPARSE()
3274 … %ldKB %ldKB\n", MatProductTypes[ptype], m, n, k, a->nz, b->nz, c->nz, bufferSize4 / 1024, bufferS… in MatProductSymbolic_SeqAIJCUSPARSE_SeqAIJCUSPARSE()
|