drivers/gpu/drm/amd/amdgpu/amdgpu_vm.c

   1 /*
   2  * Copyright 2008 Advanced Micro Devices, Inc.
   3  * Copyright 2008 Red Hat Inc.
   4  * Copyright 2009 Jerome Glisse.
   5  *
   6  * Permission is hereby granted, free of charge, to any person obtaining a
   7  * copy of this software and associated documentation files (the "Software"),
   8  * to deal in the Software without restriction, including without limitation
   9  * the rights to use, copy, modify, merge, publish, distribute, sublicense,
  10  * and/or sell copies of the Software, and to permit persons to whom the
  11  * Software is furnished to do so, subject to the following conditions:
  12  *
  13  * The above copyright notice and this permission notice shall be included in
  14  * all copies or substantial portions of the Software.
  15  *
  16  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  17  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  18  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
  19  * THE COPYRIGHT HOLDER(S) OR AUTHOR(S) BE LIABLE FOR ANY CLAIM, DAMAGES OR
  20  * OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
  21  * ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
  22  * OTHER DEALINGS IN THE SOFTWARE.
  23  *
  24  * Authors: Dave Airlie
  25  *          Alex Deucher
  26  *          Jerome Glisse
  27  */
  28 #include <drm/drmP.h>
  29 #include <drm/amdgpu_drm.h>
  30 #include "amdgpu.h"
  31 #include "amdgpu_trace.h"
  32
  33 /*
  34  * GPUVM
  35  * GPUVM is similar to the legacy gart on older asics, however
  36  * rather than there being a single global gart table
  37  * for the entire GPU, there are multiple VM page tables active
  38  * at any given time.  The VM page tables can contain a mix
  39  * vram pages and system memory pages and system memory pages
  40  * can be mapped as snooped (cached system pages) or unsnooped
  41  * (uncached system pages).
  42  * Each VM has an ID associated with it and there is a page table
  43  * associated with each VMID.  When execting a command buffer,
  44  * the kernel tells the the ring what VMID to use for that command
  45  * buffer.  VMIDs are allocated dynamically as commands are submitted.
  46  * The userspace drivers maintain their own address space and the kernel
  47  * sets up their pages tables accordingly when they submit their
  48  * command buffers and a VMID is assigned.
  49  * Cayman/Trinity support up to 8 active VMs at any given time;
  50  * SI supports 16.
  51  */
  52
  53 /**
  54  * amdgpu_vm_num_pde - return the number of page directory entries
  55  *
  56  * @adev: amdgpu_device pointer
  57  *
  58  * Calculate the number of page directory entries (cayman+).
  59  */
  60 static unsigned amdgpu_vm_num_pdes(struct amdgpu_device *adev)
  61 {
  62         return adev->vm_manager.max_pfn >> amdgpu_vm_block_size;
  63 }
  64
  65 /**
  66  * amdgpu_vm_directory_size - returns the size of the page directory in bytes
  67  *
  68  * @adev: amdgpu_device pointer
  69  *
  70  * Calculate the size of the page directory in bytes (cayman+).
  71  */
  72 static unsigned amdgpu_vm_directory_size(struct amdgpu_device *adev)
  73 {
  74         return AMDGPU_GPU_PAGE_ALIGN(amdgpu_vm_num_pdes(adev) * 8);
  75 }
  76
  77 /**
  78  * amdgpu_vm_get_bos - add the vm BOs to a validation list
  79  *
  80  * @vm: vm providing the BOs
  81  * @head: head of validation list
  82  *
  83  * Add the page directory to the list of BOs to
  84  * validate for command submission (cayman+).
  85  */
  86 struct amdgpu_bo_list_entry *amdgpu_vm_get_bos(struct amdgpu_device *adev,
  87                                           struct amdgpu_vm *vm,
  88                                           struct list_head *head)
  89 {
  90         struct amdgpu_bo_list_entry *list;
  91         unsigned i, idx;
  92
  93         mutex_lock(&vm->mutex);
  94         list = drm_malloc_ab(vm->max_pde_used + 2,
  95                              sizeof(struct amdgpu_bo_list_entry));
  96         if (!list) {
  97                 mutex_unlock(&vm->mutex);
  98                 return NULL;
  99         }
 100
 101         /* add the vm page table to the list */
 102         list[0].robj = vm->page_directory;
 103         list[0].prefered_domains = AMDGPU_GEM_DOMAIN_VRAM;
 104         list[0].allowed_domains = AMDGPU_GEM_DOMAIN_VRAM;
 105         list[0].priority = 0;
 106         list[0].tv.bo = &vm->page_directory->tbo;
 107         list[0].tv.shared = true;
 108         list_add(&list[0].tv.head, head);
 109
 110         for (i = 0, idx = 1; i <= vm->max_pde_used; i++) {
 111                 if (!vm->page_tables[i].bo)
 112                         continue;
 113
 114                 list[idx].robj = vm->page_tables[i].bo;
 115                 list[idx].prefered_domains = AMDGPU_GEM_DOMAIN_VRAM;
 116                 list[idx].allowed_domains = AMDGPU_GEM_DOMAIN_VRAM;
 117                 list[idx].priority = 0;
 118                 list[idx].tv.bo = &list[idx].robj->tbo;
 119                 list[idx].tv.shared = true;
 120                 list_add(&list[idx++].tv.head, head);
 121         }
 122         mutex_unlock(&vm->mutex);
 123
 124         return list;
 125 }
 126
 127 /**
 128  * amdgpu_vm_grab_id - allocate the next free VMID
 129  *
 130  * @vm: vm to allocate id for
 131  * @ring: ring we want to submit job to
 132  * @sync: sync object where we add dependencies
 133  *
 134  * Allocate an id for the vm, adding fences to the sync obj as necessary.
 135  *
 136  * Global mutex must be locked!
 137  */
 138 int amdgpu_vm_grab_id(struct amdgpu_vm *vm, struct amdgpu_ring *ring,
 139                       struct amdgpu_sync *sync)
 140 {
 141         struct amdgpu_fence *best[AMDGPU_MAX_RINGS] = {};
 142         struct amdgpu_vm_id *vm_id = &vm->ids[ring->idx];
 143         struct amdgpu_device *adev = ring->adev;
 144
 145         unsigned choices[2] = {};
 146         unsigned i;
 147
 148         /* check if the id is still valid */
 149         if (vm_id->id && vm_id->last_id_use &&
 150             vm_id->last_id_use == adev->vm_manager.active[vm_id->id])
 151                 return 0;
 152
 153         /* we definately need to flush */
 154         vm_id->pd_gpu_addr = ~0ll;
 155
 156         /* skip over VMID 0, since it is the system VM */
 157         for (i = 1; i < adev->vm_manager.nvm; ++i) {
 158                 struct amdgpu_fence *fence = adev->vm_manager.active[i];
 159
 160                 if (fence == NULL) {
 161                         /* found a free one */
 162                         vm_id->id = i;
 163                         trace_amdgpu_vm_grab_id(i, ring->idx);
 164                         return 0;
 165                 }
 166
 167                 if (amdgpu_fence_is_earlier(fence, best[fence->ring->idx])) {
 168                         best[fence->ring->idx] = fence;
 169                         choices[fence->ring == ring ? 0 : 1] = i;
 170                 }
 171         }
 172
 173         for (i = 0; i < 2; ++i) {
 174                 if (choices[i]) {
 175                         struct amdgpu_fence *fence;
 176
 177                         fence  = adev->vm_manager.active[choices[i]];
 178                         vm_id->id = choices[i];
 179
 180                         trace_amdgpu_vm_grab_id(choices[i], ring->idx);
 181                         return amdgpu_sync_fence(ring->adev, sync, &fence->base);
 182                 }
 183         }
 184
 185         /* should never happen */
 186         BUG();
 187         return -EINVAL;
 188 }
 189
 190 /**
 191  * amdgpu_vm_flush - hardware flush the vm
 192  *
 193  * @ring: ring to use for flush
 194  * @vm: vm we want to flush
 195  * @updates: last vm update that we waited for
 196  *
 197  * Flush the vm (cayman+).
 198  *
 199  * Global and local mutex must be locked!
 200  */
 201 void amdgpu_vm_flush(struct amdgpu_ring *ring,
 202                      struct amdgpu_vm *vm,
 203                      struct amdgpu_fence *updates)
 204 {
 205         uint64_t pd_addr = amdgpu_bo_gpu_offset(vm->page_directory);
 206         struct amdgpu_vm_id *vm_id = &vm->ids[ring->idx];
 207         struct amdgpu_fence *flushed_updates = vm_id->flushed_updates;
 208
 209         if (pd_addr != vm_id->pd_gpu_addr || !flushed_updates ||
 210             (updates && amdgpu_fence_is_earlier(flushed_updates, updates))) {
 211
 212                 trace_amdgpu_vm_flush(pd_addr, ring->idx, vm_id->id);
 213                 vm_id->flushed_updates = amdgpu_fence_ref(
 214                         amdgpu_fence_later(flushed_updates, updates));
 215                 amdgpu_fence_unref(&flushed_updates);
 216                 vm_id->pd_gpu_addr = pd_addr;
 217                 amdgpu_ring_emit_vm_flush(ring, vm_id->id, vm_id->pd_gpu_addr);
 218         }
 219 }
 220
 221 /**
 222  * amdgpu_vm_fence - remember fence for vm
 223  *
 224  * @adev: amdgpu_device pointer
 225  * @vm: vm we want to fence
 226  * @fence: fence to remember
 227  *
 228  * Fence the vm (cayman+).
 229  * Set the fence used to protect page table and id.
 230  *
 231  * Global and local mutex must be locked!
 232  */
 233 void amdgpu_vm_fence(struct amdgpu_device *adev,
 234                      struct amdgpu_vm *vm,
 235                      struct amdgpu_fence *fence)
 236 {
 237         unsigned ridx = fence->ring->idx;
 238         unsigned vm_id = vm->ids[ridx].id;
 239
 240         amdgpu_fence_unref(&adev->vm_manager.active[vm_id]);
 241         adev->vm_manager.active[vm_id] = amdgpu_fence_ref(fence);
 242
 243         amdgpu_fence_unref(&vm->ids[ridx].last_id_use);
 244         vm->ids[ridx].last_id_use = amdgpu_fence_ref(fence);
 245 }
 246
 247 /**
 248  * amdgpu_vm_bo_find - find the bo_va for a specific vm & bo
 249  *
 250  * @vm: requested vm
 251  * @bo: requested buffer object
 252  *
 253  * Find @bo inside the requested vm (cayman+).
 254  * Search inside the @bos vm list for the requested vm
 255  * Returns the found bo_va or NULL if none is found
 256  *
 257  * Object has to be reserved!
 258  */
 259 struct amdgpu_bo_va *amdgpu_vm_bo_find(struct amdgpu_vm *vm,
 260                                        struct amdgpu_bo *bo)
 261 {
 262         struct amdgpu_bo_va *bo_va;
 263
 264         list_for_each_entry(bo_va, &bo->va, bo_list) {
 265                 if (bo_va->vm == vm) {
 266                         return bo_va;
 267                 }
 268         }
 269         return NULL;
 270 }
 271
 272 /**
 273  * amdgpu_vm_update_pages - helper to call the right asic function
 274  *
 275  * @adev: amdgpu_device pointer
 276  * @ib: indirect buffer to fill with commands
 277  * @pe: addr of the page entry
 278  * @addr: dst addr to write into pe
 279  * @count: number of page entries to update
 280  * @incr: increase next addr by incr bytes
 281  * @flags: hw access flags
 282  * @gtt_flags: GTT hw access flags
 283  *
 284  * Traces the parameters and calls the right asic functions
 285  * to setup the page table using the DMA.
 286  */
 287 static void amdgpu_vm_update_pages(struct amdgpu_device *adev,
 288                                    struct amdgpu_ib *ib,
 289                                    uint64_t pe, uint64_t addr,
 290                                    unsigned count, uint32_t incr,
 291                                    uint32_t flags, uint32_t gtt_flags)
 292 {
 293         trace_amdgpu_vm_set_page(pe, addr, count, incr, flags);
 294
 295         if ((flags & AMDGPU_PTE_SYSTEM) && (flags == gtt_flags)) {
 296                 uint64_t src = adev->gart.table_addr + (addr >> 12) * 8;
 297                 amdgpu_vm_copy_pte(adev, ib, pe, src, count);
 298
 299         } else if ((flags & AMDGPU_PTE_SYSTEM) || (count < 3)) {
 300                 amdgpu_vm_write_pte(adev, ib, pe, addr,
 301                                       count, incr, flags);
 302
 303         } else {
 304                 amdgpu_vm_set_pte_pde(adev, ib, pe, addr,
 305                                       count, incr, flags);
 306         }
 307 }
 308
 309 static int amdgpu_vm_free_job(
 310         struct amdgpu_cs_parser *sched_job)
 311 {
 312         int i;
 313         for (i = 0; i < sched_job->num_ibs; i++)
 314                 amdgpu_ib_free(sched_job->adev, &sched_job->ibs[i]);
 315         kfree(sched_job->ibs);
 316         return 0;
 317 }
 318
 319 static int amdgpu_vm_run_job(
 320         struct amdgpu_cs_parser *sched_job)
 321 {
 322         amdgpu_bo_fence(sched_job->job_param.vm.bo,
 323                         sched_job->ibs[sched_job->num_ibs -1].fence, true);
 324         return 0;
 325 }
 326
 327 /**
 328  * amdgpu_vm_clear_bo - initially clear the page dir/table
 329  *
 330  * @adev: amdgpu_device pointer
 331  * @bo: bo to clear
 332  */
 333 static int amdgpu_vm_clear_bo(struct amdgpu_device *adev,
 334                               struct amdgpu_bo *bo)
 335 {
 336         struct amdgpu_ring *ring = adev->vm_manager.vm_pte_funcs_ring;
 337         struct amdgpu_cs_parser *sched_job = NULL;
 338         struct amdgpu_ib *ib;
 339         unsigned entries;
 340         uint64_t addr;
 341         int r;
 342
 343         r = amdgpu_bo_reserve(bo, false);
 344         if (r)
 345                 return r;
 346
 347         r = reservation_object_reserve_shared(bo->tbo.resv);
 348         if (r)
 349                 return r;
 350
 351         r = ttm_bo_validate(&bo->tbo, &bo->placement, true, false);
 352         if (r)
 353                 goto error_unreserve;
 354
 355         addr = amdgpu_bo_gpu_offset(bo);
 356         entries = amdgpu_bo_size(bo) / 8;
 357
 358         ib = kzalloc(sizeof(struct amdgpu_ib), GFP_KERNEL);
 359         if (!ib)
 360                 goto error_unreserve;
 361
 362         r = amdgpu_ib_get(ring, NULL, entries * 2 + 64, ib);
 363         if (r)
 364                 goto error_free;
 365
 366         ib->length_dw = 0;
 367
 368         amdgpu_vm_update_pages(adev, ib, addr, 0, entries, 0, 0, 0);
 369         amdgpu_vm_pad_ib(adev, ib);
 370         WARN_ON(ib->length_dw > 64);
 371
 372         if (amdgpu_enable_scheduler) {
 373                 int r;
 374                 uint64_t v_seq;
 375                 sched_job = amdgpu_cs_parser_create(adev, AMDGPU_FENCE_OWNER_VM,
 376                                                     adev->kernel_ctx, ib, 1);
 377                 if(!sched_job)
 378                         goto error_free;
 379                 sched_job->job_param.vm.bo = bo;
 380                 sched_job->run_job = amdgpu_vm_run_job;
 381                 sched_job->free_job = amdgpu_vm_free_job;
 382                 v_seq = atomic64_inc_return(&adev->kernel_ctx->rings[ring->idx].c_entity.last_queued_v_seq);
 383                 ib->sequence = v_seq;
 384                 amd_sched_push_job(ring->scheduler,
 385                                    &adev->kernel_ctx->rings[ring->idx].c_entity,
 386                                    sched_job);
 387                 r = amd_sched_wait_emit(&adev->kernel_ctx->rings[ring->idx].c_entity,
 388                                         v_seq,
 389                                         false,
 390                                         -1);
 391                 if (r)
 392                         DRM_ERROR("emit timeout\n");
 393
 394                 amdgpu_bo_unreserve(bo);
 395                 return 0;
 396         } else {
 397                 r = amdgpu_ib_schedule(adev, 1, ib, AMDGPU_FENCE_OWNER_VM);
 398                 if (r)
 399                         goto error_free;
 400                 amdgpu_bo_fence(bo, ib->fence, true);
 401         }
 402
 403 error_free:
 404         amdgpu_ib_free(adev, ib);
 405         kfree(ib);
 406
 407 error_unreserve:
 408         amdgpu_bo_unreserve(bo);
 409         return r;
 410 }
 411
 412 /**
 413  * amdgpu_vm_map_gart - get the physical address of a gart page
 414  *
 415  * @adev: amdgpu_device pointer
 416  * @addr: the unmapped addr
 417  *
 418  * Look up the physical address of the page that the pte resolves
 419  * to (cayman+).
 420  * Returns the physical address of the page.
 421  */
 422 uint64_t amdgpu_vm_map_gart(struct amdgpu_device *adev, uint64_t addr)
 423 {
 424         uint64_t result;
 425
 426         /* page table offset */
 427         result = adev->gart.pages_addr[addr >> PAGE_SHIFT];
 428
 429         /* in case cpu page size != gpu page size*/
 430         result |= addr & (~PAGE_MASK);
 431
 432         return result;
 433 }
 434
 435 /**
 436  * amdgpu_vm_update_pdes - make sure that page directory is valid
 437  *
 438  * @adev: amdgpu_device pointer
 439  * @vm: requested vm
 440  * @start: start of GPU address range
 441  * @end: end of GPU address range
 442  *
 443  * Allocates new page tables if necessary
 444  * and updates the page directory (cayman+).
 445  * Returns 0 for success, error for failure.
 446  *
 447  * Global and local mutex must be locked!
 448  */
 449 int amdgpu_vm_update_page_directory(struct amdgpu_device *adev,
 450                                     struct amdgpu_vm *vm)
 451 {
 452         struct amdgpu_ring *ring = adev->vm_manager.vm_pte_funcs_ring;
 453         struct amdgpu_bo *pd = vm->page_directory;
 454         uint64_t pd_addr = amdgpu_bo_gpu_offset(pd);
 455         uint32_t incr = AMDGPU_VM_PTE_COUNT * 8;
 456         uint64_t last_pde = ~0, last_pt = ~0;
 457         unsigned count = 0, pt_idx, ndw;
 458         struct amdgpu_ib *ib;
 459         struct amdgpu_cs_parser *sched_job = NULL;
 460
 461         int r;
 462
 463         /* padding, etc. */
 464         ndw = 64;
 465
 466         /* assume the worst case */
 467         ndw += vm->max_pde_used * 6;
 468
 469         /* update too big for an IB */
 470         if (ndw > 0xfffff)
 471                 return -ENOMEM;
 472
 473         ib = kzalloc(sizeof(struct amdgpu_ib), GFP_KERNEL);
 474         if (!ib)
 475                 return -ENOMEM;
 476
 477         r = amdgpu_ib_get(ring, NULL, ndw * 4, ib);
 478         if (r)
 479                 return r;
 480         ib->length_dw = 0;
 481
 482         /* walk over the address space and update the page directory */
 483         for (pt_idx = 0; pt_idx <= vm->max_pde_used; ++pt_idx) {
 484                 struct amdgpu_bo *bo = vm->page_tables[pt_idx].bo;
 485                 uint64_t pde, pt;
 486
 487                 if (bo == NULL)
 488                         continue;
 489
 490                 pt = amdgpu_bo_gpu_offset(bo);
 491                 if (vm->page_tables[pt_idx].addr == pt)
 492                         continue;
 493                 vm->page_tables[pt_idx].addr = pt;
 494
 495                 pde = pd_addr + pt_idx * 8;
 496                 if (((last_pde + 8 * count) != pde) ||
 497                     ((last_pt + incr * count) != pt)) {
 498
 499                         if (count) {
 500                                 amdgpu_vm_update_pages(adev, ib, last_pde,
 501                                                        last_pt, count, incr,
 502                                                        AMDGPU_PTE_VALID, 0);
 503                         }
 504
 505                         count = 1;
 506                         last_pde = pde;
 507                         last_pt = pt;
 508                 } else {
 509                         ++count;
 510                 }
 511         }
 512
 513         if (count)
 514                 amdgpu_vm_update_pages(adev, ib, last_pde, last_pt, count,
 515                                        incr, AMDGPU_PTE_VALID, 0);
 516
 517         if (ib->length_dw != 0) {
 518                 amdgpu_vm_pad_ib(adev, ib);
 519                 amdgpu_sync_resv(adev, &ib->sync, pd->tbo.resv, AMDGPU_FENCE_OWNER_VM);
 520                 WARN_ON(ib->length_dw > ndw);
 521
 522                 if (amdgpu_enable_scheduler) {
 523                         int r;
 524                         uint64_t v_seq;
 525                         sched_job = amdgpu_cs_parser_create(adev, AMDGPU_FENCE_OWNER_VM,
 526                                                             adev->kernel_ctx,
 527                                                             ib, 1);
 528                         if(!sched_job)
 529                                 goto error_free;
 530                         sched_job->job_param.vm.bo = pd;
 531                         sched_job->run_job = amdgpu_vm_run_job;
 532                         sched_job->free_job = amdgpu_vm_free_job;
 533                         v_seq = atomic64_inc_return(&adev->kernel_ctx->rings[ring->idx].c_entity.last_queued_v_seq);
 534                         ib->sequence = v_seq;
 535                         amd_sched_push_job(ring->scheduler,
 536                                            &adev->kernel_ctx->rings[ring->idx].c_entity,
 537                                            sched_job);
 538                         r = amd_sched_wait_emit(&adev->kernel_ctx->rings[ring->idx].c_entity,
 539                                                 v_seq,
 540                                                 false,
 541                                                 -1);
 542                         if (r)
 543                                 DRM_ERROR("emit timeout\n");
 544                 } else {
 545                         r = amdgpu_ib_schedule(adev, 1, ib, AMDGPU_FENCE_OWNER_VM);
 546                         if (r) {
 547                                 amdgpu_ib_free(adev, ib);
 548                                 return r;
 549                         }
 550                         amdgpu_bo_fence(pd, ib->fence, true);
 551                 }
 552         }
 553
 554         if (!amdgpu_enable_scheduler || ib->length_dw == 0) {
 555                 amdgpu_ib_free(adev, ib);
 556                 kfree(ib);
 557         }
 558
 559         return 0;
 560
 561 error_free:
 562         if (sched_job)
 563                 kfree(sched_job);
 564         amdgpu_ib_free(adev, ib);
 565         kfree(ib);
 566         return -ENOMEM;
 567 }
 568
 569 /**
 570  * amdgpu_vm_frag_ptes - add fragment information to PTEs
 571  *
 572  * @adev: amdgpu_device pointer
 573  * @ib: IB for the update
 574  * @pe_start: first PTE to handle
 575  * @pe_end: last PTE to handle
 576  * @addr: addr those PTEs should point to
 577  * @flags: hw mapping flags
 578  * @gtt_flags: GTT hw mapping flags
 579  *
 580  * Global and local mutex must be locked!
 581  */
 582 static void amdgpu_vm_frag_ptes(struct amdgpu_device *adev,
 583                                 struct amdgpu_ib *ib,
 584                                 uint64_t pe_start, uint64_t pe_end,
 585                                 uint64_t addr, uint32_t flags,
 586                                 uint32_t gtt_flags)
 587 {
 588         /**
 589          * The MC L1 TLB supports variable sized pages, based on a fragment
 590          * field in the PTE. When this field is set to a non-zero value, page
 591          * granularity is increased from 4KB to (1 << (12 + frag)). The PTE
 592          * flags are considered valid for all PTEs within the fragment range
 593          * and corresponding mappings are assumed to be physically contiguous.
 594          *
 595          * The L1 TLB can store a single PTE for the whole fragment,
 596          * significantly increasing the space available for translation
 597          * caching. This leads to large improvements in throughput when the
 598          * TLB is under pressure.
 599          *
 600          * The L2 TLB distributes small and large fragments into two
 601          * asymmetric partitions. The large fragment cache is significantly
 602          * larger. Thus, we try to use large fragments wherever possible.
 603          * Userspace can support this by aligning virtual base address and
 604          * allocation size to the fragment size.
 605          */
 606
 607         /* SI and newer are optimized for 64KB */
 608         uint64_t frag_flags = AMDGPU_PTE_FRAG_64KB;
 609         uint64_t frag_align = 0x80;
 610
 611         uint64_t frag_start = ALIGN(pe_start, frag_align);
 612         uint64_t frag_end = pe_end & ~(frag_align - 1);
 613
 614         unsigned count;
 615
 616         /* system pages are non continuously */
 617         if ((flags & AMDGPU_PTE_SYSTEM) || !(flags & AMDGPU_PTE_VALID) ||
 618             (frag_start >= frag_end)) {
 619
 620                 count = (pe_end - pe_start) / 8;
 621                 amdgpu_vm_update_pages(adev, ib, pe_start, addr, count,
 622                                        AMDGPU_GPU_PAGE_SIZE, flags, gtt_flags);
 623                 return;
 624         }
 625
 626         /* handle the 4K area at the beginning */
 627         if (pe_start != frag_start) {
 628                 count = (frag_start - pe_start) / 8;
 629                 amdgpu_vm_update_pages(adev, ib, pe_start, addr, count,
 630                                        AMDGPU_GPU_PAGE_SIZE, flags, gtt_flags);
 631                 addr += AMDGPU_GPU_PAGE_SIZE * count;
 632         }
 633
 634         /* handle the area in the middle */
 635         count = (frag_end - frag_start) / 8;
 636         amdgpu_vm_update_pages(adev, ib, frag_start, addr, count,
 637                                AMDGPU_GPU_PAGE_SIZE, flags | frag_flags,
 638                                gtt_flags);
 639
 640         /* handle the 4K area at the end */
 641         if (frag_end != pe_end) {
 642                 addr += AMDGPU_GPU_PAGE_SIZE * count;
 643                 count = (pe_end - frag_end) / 8;
 644                 amdgpu_vm_update_pages(adev, ib, frag_end, addr, count,
 645                                        AMDGPU_GPU_PAGE_SIZE, flags, gtt_flags);
 646         }
 647 }
 648
 649 /**
 650  * amdgpu_vm_update_ptes - make sure that page tables are valid
 651  *
 652  * @adev: amdgpu_device pointer
 653  * @vm: requested vm
 654  * @start: start of GPU address range
 655  * @end: end of GPU address range
 656  * @dst: destination address to map to
 657  * @flags: mapping flags
 658  *
 659  * Update the page tables in the range @start - @end (cayman+).
 660  *
 661  * Global and local mutex must be locked!
 662  */
 663 static int amdgpu_vm_update_ptes(struct amdgpu_device *adev,
 664                                  struct amdgpu_vm *vm,
 665                                  struct amdgpu_ib *ib,
 666                                  uint64_t start, uint64_t end,
 667                                  uint64_t dst, uint32_t flags,
 668                                  uint32_t gtt_flags)
 669 {
 670         uint64_t mask = AMDGPU_VM_PTE_COUNT - 1;
 671         uint64_t last_pte = ~0, last_dst = ~0;
 672         unsigned count = 0;
 673         uint64_t addr;
 674
 675         /* walk over the address space and update the page tables */
 676         for (addr = start; addr < end; ) {
 677                 uint64_t pt_idx = addr >> amdgpu_vm_block_size;
 678                 struct amdgpu_bo *pt = vm->page_tables[pt_idx].bo;
 679                 unsigned nptes;
 680                 uint64_t pte;
 681                 int r;
 682
 683                 amdgpu_sync_resv(adev, &ib->sync, pt->tbo.resv,
 684                                  AMDGPU_FENCE_OWNER_VM);
 685                 r = reservation_object_reserve_shared(pt->tbo.resv);
 686                 if (r)
 687                         return r;
 688
 689                 if ((addr & ~mask) == (end & ~mask))
 690                         nptes = end - addr;
 691                 else
 692                         nptes = AMDGPU_VM_PTE_COUNT - (addr & mask);
 693
 694                 pte = amdgpu_bo_gpu_offset(pt);
 695                 pte += (addr & mask) * 8;
 696
 697                 if ((last_pte + 8 * count) != pte) {
 698
 699                         if (count) {
 700                                 amdgpu_vm_frag_ptes(adev, ib, last_pte,
 701                                                     last_pte + 8 * count,
 702                                                     last_dst, flags,
 703                                                     gtt_flags);
 704                         }
 705
 706                         count = nptes;
 707                         last_pte = pte;
 708                         last_dst = dst;
 709                 } else {
 710                         count += nptes;
 711                 }
 712
 713                 addr += nptes;
 714                 dst += nptes * AMDGPU_GPU_PAGE_SIZE;
 715         }
 716
 717         if (count) {
 718                 amdgpu_vm_frag_ptes(adev, ib, last_pte,
 719                                     last_pte + 8 * count,
 720                                     last_dst, flags, gtt_flags);
 721         }
 722
 723         return 0;
 724 }
 725
 726 /**
 727  * amdgpu_vm_fence_pts - fence page tables after an update
 728  *
 729  * @vm: requested vm
 730  * @start: start of GPU address range
 731  * @end: end of GPU address range
 732  * @fence: fence to use
 733  *
 734  * Fence the page tables in the range @start - @end (cayman+).
 735  *
 736  * Global and local mutex must be locked!
 737  */
 738 static void amdgpu_vm_fence_pts(struct amdgpu_vm *vm,
 739                                 uint64_t start, uint64_t end,
 740                                 struct amdgpu_fence *fence)
 741 {
 742         unsigned i;
 743
 744         start >>= amdgpu_vm_block_size;
 745         end >>= amdgpu_vm_block_size;
 746
 747         for (i = start; i <= end; ++i)
 748                 amdgpu_bo_fence(vm->page_tables[i].bo, fence, true);
 749 }
 750
 751 static int amdgpu_vm_bo_update_mapping_run_job(
 752         struct amdgpu_cs_parser *sched_job)
 753 {
 754         struct amdgpu_fence **fence = sched_job->job_param.vm_mapping.fence;
 755         amdgpu_vm_fence_pts(sched_job->job_param.vm_mapping.vm,
 756                             sched_job->job_param.vm_mapping.start,
 757                             sched_job->job_param.vm_mapping.last + 1,
 758                             sched_job->ibs[sched_job->num_ibs -1].fence);
 759         if (fence) {
 760                 amdgpu_fence_unref(fence);
 761                 *fence = amdgpu_fence_ref(sched_job->ibs[sched_job->num_ibs -1].fence);
 762         }
 763         return 0;
 764 }
 765 /**
 766  * amdgpu_vm_bo_update_mapping - update a mapping in the vm page table
 767  *
 768  * @adev: amdgpu_device pointer
 769  * @vm: requested vm
 770  * @mapping: mapped range and flags to use for the update
 771  * @addr: addr to set the area to
 772  * @gtt_flags: flags as they are used for GTT
 773  * @fence: optional resulting fence
 774  *
 775  * Fill in the page table entries for @mapping.
 776  * Returns 0 for success, -EINVAL for failure.
 777  *
 778  * Object have to be reserved and mutex must be locked!
 779  */
 780 static int amdgpu_vm_bo_update_mapping(struct amdgpu_device *adev,
 781                                        struct amdgpu_vm *vm,
 782                                        struct amdgpu_bo_va_mapping *mapping,
 783                                        uint64_t addr, uint32_t gtt_flags,
 784                                        struct amdgpu_fence **fence)
 785 {
 786         struct amdgpu_ring *ring = adev->vm_manager.vm_pte_funcs_ring;
 787         unsigned nptes, ncmds, ndw;
 788         uint32_t flags = gtt_flags;
 789         struct amdgpu_ib *ib;
 790         struct amdgpu_cs_parser *sched_job = NULL;
 791         int r;
 792
 793         /* normally,bo_va->flags only contians READABLE and WIRTEABLE bit go here
 794          * but in case of something, we filter the flags in first place
 795          */
 796         if (!(mapping->flags & AMDGPU_PTE_READABLE))
 797                 flags &= ~AMDGPU_PTE_READABLE;
 798         if (!(mapping->flags & AMDGPU_PTE_WRITEABLE))
 799                 flags &= ~AMDGPU_PTE_WRITEABLE;
 800
 801         trace_amdgpu_vm_bo_update(mapping);
 802
 803         nptes = mapping->it.last - mapping->it.start + 1;
 804
 805         /*
 806          * reserve space for one command every (1 << BLOCK_SIZE)
 807          *  entries or 2k dwords (whatever is smaller)
 808          */
 809         ncmds = (nptes >> min(amdgpu_vm_block_size, 11)) + 1;
 810
 811         /* padding, etc. */
 812         ndw = 64;
 813
 814         if ((flags & AMDGPU_PTE_SYSTEM) && (flags == gtt_flags)) {
 815                 /* only copy commands needed */
 816                 ndw += ncmds * 7;
 817
 818         } else if (flags & AMDGPU_PTE_SYSTEM) {
 819                 /* header for write data commands */
 820                 ndw += ncmds * 4;
 821
 822                 /* body of write data command */
 823                 ndw += nptes * 2;
 824
 825         } else {
 826                 /* set page commands needed */
 827                 ndw += ncmds * 10;
 828
 829                 /* two extra commands for begin/end of fragment */
 830                 ndw += 2 * 10;
 831         }
 832
 833         /* update too big for an IB */
 834         if (ndw > 0xfffff)
 835                 return -ENOMEM;
 836
 837         ib = kzalloc(sizeof(struct amdgpu_ib), GFP_KERNEL);
 838         if (!ib)
 839                 return -ENOMEM;
 840
 841         r = amdgpu_ib_get(ring, NULL, ndw * 4, ib);
 842         if (r) {
 843                 kfree(ib);
 844                 return r;
 845         }
 846
 847         ib->length_dw = 0;
 848
 849         if (!(flags & AMDGPU_PTE_VALID)) {
 850                 unsigned i;
 851
 852                 for (i = 0; i < AMDGPU_MAX_RINGS; ++i) {
 853                         struct amdgpu_fence *f = vm->ids[i].last_id_use;
 854                         r = amdgpu_sync_fence(adev, &ib->sync, &f->base);
 855                         if (r)
 856                                 return r;
 857                 }
 858         }
 859
 860         r = amdgpu_vm_update_ptes(adev, vm, ib, mapping->it.start,
 861                                   mapping->it.last + 1, addr + mapping->offset,
 862                                   flags, gtt_flags);
 863
 864         if (r) {
 865                 amdgpu_ib_free(adev, ib);
 866                 kfree(ib);
 867                 return r;
 868         }
 869
 870         amdgpu_vm_pad_ib(adev, ib);
 871         WARN_ON(ib->length_dw > ndw);
 872
 873         if (amdgpu_enable_scheduler) {
 874                 int r;
 875                 uint64_t v_seq;
 876                 sched_job = amdgpu_cs_parser_create(adev, AMDGPU_FENCE_OWNER_VM,
 877                                                     adev->kernel_ctx, ib, 1);
 878                 if(!sched_job)
 879                         goto error_free;
 880                 sched_job->job_param.vm_mapping.vm = vm;
 881                 sched_job->job_param.vm_mapping.start = mapping->it.start;
 882                 sched_job->job_param.vm_mapping.last = mapping->it.last;
 883                 sched_job->job_param.vm_mapping.fence = fence;
 884                 sched_job->run_job = amdgpu_vm_bo_update_mapping_run_job;
 885                 sched_job->free_job = amdgpu_vm_free_job;
 886                 v_seq = atomic64_inc_return(&adev->kernel_ctx->rings[ring->idx].c_entity.last_queued_v_seq);
 887                 ib->sequence = v_seq;
 888                 amd_sched_push_job(ring->scheduler,
 889                                    &adev->kernel_ctx->rings[ring->idx].c_entity,
 890                                    sched_job);
 891                 r = amd_sched_wait_emit(&adev->kernel_ctx->rings[ring->idx].c_entity,
 892                                         v_seq,
 893                                         false,
 894                                         -1);
 895                 if (r)
 896                         DRM_ERROR("emit timeout\n");
 897         } else {
 898                 r = amdgpu_ib_schedule(adev, 1, ib, AMDGPU_FENCE_OWNER_VM);
 899                 if (r) {
 900                         amdgpu_ib_free(adev, ib);
 901                         return r;
 902                 }
 903
 904                 amdgpu_vm_fence_pts(vm, mapping->it.start,
 905                                     mapping->it.last + 1, ib->fence);
 906                 if (fence) {
 907                         amdgpu_fence_unref(fence);
 908                         *fence = amdgpu_fence_ref(ib->fence);
 909                 }
 910
 911                 amdgpu_ib_free(adev, ib);
 912                 kfree(ib);
 913         }
 914         return 0;
 915
 916 error_free:
 917         if (sched_job)
 918                 kfree(sched_job);
 919         amdgpu_ib_free(adev, ib);
 920         kfree(ib);
 921         return -ENOMEM;
 922 }
 923
 924 /**
 925  * amdgpu_vm_bo_update - update all BO mappings in the vm page table
 926  *
 927  * @adev: amdgpu_device pointer
 928  * @bo_va: requested BO and VM object
 929  * @mem: ttm mem
 930  *
 931  * Fill in the page table entries for @bo_va.
 932  * Returns 0 for success, -EINVAL for failure.
 933  *
 934  * Object have to be reserved and mutex must be locked!
 935  */
 936 int amdgpu_vm_bo_update(struct amdgpu_device *adev,
 937                         struct amdgpu_bo_va *bo_va,
 938                         struct ttm_mem_reg *mem)
 939 {
 940         struct amdgpu_vm *vm = bo_va->vm;
 941         struct amdgpu_bo_va_mapping *mapping;
 942         uint32_t flags;
 943         uint64_t addr;
 944         int r;
 945
 946         if (mem) {
 947                 addr = mem->start << PAGE_SHIFT;
 948                 if (mem->mem_type != TTM_PL_TT)
 949                         addr += adev->vm_manager.vram_base_offset;
 950         } else {
 951                 addr = 0;
 952         }
 953
 954         if (addr == bo_va->addr)
 955                 return 0;
 956
 957         flags = amdgpu_ttm_tt_pte_flags(adev, bo_va->bo->tbo.ttm, mem);
 958
 959         list_for_each_entry(mapping, &bo_va->mappings, list) {
 960                 r = amdgpu_vm_bo_update_mapping(adev, vm, mapping, addr,
 961                                                 flags, &bo_va->last_pt_update);
 962                 if (r)
 963                         return r;
 964         }
 965
 966         bo_va->addr = addr;
 967         spin_lock(&vm->status_lock);
 968         list_del_init(&bo_va->vm_status);
 969         spin_unlock(&vm->status_lock);
 970
 971         return 0;
 972 }
 973
 974 /**
 975  * amdgpu_vm_clear_freed - clear freed BOs in the PT
 976  *
 977  * @adev: amdgpu_device pointer
 978  * @vm: requested vm
 979  *
 980  * Make sure all freed BOs are cleared in the PT.
 981  * Returns 0 for success.
 982  *
 983  * PTs have to be reserved and mutex must be locked!
 984  */
 985 int amdgpu_vm_clear_freed(struct amdgpu_device *adev,
 986                           struct amdgpu_vm *vm)
 987 {
 988         struct amdgpu_bo_va_mapping *mapping;
 989         int r;
 990
 991         while (!list_empty(&vm->freed)) {
 992                 mapping = list_first_entry(&vm->freed,
 993                         struct amdgpu_bo_va_mapping, list);
 994                 list_del(&mapping->list);
 995
 996                 r = amdgpu_vm_bo_update_mapping(adev, vm, mapping, 0, 0, NULL);
 997                 kfree(mapping);
 998                 if (r)
 999                         return r;
1000
1001         }
1002         return 0;
1003
1004 }
1005
1006 /**
1007  * amdgpu_vm_clear_invalids - clear invalidated BOs in the PT
1008  *
1009  * @adev: amdgpu_device pointer
1010  * @vm: requested vm
1011  *
1012  * Make sure all invalidated BOs are cleared in the PT.
1013  * Returns 0 for success.
1014  *
1015  * PTs have to be reserved and mutex must be locked!
1016  */
1017 int amdgpu_vm_clear_invalids(struct amdgpu_device *adev,
1018                              struct amdgpu_vm *vm, struct amdgpu_sync *sync)
1019 {
1020         struct amdgpu_bo_va *bo_va = NULL;
1021         int r = 0;
1022
1023         spin_lock(&vm->status_lock);
1024         while (!list_empty(&vm->invalidated)) {
1025                 bo_va = list_first_entry(&vm->invalidated,
1026                         struct amdgpu_bo_va, vm_status);
1027                 spin_unlock(&vm->status_lock);
1028
1029                 r = amdgpu_vm_bo_update(adev, bo_va, NULL);
1030                 if (r)
1031                         return r;
1032
1033                 spin_lock(&vm->status_lock);
1034         }
1035         spin_unlock(&vm->status_lock);
1036
1037         if (bo_va)
1038                 r = amdgpu_sync_fence(adev, sync, &bo_va->last_pt_update->base);
1039
1040         return r;
1041 }
1042
1043 /**
1044  * amdgpu_vm_bo_add - add a bo to a specific vm
1045  *
1046  * @adev: amdgpu_device pointer
1047  * @vm: requested vm
1048  * @bo: amdgpu buffer object
1049  *
1050  * Add @bo into the requested vm (cayman+).
1051  * Add @bo to the list of bos associated with the vm
1052  * Returns newly added bo_va or NULL for failure
1053  *
1054  * Object has to be reserved!
1055  */
1056 struct amdgpu_bo_va *amdgpu_vm_bo_add(struct amdgpu_device *adev,
1057                                       struct amdgpu_vm *vm,
1058                                       struct amdgpu_bo *bo)
1059 {
1060         struct amdgpu_bo_va *bo_va;
1061
1062         bo_va = kzalloc(sizeof(struct amdgpu_bo_va), GFP_KERNEL);
1063         if (bo_va == NULL) {
1064                 return NULL;
1065         }
1066         bo_va->vm = vm;
1067         bo_va->bo = bo;
1068         bo_va->addr = 0;
1069         bo_va->ref_count = 1;
1070         INIT_LIST_HEAD(&bo_va->bo_list);
1071         INIT_LIST_HEAD(&bo_va->mappings);
1072         INIT_LIST_HEAD(&bo_va->vm_status);
1073
1074         mutex_lock(&vm->mutex);
1075         list_add_tail(&bo_va->bo_list, &bo->va);
1076         mutex_unlock(&vm->mutex);
1077
1078         return bo_va;
1079 }
1080
1081 /**
1082  * amdgpu_vm_bo_map - map bo inside a vm
1083  *
1084  * @adev: amdgpu_device pointer
1085  * @bo_va: bo_va to store the address
1086  * @saddr: where to map the BO
1087  * @offset: requested offset in the BO
1088  * @flags: attributes of pages (read/write/valid/etc.)
1089  *
1090  * Add a mapping of the BO at the specefied addr into the VM.
1091  * Returns 0 for success, error for failure.
1092  *
1093  * Object has to be reserved and gets unreserved by this function!
1094  */
1095 int amdgpu_vm_bo_map(struct amdgpu_device *adev,
1096                      struct amdgpu_bo_va *bo_va,
1097                      uint64_t saddr, uint64_t offset,
1098                      uint64_t size, uint32_t flags)
1099 {
1100         struct amdgpu_bo_va_mapping *mapping;
1101         struct amdgpu_vm *vm = bo_va->vm;
1102         struct interval_tree_node *it;
1103         unsigned last_pfn, pt_idx;
1104         uint64_t eaddr;
1105         int r;
1106
1107         /* validate the parameters */
1108         if (saddr & AMDGPU_GPU_PAGE_MASK || offset & AMDGPU_GPU_PAGE_MASK ||
1109             size == 0 || size & AMDGPU_GPU_PAGE_MASK) {
1110                 amdgpu_bo_unreserve(bo_va->bo);
1111                 return -EINVAL;
1112         }
1113
1114         /* make sure object fit at this offset */
1115         eaddr = saddr + size;
1116         if ((saddr >= eaddr) || (offset + size > amdgpu_bo_size(bo_va->bo))) {
1117                 amdgpu_bo_unreserve(bo_va->bo);
1118                 return -EINVAL;
1119         }
1120
1121         last_pfn = eaddr / AMDGPU_GPU_PAGE_SIZE;
1122         if (last_pfn > adev->vm_manager.max_pfn) {
1123                 dev_err(adev->dev, "va above limit (0x%08X > 0x%08X)\n",
1124                         last_pfn, adev->vm_manager.max_pfn);
1125                 amdgpu_bo_unreserve(bo_va->bo);
1126                 return -EINVAL;
1127         }
1128
1129         mutex_lock(&vm->mutex);
1130
1131         saddr /= AMDGPU_GPU_PAGE_SIZE;
1132         eaddr /= AMDGPU_GPU_PAGE_SIZE;
1133
1134         it = interval_tree_iter_first(&vm->va, saddr, eaddr - 1);
1135         if (it) {
1136                 struct amdgpu_bo_va_mapping *tmp;
1137                 tmp = container_of(it, struct amdgpu_bo_va_mapping, it);
1138                 /* bo and tmp overlap, invalid addr */
1139                 dev_err(adev->dev, "bo %p va 0x%010Lx-0x%010Lx conflict with "
1140                         "0x%010lx-0x%010lx\n", bo_va->bo, saddr, eaddr,
1141                         tmp->it.start, tmp->it.last + 1);
1142                 amdgpu_bo_unreserve(bo_va->bo);
1143                 r = -EINVAL;
1144                 goto error_unlock;
1145         }
1146
1147         mapping = kmalloc(sizeof(*mapping), GFP_KERNEL);
1148         if (!mapping) {
1149                 amdgpu_bo_unreserve(bo_va->bo);
1150                 r = -ENOMEM;
1151                 goto error_unlock;
1152         }
1153
1154         INIT_LIST_HEAD(&mapping->list);
1155         mapping->it.start = saddr;
1156         mapping->it.last = eaddr - 1;
1157         mapping->offset = offset;
1158         mapping->flags = flags;
1159
1160         list_add(&mapping->list, &bo_va->mappings);
1161         interval_tree_insert(&mapping->it, &vm->va);
1162         trace_amdgpu_vm_bo_map(bo_va, mapping);
1163
1164         bo_va->addr = 0;
1165
1166         /* Make sure the page tables are allocated */
1167         saddr >>= amdgpu_vm_block_size;
1168         eaddr >>= amdgpu_vm_block_size;
1169
1170         BUG_ON(eaddr >= amdgpu_vm_num_pdes(adev));
1171
1172         if (eaddr > vm->max_pde_used)
1173                 vm->max_pde_used = eaddr;
1174
1175         amdgpu_bo_unreserve(bo_va->bo);
1176
1177         /* walk over the address space and allocate the page tables */
1178         for (pt_idx = saddr; pt_idx <= eaddr; ++pt_idx) {
1179                 struct amdgpu_bo *pt;
1180
1181                 if (vm->page_tables[pt_idx].bo)
1182                         continue;
1183
1184                 /* drop mutex to allocate and clear page table */
1185                 mutex_unlock(&vm->mutex);
1186
1187                 r = amdgpu_bo_create(adev, AMDGPU_VM_PTE_COUNT * 8,
1188                                      AMDGPU_GPU_PAGE_SIZE, true,
1189                                      AMDGPU_GEM_DOMAIN_VRAM, 0, NULL, &pt);
1190                 if (r)
1191                         goto error_free;
1192
1193                 r = amdgpu_vm_clear_bo(adev, pt);
1194                 if (r) {
1195                         amdgpu_bo_unref(&pt);
1196                         goto error_free;
1197                 }
1198
1199                 /* aquire mutex again */
1200                 mutex_lock(&vm->mutex);
1201                 if (vm->page_tables[pt_idx].bo) {
1202                         /* someone else allocated the pt in the meantime */
1203                         mutex_unlock(&vm->mutex);
1204                         amdgpu_bo_unref(&pt);
1205                         mutex_lock(&vm->mutex);
1206                         continue;
1207                 }
1208
1209                 vm->page_tables[pt_idx].addr = 0;
1210                 vm->page_tables[pt_idx].bo = pt;
1211         }
1212
1213         mutex_unlock(&vm->mutex);
1214         return 0;
1215
1216 error_free:
1217         mutex_lock(&vm->mutex);
1218         list_del(&mapping->list);
1219         interval_tree_remove(&mapping->it, &vm->va);
1220         trace_amdgpu_vm_bo_unmap(bo_va, mapping);
1221         kfree(mapping);
1222
1223 error_unlock:
1224         mutex_unlock(&vm->mutex);
1225         return r;
1226 }
1227
1228 /**
1229  * amdgpu_vm_bo_unmap - remove bo mapping from vm
1230  *
1231  * @adev: amdgpu_device pointer
1232  * @bo_va: bo_va to remove the address from
1233  * @saddr: where to the BO is mapped
1234  *
1235  * Remove a mapping of the BO at the specefied addr from the VM.
1236  * Returns 0 for success, error for failure.
1237  *
1238  * Object has to be reserved and gets unreserved by this function!
1239  */
1240 int amdgpu_vm_bo_unmap(struct amdgpu_device *adev,
1241                        struct amdgpu_bo_va *bo_va,
1242                        uint64_t saddr)
1243 {
1244         struct amdgpu_bo_va_mapping *mapping;
1245         struct amdgpu_vm *vm = bo_va->vm;
1246
1247         saddr /= AMDGPU_GPU_PAGE_SIZE;
1248
1249         list_for_each_entry(mapping, &bo_va->mappings, list) {
1250                 if (mapping->it.start == saddr)
1251                         break;
1252         }
1253
1254         if (&mapping->list == &bo_va->mappings) {
1255                 amdgpu_bo_unreserve(bo_va->bo);
1256                 return -ENOENT;
1257         }
1258
1259         mutex_lock(&vm->mutex);
1260         list_del(&mapping->list);
1261         interval_tree_remove(&mapping->it, &vm->va);
1262         trace_amdgpu_vm_bo_unmap(bo_va, mapping);
1263
1264         if (bo_va->addr) {
1265                 /* clear the old address */
1266                 list_add(&mapping->list, &vm->freed);
1267         } else {
1268                 kfree(mapping);
1269         }
1270         mutex_unlock(&vm->mutex);
1271         amdgpu_bo_unreserve(bo_va->bo);
1272
1273         return 0;
1274 }
1275
1276 /**
1277  * amdgpu_vm_bo_rmv - remove a bo to a specific vm
1278  *
1279  * @adev: amdgpu_device pointer
1280  * @bo_va: requested bo_va
1281  *
1282  * Remove @bo_va->bo from the requested vm (cayman+).
1283  *
1284  * Object have to be reserved!
1285  */
1286 void amdgpu_vm_bo_rmv(struct amdgpu_device *adev,
1287                       struct amdgpu_bo_va *bo_va)
1288 {
1289         struct amdgpu_bo_va_mapping *mapping, *next;
1290         struct amdgpu_vm *vm = bo_va->vm;
1291
1292         list_del(&bo_va->bo_list);
1293
1294         mutex_lock(&vm->mutex);
1295
1296         spin_lock(&vm->status_lock);
1297         list_del(&bo_va->vm_status);
1298         spin_unlock(&vm->status_lock);
1299
1300         list_for_each_entry_safe(mapping, next, &bo_va->mappings, list) {
1301                 list_del(&mapping->list);
1302                 interval_tree_remove(&mapping->it, &vm->va);
1303                 trace_amdgpu_vm_bo_unmap(bo_va, mapping);
1304                 if (bo_va->addr)
1305                         list_add(&mapping->list, &vm->freed);
1306                 else
1307                         kfree(mapping);
1308         }
1309         amdgpu_fence_unref(&bo_va->last_pt_update);
1310         kfree(bo_va);
1311
1312         mutex_unlock(&vm->mutex);
1313 }
1314
1315 /**
1316  * amdgpu_vm_bo_invalidate - mark the bo as invalid
1317  *
1318  * @adev: amdgpu_device pointer
1319  * @vm: requested vm
1320  * @bo: amdgpu buffer object
1321  *
1322  * Mark @bo as invalid (cayman+).
1323  */
1324 void amdgpu_vm_bo_invalidate(struct amdgpu_device *adev,
1325                              struct amdgpu_bo *bo)
1326 {
1327         struct amdgpu_bo_va *bo_va;
1328
1329         list_for_each_entry(bo_va, &bo->va, bo_list) {
1330                 if (bo_va->addr) {
1331                         spin_lock(&bo_va->vm->status_lock);
1332                         list_del(&bo_va->vm_status);
1333                         list_add(&bo_va->vm_status, &bo_va->vm->invalidated);
1334                         spin_unlock(&bo_va->vm->status_lock);
1335                 }
1336         }
1337 }
1338
1339 /**
1340  * amdgpu_vm_init - initialize a vm instance
1341  *
1342  * @adev: amdgpu_device pointer
1343  * @vm: requested vm
1344  *
1345  * Init @vm fields (cayman+).
1346  */
1347 int amdgpu_vm_init(struct amdgpu_device *adev, struct amdgpu_vm *vm)
1348 {
1349         const unsigned align = min(AMDGPU_VM_PTB_ALIGN_SIZE,
1350                 AMDGPU_VM_PTE_COUNT * 8);
1351         unsigned pd_size, pd_entries, pts_size;
1352         int i, r;
1353
1354         for (i = 0; i < AMDGPU_MAX_RINGS; ++i) {
1355                 vm->ids[i].id = 0;
1356                 vm->ids[i].flushed_updates = NULL;
1357                 vm->ids[i].last_id_use = NULL;
1358         }
1359         mutex_init(&vm->mutex);
1360         vm->va = RB_ROOT;
1361         spin_lock_init(&vm->status_lock);
1362         INIT_LIST_HEAD(&vm->invalidated);
1363         INIT_LIST_HEAD(&vm->freed);
1364
1365         pd_size = amdgpu_vm_directory_size(adev);
1366         pd_entries = amdgpu_vm_num_pdes(adev);
1367
1368         /* allocate page table array */
1369         pts_size = pd_entries * sizeof(struct amdgpu_vm_pt);
1370         vm->page_tables = kzalloc(pts_size, GFP_KERNEL);
1371         if (vm->page_tables == NULL) {
1372                 DRM_ERROR("Cannot allocate memory for page table array\n");
1373                 return -ENOMEM;
1374         }
1375
1376         r = amdgpu_bo_create(adev, pd_size, align, true,
1377                              AMDGPU_GEM_DOMAIN_VRAM, 0,
1378                              NULL, &vm->page_directory);
1379         if (r)
1380                 return r;
1381
1382         r = amdgpu_vm_clear_bo(adev, vm->page_directory);
1383         if (r) {
1384                 amdgpu_bo_unref(&vm->page_directory);
1385                 vm->page_directory = NULL;
1386                 return r;
1387         }
1388
1389         return 0;
1390 }
1391
1392 /**
1393  * amdgpu_vm_fini - tear down a vm instance
1394  *
1395  * @adev: amdgpu_device pointer
1396  * @vm: requested vm
1397  *
1398  * Tear down @vm (cayman+).
1399  * Unbind the VM and remove all bos from the vm bo list
1400  */
1401 void amdgpu_vm_fini(struct amdgpu_device *adev, struct amdgpu_vm *vm)
1402 {
1403         struct amdgpu_bo_va_mapping *mapping, *tmp;
1404         int i;
1405
1406         if (!RB_EMPTY_ROOT(&vm->va)) {
1407                 dev_err(adev->dev, "still active bo inside vm\n");
1408         }
1409         rbtree_postorder_for_each_entry_safe(mapping, tmp, &vm->va, it.rb) {
1410                 list_del(&mapping->list);
1411                 interval_tree_remove(&mapping->it, &vm->va);
1412                 kfree(mapping);
1413         }
1414         list_for_each_entry_safe(mapping, tmp, &vm->freed, list) {
1415                 list_del(&mapping->list);
1416                 kfree(mapping);
1417         }
1418
1419         for (i = 0; i < amdgpu_vm_num_pdes(adev); i++)
1420                 amdgpu_bo_unref(&vm->page_tables[i].bo);
1421         kfree(vm->page_tables);
1422
1423         amdgpu_bo_unref(&vm->page_directory);
1424
1425         for (i = 0; i < AMDGPU_MAX_RINGS; ++i) {
1426                 amdgpu_fence_unref(&vm->ids[i].flushed_updates);
1427                 amdgpu_fence_unref(&vm->ids[i].last_id_use);
1428         }
1429
1430         mutex_destroy(&vm->mutex);
1431 }