Searched refs:bufSize2 (Results 1 – 2 of 2) sorted by relevance
| /petsc/src/mat/impls/aij/seq/seqhipsparse/ |
| H A D | aijhipsparse.hip.cxx | 2758 size_t bufSize2; in MatProductSymbolic_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE() local
2760 …t->matDescr, hipsparse_scalartype, HIPSPARSE_SPGEMM_DEFAULT, mmdata->spgemmDesc, &bufSize2, NULL)); in MatProductSymbolic_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE()
2761 PetscCallHIP(hipMalloc((void **)&mmdata->mmBuffer2, bufSize2)); in MatProductSymbolic_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE()
2763 …r, hipsparse_scalartype, HIPSPARSE_SPGEMM_DEFAULT, mmdata->spgemmDesc, &bufSize2, mmdata->mmBuffer… in MatProductSymbolic_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE()
2776 …_FMT ") are: %ldKB %ldKB\n", MatProductTypes[ptype], m, n, k, a->nz, b->nz, c->nz, bufSize2 / 1024, in MatProductSymbolic_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE()
|
| /petsc/src/mat/impls/aij/seq/seqcusparse/ |
| H A D | aijcusparse.cu | 3277 size_t bufSize2; in MatProductSymbolic_SeqAIJCUSPARSE_SeqAIJCUSPARSE() local
3279 …Cmat->matDescr, cusparse_scalartype, CUSPARSE_SPGEMM_DEFAULT, mmdata->spgemmDesc, &bufSize2, NULL); in MatProductSymbolic_SeqAIJCUSPARSE_SeqAIJCUSPARSE()
3281 PetscCallCUDA(cudaMalloc((void **)&mmdata->mmBuffer2, bufSize2)); in MatProductSymbolic_SeqAIJCUSPARSE_SeqAIJCUSPARSE()
3283 …scr, cusparse_scalartype, CUSPARSE_SPGEMM_DEFAULT, mmdata->spgemmDesc, &bufSize2, mmdata->mmBuffer… in MatProductSymbolic_SeqAIJCUSPARSE_SeqAIJCUSPARSE()
3300 …_FMT ") are: %ldKB %ldKB\n", MatProductTypes[ptype], m, n, k, a->nz, b->nz, c->nz, bufSize2 / 1024, in MatProductSymbolic_SeqAIJCUSPARSE_SeqAIJCUSPARSE()
|