Searched refs:mcu (Results 1 – 1 of 1) sorted by relevance
204 const auto mcu = MatCUPMCast(m); in SetPreallocation_() local217 if (!mcu->d_user_alloc) PetscCallCUPM(cupmFreeAsync(mcu->d_v, stream)); in SetPreallocation_()219 mcu->d_user_alloc = PETSC_TRUE; in SetPreallocation_()220 mcu->d_v = user_device_array; in SetPreallocation_()224 mcu->d_user_alloc = PETSC_FALSE; in SetPreallocation_()226 PetscCall(PetscCUPMMallocAsync(&mcu->d_v, size, stream)); in SetPreallocation_()227 PetscCall(PetscCUPMMemsetAsync(mcu->d_v, 0, size, stream)); in SetPreallocation_()245 const auto mcu = MatCUPMCast(m); in HostToDevice_() local249 if (!mcu->d_v) PetscCall(SetPreallocation(m, dctx, nullptr)); in HostToDevice_()256 const auto dest = mcu->d_v; in HostToDevice_()[all …]