Searched refs:Em (Results 1 – 2 of 2) sorted by relevance
| /petsc/src/mat/impls/aij/mpi/kokkos/ |
| H A D | mpiaijkok.kokkos.cxx | 346 PetscInt Em = A.numRows(), Fm; in MatMPIAIJKokkosReduceBegin() local
360 … PetscIntKokkosViewHost E_NzLeft_h(NoInit("E_NzLeft_h"), Em), E_RowLen_h(NoInit("E_RowLen_h"), Em); in MatMPIAIJKokkosReduceBegin()
362 for (PetscInt i = 0; i < Em; i++) { in MatMPIAIJKokkosReduceBegin()
426 for (PetscInt i = 0; i < Em; i++) Enz += E_RowLen[i]; in MatMPIAIJKokkosReduceBegin()
624 …PetscCall(MatMergeGetLaunchParameters<DefaultExecutionSpace>(Em, Enz, -1, teamSize, vectorLength, … in MatMPIAIJKokkosReduceBegin()
639 PetscInt Em = A.numRows(); in MatMPIAIJKokkosReduceBegin() local
643 PetscInt workSets = (Em + rowsPerTeam - 1) / rowsPerTeam; in MatMPIAIJKokkosReduceBegin()
650 if (i < Em) { in MatMPIAIJKokkosReduceBegin()
744 PetscInt Em = E->rmap->n; // #local rows in MatMPIAIJKokkosBcastBegin() local
759 … PetscIntKokkosViewHost E_NzLeft_h(NoInit("E_NzLeft_h"), Em), E_RowLen_h(NoInit("E_RowLen_h"), Em); in MatMPIAIJKokkosBcastBegin()
[all …]
|
| /petsc/src/ts/tutorials/ |
| H A D | ex10.c | 98 static void RDMaterialEnergy(RD rd, const RDNode *n, PetscScalar *Em, RDNode *dEm) in RDMaterialEnergy() argument
100 rd->MaterialEnergy(rd, n, Em, dEm); in RDMaterialEnergy()
115 …PetscScalar Em, alpha, alpha_t, T = n->T, T_t = 1., chi = rd->I_H / (rd->k * T), chi_t = -chi / T … in RDMaterialEnergy_Saha() local
118 Em = rd->k * T / rd->m_p * (1.5 * (1. + alpha) + alpha * chi); /* Eq 6 */ in RDMaterialEnergy_Saha()
119 if (inEm) *inEm = Em; in RDMaterialEnergy_Saha()
122 dEm->T = Em / T * T_t + rd->k * T / rd->m_p * (1.5 * alpha_t + alpha_t * chi + alpha * chi_t); in RDMaterialEnergy_Saha()
126 static void RDMaterialEnergy_Reduced(RD rd, const RDNode *n, PetscScalar *Em, RDNode *dEm) in RDMaterialEnergy_Reduced() argument
130 if (Em) *Em = (1. + alpha) * T + 0.3 * alpha; in RDMaterialEnergy_Reduced()
|