| /petsc/src/ksp/pc/impls/vpbjacobi/cuda/ |
| H A D | vpbjacobi_cuda.cu | 9 PetscInt nsize; /* sum of sizes of the point blocks */ member
21 …nt n, PetscInt nblocks, PetscInt nsize, const PetscInt *bsizes, MatScalar *diag_h) : n(n), nblocks… in PC_VPBJacobi_CUDA()
28 PetscCallCUDAVoid(cudaMalloc(&diag_d, sizeof(MatScalar) * nsize)); in PC_VPBJacobi_CUDA()
39 PetscCallCUDA(cudaMemcpy(diag_d, diag_h, sizeof(MatScalar) * nsize, cudaMemcpyHostToDevice)); in UpdateOffsetsOnDevice()
40 … PetscCall(PetscLogCpuToGpu(sizeof(PetscInt) * (2 * nblocks + 2 + n) + sizeof(MatScalar) * nsize)); in UpdateOffsetsOnDevice()
131 PetscCall(PetscLogGpuFlops(pcuda->nsize * 2)); /* FMA on entries in all blocks */ in PCApplyOrTranspose_VPBJacobi_CUDA()
164 PetscInt i, n, nblocks, nsize = 0; in PCSetUp_VPBJacobi_CUDA() local
170 for (i = 0; i < nblocks; i++) nsize += bsizes[i] * bsizes[i]; in PCSetUp_VPBJacobi_CUDA()
174 if (pcuda && (pcuda->n != n || pcuda->nblocks != nblocks || pcuda->nsize != nsize)) { in PCSetUp_VPBJacobi_CUDA()
180 PetscCallCXX(jac->spptr = new PC_VPBJacobi_CUDA(n, nblocks, nsize, bsizes, jac->diag)); in PCSetUp_VPBJacobi_CUDA()
|
| /petsc/src/ksp/pc/impls/vpbjacobi/kokkos/ |
| H A D | vpbjacobi_kok.kokkos.cxx | 15 PetscInt nsize; /* sum of sizes (elements) of the point blocks */ member
32 PC_VPBJacobi_Kokkos(PetscInt n, PetscInt nblocks, PetscInt nsize, const PetscInt *bsizes) : in PC_VPBJacobi_Kokkos()
33 n(n), nblocks(nblocks), nsize(nsize), bs_dual(NoInit("bs_dual"), nblocks + 1), in PC_VPBJacobi_Kokkos()
35 diag(NoInit("diag"), nsize), work(NoInit("work"), nsize) in PC_VPBJacobi_Kokkos()
129 PetscCall(PetscLogGpuFlops(pckok->nsize * 2)); /* FMA on entries in all blocks */ in PCApplyOrTranspose_VPBJacobi_Kokkos()
148 PetscInt i, nlocal, nblocks, nsize = 0; in PCSetUp_VPBJacobi_Kokkos() local
163 nsize += bsizes[i] * bsizes[i]; in PCSetUp_VPBJacobi_Kokkos()
171 if (pckok && (pckok->n != nlocal || pckok->nblocks != nblocks || pckok->nsize != nsize)) { in PCSetUp_VPBJacobi_Kokkos()
178 PetscCallCXX(pckok = new PC_VPBJacobi_Kokkos(nlocal, nblocks, nsize, bsizes)); in PCSetUp_VPBJacobi_Kokkos()
|
| /petsc/src/binding/petsc4py/src/petsc4py/PETSc/ |
| H A D | Partitioner.pyx | 176 cdef PetscInt nsize = 0
178 sizes = iarray_i(sizes, &nsize, &csizes)
179 if nsize != cnumProcs:
181 numProcs, toInt(nsize))
|
| /petsc/src/sys/memory/ |
| H A D | mtr.c | 164 size_t nsize; in PetscTrMallocDefault() local
174 nsize = (a + (PETSC_MEMALIGN - 1)) & ~(PETSC_MEMALIGN - 1); in PetscTrMallocDefault()
175 …PetscCall(PetscMallocAlign(nsize + sizeof(TrSPACE) + sizeof(PetscInt), clear, lineno, function, fi… in PetscTrMallocDefault()
184 head->size = nsize; in PetscTrMallocDefault()
192 *(PetscInt *)(inew + nsize) = COOKIE_VALUE; in PetscTrMallocDefault()
238 PetscLogMallocLength[PetscLogMalloc] = nsize; in PetscTrMallocDefault()
326 size_t nsize; in PetscTrReallocDefault() local
382 nsize = (len + (PETSC_MEMALIGN - 1)) & ~(PETSC_MEMALIGN - 1); in PetscTrReallocDefault()
383 …PetscCall(PetscReallocAlign(nsize + sizeof(TrSPACE) + sizeof(PetscInt), lineno, function, filename… in PetscTrReallocDefault()
392 head->size = nsize; in PetscTrReallocDefault()
[all …]
|
| /petsc/src/dm/partitioner/impls/multistage/ |
| H A D | mspart.c | 152 PetscMPIInt size, nsize, isize, rank, nrank, i, l, n, m; in PetscPartitionerMultistage_CreateStages() local
195 PetscCallMPI(MPI_Comm_size(ncomm, &nsize)); in PetscPartitionerMultistage_CreateStages()
196 if (nsize == size) { /* one node */ in PetscPartitionerMultistage_CreateStages()
209 PetscCall(PetscMalloc2(size, &ranks, nsize, &granks)); in PetscPartitionerMultistage_CreateStages()
210 for (i = 0; i < nsize; i++) ranks[i] = i; in PetscPartitionerMultistage_CreateStages()
213 PetscCallMPI(MPI_Group_translate_ranks(ngroup, nsize, ranks, ggroup, granks)); in PetscPartitionerMultistage_CreateStages()
214 PetscCallMPI(MPI_Group_incl(ggroup, nsize, granks, &lgroup[1])); in PetscPartitionerMultistage_CreateStages()
|
| /petsc/src/ksp/ksp/impls/cg/pipecgrr/ |
| H A D | pipecgrr.c | 22 PetscInt i = 0, replace = 0, nsize; in KSPSolve_PIPECGRR() local
99 PetscCall(VecGetSize(B, &nsize)); in KSPSolve_PIPECGRR()
100 nsi = (PetscReal)nsize; in KSPSolve_PIPECGRR()
|
| /petsc/src/ksp/pc/impls/vpbjacobi/ |
| H A D | vpbjacobi.c | 213 PetscInt i, nsize = 0, nlocal; in PCSetUp_VPBJacobi_Host() local
226 nsize += bsizes[i] * bsizes[i]; in PCSetUp_VPBJacobi_Host()
228 PetscCall(PetscMalloc1(nsize, &jac->diag)); in PCSetUp_VPBJacobi_Host()
|
| /petsc/src/sys/classes/draw/impls/image/ |
| H A D | drawimage.c | 536 PetscInt size[2], nsize = 2; in PetscDrawCreate_Image() local
554 …rray(((PetscObject)draw)->options, ((PetscObject)draw)->prefix, "-draw_size", size, &nsize, &set)); in PetscDrawCreate_Image()
555 if (set && nsize == 1) size[1] = size[0]; in PetscDrawCreate_Image()
|
| /petsc/config/ |
| H A D | petsc_harness.sh | 88 n ) nsize="$OPTARG" ;;
|
| /petsc/doc/developers/ |
| H A D | testing.md | 117 - **nsize**: (*Optional*; *Default:* `nsize=1`)
119 - This integer is passed to mpiexec; i.e., `mpiexec -n nsize`
275 It is supported within `nsize` and `args`. For example,
278 nsize: {{1 2 4}}
376 nsize: 2
|