Lines Matching refs:tid
287 for (PetscInt tid = 0; tid < numthreads; tid++) { in gridToParticles_private() local
288 const PetscInt v_id = v_id_0 + tid, glb_v_id = global_vertex_id + v_id; in gridToParticles_private()
293 …PACK_IDX(v_id, grid)], globXArray[LAND_PACK_IDX(v_id, grid)], t_fhat[grid][tid], globMpArray[LAND_… in gridToParticles_private()
301 for (PetscInt tid = 0; tid < numthreads; tid++) { in gridToParticles_private() local
302 const PetscInt v_id = v_id_0 + tid, glb_v_id = global_vertex_id + v_id; in gridToParticles_private()
518 for (PetscInt tid = 0; tid < numthreads; tid++) { in go() local
520 PetscCall(VecDuplicate(subX, &t_fhat[grid][tid])); in go()
521 PetscCall(KSPCreate(PETSC_COMM_SELF, &t_ksp[grid][tid])); in go()
522 PetscCall(KSPSetType(t_ksp[grid][tid], KSPCG)); in go()
523 PetscCall(KSPGetPC(t_ksp[grid][tid], &pc)); in go()
525 PetscCall(KSPSetOptionsPrefix(t_ksp[grid][tid], "ptof_")); in go()
526 PetscCall(KSPSetOperators(t_ksp[grid][tid], g_Mass[grid], g_Mass[grid])); in go()
527 PetscCall(KSPSetFromOptions(t_ksp[grid][tid])); in go()
549 for (PetscInt tid = 0; tid < numthreads; tid++) { in go() local
550 const PetscInt v_id = v_id_0 + tid, glb_v_id = global_vertex_id_0 + v_id; in go()
567 Np_t[grid][tid] = NN; in go()
569 …Call(PetscMalloc4(NN, &xx_t[grid][tid], NN, &yy_t[grid][tid], NN, &wp_t[grid][tid], dim == 2 ? 1 :… in go()
581 wp_t[grid][tid][pp] = 0; in go()
583 xx_t[grid][tid][pp] = lo[0] + hp[0] / 2.0 + pi * hp[0]; in go()
584 yy_t[grid][tid][pp] = lo[1] + hp[1] / 2.0 + pj * hp[1]; in go()
585 if (dim == 3) zz_t[grid][tid][pp] = lo[2] + hp[2] / 2.0 + pk * hp[2]; in go()
586 … PetscReal x[] = {xx_t[grid][tid][pp], yy_t[grid][tid][pp], dim == 2 ? 0 : zz_t[grid][tid][pp]}; in go()
588 …if (ctx->sphere && PetscSqrtReal(PetscSqr(xx_t[grid][tid][pp]) + PetscSqr(yy_t[grid][tid][pp])) > … in go()
589 wp_t[grid][tid][pp] = 0; in go()
591 maxwellian(dim, x, kT_m, vole, p_shift, &wp_t[grid][tid][pp]); in go()
593 …maxwellian(dim, x, kT_m, vole, -p_shift, &wp_t[grid][tid][pp]); // symmetric shift of electron pla… in go()
605 xx_t[grid][tid][pp] = mag * PetscCosReal(2.0 * PETSC_PI * u2); in go()
606 yy_t[grid][tid][pp] = mag * PetscSinReal(2.0 * PETSC_PI * u2); in go()
607 … if (dim == 2 && xx_t[grid][tid][pp] < lo[0]) xx_t[grid][tid][pp] = -xx_t[grid][tid][pp]; in go()
608 if (dim == 3) zz_t[grid][tid][pp] = lo[2] + hp[2] / 2.0 + pk * hp[2]; in go()
610 … if (dim == 2 && xx_t[grid][tid][pp] < 0) xx_t[grid][tid][pp] = -xx_t[grid][tid][pp]; // ??? in go()
612 … while (zz_t[grid][tid][pp] >= hi[2] || zz_t[grid][tid][pp] <= lo[2]) zz_t[grid][tid][pp] *= .9; in go()
614 … while (xx_t[grid][tid][pp] >= hi[0] || xx_t[grid][tid][pp] <= lo[0]) xx_t[grid][tid][pp] *= .9; in go()
615 … while (yy_t[grid][tid][pp] >= hi[1] || yy_t[grid][tid][pp] <= lo[1]) yy_t[grid][tid][pp] *= .9; in go()
618 …while (PetscSqrtReal(PetscSqr(xx_t[grid][tid][pp]) + PetscSqr(yy_t[grid][tid][pp])) > 0.92 * hi[0]… in go()
619 xx_t[grid][tid][pp] *= .9; in go()
620 yy_t[grid][tid][pp] *= .9; in go()
625 if (dim == 3) zz_t[grid][tid][pp] += p_shift; in go()
626 else yy_t[grid][tid][pp] += p_shift; in go()
627 …wp_t[grid][tid][pp] += ctx->n[grid] / NNreal * PetscSqrtReal(ctx->masses[ctx->species_offset[grid]… in go()
634 … PetscReal x[] = {xx_t[grid][tid][pp], yy_t[grid][tid][pp], dim == 2 ? 0 : zz_t[grid][tid][pp]}; in go()
635 …v2 = 0, fact = dim == 2 ? 2.0 * PETSC_PI * x[0] : 1, w = fact * wp_t[grid][tid][pp] * ctx->n_0 * c… in go()
647 wp_t[grid][tid][pp] = 0; in go()
648 xx_t[grid][tid][pp] = 1.e-7; in go()
649 yy_t[grid][tid][pp++] = hi[1] - 5.e-7; in go()
650 wp_t[grid][tid][pp] = 0; in go()
651 xx_t[grid][tid][pp] = hi[0] - 5.e-7; in go()
652 yy_t[grid][tid][pp++] = 0; in go()
653 wp_t[grid][tid][pp] = 0; in go()
654 xx_t[grid][tid][pp] = 1.e-7; in go()
655 yy_t[grid][tid][pp++] = lo[1] + 5.e-7; in go()
658 … pj = 0; pj < 6; pj++) xx_t[grid][tid][p0 + pj] = yy_t[grid][tid][p0 + pj] = zz_t[grid][tid][p0 + … in go()
659 xx_t[grid][tid][p0 + 0] = lo[0]; in go()
660 xx_t[grid][tid][p0 + 1] = hi[0]; in go()
661 yy_t[grid][tid][p0 + 2] = lo[1]; in go()
662 yy_t[grid][tid][p0 + 3] = hi[1]; in go()
663 zz_t[grid][tid][p0 + 4] = lo[2]; in go()
664 zz_t[grid][tid][p0 + 5] = hi[2]; in go()
675 … ? 2.0 * PETSC_PI * xx_t[grid][tid][pp] : 1, w = fact * ctx->n_0 * ctx->masses[ctx->species_offset… in go()
686 for (PetscInt tid = 0; tid < numthreads; tid++) { in go() local
687 const PetscInt v_id = v_id_0 + tid, glb_v_id = global_vertex_id_0 + v_id; in go()
705 for (PetscInt tid = 0; tid < numthreads; tid++) { in go() local
706 const PetscInt v_id = v_id_0 + tid, glb_v_id = global_vertex_id_0 + v_id; in go()
713 …err_t = makeSwarm(sw, dim, Np_t[grid][tid], xx_t[grid][tid], yy_t[grid][tid], zz_t[grid… in go()
718 for (PetscInt tid = 0; tid < numthreads; tid++) { in go() local
719 const PetscInt v_id = v_id_0 + tid, glb_v_id = global_vertex_id_0 + v_id; in go()
732 for (PetscInt tid = 0; tid < numthreads; tid++) { in go() local
733 const PetscInt v_id = v_id_0 + tid, glb_v_id = global_vertex_id_0 + v_id; in go()
739 Vec subX = globXArray[LAND_PACK_IDX(v_id, grid)], work = t_fhat[grid][tid]; in go()
741 …ierr_t = particlesToGrid(dm, sw, tid, dim, wp_t[grid][tid], subX, globMpArray[LAND_PACK_IDX(v_id, … in go()
745 ierr_t = KSPSolve(t_ksp[grid][tid], work, subX); in go()
752 for (PetscInt tid = 0; tid < numthreads; tid++) { in go() local
753 const PetscInt v_id = v_id_0 + tid, glb_v_id = global_vertex_id_0 + v_id; in go()
756 … PetscCall(PetscFree4(xx_t[grid][tid], yy_t[grid][tid], wp_t[grid][tid], zz_t[grid][tid])); in go()
852 for (PetscInt tid = 0; tid < numthreads; tid++) { in go() local
853 const PetscInt v_id = v_id_0 + tid, glb_v_id = global_vertex_id_0 + v_id; in go()
875 Np_t[grid][tid] = NN; in go()
876 …Call(PetscMalloc4(NN, &xx_t[grid][tid], NN, &yy_t[grid][tid], NN, &wp_t[grid][tid], dim == 2 ? 1 :… in go()
885 wp_t[grid][tid][pp] = 0; in go()
886 xx_t[grid][tid][pp] = lo[0] + hp[0] / 2.0 + pi * hp[0]; in go()
887 yy_t[grid][tid][pp] = lo[1] + hp[1] / 2.0 + pj * hp[1]; in go()
888 if (dim == 3) zz_t[grid][tid][pp] = lo[2] + hp[2] / 2.0 + pk * hp[2]; in go()
894 wp_t[grid][tid][pp] = 0; in go()
895 xx_t[grid][tid][pp] = 1.e-7; in go()
896 yy_t[grid][tid][pp++] = hi[1] - 5.e-7; in go()
897 wp_t[grid][tid][pp] = 0; in go()
898 xx_t[grid][tid][pp] = hi[0] - 5.e-7; in go()
899 yy_t[grid][tid][pp++] = 0; in go()
900 wp_t[grid][tid][pp] = 0; in go()
901 xx_t[grid][tid][pp] = 1.e-7; in go()
902 yy_t[grid][tid][pp++] = lo[1] + 5.e-7; in go()
905 … pj = 0; pj < 6; pj++) xx_t[grid][tid][p0 + pj] = yy_t[grid][tid][p0 + pj] = zz_t[grid][tid][p0 + … in go()
906 xx_t[grid][tid][p0 + 0] = lo[0]; in go()
907 xx_t[grid][tid][p0 + 1] = hi[0]; in go()
908 yy_t[grid][tid][p0 + 2] = lo[1]; in go()
909 yy_t[grid][tid][p0 + 3] = hi[1]; in go()
910 zz_t[grid][tid][p0 + 4] = lo[2]; in go()
911 zz_t[grid][tid][p0 + 5] = hi[2]; in go()
917 for (PetscInt tid = 0; tid < numthreads; tid++) { in go() local
918 const PetscInt v_id = v_id_0 + tid, glb_v_id = global_vertex_id_0 + v_id; in go()
942 for (PetscInt tid = 0; tid < numthreads; tid++) { in go() local
943 const PetscInt v_id = v_id_0 + tid, glb_v_id = global_vertex_id_0 + v_id; in go()
950 …err_t = makeSwarm(sw, dim, Np_t[grid][tid], xx_t[grid][tid], yy_t[grid][tid], zz_t[grid… in go()
957 for (PetscInt tid = 0; tid < numthreads; tid++) { in go() local
958 const PetscInt v_id = v_id_0 + tid, glb_v_id = global_vertex_id_0 + v_id; in go()
972 for (PetscInt tid = 0; tid < numthreads; tid++) { in go() local
973 const PetscInt v_id = v_id_0 + tid, glb_v_id = global_vertex_id_0 + v_id; in go()
976 … PetscCall(PetscFree4(xx_t[grid][tid], yy_t[grid][tid], wp_t[grid][tid], zz_t[grid][tid])); in go()
1011 for (PetscInt tid = 0; tid < numthreads; tid++) { in go() local
1012 const PetscInt v_id = v_id_0 + tid, glb_v_id = global_vertex_id_0 + v_id; in go()
1030 for (PetscInt tid = 0; tid < numthreads; tid++) { in go() local
1031 PetscCall(VecDestroy(&t_fhat[grid][tid])); in go()
1032 PetscCall(KSPDestroy(&t_ksp[grid][tid])); in go()