Lines Matching refs:offloadmask

229   m->offloadmask = PETSC_OFFLOAD_GPU;  in SetPreallocation_()
238 …const auto copy = m->offloadmask == PETSC_OFFLOAD_CPU || m->offloadmask == PETSC_OFFLOAD_UNALLOCA… in HostToDevice_()
266 m->offloadmask = PETSC_OFFLOAD_BOTH; in HostToDevice_()
276 const auto copy = m->offloadmask == PETSC_OFFLOAD_GPU; in DeviceToHost_()
302 m->offloadmask = PETSC_OFFLOAD_BOTH; in DeviceToHost_()
664 const auto xisdevice = xiscupm && PetscOffloadDevice(x->offloadmask); in MatSolve_Factored_Dispatch_()
752 const auto bisdevice = biscupm && PetscOffloadDevice(B->offloadmask); in MatMatSolve_Factored_Dispatch_()
936 B->offloadmask = PETSC_OFFLOAD_CPU; in Convert_Dispatch_()
942 B->offloadmask = PETSC_OFFLOAD_UNALLOCATED; // REVIEW ME: why not offload host?? in Convert_Dispatch_()
1004 if (!MatIMPLCast(A)->user_alloc) A->offloadmask = PETSC_OFFLOAD_CPU; in Destroy()
1184 m->offloadmask = hostmem ? PETSC_OFFLOAD_CPU : PETSC_OFFLOAD_GPU; in GetArray()
1198 m->offloadmask = PetscMemTypeHost(mtype) ? PETSC_OFFLOAD_CPU : PETSC_OFFLOAD_GPU; in RestoreArray()
1607 …const auto hopt = (opt == MAT_COPY_VALUES && A->offloadmask != PETSC_OFFLOAD_CPU) ? MAT_DO… in Duplicate()
1662 const auto offloadmask = A->offloadmask; in GetColumnVector() local
1676 if (PetscOffloadDevice(offloadmask)) { in GetColumnVector()
1686 if (PetscOffloadUnallocated(offloadmask) || PetscOffloadHost(offloadmask)) { in GetColumnVector()
1688 } else if (PetscOffloadDevice(offloadmask)) { in GetColumnVector()
1863 cmat->offloadmask = PETSC_OFFLOAD_GPU; in GetSubMatrix()
1867 cmat->offloadmask = A->offloadmask; in GetSubMatrix()
1887 was_offload_host = cmat->offloadmask == PETSC_OFFLOAD_CPU; in RestoreSubMatrix()
1903 A->offloadmask = was_offload_host ? PETSC_OFFLOAD_CPU : PETSC_OFFLOAD_GPU; in RestoreSubMatrix()
1906 cmat->offloadmask = PETSC_OFFLOAD_UNALLOCATED; in RestoreSubMatrix()