Searched refs:gridSize (Results 1 – 4 of 4) sorted by relevance
| /petsc/src/dm/impls/patch/tests/ |
| H A D | ex1.c | 17 MatStencil gridSize; /* Size of patch grid */ member
23 PetscInt patchSize, commSize, gridSize; in ProcessOptions() local
30 gridSize = 1; in ProcessOptions()
37 …oundedInt("-grid_size", "The grid size in each dimension", "ex1.c", gridSize, &gridSize, NULL, 1)); in ProcessOptions()
42 options->gridSize.i = options->gridSize.j = options->gridSize.k = 1; in ProcessOptions()
46 options->gridSize.i = gridSize; in ProcessOptions()
51 options->gridSize.j = gridSize; in ProcessOptions()
56 options->gridSize.k = gridSize; in ProcessOptions()
69 …(DMPatchCreateGrid(PETSC_COMM_WORLD, user.dim, user.patchSize, user.commSize, user.gridSize, &dm)); in main()
|
| /petsc/src/dm/impls/patch/ |
| H A D | patchcreate.c | 100 …PI_Comm comm, PetscInt dim, MatStencil patchSize, MatStencil commSize, MatStencil gridSize, DM *dm) in DMPatchCreateGrid() argument
110 gridSize.j = 1; in DMPatchCreateGrid()
114 gridSize.k = 1; in DMPatchCreateGrid()
120 PetscCall(DMDASetSizes(da, gridSize.i, gridSize.j, gridSize.k)); in DMPatchCreateGrid()
|
| /petsc/src/ksp/pc/impls/pbjacobi/cuda/ |
| H A D | pbjacobi_cuda.cu | 8 const PetscInt gridSize = gridDim.x * blockDim.x; in MatMultBatched() local
13 for (; row < bs * mbs; row += gridSize) { in MatMultBatched()
53 PetscInt gridSize = PetscMin((bs * mbs + 255) / 256, 2147483647); /* <= 2^31-1 */ in PCApplyOrTranspose_PBJacobi_CUDA() local
54 …MatMultBatched<<<gridSize, 256>>>(bs, mbs, A, xx, yy, op == CUBLAS_OP_T ? PETSC_TRUE : PETSC_FALSE… in PCApplyOrTranspose_PBJacobi_CUDA()
|
| /petsc/src/ksp/pc/impls/vpbjacobi/cuda/ |
| H A D | vpbjacobi_cuda.cu | 83 const PetscInt gridSize = gridDim.x * blockDim.x; in MatMultBatched() local
88 for (; tid < n; tid += gridSize) { in MatMultBatched()
123 PetscInt gridSize = PetscMin((n + 255) / 256, 2147483647); /* <= 2^31-1 */ in PCApplyOrTranspose_VPBJacobi_CUDA() local
126 …MatMultBatched<<<gridSize, 256>>>(n, pcuda->bs_d, pcuda->bs2_d, pcuda->matIdx_d, pcuda->diag_d, xx… in PCApplyOrTranspose_VPBJacobi_CUDA()
|