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:
Rajneesh Bhardwaj
2023-02-27 20:08:29 -05:00
committed by Alex Deucher
parent 1e03322cfe
commit fcfefd85f1
3 changed files with 26 additions and 0 deletions

View File

@@ -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