Skip to content

Commit

Permalink
drm/i915/selftests: Prepare gtt tests for obj->mm.lock removal
Browse files Browse the repository at this point in the history
We need to lock the global gtt dma_resv, use i915_vm_lock_objects
to handle this correctly. Add ww handling for this where required.

Add the object lock around unpin/put pages, and use the unlocked
versions of pin_pages and pin_map where required.

Signed-off-by: Maarten Lankhorst <[email protected]>
Reviewed-by: Thomas Hellström <[email protected]>
Signed-off-by: Daniel Vetter <[email protected]>
Link: https://patchwork.freedesktop.org/patch/msgid/[email protected]
  • Loading branch information
mlankhorst authored and danvet committed Mar 24, 2021
1 parent b91e1b1 commit 480ae79
Showing 1 changed file with 67 additions and 25 deletions.
92 changes: 67 additions & 25 deletions drivers/gpu/drm/i915/selftests/i915_gem_gtt.c
Original file line number Diff line number Diff line change
Expand Up @@ -130,7 +130,7 @@ fake_dma_object(struct drm_i915_private *i915, u64 size)
obj->cache_level = I915_CACHE_NONE;

/* Preallocate the "backing storage" */
if (i915_gem_object_pin_pages(obj))
if (i915_gem_object_pin_pages_unlocked(obj))
goto err_obj;

