1790 	struct kfd_criu_device_bucket *device_buckets = NULL;  in criu_checkpoint_devices()  local
1793 	device_buckets = kvzalloc(num_devices * sizeof(*device_buckets), GFP_KERNEL);  in criu_checkpoint_devices()
1794 	if (!device_buckets) {  in criu_checkpoint_devices()
1808 		device_buckets[i].user_gpu_id = pdd->user_gpu_id;  in criu_checkpoint_devices()
1809 		device_buckets[i].actual_gpu_id = pdd->dev->id;  in criu_checkpoint_devices()
1817 	ret = copy_to_user(user_addr, device_buckets, num_devices * sizeof(*device_buckets));  in criu_checkpoint_devices()
1834 	kvfree(device_buckets);  in criu_checkpoint_devices()
2188 	struct kfd_criu_device_bucket *device_buckets;  in criu_restore_devices()  local
2199 	device_buckets = kmalloc_array(args->num_devices, sizeof(*device_buckets), GFP_KERNEL);  in criu_restore_devices()
2200 	if (!device_buckets)  in criu_restore_devices()
2203 	ret = copy_from_user(device_buckets, (void __user *)args->devices,  in criu_restore_devices()
2204 				args->num_devices * sizeof(*device_buckets));  in criu_restore_devices()
2218 		if (!device_buckets[i].user_gpu_id) {  in criu_restore_devices()
2224 		dev = kfd_device_by_id(device_buckets[i].actual_gpu_id);  in criu_restore_devices()
2227 				device_buckets[i].actual_gpu_id);  in criu_restore_devices()
2235 					device_buckets[i].actual_gpu_id);  in criu_restore_devices()
2239 		pdd->user_gpu_id = device_buckets[i].user_gpu_id;  in criu_restore_devices()
2241 		drm_file = fget(device_buckets[i].drm_fd);  in criu_restore_devices()
2244 				device_buckets[i].drm_fd);  in criu_restore_devices()
2287 	kfree(device_buckets);  in criu_restore_devices()