| /petsc/src/ksp/ksp/impls/hpddm/cuda/ |
| H A D | hpddm.cu | 9 const PetscInt N = data->op->getDof() * n; in KSPSolve_HPDDM_CUDA_Private() local 24 PetscCall(PetscMalloc1(2 * N, &host_ptr)); in KSPSolve_HPDDM_CUDA_Private() 25 PetscCallCUDA(cudaMalloc((void **)&ptr, 2 * N * sizeof(K))); in KSPSolve_HPDDM_CUDA_Private() 27 dptr[1] = thrust::device_pointer_cast(ptr + N); in KSPSolve_HPDDM_CUDA_Private() 28 thrust::copy_n(thrust::cuda::par.on(PetscDefaultCudaStream), db, N, dptr[0]); in KSPSolve_HPDDM_CUDA_Private() 29 thrust::copy_n(thrust::cuda::par.on(PetscDefaultCudaStream), dx, N, dptr[1]); in KSPSolve_HPDDM_CUDA_Private() 30 PetscCallCUDA(cudaMemcpy(host_ptr, ptr, 2 * N * sizeof(K), cudaMemcpyDeviceToHost)); in KSPSolve_HPDDM_CUDA_Private() 31 PetscCall(HPDDM::IterativeMethod::solve(*data->op, host_ptr, host_ptr + N, n, comm)); in KSPSolve_HPDDM_CUDA_Private() 32 PetscCallCUDA(cudaMemcpy(ptr + N, host_ptr + N, N * sizeof(K), cudaMemcpyHostToDevice)); in KSPSolve_HPDDM_CUDA_Private() 33 thrust::copy_n(thrust::cuda::par.on(PetscDefaultCudaStream), dptr[1], N, dx); in KSPSolve_HPDDM_CUDA_Private() [all …]
|
| /petsc/src/vec/is/tests/output/ |
| H A D | ex7_ltog_info_nsize-5_bs-1_test-0.out | 10 [0] Local N 0 11 [1] Local N 0 12 [2] Local N 0 13 [3] Local N 0 14 [4] Local N 0 22 [0] Local N 0 23 [1] Local N 0 24 [2] Local N 0 25 [3] Local N 0 26 [4] Local N 0 [all …]
|
| H A D | ex7_ltog_info_nsize-5_bs-3_test-0.out | 10 [0] Local N 0 11 [1] Local N 0 12 [2] Local N 0 13 [3] Local N 0 14 [4] Local N 0 22 [0] Local N 0 23 [1] Local N 0 24 [2] Local N 0 25 [3] Local N 0 26 [4] Local N 0 [all …]
|
| H A D | ex7_ltog_info_nsize-4_bs-3_test-0.out | 9 [0] Local N 0 10 [1] Local N 0 11 [2] Local N 0 12 [3] Local N 0 19 [0] Local N 0 20 [1] Local N 0 21 [2] Local N 0 22 [3] Local N 0 25 [0] Local N 0 26 [1] Local N 0 [all …]
|
| H A D | ex7_ltog_info_nsize-4_bs-1_test-0.out | 9 [0] Local N 0 10 [1] Local N 0 11 [2] Local N 0 12 [3] Local N 0 19 [0] Local N 0 20 [1] Local N 0 21 [2] Local N 0 22 [3] Local N 0 25 [0] Local N 0 26 [1] Local N 0 [all …]
|
| H A D | ex7_ltog_info_nsize-3_bs-3_test-0.out | 8 [0] Local N 0 9 [1] Local N 0 10 [2] Local N 0 16 [0] Local N 0 17 [1] Local N 0 18 [2] Local N 0 21 [0] Local N 0 22 [1] Local N 0 23 [2] Local N 0
|
| H A D | ex7_ltog_info_nsize-3_bs-1_test-0.out | 8 [0] Local N 0 9 [1] Local N 0 10 [2] Local N 0 16 [0] Local N 0 17 [1] Local N 0 18 [2] Local N 0 21 [0] Local N 0 22 [1] Local N 0 23 [2] Local N 0
|
| /petsc/src/sys/utils/ |
| H A D | psplit.c | 26 PetscErrorCode PetscSplitOwnershipBlock(MPI_Comm comm, PetscInt bs, PetscInt *n, PetscInt *N) in PetscSplitOwnershipBlock() argument 31 …PetscCheck(*N != PETSC_DECIDE || *n != PETSC_DECIDE, PETSC_COMM_SELF, PETSC_ERR_ARG_INCOMP, "Both … in PetscSplitOwnershipBlock() 33 if (*N == PETSC_DECIDE) { in PetscSplitOwnershipBlock() 35 PetscCallMPI(MPIU_Allreduce(n, N, 1, MPIU_INT, MPI_SUM, comm)); in PetscSplitOwnershipBlock() 37 PetscInt Nbs = *N / bs; in PetscSplitOwnershipBlock() 67 PetscErrorCode PetscSplitOwnership(MPI_Comm comm, PetscInt *n, PetscInt *N) in PetscSplitOwnership() argument 72 …PetscCheck(*N != PETSC_DECIDE || *n != PETSC_DECIDE, PETSC_COMM_SELF, PETSC_ERR_ARG_INCOMP, "Both … in PetscSplitOwnership() 76 l[1] = (*N == PETSC_DECIDE) ? 1 : 0; in PetscSplitOwnership() 83 if (*N == PETSC_DECIDE) { in PetscSplitOwnership() 84 PetscCallMPI(MPIU_Allreduce(n, N, 1, MPIU_INT, MPI_SUM, comm)); in PetscSplitOwnership() [all …]
|
| /petsc/src/mat/impls/transpose/ |
| H A D | transm.c | 3 static PetscErrorCode MatMult_Transpose(Mat N, Vec x, Vec y) in MatMult_Transpose() argument 8 PetscCall(MatShellGetContext(N, &A)); in MatMult_Transpose() 13 static PetscErrorCode MatMultTranspose_Transpose(Mat N, Vec x, Vec y) in MatMultTranspose_Transpose() argument 18 PetscCall(MatShellGetContext(N, &A)); in MatMultTranspose_Transpose() 23 static PetscErrorCode MatSolve_Transpose_LU(Mat N, Vec b, Vec x) in MatSolve_Transpose_LU() argument 28 PetscCall(MatShellGetContext(N, &A)); in MatSolve_Transpose_LU() 33 static PetscErrorCode MatSolveAdd_Transpose_LU(Mat N, Vec b, Vec y, Vec x) in MatSolveAdd_Transpose_LU() argument 38 PetscCall(MatShellGetContext(N, &A)); in MatSolveAdd_Transpose_LU() 43 static PetscErrorCode MatSolveTranspose_Transpose_LU(Mat N, Vec b, Vec x) in MatSolveTranspose_Transpose_LU() argument 48 PetscCall(MatShellGetContext(N, &A)); in MatSolveTranspose_Transpose_LU() [all …]
|
| H A D | htransm.c | 218 static PetscErrorCode MatMult_HT(Mat N, Vec x, Vec y) in MatMult_HT() argument 223 PetscCall(MatShellGetContext(N, &A)); in MatMult_HT() 228 static PetscErrorCode MatMultHermitianTranspose_HT(Mat N, Vec x, Vec y) in MatMultHermitianTranspose_HT() argument 233 PetscCall(MatShellGetContext(N, &A)); in MatMultHermitianTranspose_HT() 238 static PetscErrorCode MatSolve_HT_LU(Mat N, Vec b, Vec x) in MatSolve_HT_LU() argument 244 PetscCall(MatShellGetContext(N, &A)); in MatSolve_HT_LU() 254 static PetscErrorCode MatSolveAdd_HT_LU(Mat N, Vec b, Vec y, Vec x) in MatSolveAdd_HT_LU() argument 260 PetscCall(MatShellGetContext(N, &A)); in MatSolveAdd_HT_LU() 274 static PetscErrorCode MatMatSolve_HT_LU(Mat N, Mat B, Mat X) in MatMatSolve_HT_LU() argument 279 PetscCall(MatShellGetContext(N, &A)); in MatMatSolve_HT_LU() [all …]
|
| /petsc/src/benchmarks/streams/ |
| H A D | CUDAVersion.cu | 29 #define N 10000000 macro 433 PetscCallCUDA(cudaMalloc((void **)&d_a, sizeof(float) * N)); in runStream() 434 PetscCallCUDA(cudaMalloc((void **)&d_b, sizeof(float) * N)); in runStream() 435 PetscCallCUDA(cudaMalloc((void **)&d_c, sizeof(float) * N)); in runStream() 440 dim3 dimGrid(N / dimBlock.x); /* (N/dimBlock.x,1,1) */ in runStream() 441 if (N % dimBlock.x != 0) dimGrid.x += 1; in runStream() 444 set_array<<<dimGrid, dimBlock>>>(d_a, 2.f, N); in runStream() 445 set_array<<<dimGrid, dimBlock>>>(d_b, .5f, N); in runStream() 446 set_array<<<dimGrid, dimBlock>>>(d_c, .5f, N); in runStream() 454 STREAM_Copy<<<dimGrid, dimBlock>>>(d_a, d_c, N); in runStream() [all …]
|
| H A D | SSEVersion.c | 50 #define N 4000000 macro 64 double a[N + OFFSET], b[N + OFFSET], c[N + OFFSET]; 73 …uble bytes[4] = {2 * sizeof(double) * N, 2 * sizeof(double) * N, 3 * sizeof(double) * N, 3 * siz… in main() 93 PetscPrintf(PETSC_COMM_WORLD, "Array size = %d, Offset = %d\n", N, OFFSET); in main() 94 …PetscPrintf(PETSC_COMM_WORLD, "Total memory required = %.1f MB per process.\n", (3 * N * BytesPerW… in main() 102 posix_memalign((void **)&a, 64, N * sizeof(double)); in main() 103 posix_memalign((void **)&b, 64, N * sizeof(double)); in main() 104 posix_memalign((void **)&c, 64, N * sizeof(double)); in main() 106 a = malloc(N * sizeof(double)); in main() 107 b = malloc(N * sizeof(double)); in main() [all …]
|
| /petsc/src/dm/dt/interface/ |
| H A D | dtaltv.c | 75 PetscErrorCode PetscDTAltVApply(PetscInt N, PetscInt k, const PetscReal *w, const PetscReal *v, Pet… in PetscDTAltVApply() argument 78 PetscCheck(N >= 0, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "invalid dimension"); in PetscDTAltVApply() 79 PetscCheck(k >= 0 && k <= N, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "invalid form degree"); in PetscDTAltVApply() 80 if (N <= 3) { in PetscDTAltVApply() 84 if (N == 1) { in PetscDTAltVApply() 86 } else if (N == 2) { in PetscDTAltVApply() 109 PetscCall(PetscDTBinomialInt(N, k, &Nk)); in PetscDTAltVApply() 114 PetscCall(PetscDTEnumSubset(N, k, i, subset)); in PetscDTAltVApply() 121 for (l = 0; l < k; l++) prod *= v[perm[l] * N + subset[l]]; in PetscDTAltVApply() 150 PetscErrorCode PetscDTAltVWedge(PetscInt N, PetscInt j, PetscInt k, const PetscReal *a, const Petsc… in PetscDTAltVWedge() argument [all …]
|
| /petsc/include/petsc/private/kernels/ |
| H A D | blocktranspose.h | 6 #define PetscKernel_A_gets_transpose_A_BODY(a, N) \ argument 8 for (i = 0; i < N; i++) { \ 9 for (j = i + 1; j < N; j++) { \ 10 MatScalar t = a[i * N + j]; \ 11 a[i * N + j] = a[j * N + i]; \ 12 a[j * N + i] = t; \ 17 static inline PetscErrorCode PetscKernel_A_gets_transpose_A_N(MatScalar *a, PetscInt N) in PetscKernel_A_gets_transpose_A_N() argument 19 PetscKernel_A_gets_transpose_A_BODY(a, N); in PetscKernel_A_gets_transpose_A_N() 21 #define PetscKernel_A_gets_transpose_A_DECLARE(N) \ argument 22 static inline PetscErrorCode PetscKernel_A_gets_transpose_A_##N(MatScalar *a) \ [all …]
|
| /petsc/src/dm/dt/tests/ |
| H A D | ex7.c | 6 static PetscErrorCode CheckPullback(PetscInt N, PetscInt M, const PetscReal *L, PetscInt k, const P… in CheckPullback() argument 17 PetscCall(PetscDTBinomialInt(N, k, &Nk)); in CheckPullback() 28 PetscCall(PetscDTAltVPullback(N, M, L, negative ? -k : k, w, Lstarw)); in CheckPullback() 29 PetscCall(PetscDTAltVPullbackMatrix(N, M, L, negative ? -k : k, Lstar)); in CheckPullback() 34 PetscCall(PetscDTAltVStar(N, N - k, 1, Lstarw, sLsw)); in CheckPullback() 35 PetscCall(PetscDTAltVApply(N, k, sLsw, x, &Lstarwx)); in CheckPullback() 38 PetscCall(PetscDTAltVApply(N, k, Lstarw, x, &Lstarwx)); in CheckPullback() 44 for (j = 0; j < N; j++) sum += L[i * N + j] * x[l * N + j]; in CheckPullback() 62 if (M * N > 0) PetscCall(PetscRealView(M * N, L, viewer)); in CheckPullback() 104 PetscInt k, N = n[i]; in main() local [all …]
|
| /petsc/src/mat/impls/submat/ |
| H A D | submat.c | 11 static PetscErrorCode MatScale_SubMatrix(Mat N, PetscScalar a) in MatScale_SubMatrix() argument 13 Mat_SubVirtual *Na = (Mat_SubVirtual *)N->data; in MatScale_SubMatrix() 20 static PetscErrorCode MatShift_SubMatrix(Mat N, PetscScalar a) in MatShift_SubMatrix() argument 22 Mat_SubVirtual *Na = (Mat_SubVirtual *)N->data; in MatShift_SubMatrix() 29 static PetscErrorCode MatDiagonalScale_SubMatrix(Mat N, Vec left, Vec right) in MatDiagonalScale_SubMatrix() argument 31 Mat_SubVirtual *Na = (Mat_SubVirtual *)N->data; in MatDiagonalScale_SubMatrix() 48 static PetscErrorCode MatGetDiagonal_SubMatrix(Mat N, Vec d) in MatGetDiagonal_SubMatrix() argument 50 Mat_SubVirtual *Na = (Mat_SubVirtual *)N->data; in MatGetDiagonal_SubMatrix() 59 static PetscErrorCode MatMult_SubMatrix(Mat N, Vec x, Vec y) in MatMult_SubMatrix() argument 61 Mat_SubVirtual *Na = (Mat_SubVirtual *)N->data; in MatMult_SubMatrix() [all …]
|
| /petsc/src/mat/impls/normal/ |
| H A D | normmh.c | 88 static PetscErrorCode MatMult_NormalHermitian(Mat N, Vec x, Vec y) in MatMult_NormalHermitian() argument 93 PetscCall(MatShellGetContext(N, &Na)); in MatMult_NormalHermitian() 99 static PetscErrorCode MatDestroy_NormalHermitian(Mat N) in MatDestroy_NormalHermitian() argument 104 PetscCall(MatShellGetContext(N, &Na)); in MatDestroy_NormalHermitian() 109 PetscCall(PetscObjectComposeFunction((PetscObject)N, "MatNormalHermitianGetMat_C", NULL)); in MatDestroy_NormalHermitian() 111 PetscCall(PetscObjectComposeFunction((PetscObject)N, "MatNormalGetMat_C", NULL)); in MatDestroy_NormalHermitian() 113 PetscCall(PetscObjectComposeFunction((PetscObject)N, "MatConvert_normalh_seqaij_C", NULL)); in MatDestroy_NormalHermitian() 114 PetscCall(PetscObjectComposeFunction((PetscObject)N, "MatConvert_normalh_mpiaij_C", NULL)); in MatDestroy_NormalHermitian() 116 PetscCall(PetscObjectComposeFunction((PetscObject)N, "MatConvert_normalh_hypre_C", NULL)); in MatDestroy_NormalHermitian() 118 PetscCall(PetscObjectComposeFunction((PetscObject)N, "MatShellSetContext_C", NULL)); in MatDestroy_NormalHermitian() [all …]
|
| H A D | normm.c | 105 static PetscErrorCode MatMult_Normal(Mat N, Vec x, Vec y) in MatMult_Normal() argument 110 PetscCall(MatShellGetContext(N, &Na)); in MatMult_Normal() 116 static PetscErrorCode MatDestroy_Normal(Mat N) in MatDestroy_Normal() argument 121 PetscCall(MatShellGetContext(N, &Na)); in MatDestroy_Normal() 126 PetscCall(PetscObjectComposeFunction((PetscObject)N, "MatNormalGetMat_C", NULL)); in MatDestroy_Normal() 127 PetscCall(PetscObjectComposeFunction((PetscObject)N, "MatConvert_normal_seqaij_C", NULL)); in MatDestroy_Normal() 128 PetscCall(PetscObjectComposeFunction((PetscObject)N, "MatConvert_normal_mpiaij_C", NULL)); in MatDestroy_Normal() 130 PetscCall(PetscObjectComposeFunction((PetscObject)N, "MatConvert_normal_hypre_C", NULL)); in MatDestroy_Normal() 132 …PetscCall(PetscObjectComposeFunction((PetscObject)N, "MatProductSetFromOptions_normal_seqdense_C",… in MatDestroy_Normal() 133 …PetscCall(PetscObjectComposeFunction((PetscObject)N, "MatProductSetFromOptions_normal_mpidense_C",… in MatDestroy_Normal() [all …]
|
| /petsc/lib/petsc/bin/maint/abi-compliance-checker/modules/Internals/ |
| H A D | Basic.pm | 326 my $N = $_[0]; 327 if(not $N) { 328 $N = 1; 331 $N = int($N)+1; 333 if($N>3) { 334 return $N."th"; 336 elsif($N==1) { 339 elsif($N==2) { 342 elsif($N==3) { 346 return $N; [all …]
|
| /petsc/src/vec/is/ao/impls/basic/ |
| H A D | aobasic.c | 28 …cCall(PetscViewerASCIIPrintf(viewer, "Number of elements in ordering %" PetscInt_FMT "\n", ao->N)); in AOView_Basic() 30 …for (i = 0; i < ao->N; i++) PetscCall(PetscViewerASCIIPrintf(viewer, "%3" PetscInt_FMT " %3" Pets… in AOView_Basic() 49 PetscInt i, N = ao->N; in AOPetscToApplication_Basic() local 54 if (ia[i] >= 0 && ia[i] < N) { in AOPetscToApplication_Basic() 65 PetscInt i, N = ao->N; in AOApplicationToPetsc_Basic() local 70 if (ia[i] >= 0 && ia[i] < N) { in AOApplicationToPetsc_Basic() 86 PetscCall(PetscMalloc1(ao->N * block, &temp)); in AOPetscToApplicationPermuteInt_Basic() 87 for (i = 0; i < ao->N; i++) { in AOPetscToApplicationPermuteInt_Basic() 90 PetscCall(PetscArraycpy(array, temp, ao->N * block)); in AOPetscToApplicationPermuteInt_Basic() 102 PetscCall(PetscMalloc1(ao->N * block, &temp)); in AOApplicationToPetscPermuteInt_Basic() [all …]
|
| /petsc/src/vec/is/ao/impls/mapping/ |
| H A D | aomapping.c | 10 PetscInt N; member 39 …ll(PetscViewerASCIIPrintf(viewer, "Number of elements in ordering %" PetscInt_FMT "\n", aomap->N)); in AOView_Mapping() 41 …for (i = 0; i < aomap->N; i++) PetscCall(PetscViewerASCIIPrintf(viewer, "%" PetscInt_FMT " %" Pe… in AOView_Mapping() 52 PetscInt N = aomap->N; in AOPetscToApplication_Mapping() local 68 high = N - 1; in AOPetscToApplication_Mapping() 87 PetscInt N = aomap->N; in AOApplicationToPetsc_Mapping() local 103 high = N - 1; in AOApplicationToPetsc_Mapping() 159 high = aomap->N - 1; in AOMappingHasApplicationIndex() 203 high = aomap->N - 1; in AOMappingHasPetscIndex() 246 PetscCount N; in AOCreateMapping() local [all …]
|
| /petsc/src/binding/petsc4py/demo/legacy/poisson3d/ |
| H A D | del2mat.py | 9 self.N = (n, n, n) 13 N = self.N 15 grid_eqs = N[0]*N[1]*N[2] 21 N, F = self.N, self.F 23 xx = x.getArray(readonly=1).reshape(N, order='f') 24 yy = y.getArray(readonly=0).reshape(N, order='f')
|
| /petsc/src/mat/impls/lrc/ |
| H A D | lrc.c | 13 static PetscErrorCode MatMult_LRC_kernel(Mat N, Vec x, Vec y, PetscBool transpose) in MatMult_LRC_kernel() argument 15 Mat_LRC *Na = (Mat_LRC *)N->data; in MatMult_LRC_kernel() 22 PetscCallMPI(MPI_Comm_size(PetscObjectComm((PetscObject)N), &size)); in MatMult_LRC_kernel() 59 …etscCallMPI(MPIU_Allreduce(w1, w2, nwork, MPIU_SCALAR, MPIU_SUM, PetscObjectComm((PetscObject)N))); in MatMult_LRC_kernel() 86 static PetscErrorCode MatMult_LRC(Mat N, Vec x, Vec y) in MatMult_LRC() argument 89 PetscCall(MatMult_LRC_kernel(N, x, y, PETSC_FALSE)); in MatMult_LRC() 93 static PetscErrorCode MatMultTranspose_LRC(Mat N, Vec x, Vec y) in MatMultTranspose_LRC() argument 96 PetscCall(MatMult_LRC_kernel(N, x, y, PETSC_TRUE)); in MatMultTranspose_LRC() 100 static PetscErrorCode MatDestroy_LRC(Mat N) in MatDestroy_LRC() argument 102 Mat_LRC *Na = (Mat_LRC *)N->data; in MatDestroy_LRC() [all …]
|
| /petsc/src/sys/tests/ |
| H A D | ex42.c | 10 PetscInt N = 0, i, j, n; in main() local 18 PetscCall(PetscOptionsGetInt(NULL, NULL, "-N", &N, NULL)); in main() 25 for (i = 0; i < N; ++i) { in main() 26 for (j = 0; j < N; ++j) { in main() 27 PetscInt key = i + j * N; in main() 36 for (i = 0; i < N; ++i) { in main() 37 for (j = 0; j < N; ++j) { in main() 38 PetscInt key = i + j * N; in main() 45 for (i = 0; i < N; ++i) { in main() 46 for (j = 0; j < N; ++j) { in main() [all …]
|
| /petsc/src/snes/tutorials/ |
| H A D | ex31.c | 32 PetscInt N; member 56 blasius->N = 10; /* Number of Chebyshev terms */ in main() 64 …PetscCall(PetscOptionsInt("-N", "Number of Chebyshev terms for f", "", blasius->N, &blasius->N, NU… in main() 68 PetscCall(PetscMalloc2(blasius->N - 3, &blasius->x, blasius->N - 3, &weight)); in main() 69 PetscCall(PetscDTGaussQuadrature(blasius->N - 3, -1., 1., blasius->x, weight)); in main() 83 PetscCall(VecSetSizes(x, PETSC_DECIDE, 2 * blasius->N - 1)); in main() 129 static void ChebyshevEval(PetscInt N, const PetscScalar *Tf, PetscReal x, PetscReal dx_deta, PetscS… in ChebyshevEval() argument 138 for (int i = 3; i < N; i++) { in ChebyshevEval() 165 PetscInt N = blasius->N; in FormFunction() local 177 Th = Tf + N; in FormFunction() [all …]
|