| /petsc/src/vec/vec/tests/ |
| H A D | ex61.c | 21 } coo[3] = { in main() local
41 for (k = 0; k < coo[rank].n; k++) { in main()
42 PetscScalar val = (PetscScalar)coo[rank].i[k]; in main()
43 PetscCall(VecSetValues(x, 1, &coo[rank].i[k], &val, ADD_VALUES)); in main()
53 PetscCall(VecSetPreallocationCOO(y, coo[rank].n, coo[rank].i)); in main()
55 PetscCall(PetscMalloc1(coo[rank].n, &vals)); in main()
56 for (k = 0; k < coo[rank].n; k++) vals[k] = (PetscScalar)coo[rank].i[k]; in main()
|
| /petsc/src/mat/impls/aij/mpi/mpihipsparse/ |
| H A D | mpiaijhipsparse.hip.cxx | 23 MatCOOStruct_MPIAIJ *coo = *(MatCOOStruct_MPIAIJ **)data; in MatCOOStructDestroy_MPIAIJHIPSPARSE() local
26 PetscCall(PetscSFDestroy(&coo->sf)); in MatCOOStructDestroy_MPIAIJHIPSPARSE()
27 PetscCallHIP(hipFree(coo->Ajmap1)); in MatCOOStructDestroy_MPIAIJHIPSPARSE()
28 PetscCallHIP(hipFree(coo->Aperm1)); in MatCOOStructDestroy_MPIAIJHIPSPARSE()
29 PetscCallHIP(hipFree(coo->Bjmap1)); in MatCOOStructDestroy_MPIAIJHIPSPARSE()
30 PetscCallHIP(hipFree(coo->Bperm1)); in MatCOOStructDestroy_MPIAIJHIPSPARSE()
31 PetscCallHIP(hipFree(coo->Aimap2)); in MatCOOStructDestroy_MPIAIJHIPSPARSE()
32 PetscCallHIP(hipFree(coo->Ajmap2)); in MatCOOStructDestroy_MPIAIJHIPSPARSE()
33 PetscCallHIP(hipFree(coo->Aperm2)); in MatCOOStructDestroy_MPIAIJHIPSPARSE()
34 PetscCallHIP(hipFree(coo->Bimap2)); in MatCOOStructDestroy_MPIAIJHIPSPARSE()
[all …]
|
| /petsc/src/mat/impls/aij/mpi/mpicusparse/ |
| H A D | mpiaijcusparse.cu | 23 MatCOOStruct_MPIAIJ *coo = *(MatCOOStruct_MPIAIJ **)data; in MatCOOStructDestroy_MPIAIJCUSPARSE() local
26 PetscCall(PetscSFDestroy(&coo->sf)); in MatCOOStructDestroy_MPIAIJCUSPARSE()
27 PetscCallCUDA(cudaFree(coo->Ajmap1)); in MatCOOStructDestroy_MPIAIJCUSPARSE()
28 PetscCallCUDA(cudaFree(coo->Aperm1)); in MatCOOStructDestroy_MPIAIJCUSPARSE()
29 PetscCallCUDA(cudaFree(coo->Bjmap1)); in MatCOOStructDestroy_MPIAIJCUSPARSE()
30 PetscCallCUDA(cudaFree(coo->Bperm1)); in MatCOOStructDestroy_MPIAIJCUSPARSE()
31 PetscCallCUDA(cudaFree(coo->Aimap2)); in MatCOOStructDestroy_MPIAIJCUSPARSE()
32 PetscCallCUDA(cudaFree(coo->Ajmap2)); in MatCOOStructDestroy_MPIAIJCUSPARSE()
33 PetscCallCUDA(cudaFree(coo->Aperm2)); in MatCOOStructDestroy_MPIAIJCUSPARSE()
34 PetscCallCUDA(cudaFree(coo->Bimap2)); in MatCOOStructDestroy_MPIAIJCUSPARSE()
[all …]
|
| /petsc/src/mat/tutorials/ |
| H A D | ex18.c | 43 PetscCall(PetscFree(fe->coo)); in DestroyFEStruct()
72 PetscCall(PetscMalloc1(fe->Ne, &fe->coo)); in CreateMatrix()
73 fe->coo[0] = 0; in CreateMatrix()
74 for (PetscInt e = 1; e < fe->Ne; e++) fe->coo[e] = fe->coo[e - 1] + 3 * 3; in CreateMatrix()
107 s = v + fe->coo[e]; /* point to location in COO of current element stiffness */ in FillMatrixCPUCOO()
|
| H A D | ex18.h | 10 …PetscInt *coo; /* offset into the matrices COO array for the start of each element stiffness … member
|
| /petsc/src/mat/impls/aij/mpi/kokkos/ |
| H A D | mpiaijkok.kokkos.cxx | 1585 MatCOOStruct_MPIAIJKokkos *coo; in MatSetValuesCOO_MPIAIJKokkos() local
1590 PetscCall(PetscContainerGetPointer(container, &coo)); in MatSetValuesCOO_MPIAIJKokkos()
1592 const auto &n = coo->n; in MatSetValuesCOO_MPIAIJKokkos()
1593 const auto &Annz = coo->Annz; in MatSetValuesCOO_MPIAIJKokkos()
1594 const auto &Annz2 = coo->Annz2; in MatSetValuesCOO_MPIAIJKokkos()
1595 const auto &Bnnz = coo->Bnnz; in MatSetValuesCOO_MPIAIJKokkos()
1596 const auto &Bnnz2 = coo->Bnnz2; in MatSetValuesCOO_MPIAIJKokkos()
1597 const auto &vsend = coo->sendbuf; in MatSetValuesCOO_MPIAIJKokkos()
1598 const auto &v2 = coo->recvbuf; in MatSetValuesCOO_MPIAIJKokkos()
1599 const auto &Ajmap1 = coo->Ajmap1; in MatSetValuesCOO_MPIAIJKokkos()
[all …]
|
| /petsc/src/mat/impls/aij/mpi/ |
| H A D | mpiaij.c | 6319 MatCOOStruct_MPIAIJ *coo = *(MatCOOStruct_MPIAIJ **)data; in MatCOOStructDestroy_MPIAIJ() local
6322 PetscCall(PetscSFDestroy(&coo->sf)); in MatCOOStructDestroy_MPIAIJ()
6323 PetscCall(PetscFree(coo->Aperm1)); in MatCOOStructDestroy_MPIAIJ()
6324 PetscCall(PetscFree(coo->Bperm1)); in MatCOOStructDestroy_MPIAIJ()
6325 PetscCall(PetscFree(coo->Ajmap1)); in MatCOOStructDestroy_MPIAIJ()
6326 PetscCall(PetscFree(coo->Bjmap1)); in MatCOOStructDestroy_MPIAIJ()
6327 PetscCall(PetscFree(coo->Aimap2)); in MatCOOStructDestroy_MPIAIJ()
6328 PetscCall(PetscFree(coo->Bimap2)); in MatCOOStructDestroy_MPIAIJ()
6329 PetscCall(PetscFree(coo->Aperm2)); in MatCOOStructDestroy_MPIAIJ()
6330 PetscCall(PetscFree(coo->Bperm2)); in MatCOOStructDestroy_MPIAIJ()
[all …]
|
| /petsc/src/mat/impls/aij/seq/ |
| H A D | aij.c | 4544 MatCOOStruct_SeqAIJ *coo = *(MatCOOStruct_SeqAIJ **)data; in MatCOOStructDestroy_SeqAIJ() local
4547 PetscCall(PetscFree(coo->perm)); in MatCOOStructDestroy_SeqAIJ()
4548 PetscCall(PetscFree(coo->jmap)); in MatCOOStructDestroy_SeqAIJ()
4549 PetscCall(PetscFree(coo)); in MatCOOStructDestroy_SeqAIJ()
4565 MatCOOStruct_SeqAIJ *coo; in MatSetPreallocationCOO_SeqAIJ() local
4729 PetscCall(PetscMalloc1(1, &coo)); in MatSetPreallocationCOO_SeqAIJ()
4730 PetscCall(PetscIntCast(nnz, &coo->nz)); in MatSetPreallocationCOO_SeqAIJ()
4731 coo->n = coo_n; in MatSetPreallocationCOO_SeqAIJ()
4732 coo->Atot = coo_n - nneg; // Annz is seqaij->nz, so no need to record that again in MatSetPreallocationCOO_SeqAIJ()
4733 coo->jmap = jmap; // of length nnz+1 in MatSetPreallocationCOO_SeqAIJ()
[all …]
|
| /petsc/src/mat/impls/aij/seq/kokkos/ |
| H A D | aijkok.kokkos.cxx | 1399 MatCOOStruct_SeqAIJKokkos *coo; in MatSetValuesCOO_SeqAIJKokkos() local
1403 PetscCall(PetscContainerGetPointer(container, &coo)); in MatSetValuesCOO_SeqAIJKokkos()
1405 const auto &n = coo->n; in MatSetValuesCOO_SeqAIJKokkos()
1406 const auto &Annz = coo->nz; in MatSetValuesCOO_SeqAIJKokkos()
1407 const auto &jmap = coo->jmap; in MatSetValuesCOO_SeqAIJKokkos()
1408 const auto &perm = coo->perm; in MatSetValuesCOO_SeqAIJKokkos()
|
| /petsc/src/mat/impls/aij/seq/seqhipsparse/ |
| H A D | aijhipsparse.hip.cxx | 3721 MatCOOStruct_SeqAIJ *coo = *(MatCOOStruct_SeqAIJ **)data; in MatCOOStructDestroy_SeqAIJHIPSPARSE() local
3724 PetscCallHIP(hipFree(coo->perm)); in MatCOOStructDestroy_SeqAIJHIPSPARSE()
3725 PetscCallHIP(hipFree(coo->jmap)); in MatCOOStructDestroy_SeqAIJHIPSPARSE()
3726 PetscCall(PetscFree(coo)); in MatCOOStructDestroy_SeqAIJHIPSPARSE()
3790 MatCOOStruct_SeqAIJ *coo; in MatSetValuesCOO_SeqAIJHIPSPARSE() local
3796 PetscCall(PetscContainerGetPointer(container, &coo)); in MatSetValuesCOO_SeqAIJHIPSPARSE()
3800 PetscCallHIP(hipMalloc((void **)&v1, coo->n * sizeof(PetscScalar))); in MatSetValuesCOO_SeqAIJHIPSPARSE()
3801 PetscCallHIP(hipMemcpy((void *)v1, v, coo->n * sizeof(PetscScalar), hipMemcpyHostToDevice)); in MatSetValuesCOO_SeqAIJHIPSPARSE()
3809 …((Annz + 255) / 256), dim3(256), 0, PetscDefaultHipStream, v1, Annz, coo->jmap, coo->perm, imode, … in MatSetValuesCOO_SeqAIJHIPSPARSE()
|
| /petsc/src/mat/impls/aij/seq/seqcusparse/ |
| H A D | aijcusparse.cu | 4363 MatCOOStruct_SeqAIJ *coo = *(MatCOOStruct_SeqAIJ **)ctx; in MatCOOStructDestroy_SeqAIJCUSPARSE() local
4366 PetscCallCUDA(cudaFree(coo->perm)); in MatCOOStructDestroy_SeqAIJCUSPARSE()
4367 PetscCallCUDA(cudaFree(coo->jmap)); in MatCOOStructDestroy_SeqAIJCUSPARSE()
4368 PetscCall(PetscFree(coo)); in MatCOOStructDestroy_SeqAIJCUSPARSE()
4433 MatCOOStruct_SeqAIJ *coo; in MatSetValuesCOO_SeqAIJCUSPARSE() local
4439 PetscCall(PetscContainerGetPointer(container, &coo)); in MatSetValuesCOO_SeqAIJCUSPARSE()
4443 PetscCallCUDA(cudaMalloc((void **)&v1, coo->n * sizeof(PetscScalar))); in MatSetValuesCOO_SeqAIJCUSPARSE()
4444 PetscCallCUDA(cudaMemcpy((void *)v1, v, coo->n * sizeof(PetscScalar), cudaMemcpyHostToDevice)); in MatSetValuesCOO_SeqAIJCUSPARSE()
4452 MatAddCOOValues<<<((int)(Annz + 255) / 256), 256>>>(v1, Annz, coo->jmap, coo->perm, imode, Aa); in MatSetValuesCOO_SeqAIJCUSPARSE()
|