1117 		pdd->drm_priv, (struct kgd_mem **) &mem, &offset,  in kfd_ioctl_alloc_memory_of_gpu()
1148 	amdgpu_amdkfd_gpuvm_free_memory_of_gpu(dev->adev, (struct kgd_mem *)mem,  in kfd_ioctl_alloc_memory_of_gpu()
1192 				(struct kgd_mem *)mem, pdd->drm_priv, &size);  in kfd_ioctl_free_memory_of_gpu()
1279 			peer_pdd->dev->adev, (struct kgd_mem *)mem,  in kfd_ioctl_map_memory_to_gpu()
1290 			       ((struct kgd_mem *)mem)->domain);  in kfd_ioctl_map_memory_to_gpu()
1298 	err = amdgpu_amdkfd_gpuvm_sync_memory(dev->adev, (struct kgd_mem *) mem, true);  in kfd_ioctl_map_memory_to_gpu()
1379 			peer_pdd->dev->adev, (struct kgd_mem *)mem, peer_pdd->drm_priv);  in kfd_ioctl_unmap_memory_from_gpu()
1391 				(struct kgd_mem *) mem, true);  in kfd_ioctl_unmap_memory_from_gpu()
1543 					      (struct kgd_mem **)&mem, &size,  in kfd_ioctl_import_dmabuf()
1562 	amdgpu_amdkfd_gpuvm_free_memory_of_gpu(pdd->dev->adev, (struct kgd_mem *)mem,  in kfd_ioctl_import_dmabuf()
1748 			struct kgd_mem *kgd_mem = (struct kgd_mem *)mem;  in get_process_num_bos()  local
1750 			if ((uint64_t)kgd_mem->va > pdd->gpuvm_base)  in get_process_num_bos()
1808 		struct kgd_mem *kgd_mem;  in criu_checkpoint_bos()  local
1820 			kgd_mem = (struct kgd_mem *)mem;  in criu_checkpoint_bos()
1821 			dumper_bo = kgd_mem->bo;  in criu_checkpoint_bos()
1823 			if ((uint64_t)kgd_mem->va <= pdd->gpuvm_base)  in criu_checkpoint_bos()
1830 			bo_bucket->addr = (uint64_t)kgd_mem->va;  in criu_checkpoint_bos()
1832 			bo_bucket->alloc_flags = (uint32_t)kgd_mem->alloc_flags;  in criu_checkpoint_bos()
1866 				if (amdgpu_amdkfd_bo_mapped_to_dev(p->pdds[i]->dev->adev, kgd_mem))  in criu_checkpoint_bos()
2188 				      struct kgd_mem **kgd_mem)  in criu_restore_memory_of_gpu()  argument
2218 						      bo_bucket->size, pdd->drm_priv, kgd_mem,  in criu_restore_memory_of_gpu()
2229 	idr_handle = idr_alloc(&pdd->alloc_idr, *kgd_mem, bo_priv->idr_handle,  in criu_restore_memory_of_gpu()
2234 		amdgpu_amdkfd_gpuvm_free_memory_of_gpu(pdd->dev->adev, *kgd_mem, pdd->drm_priv,  in criu_restore_memory_of_gpu()
2258 	struct kgd_mem *kgd_mem;  in criu_restore_bo()  local
2272 	ret = criu_restore_memory_of_gpu(pdd, bo_bucket, bo_priv, &kgd_mem);  in criu_restore_bo()
2294 		ret = amdgpu_amdkfd_gpuvm_map_memory_to_gpu(peer->adev, kgd_mem,  in criu_restore_bo()
2306 		ret = criu_get_prime_handle(&kgd_mem->bo->tbo.base, DRM_RDWR,  in criu_restore_bo()