pgmap owner member at the svm migrate init could be referenced
to either adev or hive, depending on device topology.

Signed-off-by: Alex Sierra <alex.sie...@amd.com>
---
 drivers/gpu/drm/amd/amdkfd/kfd_migrate.c | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/drivers/gpu/drm/amd/amdkfd/kfd_migrate.c 
b/drivers/gpu/drm/amd/amdkfd/kfd_migrate.c
index 6b810863f6ba..21d92de235be 100644
--- a/drivers/gpu/drm/amd/amdkfd/kfd_migrate.c
+++ b/drivers/gpu/drm/amd/amdkfd/kfd_migrate.c
@@ -426,7 +426,7 @@ svm_migrate_vma_to_vram(struct amdgpu_device *adev, struct 
svm_range *prange,
        migrate.start = start;
        migrate.end = end;
        migrate.flags = MIGRATE_VMA_SELECT_SYSTEM;
-       migrate.pgmap_owner = adev;
+       migrate.pgmap_owner = (adev->hive)?(void *)adev->hive:(void *)adev;
 
        size = 2 * sizeof(*migrate.src) + sizeof(uint64_t) + sizeof(dma_addr_t);
        size *= npages;
@@ -641,7 +641,7 @@ svm_migrate_vma_to_ram(struct amdgpu_device *adev, struct 
svm_range *prange,
        migrate.start = start;
        migrate.end = end;
        migrate.flags = MIGRATE_VMA_SELECT_DEVICE_PRIVATE;
-       migrate.pgmap_owner = adev;
+       migrate.pgmap_owner = (adev->hive)?(void *)adev->hive:(void *)adev;
 
        size = 2 * sizeof(*migrate.src) + sizeof(uint64_t) + sizeof(dma_addr_t);
        size *= npages;
@@ -907,7 +907,7 @@ int svm_migrate_init(struct amdgpu_device *adev)
        pgmap->range.start = res->start;
        pgmap->range.end = res->end;
        pgmap->ops = &svm_migrate_pgmap_ops;
-       pgmap->owner = adev;
+       pgmap->owner = (adev->hive)?adev->hive:adev;
        pgmap->flags = MIGRATE_VMA_SELECT_DEVICE_PRIVATE;
        r = devm_memremap_pages(adev->dev, pgmap);
        if (IS_ERR(r)) {
-- 
2.17.1

_______________________________________________
amd-gfx mailing list
amd-gfx@lists.freedesktop.org
https://lists.freedesktop.org/mailman/listinfo/amd-gfx

Reply via email to