1777 	struct kfd_criu_device_bucket *device_buckets = NULL;  in criu_checkpoint_devices()  local
1780 	device_buckets = kvzalloc(num_devices * sizeof(*device_buckets), GFP_KERNEL);  in criu_checkpoint_devices()
1781 	if (!device_buckets) {  in criu_checkpoint_devices()
1795 		device_buckets[i].user_gpu_id = pdd->user_gpu_id;  in criu_checkpoint_devices()
1796 		device_buckets[i].actual_gpu_id = pdd->dev->id;  in criu_checkpoint_devices()
1804 	ret = copy_to_user(user_addr, device_buckets, num_devices * sizeof(*device_buckets));  in criu_checkpoint_devices()
1821 	kvfree(device_buckets);  in criu_checkpoint_devices()
2192 	struct kfd_criu_device_bucket *device_buckets;  in criu_restore_devices()  local
2203 	device_buckets = kmalloc_array(args->num_devices, sizeof(*device_buckets), GFP_KERNEL);  in criu_restore_devices()
2204 	if (!device_buckets)  in criu_restore_devices()
2207 	ret = copy_from_user(device_buckets, (void __user *)args->devices,  in criu_restore_devices()
2208 				args->num_devices * sizeof(*device_buckets));  in criu_restore_devices()
2222 		if (!device_buckets[i].user_gpu_id) {  in criu_restore_devices()
2228 		dev = kfd_device_by_id(device_buckets[i].actual_gpu_id);  in criu_restore_devices()
2231 				device_buckets[i].actual_gpu_id);  in criu_restore_devices()
2239 					device_buckets[i].actual_gpu_id);  in criu_restore_devices()
2243 		pdd->user_gpu_id = device_buckets[i].user_gpu_id;  in criu_restore_devices()
2245 		drm_file = fget(device_buckets[i].drm_fd);  in criu_restore_devices()
2248 				device_buckets[i].drm_fd);  in criu_restore_devices()
2291 	kfree(device_buckets);  in criu_restore_devices()