i915_gem_object_unpin_pages(obj);
Expand All @@ -146,6 +146,7 @@ static int igt_ppgtt_alloc(void *arg)
{
struct drm_i915_private *dev_priv = arg;
struct i915_ppgtt *ppgtt;
struct i915_gem_ww_ctx ww;
u64 size, last, limit;
int err = 0;

Expand All @@ -171,6 +172,12 @@ static int igt_ppgtt_alloc(void *arg)
limit = totalram_pages() << PAGE_SHIFT;
limit = min(ppgtt->vm.total, limit);

i915_gem_ww_ctx_init(&ww, false);
retry:
err = i915_vm_lock_objects(&ppgtt->vm, &ww);
if (err)
goto err_ppgtt_cleanup;

/* Check we can allocate the entire range */
for (size = 4096; size <= limit; size <<= 2) {
struct i915_vm_pt_stash stash = {};
Expand Down Expand Up @@ -215,6 +222,13 @@ static int igt_ppgtt_alloc(void *arg)
}

err_ppgtt_cleanup:
if (err == -EDEADLK) {
err = i915_gem_ww_ctx_backoff(&ww);
if (!err)
goto retry;
}
i915_gem_ww_ctx_fini(&ww);

i915_vm_put(&ppgtt->vm);
return err;
}
Expand Down Expand Up @@ -276,7 +290,7 @@ static int lowlevel_hole(struct i915_address_space *vm,

GEM_BUG_ON(obj->base.size != BIT_ULL(size));

if (i915_gem_object_pin_pages(obj)) {
if (i915_gem_object_pin_pages_unlocked(obj)) {
i915_gem_object_put(obj);
kfree(order);
break;
Expand All @@ -297,20 +311,36 @@ static int lowlevel_hole(struct i915_address_space *vm,

if (vm->allocate_va_range) {
struct i915_vm_pt_stash stash = {};
struct i915_gem_ww_ctx ww;
int err;

i915_gem_ww_ctx_init(&ww, false);
retry:
err = i915_vm_lock_objects(vm, &ww);
if (err)
goto alloc_vm_end;

err = -ENOMEM;
if (i915_vm_alloc_pt_stash(vm, &stash,
BIT_ULL(size)))
break;

if (i915_vm_pin_pt_stash(vm, &stash)) {
i915_vm_free_pt_stash(vm, &stash);
break;
}
goto alloc_vm_end;

vm->allocate_va_range(vm, &stash,
addr, BIT_ULL(size));
err = i915_vm_pin_pt_stash(vm, &stash);
if (!err)
vm->allocate_va_range(vm, &stash,
addr, BIT_ULL(size));

i915_vm_free_pt_stash(vm, &stash);
alloc_vm_end:
if (err == -EDEADLK) {
err = i915_gem_ww_ctx_backoff(&ww);
if (!err)
goto retry;
}
i915_gem_ww_ctx_fini(&ww);

if (err)
break;
}

mock_vma->pages = obj->mm.pages;
Expand Down Expand Up @@ -1166,7 +1196,7 @@ static int igt_ggtt_page(void *arg)
if (IS_ERR(obj))
return PTR_ERR(obj);

err = i915_gem_object_pin_pages(obj);
err = i915_gem_object_pin_pages_unlocked(obj);
if (err)
goto out_free;

Expand Down Expand Up @@ -1333,7 +1363,7 @@ static int igt_gtt_reserve(void *arg)
goto out;
}

err = i915_gem_object_pin_pages(obj);
err = i915_gem_object_pin_pages_unlocked(obj);
if (err) {
i915_gem_object_put(obj);
goto out;
Expand Down Expand Up @@ -1385,7 +1415,7 @@ static int igt_gtt_reserve(void *arg)
goto out;
}

err = i915_gem_object_pin_pages(obj);
err = i915_gem_object_pin_pages_unlocked(obj);
if (err) {
i915_gem_object_put(obj);
goto out;
Expand Down Expand Up @@ -1549,7 +1579,7 @@ static int igt_gtt_insert(void *arg)
goto out;
}

err = i915_gem_object_pin_pages(obj);
err = i915_gem_object_pin_pages_unlocked(obj);
if (err) {
i915_gem_object_put(obj);
goto out;
Expand Down Expand Up @@ -1658,7 +1688,7 @@ static int igt_gtt_insert(void *arg)
goto out;
}

err = i915_gem_object_pin_pages(obj);
err = i915_gem_object_pin_pages_unlocked(obj);
if (err) {
i915_gem_object_put(obj);
goto out;
Expand Down Expand Up @@ -1829,7 +1859,7 @@ static int igt_cs_tlb(void *arg)
goto out_vm;
}

batch = i915_gem_object_pin_map(bbe, I915_MAP_WC);
batch = i915_gem_object_pin_map_unlocked(bbe, I915_MAP_WC);
if (IS_ERR(batch)) {
err = PTR_ERR(batch);
goto out_put_bbe;
Expand All @@ -1845,7 +1875,7 @@ static int igt_cs_tlb(void *arg)
}

/* Track the execution of each request by writing into different slot */
batch = i915_gem_object_pin_map(act, I915_MAP_WC);
batch = i915_gem_object_pin_map_unlocked(act, I915_MAP_WC);
if (IS_ERR(batch)) {
err = PTR_ERR(batch);
goto out_put_act;
Expand Down Expand Up @@ -1892,7 +1922,7 @@ static int igt_cs_tlb(void *arg)
goto out_put_out;
GEM_BUG_ON(vma->node.start != vm->total - PAGE_SIZE);

result = i915_gem_object_pin_map(out, I915_MAP_WB);
result = i915_gem_object_pin_map_unlocked(out, I915_MAP_WB);
if (IS_ERR(result)) {
err = PTR_ERR(result);
goto out_put_out;
Expand All @@ -1908,6 +1938,7 @@ static int igt_cs_tlb(void *arg)
while (!__igt_timeout(end_time, NULL)) {
struct i915_vm_pt_stash stash = {};
struct i915_request *rq;
struct i915_gem_ww_ctx ww;
u64 offset;

offset = igt_random_offset(&prng,
Expand All @@ -1926,19 +1957,30 @@ static int igt_cs_tlb(void *arg)
if (err)
goto end;

i915_gem_ww_ctx_init(&ww, false);
retry:
err = i915_vm_lock_objects(vm, &ww);
if (err)
goto end_ww;

err = i915_vm_alloc_pt_stash(vm, &stash, chunk_size);
if (err)
goto end;
goto end_ww;

err = i915_vm_pin_pt_stash(vm, &stash);
if (err) {
i915_vm_free_pt_stash(vm, &stash);
goto end;
}

vm->allocate_va_range(vm, &stash, offset, chunk_size);
if (!err)
vm->allocate_va_range(vm, &stash, offset, chunk_size);

i915_vm_free_pt_stash(vm, &stash);
end_ww:
if (err == -EDEADLK) {
err = i915_gem_ww_ctx_backoff(&ww);
if (!err)
goto retry;
}
i915_gem_ww_ctx_fini(&ww);
if (err)
goto end;

/* Prime the TLB with the dummy pages */
for (i = 0; i < count; i++) {
Expand Down

0 comments on commit 480ae79

Please sign in to comment.