Extend checkpoint logic to allow inclusion of VRAM BOs that do not have a VA attached
Signed-off-by: Ramesh Errabolu <[email protected]> --- drivers/gpu/drm/amd/amdkfd/kfd_chardev.c | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/drivers/gpu/drm/amd/amdkfd/kfd_chardev.c b/drivers/gpu/drm/amd/amdkfd/kfd_chardev.c index 40ac093b5035..44c647c82070 100644 --- a/drivers/gpu/drm/amd/amdkfd/kfd_chardev.c +++ b/drivers/gpu/drm/amd/amdkfd/kfd_chardev.c @@ -1845,7 +1845,8 @@ static uint32_t get_process_num_bos(struct kfd_process *p) idr_for_each_entry(&pdd->alloc_idr, mem, id) { struct kgd_mem *kgd_mem = (struct kgd_mem *)mem; - if ((uint64_t)kgd_mem->va > pdd->gpuvm_base) + if (((uint64_t)kgd_mem->va > pdd->gpuvm_base) || + !kgd_mem->va) num_of_bos++; } } @@ -1917,7 +1918,11 @@ static int criu_checkpoint_bos(struct kfd_process *p, kgd_mem = (struct kgd_mem *)mem; dumper_bo = kgd_mem->bo; - if ((uint64_t)kgd_mem->va <= pdd->gpuvm_base) + /* Skip checkpointing BOs that are used for Trap handler + * code and state. Currently, these BOs have a VA that + * is less GPUVM Base + */ + if (((uint64_t)kgd_mem->va <= pdd->gpuvm_base) && kgd_mem->va) continue; bo_bucket = &bo_buckets[bo_index]; -- 2.25.1
