amdgpu: add a function to find bo by cpu mapping (v2)
Userspace needs to know if the user memory is from BO or malloc. v2: update mutex range and rebase Signed-off-by: Junwei Zhang <Jerry.Zhang@amd.com> Reviewed-by: Christian König <christian.koenig@amd.com>main
parent
f49dccbb24
commit
4d454424e1
|
@ -677,6 +677,29 @@ int amdgpu_create_bo_from_user_mem(amdgpu_device_handle dev,
|
||||||
void *cpu, uint64_t size,
|
void *cpu, uint64_t size,
|
||||||
amdgpu_bo_handle *buf_handle);
|
amdgpu_bo_handle *buf_handle);
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Validate if the user memory comes from BO
|
||||||
|
*
|
||||||
|
* \param dev - [in] Device handle. See #amdgpu_device_initialize()
|
||||||
|
* \param cpu - [in] CPU address of user allocated memory which we
|
||||||
|
* want to map to GPU address space (make GPU accessible)
|
||||||
|
* (This address must be correctly aligned).
|
||||||
|
* \param size - [in] Size of allocation (must be correctly aligned)
|
||||||
|
* \param buf_handle - [out] Buffer handle for the userptr memory
|
||||||
|
* if the user memory is not from BO, the buf_handle will be NULL.
|
||||||
|
* \param offset_in_bo - [out] offset in this BO for this user memory
|
||||||
|
*
|
||||||
|
*
|
||||||
|
* \return 0 on success\n
|
||||||
|
* <0 - Negative POSIX Error code
|
||||||
|
*
|
||||||
|
*/
|
||||||
|
int amdgpu_find_bo_by_cpu_mapping(amdgpu_device_handle dev,
|
||||||
|
void *cpu,
|
||||||
|
uint64_t size,
|
||||||
|
amdgpu_bo_handle *buf_handle,
|
||||||
|
uint64_t *offset_in_bo);
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Free previosuly allocated memory
|
* Free previosuly allocated memory
|
||||||
*
|
*
|
||||||
|
|
|
@ -529,6 +529,45 @@ int amdgpu_bo_wait_for_idle(amdgpu_bo_handle bo,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
int amdgpu_find_bo_by_cpu_mapping(amdgpu_device_handle dev,
|
||||||
|
void *cpu,
|
||||||
|
uint64_t size,
|
||||||
|
amdgpu_bo_handle *buf_handle,
|
||||||
|
uint64_t *offset_in_bo)
|
||||||
|
{
|
||||||
|
int i;
|
||||||
|
struct amdgpu_bo *bo;
|
||||||
|
|
||||||
|
if (cpu == NULL || size == 0)
|
||||||
|
return -EINVAL;
|
||||||
|
|
||||||
|
/*
|
||||||
|
* Workaround for a buggy application which tries to import previously
|
||||||
|
* exposed CPU pointers. If we find a real world use case we should
|
||||||
|
* improve that by asking the kernel for the right handle.
|
||||||
|
*/
|
||||||
|
pthread_mutex_lock(&dev->bo_table_mutex);
|
||||||
|
for (i = 0; i < dev->bo_handles.max_key; i++) {
|
||||||
|
bo = handle_table_lookup(&dev->bo_handles, i);
|
||||||
|
if (!bo || !bo->cpu_ptr || size > bo->alloc_size)
|
||||||
|
continue;
|
||||||
|
if (cpu >= bo->cpu_ptr && cpu < (bo->cpu_ptr + bo->alloc_size))
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (i < dev->bo_handles.max_key) {
|
||||||
|
atomic_inc(&bo->refcount);
|
||||||
|
*buf_handle = bo;
|
||||||
|
*offset_in_bo = cpu - bo->cpu_ptr;
|
||||||
|
} else {
|
||||||
|
*buf_handle = NULL;
|
||||||
|
*offset_in_bo = 0;
|
||||||
|
}
|
||||||
|
pthread_mutex_unlock(&dev->bo_table_mutex);
|
||||||
|
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
int amdgpu_create_bo_from_user_mem(amdgpu_device_handle dev,
|
int amdgpu_create_bo_from_user_mem(amdgpu_device_handle dev,
|
||||||
void *cpu,
|
void *cpu,
|
||||||
uint64_t size,
|
uint64_t size,
|
||||||
|
|
Loading…
Reference in New Issue