Searched refs:xmask (Results 1 – 3 of 3) sorted by relevance
| /petsc/src/vec/vec/impls/seq/cupm/ |
| H A D | vecseqcupm_impl.hpp | 1424 const auto xmask = xin->offloadmask; in CopyAsync() local
1437 mode = PetscOffloadDevice(xmask) ? cupmMemcpyDeviceToDevice : cupmMemcpyHostToHost; in CopyAsync()
1449 mode = PetscOffloadHost(xmask) ? cupmMemcpyHostToDevice : cupmMemcpyDeviceToDevice; in CopyAsync()
1451 mode = PetscOffloadHost(xmask) ? cupmMemcpyHostToHost : cupmMemcpyDeviceToHost; in CopyAsync()
1457 mode = PetscOffloadDevice(xmask) ? cupmMemcpyDeviceToDevice : cupmMemcpyHostToDevice; in CopyAsync()
|
| /petsc/src/mat/impls/baij/mpi/ |
| H A D | mpibaij.c | 1664 Vec xmask, lmask; in MatZeroRowsColumns_MPIBAIJ() local
1696 PetscCall(MatCreateVecs(A, &xmask, NULL)); in MatZeroRowsColumns_MPIBAIJ()
1698 PetscCall(VecGetArray(xmask, &bb)); in MatZeroRowsColumns_MPIBAIJ()
1700 PetscCall(VecRestoreArray(xmask, &bb)); in MatZeroRowsColumns_MPIBAIJ()
1701 PetscCall(VecScatterBegin(l->Mvctx, xmask, lmask, ADD_VALUES, SCATTER_FORWARD)); in MatZeroRowsColumns_MPIBAIJ()
1702 PetscCall(VecScatterEnd(l->Mvctx, xmask, lmask, ADD_VALUES, SCATTER_FORWARD)); in MatZeroRowsColumns_MPIBAIJ()
1703 PetscCall(VecDestroy(&xmask)); in MatZeroRowsColumns_MPIBAIJ()
|
| /petsc/src/mat/impls/aij/mpi/ |
| H A D | mpiaij.c | 944 Vec xmask, lmask; in MatZeroRowsColumns_MPIAIJ() local
975 PetscCall(MatCreateVecs(A, &xmask, NULL)); in MatZeroRowsColumns_MPIAIJ()
977 PetscCall(VecGetArray(xmask, &bb)); in MatZeroRowsColumns_MPIAIJ()
979 PetscCall(VecRestoreArray(xmask, &bb)); in MatZeroRowsColumns_MPIAIJ()
980 PetscCall(VecScatterBegin(l->Mvctx, xmask, lmask, ADD_VALUES, SCATTER_FORWARD)); in MatZeroRowsColumns_MPIAIJ()
981 PetscCall(VecScatterEnd(l->Mvctx, xmask, lmask, ADD_VALUES, SCATTER_FORWARD)); in MatZeroRowsColumns_MPIAIJ()
982 PetscCall(VecDestroy(&xmask)); in MatZeroRowsColumns_MPIAIJ()
|