Searched refs:Enz (Results 1 – 1 of 1) sorted by relevance
| /petsc/src/mat/impls/aij/mpi/kokkos/ |
| H A D | mpiaijkok.kokkos.cxx | 422 PetscInt nleaves = 0, Enz = 0; // leaves are nonzeros I will send in MatMPIAIJKokkosReduceBegin() local
426 for (PetscInt i = 0; i < Em; i++) Enz += E_RowLen[i]; in MatMPIAIJKokkosReduceBegin()
427 …PetscAssert(A.nnz() + B.nnz() == Enz, comm, PETSC_ERR_PLIB, "Enz should be equal to sum of nnz of … in MatMPIAIJKokkosReduceBegin()
428 PetscCallMPI(PetscMalloc1(Enz, &iremote)); // no free, since we give ownership to reduceSF in MatMPIAIJKokkosReduceBegin()
439 PetscCheck(nleaves == Enz, comm, PETSC_ERR_PLIB, "nleaves should be equal to Enz"); in MatMPIAIJKokkosReduceBegin()
624 …PetscCall(MatMergeGetLaunchParameters<DefaultExecutionSpace>(Em, Enz, -1, teamSize, vectorLength, … in MatMPIAIJKokkosReduceBegin()
|