Searched refs:hostmem (Results 1 – 4 of 4) sorted by relevance
574 constexpr auto hostmem = PetscMemTypeHost(mtype); in GetArray() local584 … if (((oldmask == PETSC_OFFLOAD_GPU) && hostmem) || ((oldmask == PETSC_OFFLOAD_CPU) && !hostmem)) { in GetArray()588 PetscCall((hostmem ? CopyToHost_ : CopyToDevice_)(dctx, v, force)); in GetArray()591 PetscCall((hostmem ? HostAllocateCheck_ : DeviceAllocateCheck_)(dctx, v)); in GetArray()593 *a = hostmem ? VecIMPLCast(v)->array : VecCUPMCast(v)->array_d; in GetArray()598 if (hostmem) { in GetArray()613 mask = hostmem ? PETSC_OFFLOAD_CPU : PETSC_OFFLOAD_GPU; in GetArray()
1160 constexpr auto hostmem = PetscMemTypeHost(mtype); in GetArray() local1166 if (hostmem) { in GetArray()1184 m->offloadmask = hostmem ? PETSC_OFFLOAD_CPU : PETSC_OFFLOAD_GPU; in GetArray()
757 hostmem = (device_type == kDLCPU)770 if hostmem:777 if hostmem:784 if hostmem:
6000 hostmem = (device_type == kDLCPU)6013 if hostmem:6020 if hostmem:6027 if hostmem: