| /petsc/src/mat/tests/ |
| H A D | ex142.c | 29 PetscScalar *x_array, *y_array, *z_array; in main() local
72 PetscCall(VecGetArray(y, &y_array)); in main()
82 fplan = fftw_plan_dft_r2c_1d(dim[0], (double *)x_array, (fftw_complex *)y_array, flags); in main()
83 bplan = fftw_plan_dft_c2r_1d(dim[0], (fftw_complex *)y_array, (double *)z_array, flags); in main()
86 … fplan = fftw_plan_dft_r2c_2d(dim[0], dim[1], (double *)x_array, (fftw_complex *)y_array, flags); in main()
87 … bplan = fftw_plan_dft_c2r_2d(dim[0], dim[1], (fftw_complex *)y_array, (double *)z_array, flags); in main()
90 … = fftw_plan_dft_r2c_3d(dim[0], dim[1], dim[2], (double *)x_array, (fftw_complex *)y_array, flags); in main()
91 …bplan = fftw_plan_dft_c2r_3d(dim[0], dim[1], dim[2], (fftw_complex *)y_array, (double *)z_array, f… in main()
94 fplan = fftw_plan_dft_r2c(DIM, (int *)dim, (double *)x_array, (fftw_complex *)y_array, flags); in main()
95 bplan = fftw_plan_dft_c2r(DIM, (int *)dim, (fftw_complex *)y_array, (double *)z_array, flags); in main()
[all …]
|
| /petsc/src/mat/impls/cufft/ |
| H A D | cufft.cu | 21 PetscScalar *x_array, *y_array; in MatMult_SeqCUFFT() local
25 PetscCall(VecGetArray(y, &y_array)); in MatMult_SeqCUFFT()
47 …PetscCallCUDA(cudaMemcpy(y_array, devArray, sizeof(cufftComplex) * dim[ndim], cudaMemcpyDeviceToHo… in MatMult_SeqCUFFT()
48 PetscCall(VecRestoreArray(y, &y_array)); in MatMult_SeqCUFFT()
58 PetscScalar *x_array, *y_array; in MatMultTranspose_SeqCUFFT() local
62 PetscCall(VecGetArray(y, &y_array)); in MatMultTranspose_SeqCUFFT()
84 …PetscCallCUDA(cudaMemcpy(y_array, devArray, sizeof(cufftComplex) * dim[ndim], cudaMemcpyDeviceToHo… in MatMultTranspose_SeqCUFFT()
85 PetscCall(VecRestoreArray(y, &y_array)); in MatMultTranspose_SeqCUFFT()
|
| /petsc/src/mat/impls/fft/fftw/ |
| H A D | fftw.c | 36 PetscScalar *y_array; in MatMult_SeqFFTW() local
56 PetscCall(VecGetArray(y, &y_array)); in MatMult_SeqFFTW()
61 …fftw->p_forward = fftw_plan_dft_1d(dim[0], (fftw_complex *)x_array, (fftw_complex *)y_array, FFTW_… in MatMult_SeqFFTW()
63 …fftw->p_forward = fftw_plan_dft_r2c_1d(dim[0], (double *)x_array, (fftw_complex *)y_array, fftw->p… in MatMult_SeqFFTW()
68 …ftw_plan_dft_2d(dim[0], dim[1], (fftw_complex *)x_array, (fftw_complex *)y_array, FFTW_FORWARD, ff… in MatMult_SeqFFTW()
70 …d = fftw_plan_dft_r2c_2d(dim[0], dim[1], (double *)x_array, (fftw_complex *)y_array, fftw->p_flag); in MatMult_SeqFFTW()
75 …_dft_3d(dim[0], dim[1], dim[2], (fftw_complex *)x_array, (fftw_complex *)y_array, FFTW_FORWARD, ff… in MatMult_SeqFFTW()
77 …_plan_dft_r2c_3d(dim[0], dim[1], dim[2], (double *)x_array, (fftw_complex *)y_array, fftw->p_flag); in MatMult_SeqFFTW()
92 …fftw_iodim64 *)iodims, 0, NULL, (fftw_complex *)x_array, (fftw_complex *)y_array, FFTW_FORWARD, ff… in MatMult_SeqFFTW()
102 … (fftw_iodim *)iodims, 0, NULL, (fftw_complex *)x_array, (fftw_complex *)y_array, FFTW_FORWARD, ff… in MatMult_SeqFFTW()
[all …]
|
| /petsc/src/binding/petsc4py/demo/regressor/ |
| H A D | test_regressor.py | 20 y_array = [0.2, -0.3, -0.8, -0.3, 1.2] variable
36 y.setValues(rows_ix,y_array,addv=False)
|
| /petsc/src/ml/regressor/tests/ |
| H A D | ex_sharks.c | 22 …PetscScalar y_array[20] = {98, 53, 39, 127, 73, 42, 71, 61, 83, 74, 85, 82, 62, 60, 43, 69, 67, 69… in main() local
61 PetscCall(VecSetValues(y, 20, rows_ix, y_array, INSERT_VALUES)); in main()
|
| H A D | ex2.c | 21 PetscScalar y_array[5] = {1.0, 0.5, 0, 0.5, 2}; in main() local
39 PetscCall(VecSetValues(y, 5, rows_ix, y_array, INSERT_VALUES)); in main()
|
| H A D | ex1.c | 21 PetscScalar y_array[5] = {0.20000, -0.30000, -0.80000, -0.30000, 1.20000}; in main() local
39 PetscCall(VecSetValues(y, 5, rows_ix, y_array, INSERT_VALUES)); in main()
|
| /petsc/src/ksp/pc/impls/bjacobi/ |
| H A D | bjacobi.c | 647 PetscScalar *y_array; in PCApplySymmetricLeft_BJacobi_Singleblock() local
659 PetscCall(VecGetArray(y, &y_array)); in PCApplySymmetricLeft_BJacobi_Singleblock()
661 PetscCall(VecPlaceArray(bjac->y, y_array)); in PCApplySymmetricLeft_BJacobi_Singleblock()
669 PetscCall(VecRestoreArray(y, &y_array)); in PCApplySymmetricLeft_BJacobi_Singleblock()
677 PetscScalar *y_array; in PCApplySymmetricRight_BJacobi_Singleblock() local
689 PetscCall(VecGetArray(y, &y_array)); in PCApplySymmetricRight_BJacobi_Singleblock()
691 PetscCall(VecPlaceArray(bjac->y, y_array)); in PCApplySymmetricRight_BJacobi_Singleblock()
702 PetscCall(VecRestoreArray(y, &y_array)); in PCApplySymmetricRight_BJacobi_Singleblock()
710 PetscScalar *y_array; in PCApplyTranspose_BJacobi_Singleblock() local
721 PetscCall(VecGetArray(y, &y_array)); in PCApplyTranspose_BJacobi_Singleblock()
[all …]
|
| /petsc/src/mat/impls/dense/seq/cupm/ |
| H A D | matseqdensecupm.hpp | 648 PetscScalar *y_array = nullptr; in MatSolve_Factored_Dispatch_() local
679 PetscCall(VecCUPMGetArrayWriteAsync<T>(use_y_array_directly ? y : workvec, &y_array, dctx)); in MatSolve_Factored_Dispatch_()
680 PetscCall(PetscCUPMMemcpyAsync(y_array, x_array, m, copy_mode, stream)); in MatSolve_Factored_Dispatch_()
689 …PetscCall(Solver{}.template Solve<transpose>(A, cupmScalarPtrCast(y_array), m, m, 1, k, dctx, stre… in MatSolve_Factored_Dispatch_()
693 PetscCall(VecCUPMRestoreArrayWriteAsync<T>(y, &y_array, dctx)); in MatSolve_Factored_Dispatch_()
706 PetscCall(PetscCUPMMemcpyAsync(yv, y_array, k, copy_mode, stream)); in MatSolve_Factored_Dispatch_()
712 PetscCall(VecCUPMRestoreArrayWriteAsync<T>(workvec, &y_array)); in MatSolve_Factored_Dispatch_()
|