| /petsc/src/dm/impls/da/ |
| H A D | da3.c | 56 … w %" PetscInt_FMT " s %" PetscInt_FMT "\n", rank, dd->M, dd->N, dd->P, dd->m, dd->n, dd->p, dd->w… in DMView_DA_3d() 196 PetscMPIInt m, n, p; in DMSetUp_DA_3D() local 228 PetscCall(PetscMPIIntCast(dd->m, &m)); in DMSetUp_DA_3D() 235 if (m != PETSC_DECIDE) { in DMSetUp_DA_3D() 236 …PetscCheck(m >= 1, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Non-positive number of processors i… in DMSetUp_DA_3D() 237 …PetscCheck(m <= size, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Too many processors in X directi… in DMSetUp_DA_3D() 247 …PetscCheck(m <= 0 || n <= 0 || p <= 0 || m * n * p == size, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRA… in DMSetUp_DA_3D() 250 if (m == PETSC_DECIDE && n != PETSC_DECIDE && p != PETSC_DECIDE) { in DMSetUp_DA_3D() 251 m = size / (n * p); in DMSetUp_DA_3D() 252 } else if (m != PETSC_DECIDE && n == PETSC_DECIDE && p != PETSC_DECIDE) { in DMSetUp_DA_3D() [all …]
|
| H A D | da2.c | 50 …t_FMT " w %" PetscInt_FMT " s %" PetscInt_FMT "\n", rank, dd->M, dd->N, dd->m, dd->n, dd->w, dd->s… in DMView_DA_2d() 192 PetscMPIInt m, n; in DMSetUp_DA_2D() local 217 PetscCall(PetscMPIIntCast(dd->m, &m)); in DMSetUp_DA_2D() 224 if (m != PETSC_DECIDE) { in DMSetUp_DA_2D() 225 …PetscCheck(m >= 1, comm, PETSC_ERR_ARG_OUTOFRANGE, "Non-positive number of processors in X directi… in DMSetUp_DA_2D() 226 …PetscCheck(m <= size, comm, PETSC_ERR_ARG_OUTOFRANGE, "Too many processors in X direction: %d %d",… in DMSetUp_DA_2D() 233 if (m == PETSC_DECIDE || n == PETSC_DECIDE) { in DMSetUp_DA_2D() 235 m = size / n; in DMSetUp_DA_2D() 236 } else if (m != PETSC_DECIDE) { in DMSetUp_DA_2D() 237 n = size / m; in DMSetUp_DA_2D() [all …]
|
| /petsc/src/mat/tests/ |
| H A D | ex226.c | 6 PetscInt global_index(PetscInt i, PetscInt j, PetscInt k, PetscInt m, PetscInt n) in global_index() argument 8 return i + j * m + k * m * n; in global_index() 14 PetscInt i, M, N, Istart, Iend, n = 7, j, J, Ii, m = 8, k, o = 1; in main() local 22 PetscCall(PetscOptionsGetInt(NULL, NULL, "-m", &m, NULL)); in main() 29 M = N = m * n * o; in main() 36 …PetscCheck(o >= 1 && m > 1 && n >= 1, PETSC_COMM_WORLD, PETSC_ERR_USER, "Dimensions need to be lar… in main() 46 k = Ii / (m * n); in main() 47 j = (Ii - k * m * n) / m; in main() 48 i = (Ii - k * m * n - j * m); in main() 50 J = global_index(i - 1, j, k, m, n); in main() [all …]
|
| H A D | ex201f.F90 | 153 Mat m, mt 169 PetscCallA(MatCreateShell(PETSC_COMM_SELF, i12, i12, i12, i12, i0, m, ierr)) 170 PetscCallA(MatShellSetManageScalingShifts(m, ierr)) 171 PetscCallA(MatAssemblyBegin(m, MAT_FINAL_ASSEMBLY, ierr)) 172 PetscCallA(MatAssemblyEnd(m, MAT_FINAL_ASSEMBLY, ierr)) 175 PetscCallA(MatShellSetOperation(m, op, mymatmult, ierr)) 177 PetscCallA(MatShellSetOperation(m, op, mymatmultadd, ierr)) 179 PetscCallA(MatShellSetOperation(m, op, mymatmulttranspose, ierr)) 181 PetscCallA(MatShellSetOperation(m, op, mymatmulthermitiantranspose, ierr)) 183 PetscCallA(MatShellSetOperation(m, op, mymatmulttransposeadd, ierr)) [all …]
|
| H A D | ex36f.F90 | 23 PetscInt n, m 28 m = 5 33 PetscCall(MatSetSizes(A, m, n, m, n, ierr)) 41 PetscCall(FillUpMatrix(m, n, aa)) 63 PetscInt n, m 65 parameter(m=5, n=4) 67 PetscScalar aa(m, n) 70 PetscCall(MatCreateSeqDense(PETSC_COMM_SELF, m, n, aa, A, ierr)) 73 PetscCall(FillUpMatrix(m, n, aa)) 88 subroutine FillUpMatrix(m, n, X) argument [all …]
|
| H A D | ex105f.F90 | 10 Mat m 15 PetscCallA(MatCreate(PETSC_COMM_WORLD, m, ierr)) 16 PetscCallA(MatSetSizes(m, PETSC_DECIDE, PETSC_DECIDE, twelve, twelve, ierr)) 17 PetscCallA(MatSetFromOptions(m, ierr)) 18 …PetscCallA(MatMPIAIJSetPreallocation(m, PETSC_DEFAULT_INTEGER, PETSC_NULL_INTEGER_ARRAY, PETSC_DEF… 23 PetscCallA(MatSetValuesMPIAIJ(m, one, [i], one, [i], [value], ADD_VALUES, ierr)) 26 PetscCallA(MatSetValuesMPIAIJ(m, one, [i], one, [j], [value], ADD_VALUES, ierr)) 29 PetscCallA(MatSetValuesMPIAIJ(m, one, [i], one, [j], [value], ADD_VALUES, ierr)) 30 PetscCallA(MatAssemblyBegin(m, MAT_FINAL_ASSEMBLY, ierr)) 31 PetscCallA(MatAssemblyEnd(m, MAT_FINAL_ASSEMBLY, ierr)) [all …]
|
| H A D | ex85f.F90 | 12 Mat m 18 PetscCallA(MatCreate(PETSC_COMM_SELF, m, ierr)) 19 PetscCallA(MatSetSizes(m, twelve, twelve, twelve, twelve, ierr)) 20 PetscCallA(MatSetFromOptions(m, ierr)) 21 PetscCallA(MatSetUp(m, ierr)) 26 PetscCallA(MatSetValues(m, one, [i], one, [i], [value], INSERT_VALUES, ierr)) 27 PetscCallA(MatAssemblyBegin(m, MAT_FINAL_ASSEMBLY, ierr)) 28 PetscCallA(MatAssemblyEnd(m, MAT_FINAL_ASSEMBLY, ierr)) 31 PetscCallA(MatGetDiagonal(m, v, ierr)) 34 PetscCallA(MatDestroy(m, ierr))
|
| H A D | ex126f.F90 | 12 PetscInt i, j, II, JJ, m 23 m = 10 30 PetscCallA(PetscOptionsGetInt(PETSC_NULL_OPTIONS, PETSC_NULL_CHARACTER, '-m', m, flg, ierr)) 34 PetscCallA(MatSetSizes(A, PETSC_DECIDE, PETSC_DECIDE, m*m, m*m, ierr)) 44 i = II/m 45 j = II - i*m 47 JJ = II - m 50 if (i < m - 1) then 51 JJ = II + m 58 if (j < m - 1) then [all …]
|
| /petsc/src/ts/tests/ |
| H A D | ex4.c | 21 PetscInt m; /* the number of mesh points in x-direction */ member 60 data.m = 9; in main() 64 data.dx = 1.0 / (data.m + 1.0); in main() 66 mn = (data.m) * (data.n); in main() 198 PetscInt m, row, col; in Initial() local 205 m = data->m; in Initial() 217 row = 1 + (mybase + i) - ((mybase + i) / m) * m; in Initial() 218 col = (mybase + i) / m + 1; in Initial() 276 PetscInt m, n, mn; in RHSJacobian() local 280 m = data->m; in RHSJacobian() [all …]
|
| /petsc/src/ksp/ksp/tutorials/ |
| H A D | ex3.c | 34 PetscInt idx[4], count, *rows, i, m = 5, start, end, its; in main() local 38 PetscCall(PetscOptionsGetInt(NULL, NULL, "-m", &m, NULL)); in main() 39 N = (m + 1) * (m + 1); in main() 40 M = m * m; in main() 41 h = 1.0 / m; in main() 67 idx[0] = (m + 1) * (i / m) + (i % m); in main() 69 idx[2] = idx[1] + m + 1; in main() 94 x = h * (i % m); in main() 95 y = h * (i / m); in main() 97 idx[0] = (m + 1) * (i / m) + (i % m); in main() [all …]
|
| H A D | ex13f90.F90 | 9 PetscInt m 15 subroutine UserInitializeLinearSolver(m, n, userctx, ierr) argument 17 PetscInt m, n 35 hx2 = (m + 1)*(m + 1) 37 Ntot = m*n 62 userctx%m = m 85 PetscInt m, n, one 94 m = userctx%m 111 do i = 1, m 113 JJ = II - m [all …]
|
| /petsc/src/ts/tutorials/autodiff/adolc-utils/ |
| H A D | drivers.cxx | 34 PetscInt i, j, m = adctx->m, n = adctx->n, p = adctx->p; in PetscAdolcComputeRHSJacobian() local 38 PetscCall(AdolcMalloc2(m, p, &J)); in PetscAdolcComputeRHSJacobian() 39 if (adctx->Seed) fov_forward(tag, m, n, p, u_vec, adctx->Seed, NULL, J); in PetscAdolcComputeRHSJacobian() 40 else jacobian(tag, m, n, u_vec, J); in PetscAdolcComputeRHSJacobian() 42 PetscCall(RecoverJacobian(A, INSERT_VALUES, m, p, adctx->Rec, J, NULL)); in PetscAdolcComputeRHSJacobian() 44 for (i = 0; i < m; i++) { in PetscAdolcComputeRHSJacobian() 72 PetscInt i, j, m = adctx->m, n = adctx->n, p = adctx->p; in PetscAdolcComputeRHSJacobianLocal() local 76 PetscCall(AdolcMalloc2(m, p, &J)); in PetscAdolcComputeRHSJacobianLocal() 77 if (adctx->Seed) fov_forward(tag, m, n, p, u_vec, adctx->Seed, NULL, J); in PetscAdolcComputeRHSJacobianLocal() 78 else jacobian(tag, m, n, u_vec, J); in PetscAdolcComputeRHSJacobianLocal() [all …]
|
| /petsc/src/ksp/ksp/tests/ |
| H A D | ex4.c | 40 PetscInt i, m = 2, N, M, its, idx[4], count, *rows; in main() local 51 PetscCall(PetscOptionsGetInt(NULL, NULL, "-m", &m, NULL)); in main() 53 N = (m + 1) * (m + 1); /* dimension of matrix */ in main() 54 M = m * m; /* number of elements */ in main() 55 h = 1.0 / m; /* mesh width */ in main() 74 idx[0] = (m + 1) * (i / m) + (i % m); in main() 76 idx[2] = idx[1] + m + 1; in main() 101 x = h * (i % m); in main() 102 y = h * (i / m); in main() 104 idx[0] = (m + 1) * (i / m) + (i % m); in main() [all …]
|
| H A D | ex3.c | 46 PetscInt i, m = 5, N, start, end, M, its; in main() local 57 PetscCall(PetscOptionsGetInt(NULL, NULL, "-m", &m, NULL)); in main() 60 N = (m + 1) * (m + 1); /* dimension of matrix */ in main() 61 M = m * m; /* number of elements */ in main() 62 h = 1.0 / m; /* mesh width */ in main() 78 idx[0] = (m + 1) * (i / m) + (i % m); in main() 80 idx[2] = idx[1] + m + 1; in main() 101 x = h * (i % m); in main() 102 y = h * (i / m); in main() 104 idx[0] = (m + 1) * (i / m) + (i % m); in main() [all …]
|
| /petsc/src/mat/impls/aij/seq/crl/ftn-kernels/ |
| H A D | fmultcrl.F90 | 10 PetscInt, intent(in) :: m, rmax, icols(m, rmax) 11 PetscScalar, intent(in) :: x(0:m - 1), acols(m, rmax) 12 PetscScalar, intent(out) :: y(m) 16 y(1:m) = acols(1:m, 1)*x(icols(1:m, 1)) 18 y(1:m) = y(1:m) + acols(1:m, i)*x(icols(1:m, i))
|
| /petsc/src/ksp/ksp/utils/lmvm/dense/ |
| H A D | cd_utils.c | 16 PetscInt m = lmvm->m; in VecCyclicShift() local 36 PetscCall(PetscArraycpy(dest, &src[d], m - d)); in VecCyclicShift() 37 PetscCall(PetscArraycpy(&dest[m - d], src, d)); in VecCyclicShift() 42 PetscCall(PetscDeviceRegisterMemory(dest, dest_memtype, m * sizeof(*dest))); in VecCyclicShift() 43 PetscCall(PetscDeviceRegisterMemory(src, src_memtype, m * sizeof(*src))); in VecCyclicShift() 44 PetscCall(PetscDeviceArrayCopy(dctx, dest, &src[d], m - d)); in VecCyclicShift() 45 PetscCall(PetscDeviceArrayCopy(dctx, &dest[m - d], src, d)); in VecCyclicShift() 52 static inline PetscInt recycle_index(PetscInt m, PetscInt idx) in recycle_index() argument 54 return idx % m; in recycle_index() 57 static inline PetscInt oldest_update(PetscInt m, PetscInt idx) in oldest_update() argument [all …]
|
| /petsc/src/tao/leastsquares/tutorials/ |
| H A D | cs1.c | 210 PetscInt m, n; in EvaluateFunction() local 219 for (m = 0; m < M; m++) { in EvaluateFunction() 220 f[m] = -b[m]; in EvaluateFunction() 221 for (n = 0; n < N; n++) f[m] += user->A[m][n] * x[n]; in EvaluateFunction() 234 PetscInt m, n; in EvaluateJacobian() local 241 for (m = 0; m < M; ++m) { in EvaluateJacobian() 242 for (n = 0; n < N; ++n) user->J[m][n] = user->A[m][n]; in EvaluateJacobian() 279 PetscInt m, n, k; /* loop index for M,N,K dimension. */ in InitializeUserData() local 283 m = 0; in InitializeUserData() 284 b[m++] = 0.28; in InitializeUserData() [all …]
|
| /petsc/share/petsc/matlab/ |
| H A D | makefile | 3 …iles.m laplacian.m launch.m PetscBagRead.m PetscBinaryRead.m PetscBinaryWrite.m PetscReadBinaryMat…
|
| /petsc/src/binding/petsc4py/demo/legacy/bratu2d/ |
| H A D | bratu2df90.f90 | 8 integer :: m, n local 10 real(kind=8), intent(in), target :: x(m,n) 11 real(kind=8), intent(inout) :: f(m,n) 16 u => x(2:m-1, 2:n-1) ! center 17 uN => x(2:m-1, 1:n-2) ! north 18 uS => x(2:m-1, 3:n ) ! south 19 uW => x(1:m-2, 2:n-1) ! west 20 uE => x(3:m, 2:n-1) ! east 22 hx = 1.0/(m-1) ! x grid spacing 25 f(2:m-1, 2:n-1) = &
|
| /petsc/include/ |
| H A D | petscdevicetypes.h | 63 #define PetscMemTypeHost(m) (((m) & 0x1) == PETSC_MEMTYPE_HOST) argument 64 #define PetscMemTypeDevice(m) (((m) & 0x1) == PETSC_MEMTYPE_DEVICE) argument 65 #define PetscMemTypeCUDA(m) (((m) & 0xF) == PETSC_MEMTYPE_CUDA) argument 66 #define PetscMemTypeHIP(m) (((m) & 0xF) == PETSC_MEMTYPE_HIP) argument 67 #define PetscMemTypeSYCL(m) (((m) & 0xF) == PETSC_MEMTYPE_SYCL) argument 68 #define PetscMemTypeNVSHMEM(m) ((m) == PETSC_MEMTYPE_NVSHMEM) argument 147 #define PetscOffloadUnallocated(m) ((m) == PETSC_OFFLOAD_UNALLOCATED) argument 148 #define PetscOffloadHost(m) (((m) & PETSC_OFFLOAD_CPU) == PETSC_OFFLOAD_CPU) argument 149 #define PetscOffloadDevice(m) (((m) & PETSC_OFFLOAD_GPU) == PETSC_OFFLOAD_GPU) argument 150 #define PetscOffloadBoth(m) ((m) == PETSC_OFFLOAD_BOTH) argument [all …]
|
| /petsc/src/ksp/ksp/utils/lmvm/blas_cyclic/ |
| H A D | blas_cyclic.c | 16 static inline void AXPBY_Private(PetscInt m, PetscScalar alpha, const PetscScalar x[], PetscScalar … in AXPBY_Private() argument 18 for (PetscInt i = 0; i < m; i++) y[i * y_stride] = alpha * x[i] + beta * y[i * y_stride]; in AXPBY_Private() 21 static PetscErrorCode AXPBYCylic_Private(PetscInt m, PetscInt oldest, PetscInt next, PetscScalar al… in AXPBYCylic_Private() argument 23 PetscInt i_oldest = oldest % m; in AXPBYCylic_Private() 24 PetscInt i_next = ((next - 1) % m) + 1; in AXPBYCylic_Private() 27 if (next - oldest == m) { in AXPBYCylic_Private() 28 AXPBY_Private(m, alpha, x, beta, y, y_stride); in AXPBYCylic_Private() 33 AXPBY_Private(m - i_oldest, alpha, &x[i_oldest], beta, &y[i_oldest * y_stride], y_stride); in AXPBYCylic_Private() 42 PetscInt m, m_local; in VecAXPBYCyclic() local 53 PetscCall(VecGetSize(x, &m)); in VecAXPBYCyclic() [all …]
|
| /petsc/src/ksp/pc/tests/ |
| H A D | ex6.c | 45 PetscInt i, m = 2, N, M, idx[4], Nsub1, Nsub2, ol = 1, x1, x2; in main() local 53 PetscCall(PetscOptionsGetInt(NULL, NULL, "-m", &m, NULL)); in main() 54 N = (m + 1) * (m + 1); /* dimension of matrix */ in main() 55 M = m * m; /* number of elements */ in main() 56 h = 1.0 / m; /* mesh width */ in main() 57 x1 = (m + 1) / 2; in main() 68 idx[0] = (m + 1) * (i / m) + (i % m); in main() 70 idx[2] = idx[1] + m + 1; in main() 77 for (ol = 0; ol < m + 2; ++ol) { in main() 78 PetscCall(PCASMCreateSubdomains2D(m + 1, m + 1, x1, x2, 1, 0, &Nsub1, &is1, &islocal1)); in main() [all …]
|
| /petsc/src/mat/impls/aij/seq/aijmkl/ |
| H A D | aijmkl.h | 17 …#define mkl_cspblas_xcsrgemv(transa, m, a, ia, ja, x, y) mkl_cspblas_scsrgemv(transa, m, a, ia, ja… argument 19 …#define mkl_cspblas_xcsrgemv(transa, m, a, ia, ja, x, y) mkl_cspblas_dcsrgemv(transa, m, a, ia, ja… argument 23 …#define mkl_cspblas_xcsrgemv(transa, m, a, ia, ja, x, y) mkl_cspblas_ccsrgemv(transa, m, a, ia, ja… argument 25 …#define mkl_cspblas_xcsrgemv(transa, m, a, ia, ja, x, y) mkl_cspblas_zcsrgemv(transa, m, a, ia, ja… argument 33 …#define mkl_xcsrmv(transa, m, k, alpha, matdescra, val, indx, pntrb, pntre, x, beta, y) mkl_scsrmv… argument 35 …#define mkl_xcsrmv(transa, m, k, alpha, matdescra, val, indx, pntrb, pntre, x, beta, y) mkl_dcsrmv… argument 39 …#define mkl_xcsrmv(transa, m, k, alpha, matdescra, val, indx, pntrb, pntre, x, beta, y) mkl_ccsrmv… argument 41 …#define mkl_xcsrmv(transa, m, k, alpha, matdescra, val, indx, pntrb, pntre, x, beta, y) mkl_zcsrmv… argument
|
| /petsc/src/mat/impls/nest/ftn-custom/ |
| H A D | zmatnestf.c | 16 Mat *m, *tmp; in matcreatenest_() local 22 *ierr = PetscMalloc1((*nr) * (*nc), &m); in matcreatenest_() 32 m[i] = (tmp == NULL ? NULL : a[i]); in matcreatenest_() 34 *ierr = MatCreateNest(MPI_Comm_f2c(*comm), *nr, is_row, *nc, is_col, m, B); in matcreatenest_() 36 *ierr = PetscFree(m); in matcreatenest_() 41 Mat *m, *tmp; in matnestsetsubmats_() local 48 *ierr = PetscMalloc1((*nr) * (*nc), &m); in matnestsetsubmats_() 60 m[i] = (tmp == NULL ? NULL : a[i]); in matnestsetsubmats_() 62 *ierr = MatNestSetSubMats(*B, *nr, is_row, *nc, is_col, m); in matnestsetsubmats_() 64 *ierr = PetscFree(m); in matnestsetsubmats_() [all …]
|
| /petsc/src/mat/impls/aij/seq/crl/ |
| H A D | crl.c | 35 PetscInt m = A->rmap->n; /* Number of rows in the matrix. */ in MatSeqAIJCRL_create_aijcrl() local 43 aijcrl->m = A->rmap->n; in MatSeqAIJCRL_create_aijcrl() 47 PetscCall(PetscMalloc2(rmax * m, &aijcrl->acols, rmax * m, &aijcrl->icols)); in MatSeqAIJCRL_create_aijcrl() 50 for (i = 0; i < m; i++) { in MatSeqAIJCRL_create_aijcrl() 52 acols[j * m + i] = *aa++; in MatSeqAIJCRL_create_aijcrl() 53 icols[j * m + i] = *aj++; in MatSeqAIJCRL_create_aijcrl() 56 acols[j * m + i] = 0.0; in MatSeqAIJCRL_create_aijcrl() 57 icols[j * m + i] = (j) ? icols[(j - 1) * m + i] : 0; /* handle case where row is EMPTY */ in MatSeqAIJCRL_create_aijcrl() 60 …iply %g. Rmax= %" PetscInt_FMT "\n", 1.0 - ((double)a->nz) / PetscMax((double)rmax * m, 1), rmax)); in MatSeqAIJCRL_create_aijcrl() 89 PetscInt m = aijcrl->m; /* Number of rows in the matrix. */ in MatMult_AIJCRL() local [all …]
|