drm/amdkfd: Native mode memory partition support
For native mode, after amdgpu_bo is created on CPU domain, then call amdgpu_ttm_tt_set_mem_pool to select the TTM pool using bo->mem_id. ttm_bo_validate will allocate the memory to the correct memory partition before mapping to GPUs. Reviewed-by: Felix Kuehling <Felix.Kuehling@amd.com> Acked-and-tested-by: Mukul Joshi <mukul.joshi@amd.com> Signed-off-by: Philip Yang <Philip.Yang@amd.com> Signed-off-by: Rajneesh Bhardwaj <rajneesh.bhardwaj@amd.com> Signed-off-by: Alex Deucher <alexander.deucher@amd.com>
This commit is contained in:
committed by
Alex Deucher
parent
1e03322cfe
commit
fcfefd85f1
@@ -1155,6 +1155,24 @@ static void amdgpu_ttm_tt_unpopulate(struct ttm_device *bdev,
|
||||
return ttm_pool_free(pool, ttm);
|
||||
}
|
||||
|
||||
/**
|
||||
* amdgpu_ttm_tt_set_mem_pool - Set the TTM memory pool for the TTM BO
|
||||
* @tbo: The ttm_buffer_object that backs the VRAM bo
|
||||
* @mem_id: to select the initialized ttm pool corresponding to the memory partition
|
||||
*/
|
||||
int amdgpu_ttm_tt_set_mem_pool(struct ttm_buffer_object *tbo, int mem_id)
|
||||
{
|
||||
struct ttm_tt *ttm = tbo->ttm;
|
||||
struct amdgpu_ttm_tt *gtt;
|
||||
|
||||
if (!ttm && !ttm_tt_is_populated(ttm))
|
||||
return -EINVAL;
|
||||
|
||||
gtt = ttm_to_amdgpu_ttm_tt(ttm);
|
||||
gtt->pool_id = mem_id;
|
||||
return 0;
|
||||
}
|
||||
|
||||
/**
|
||||
* amdgpu_ttm_tt_get_userptr - Return the userptr GTT ttm_tt for the current
|
||||
* task
|
||||
|
||||
Reference in New Issue
Block a user