No functional change for now, as we always allocate a single entity.

---
v4: stop using adev->sdma.num_instances
---

Signed-off-by: Pierre-Eric Pelloux-Prayer <[email protected]>
Acked-by: Felix Kuehling <[email protected]>
---
 drivers/gpu/drm/amd/amdgpu/amdgpu_ttm.c  | 35 +++++++++++++++---------
 drivers/gpu/drm/amd/amdgpu/amdgpu_ttm.h  |  3 +-
 drivers/gpu/drm/amd/amdkfd/kfd_migrate.c |  2 +-
 3 files changed, 25 insertions(+), 15 deletions(-)

diff --git a/drivers/gpu/drm/amd/amdgpu/amdgpu_ttm.c 
b/drivers/gpu/drm/amd/amdgpu/amdgpu_ttm.c
index 71316b3d4a29..75cb354084a0 100644
--- a/drivers/gpu/drm/amd/amdgpu/amdgpu_ttm.c
+++ b/drivers/gpu/drm/amd/amdgpu/amdgpu_ttm.c
@@ -399,7 +399,7 @@ static int amdgpu_move_blit(struct ttm_buffer_object *bo,
        dst.offset = 0;
 
        r = amdgpu_ttm_copy_mem_to_mem(adev,
-                                      &adev->mman.move_entity,
+                                      &adev->mman.move_entities[0],
                                       &src, &dst,
                                       new_mem->size,
                                       amdgpu_bo_encrypted(abo),
@@ -412,7 +412,7 @@ static int amdgpu_move_blit(struct ttm_buffer_object *bo,
            (abo->flags & AMDGPU_GEM_CREATE_VRAM_WIPE_ON_RELEASE)) {
                struct dma_fence *wipe_fence = NULL;
 
-               r = amdgpu_fill_buffer(&adev->mman.move_entity,
+               r = amdgpu_fill_buffer(&adev->mman.move_entities[0],
                                       abo, 0, NULL, &wipe_fence,
                                       AMDGPU_KERNEL_JOB_ID_MOVE_BLIT);
                if (r) {
@@ -2346,7 +2346,7 @@ void amdgpu_ttm_fini(struct amdgpu_device *adev)
 void amdgpu_ttm_set_buffer_funcs_status(struct amdgpu_device *adev, bool 
enable)
 {
        struct ttm_resource_manager *man = ttm_manager_type(&adev->mman.bdev, 
TTM_PL_VRAM);
-       u32 num_clear_entities;
+       u32 num_clear_entities, num_move_entities;
        uint64_t size;
        int r, i, j;
 
@@ -2364,6 +2364,7 @@ void amdgpu_ttm_set_buffer_funcs_status(struct 
amdgpu_device *adev, bool enable)
                }
 
                num_clear_entities = 1;
+               num_move_entities = 1;
                ring = adev->mman.buffer_funcs_ring;
                sched = &ring->sched;
                r = amdgpu_ttm_buffer_entity_init(&adev->mman.gtt_mgr,
@@ -2400,14 +2401,20 @@ void amdgpu_ttm_set_buffer_funcs_status(struct 
amdgpu_device *adev, bool enable)
                        }
                }
 
-               r = amdgpu_ttm_buffer_entity_init(&adev->mman.gtt_mgr,
-                                                 &adev->mman.move_entity,
-                                                 DRM_SCHED_PRIORITY_NORMAL,
-                                                 &sched, 1, 2);
-               if (r < 0) {
-                       dev_err(adev->dev,
-                               "Failed setting up TTM BO move entity (%d)\n", 
r);
-                       goto error_free_clear_entities;
+               adev->mman.num_move_entities = num_move_entities;
+               for (i = 0; i < num_move_entities; i++) {
+                       r = amdgpu_ttm_buffer_entity_init(
+                               &adev->mman.gtt_mgr,
+                               &adev->mman.move_entities[i],
+                               DRM_SCHED_PRIORITY_NORMAL, &sched, 1, 2);
+
+                       if (r < 0) {
+                               for (j = 0; j < i; j++)
+                                       amdgpu_ttm_buffer_entity_fini(
+                                               &adev->mman.gtt_mgr, 
&adev->mman.move_entities[j]);
+                               adev->mman.num_move_entities = 0;
+                               goto error_free_clear_entities;
+                       }
                }
        } else {
                amdgpu_ttm_buffer_entity_fini(&adev->mman.gtt_mgr,
@@ -2415,8 +2422,9 @@ void amdgpu_ttm_set_buffer_funcs_status(struct 
amdgpu_device *adev, bool enable)
                for (i = 0; i < adev->mman.num_clear_entities; i++)
                        amdgpu_ttm_buffer_entity_fini(&adev->mman.gtt_mgr,
                                                      
&adev->mman.clear_entities[i]);
-               amdgpu_ttm_buffer_entity_fini(&adev->mman.gtt_mgr,
-                                             &adev->mman.move_entity);
+               for (i = 0; i < adev->mman.num_move_entities; i++)
+                       amdgpu_ttm_buffer_entity_fini(&adev->mman.gtt_mgr,
+                                                     
&adev->mman.move_entities[i]);
                /* Drop all the old fences since re-creating the scheduler 
entities
                 * will allocate new contexts.
                 */
@@ -2424,6 +2432,7 @@ void amdgpu_ttm_set_buffer_funcs_status(struct 
amdgpu_device *adev, bool enable)
                kfree(adev->mman.clear_entities);
                adev->mman.clear_entities = NULL;
                adev->mman.num_clear_entities = 0;
+               adev->mman.num_move_entities = 0;
        }
 
        /* this just adjusts TTM size idea, which sets lpfn to the correct 
value */
diff --git a/drivers/gpu/drm/amd/amdgpu/amdgpu_ttm.h 
b/drivers/gpu/drm/amd/amdgpu/amdgpu_ttm.h
index e98d458b8029..cd24ca851b6d 100644
--- a/drivers/gpu/drm/amd/amdgpu/amdgpu_ttm.h
+++ b/drivers/gpu/drm/amd/amdgpu/amdgpu_ttm.h
@@ -72,9 +72,10 @@ struct amdgpu_mman {
 
        /* @default_entity: for workarounds, has no gart windows */
        struct amdgpu_ttm_buffer_entity default_entity;
-       struct amdgpu_ttm_buffer_entity move_entity;
        struct amdgpu_ttm_buffer_entity *clear_entities;
        u32 num_clear_entities;
+       struct amdgpu_ttm_buffer_entity move_entities[TTM_NUM_MOVE_FENCES];
+       u32 num_move_entities;
 
        struct amdgpu_vram_mgr vram_mgr;
        struct amdgpu_gtt_mgr gtt_mgr;
diff --git a/drivers/gpu/drm/amd/amdkfd/kfd_migrate.c 
b/drivers/gpu/drm/amd/amdkfd/kfd_migrate.c
index b3d304aab686..7b089d22c367 100644
--- a/drivers/gpu/drm/amd/amdkfd/kfd_migrate.c
+++ b/drivers/gpu/drm/amd/amdkfd/kfd_migrate.c
@@ -136,7 +136,7 @@ svm_migrate_copy_memory_gart(struct amdgpu_device *adev, 
dma_addr_t *sys,
        u64 size;
        int r;
 
-       entity = &adev->mman.move_entity;
+       entity = &adev->mman.move_entities[0];
 
        mutex_lock(&entity->lock);
 
-- 
2.43.0

Reply via email to