Searched refs:PetscOffloadDevice (Results 1 – 3 of 3) sorted by relevance
149 #define PetscOffloadDevice(m) (((m) & PETSC_OFFLOAD_GPU) == PETSC_OFFLOAD_GPU) macro163 static_assert(!PetscOffloadDevice(PETSC_OFFLOAD_UNALLOCATED), "");164 static_assert(!PetscOffloadDevice(PETSC_OFFLOAD_CPU), "");165 static_assert(PetscOffloadDevice(PETSC_OFFLOAD_GPU), "");166 static_assert(PetscOffloadDevice(PETSC_OFFLOAD_BOTH), "");167 static_assert(!PetscOffloadDevice(PETSC_OFFLOAD_KOKKOS), "");390 if (PetscOffloadDevice(dest)) { in PetscOffloadMaskToDeviceCopyMode()
664 const auto xisdevice = xiscupm && PetscOffloadDevice(x->offloadmask); in MatSolve_Factored_Dispatch_()752 const auto bisdevice = biscupm && PetscOffloadDevice(B->offloadmask); in MatMatSolve_Factored_Dispatch_()1676 if (PetscOffloadDevice(offloadmask)) { in GetColumnVector()1688 } else if (PetscOffloadDevice(offloadmask)) { in GetColumnVector()
1437 mode = PetscOffloadDevice(xmask) ? cupmMemcpyDeviceToDevice : cupmMemcpyHostToHost; in CopyAsync()1457 mode = PetscOffloadDevice(xmask) ? cupmMemcpyDeviceToDevice : cupmMemcpyHostToDevice; in CopyAsync()