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

   1 /*
   2  * Copyright 2009 Jerome Glisse.
   3  * All Rights Reserved.
   4  *
   5  * Permission is hereby granted, free of charge, to any person obtaining a
   6  * copy of this software and associated documentation files (the
   7  * "Software"), to deal in the Software without restriction, including
   8  * without limitation the rights to use, copy, modify, merge, publish,
   9  * distribute, sub license, and/or sell copies of the Software, and to
  10  * permit persons to whom the Software is furnished to do so, subject to
  11  * the following conditions:
  12  *
  13  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  14  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  15  * FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT. IN NO EVENT SHALL
  16  * THE COPYRIGHT HOLDERS, AUTHORS AND/OR ITS SUPPLIERS BE LIABLE FOR ANY CLAIM,
  17  * DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR
  18  * OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE
  19  * USE OR OTHER DEALINGS IN THE SOFTWARE.
  20  *
  21  * The above copyright notice and this permission notice (including the
  22  * next paragraph) shall be included in all copies or substantial portions
  23  * of the Software.
  24  *
  25  */
  26 /*
  27  * Authors:
  28  *    Jerome Glisse <glisse@freedesktop.org>
  29  *    Thomas Hellstrom <thomas-at-tungstengraphics-dot-com>
  30  *    Dave Airlie
  31  */
  32 #include <linux/list.h>
  33 #include <linux/slab.h>
  34 #include <drm/drmP.h>
  35 #include <drm/amdgpu_drm.h>
  36 #include <drm/drm_cache.h>
  37 #include "amdgpu.h"
  38 #include "amdgpu_trace.h"
  39
  40
  41
  42 static u64 amdgpu_get_vis_part_size(struct amdgpu_device *adev,
  43                                                 struct ttm_mem_reg *mem)
  44 {
  45         if (mem->start << PAGE_SHIFT >= adev->mc.visible_vram_size)
  46                 return 0;
  47
  48         return ((mem->start << PAGE_SHIFT) + mem->size) >
  49                 adev->mc.visible_vram_size ?
  50                 adev->mc.visible_vram_size - (mem->start << PAGE_SHIFT) :
  51                 mem->size;
  52 }
  53
  54 static void amdgpu_update_memory_usage(struct amdgpu_device *adev,
  55                        struct ttm_mem_reg *old_mem,
  56                        struct ttm_mem_reg *new_mem)
  57 {
  58         u64 vis_size;
  59         if (!adev)
  60                 return;
  61
  62         if (new_mem) {
  63                 switch (new_mem->mem_type) {
  64                 case TTM_PL_TT:
  65                         atomic64_add(new_mem->size, &adev->gtt_usage);
  66                         break;
  67                 case TTM_PL_VRAM:
  68                         atomic64_add(new_mem->size, &adev->vram_usage);
  69                         vis_size = amdgpu_get_vis_part_size(adev, new_mem);
  70                         atomic64_add(vis_size, &adev->vram_vis_usage);
  71                         break;
  72                 }
  73         }
  74
  75         if (old_mem) {
  76                 switch (old_mem->mem_type) {
  77                 case TTM_PL_TT:
  78                         atomic64_sub(old_mem->size, &adev->gtt_usage);
  79                         break;
  80                 case TTM_PL_VRAM:
  81                         atomic64_sub(old_mem->size, &adev->vram_usage);
  82                         vis_size = amdgpu_get_vis_part_size(adev, old_mem);
  83                         atomic64_sub(vis_size, &adev->vram_vis_usage);
  84                         break;
  85                 }
  86         }
  87 }
  88
  89 static void amdgpu_ttm_bo_destroy(struct ttm_buffer_object *tbo)
  90 {
  91         struct amdgpu_bo *bo;
  92
  93         bo = container_of(tbo, struct amdgpu_bo, tbo);
  94
  95         amdgpu_update_memory_usage(bo->adev, &bo->tbo.mem, NULL);
  96
  97         drm_gem_object_release(&bo->gem_base);
  98         amdgpu_bo_unref(&bo->parent);
  99         if (!list_empty(&bo->shadow_list)) {
 100                 mutex_lock(&bo->adev->shadow_list_lock);
 101                 list_del_init(&bo->shadow_list);
 102                 mutex_unlock(&bo->adev->shadow_list_lock);
 103         }
 104         kfree(bo->metadata);
 105         kfree(bo);
 106 }
 107
 108 bool amdgpu_ttm_bo_is_amdgpu_bo(struct ttm_buffer_object *bo)
 109 {
 110         if (bo->destroy == &amdgpu_ttm_bo_destroy)
 111                 return true;
 112         return false;
 113 }
 114
 115 static void amdgpu_ttm_placement_init(struct amdgpu_device *adev,
 116                                       struct ttm_placement *placement,
 117                                       struct ttm_place *places,
 118                                       u32 domain, u64 flags)
 119 {
 120         u32 c = 0;
 121
 122         if (domain & AMDGPU_GEM_DOMAIN_VRAM) {
 123                 unsigned visible_pfn = adev->mc.visible_vram_size >> PAGE_SHIFT;
 124
 125                 if (flags & AMDGPU_GEM_CREATE_NO_CPU_ACCESS &&
 126                     !(flags & AMDGPU_GEM_CREATE_CPU_ACCESS_REQUIRED) &&
 127                     adev->mc.visible_vram_size < adev->mc.real_vram_size) {
 128                         places[c].fpfn = visible_pfn;
 129                         places[c].lpfn = 0;
 130                         places[c].flags = TTM_PL_FLAG_WC |
 131                                 TTM_PL_FLAG_UNCACHED | TTM_PL_FLAG_VRAM |
 132                                 TTM_PL_FLAG_TOPDOWN;
 133                         c++;
 134                 }
 135
 136                 places[c].fpfn = 0;
 137                 places[c].lpfn = 0;
 138                 places[c].flags = TTM_PL_FLAG_WC | TTM_PL_FLAG_UNCACHED |
 139                         TTM_PL_FLAG_VRAM;
 140                 if (flags & AMDGPU_GEM_CREATE_CPU_ACCESS_REQUIRED)
 141                         places[c].lpfn = visible_pfn;
 142                 else
 143                         places[c].flags |= TTM_PL_FLAG_TOPDOWN;
 144                 c++;
 145         }
 146
 147         if (domain & AMDGPU_GEM_DOMAIN_GTT) {
 148                 places[c].fpfn = 0;
 149                 places[c].lpfn = 0;
 150                 places[c].flags = TTM_PL_FLAG_TT;
 151                 if (flags & AMDGPU_GEM_CREATE_CPU_GTT_USWC)
 152                         places[c].flags |= TTM_PL_FLAG_WC |
 153                                 TTM_PL_FLAG_UNCACHED;
 154                 else
 155                         places[c].flags |= TTM_PL_FLAG_CACHED;
 156                 c++;
 157         }
 158
 159         if (domain & AMDGPU_GEM_DOMAIN_CPU) {
 160                 places[c].fpfn = 0;
 161                 places[c].lpfn = 0;
 162                 places[c].flags = TTM_PL_FLAG_SYSTEM;
 163                 if (flags & AMDGPU_GEM_CREATE_CPU_GTT_USWC)
 164                         places[c].flags |= TTM_PL_FLAG_WC |
 165                                 TTM_PL_FLAG_UNCACHED;
 166                 else
 167                         places[c].flags |= TTM_PL_FLAG_CACHED;
 168                 c++;
 169         }
 170
 171         if (domain & AMDGPU_GEM_DOMAIN_GDS) {
 172                 places[c].fpfn = 0;
 173                 places[c].lpfn = 0;
 174                 places[c].flags = TTM_PL_FLAG_UNCACHED | AMDGPU_PL_FLAG_GDS;
 175                 c++;
 176         }
 177
 178         if (domain & AMDGPU_GEM_DOMAIN_GWS) {
 179                 places[c].fpfn = 0;
 180                 places[c].lpfn = 0;
 181                 places[c].flags = TTM_PL_FLAG_UNCACHED | AMDGPU_PL_FLAG_GWS;
 182                 c++;
 183         }
 184
 185         if (domain & AMDGPU_GEM_DOMAIN_OA) {
 186                 places[c].fpfn = 0;
 187                 places[c].lpfn = 0;
 188                 places[c].flags = TTM_PL_FLAG_UNCACHED | AMDGPU_PL_FLAG_OA;
 189                 c++;
 190         }
 191
 192         if (!c) {
 193                 places[c].fpfn = 0;
 194                 places[c].lpfn = 0;
 195                 places[c].flags = TTM_PL_MASK_CACHING | TTM_PL_FLAG_SYSTEM;
 196                 c++;
 197         }
 198
 199         placement->num_placement = c;
 200         placement->placement = places;
 201
 202         placement->num_busy_placement = c;
 203         placement->busy_placement = places;
 204 }
 205
 206 void amdgpu_ttm_placement_from_domain(struct amdgpu_bo *abo, u32 domain)
 207 {
 208         amdgpu_ttm_placement_init(abo->adev, &abo->placement,
 209                                   abo->placements, domain, abo->flags);
 210 }
 211
 212 static void amdgpu_fill_placement_to_bo(struct amdgpu_bo *bo,
 213                                         struct ttm_placement *placement)
 214 {
 215         BUG_ON(placement->num_placement > (AMDGPU_GEM_DOMAIN_MAX + 1));
 216
 217         memcpy(bo->placements, placement->placement,
 218                placement->num_placement * sizeof(struct ttm_place));
 219         bo->placement.num_placement = placement->num_placement;
 220         bo->placement.num_busy_placement = placement->num_busy_placement;
 221         bo->placement.placement = bo->placements;
 222         bo->placement.busy_placement = bo->placements;
 223 }
 224
 225 /**
 226  * amdgpu_bo_create_kernel - create BO for kernel use
 227  *
 228  * @adev: amdgpu device object
 229  * @size: size for the new BO
 230  * @align: alignment for the new BO
 231  * @domain: where to place it
 232  * @bo_ptr: resulting BO
 233  * @gpu_addr: GPU addr of the pinned BO
 234  * @cpu_addr: optional CPU address mapping
 235  *
 236  * Allocates and pins a BO for kernel internal use.
 237  *
 238  * Returns 0 on success, negative error code otherwise.
 239  */
 240 int amdgpu_bo_create_kernel(struct amdgpu_device *adev,
 241                             unsigned long size, int align,
 242                             u32 domain, struct amdgpu_bo **bo_ptr,
 243                             u64 *gpu_addr, void **cpu_addr)
 244 {
 245         int r;
 246
 247         r = amdgpu_bo_create(adev, size, align, true, domain,
 248                              AMDGPU_GEM_CREATE_CPU_ACCESS_REQUIRED,
 249                              NULL, NULL, bo_ptr);
 250         if (r) {
 251                 dev_err(adev->dev, "(%d) failed to allocate kernel bo\n", r);
 252                 return r;
 253         }
 254
 255         r = amdgpu_bo_reserve(*bo_ptr, false);
 256         if (r) {
 257                 dev_err(adev->dev, "(%d) failed to reserve kernel bo\n", r);
 258                 goto error_free;
 259         }
 260
 261         r = amdgpu_bo_pin(*bo_ptr, domain, gpu_addr);
 262         if (r) {
 263                 dev_err(adev->dev, "(%d) kernel bo pin failed\n", r);
 264                 goto error_unreserve;
 265         }
 266
 267         if (cpu_addr) {
 268                 r = amdgpu_bo_kmap(*bo_ptr, cpu_addr);
 269                 if (r) {
 270                         dev_err(adev->dev, "(%d) kernel bo map failed\n", r);
 271                         goto error_unreserve;
 272                 }
 273         }
 274
 275         amdgpu_bo_unreserve(*bo_ptr);
 276
 277         return 0;
 278
 279 error_unreserve:
 280         amdgpu_bo_unreserve(*bo_ptr);
 281
 282 error_free:
 283         amdgpu_bo_unref(bo_ptr);
 284
 285         return r;
 286 }
 287
 288 /**
 289  * amdgpu_bo_free_kernel - free BO for kernel use
 290  *
 291  * @bo: amdgpu BO to free
 292  *
 293  * unmaps and unpin a BO for kernel internal use.
 294  */
 295 void amdgpu_bo_free_kernel(struct amdgpu_bo **bo, u64 *gpu_addr,
 296                            void **cpu_addr)
 297 {
 298         if (*bo == NULL)
 299                 return;
 300
 301         if (likely(amdgpu_bo_reserve(*bo, false) == 0)) {
 302                 if (cpu_addr)
 303                         amdgpu_bo_kunmap(*bo);
 304
 305                 amdgpu_bo_unpin(*bo);
 306                 amdgpu_bo_unreserve(*bo);
 307         }
 308         amdgpu_bo_unref(bo);
 309
 310         if (gpu_addr)
 311                 *gpu_addr = 0;
 312
 313         if (cpu_addr)
 314                 *cpu_addr = NULL;
 315 }
 316
 317 int amdgpu_bo_create_restricted(struct amdgpu_device *adev,
 318                                 unsigned long size, int byte_align,
 319                                 bool kernel, u32 domain, u64 flags,
 320                                 struct sg_table *sg,
 321                                 struct ttm_placement *placement,
 322                                 struct reservation_object *resv,
 323                                 struct amdgpu_bo **bo_ptr)
 324 {
 325         struct amdgpu_bo *bo;
 326         enum ttm_bo_type type;
 327         unsigned long page_align;
 328         size_t acc_size;
 329         int r;
 330
 331         page_align = roundup(byte_align, PAGE_SIZE) >> PAGE_SHIFT;
 332         size = ALIGN(size, PAGE_SIZE);
 333
 334         if (kernel) {
 335                 type = ttm_bo_type_kernel;
 336         } else if (sg) {
 337                 type = ttm_bo_type_sg;
 338         } else {
 339                 type = ttm_bo_type_device;
 340         }
 341         *bo_ptr = NULL;
 342
 343         acc_size = ttm_bo_dma_acc_size(&adev->mman.bdev, size,
 344                                        sizeof(struct amdgpu_bo));
 345
 346         bo = kzalloc(sizeof(struct amdgpu_bo), GFP_KERNEL);
 347         if (bo == NULL)
 348                 return -ENOMEM;
 349         r = drm_gem_object_init(adev->ddev, &bo->gem_base, size);
 350         if (unlikely(r)) {
 351                 kfree(bo);
 352                 return r;
 353         }
 354         bo->adev = adev;
 355         INIT_LIST_HEAD(&bo->shadow_list);
 356         INIT_LIST_HEAD(&bo->va);
 357         bo->prefered_domains = domain & (AMDGPU_GEM_DOMAIN_VRAM |
 358                                          AMDGPU_GEM_DOMAIN_GTT |
 359                                          AMDGPU_GEM_DOMAIN_CPU |
 360                                          AMDGPU_GEM_DOMAIN_GDS |
 361                                          AMDGPU_GEM_DOMAIN_GWS |
 362                                          AMDGPU_GEM_DOMAIN_OA);
 363         bo->allowed_domains = bo->prefered_domains;
 364         if (!kernel && bo->allowed_domains == AMDGPU_GEM_DOMAIN_VRAM)
 365                 bo->allowed_domains |= AMDGPU_GEM_DOMAIN_GTT;
 366
 367         bo->flags = flags;
 368
 369         /* For architectures that don't support WC memory,
 370          * mask out the WC flag from the BO
 371          */
 372         if (!drm_arch_can_wc_memory())
 373                 bo->flags &= ~AMDGPU_GEM_CREATE_CPU_GTT_USWC;
 374
 375         amdgpu_fill_placement_to_bo(bo, placement);
 376         /* Kernel allocation are uninterruptible */
 377         r = ttm_bo_init(&adev->mman.bdev, &bo->tbo, size, type,
 378                         &bo->placement, page_align, !kernel, NULL,
 379                         acc_size, sg, resv, &amdgpu_ttm_bo_destroy);
 380         if (unlikely(r != 0)) {
 381                 return r;
 382         }
 383
 384         if (flags & AMDGPU_GEM_CREATE_VRAM_CLEARED &&
 385             bo->tbo.mem.placement & TTM_PL_FLAG_VRAM) {
 386                 struct fence *fence;
 387
 388                 if (adev->mman.buffer_funcs_ring == NULL ||
 389                    !adev->mman.buffer_funcs_ring->ready) {
 390                         r = -EBUSY;
 391                         goto fail_free;
 392                 }
 393
 394                 r = amdgpu_bo_reserve(bo, false);
 395                 if (unlikely(r != 0))
 396                         goto fail_free;
 397
 398                 amdgpu_ttm_placement_from_domain(bo, AMDGPU_GEM_DOMAIN_VRAM);
 399                 r = ttm_bo_validate(&bo->tbo, &bo->placement, false, false);
 400                 if (unlikely(r != 0))
 401                         goto fail_unreserve;
 402
 403                 amdgpu_fill_buffer(bo, 0, bo->tbo.resv, &fence);
 404                 amdgpu_bo_fence(bo, fence, false);
 405                 amdgpu_bo_unreserve(bo);
 406                 fence_put(bo->tbo.moving);
 407                 bo->tbo.moving = fence_get(fence);
 408                 fence_put(fence);
 409         }
 410         *bo_ptr = bo;
 411
 412         trace_amdgpu_bo_create(bo);
 413
 414         return 0;
 415
 416 fail_unreserve:
 417         amdgpu_bo_unreserve(bo);
 418 fail_free:
 419         amdgpu_bo_unref(&bo);
 420         return r;
 421 }
 422
 423 static int amdgpu_bo_create_shadow(struct amdgpu_device *adev,
 424                                    unsigned long size, int byte_align,
 425                                    struct amdgpu_bo *bo)
 426 {
 427         struct ttm_placement placement = {0};
 428         struct ttm_place placements[AMDGPU_GEM_DOMAIN_MAX + 1];
 429         int r;
 430
 431         if (bo->shadow)
 432                 return 0;
 433
 434         bo->flags |= AMDGPU_GEM_CREATE_SHADOW;
 435         memset(&placements, 0,
 436                (AMDGPU_GEM_DOMAIN_MAX + 1) * sizeof(struct ttm_place));
 437
 438         amdgpu_ttm_placement_init(adev, &placement,
 439                                   placements, AMDGPU_GEM_DOMAIN_GTT,
 440                                   AMDGPU_GEM_CREATE_CPU_GTT_USWC);
 441
 442         r = amdgpu_bo_create_restricted(adev, size, byte_align, true,
 443                                         AMDGPU_GEM_DOMAIN_GTT,
 444                                         AMDGPU_GEM_CREATE_CPU_GTT_USWC,
 445                                         NULL, &placement,
 446                                         bo->tbo.resv,
 447                                         &bo->shadow);
 448         if (!r) {
 449                 bo->shadow->parent = amdgpu_bo_ref(bo);
 450                 mutex_lock(&adev->shadow_list_lock);
 451                 list_add_tail(&bo->shadow_list, &adev->shadow_list);
 452                 mutex_unlock(&adev->shadow_list_lock);
 453         }
 454
 455         return r;
 456 }
 457
 458 int amdgpu_bo_create(struct amdgpu_device *adev,
 459                      unsigned long size, int byte_align,
 460                      bool kernel, u32 domain, u64 flags,
 461                      struct sg_table *sg,
 462                      struct reservation_object *resv,
 463                      struct amdgpu_bo **bo_ptr)
 464 {
 465         struct ttm_placement placement = {0};
 466         struct ttm_place placements[AMDGPU_GEM_DOMAIN_MAX + 1];
 467         int r;
 468
 469         memset(&placements, 0,
 470                (AMDGPU_GEM_DOMAIN_MAX + 1) * sizeof(struct ttm_place));
 471
 472         amdgpu_ttm_placement_init(adev, &placement,
 473                                   placements, domain, flags);
 474
 475         r = amdgpu_bo_create_restricted(adev, size, byte_align, kernel,
 476                                         domain, flags, sg, &placement,
 477                                         resv, bo_ptr);
 478         if (r)
 479                 return r;
 480
 481         if (amdgpu_need_backup(adev) && (flags & AMDGPU_GEM_CREATE_SHADOW)) {
 482                 r = amdgpu_bo_create_shadow(adev, size, byte_align, (*bo_ptr));
 483                 if (r)
 484                         amdgpu_bo_unref(bo_ptr);
 485         }
 486
 487         return r;
 488 }
 489
 490 int amdgpu_bo_backup_to_shadow(struct amdgpu_device *adev,
 491                                struct amdgpu_ring *ring,
 492                                struct amdgpu_bo *bo,
 493                                struct reservation_object *resv,
 494                                struct fence **fence,
 495                                bool direct)
 496
 497 {
 498         struct amdgpu_bo *shadow = bo->shadow;
 499         uint64_t bo_addr, shadow_addr;
 500         int r;
 501
 502         if (!shadow)
 503                 return -EINVAL;
 504
 505         bo_addr = amdgpu_bo_gpu_offset(bo);
 506         shadow_addr = amdgpu_bo_gpu_offset(bo->shadow);
 507
 508         r = reservation_object_reserve_shared(bo->tbo.resv);
 509         if (r)
 510                 goto err;
 511
 512         r = amdgpu_copy_buffer(ring, bo_addr, shadow_addr,
 513                                amdgpu_bo_size(bo), resv, fence,
 514                                direct);
 515         if (!r)
 516                 amdgpu_bo_fence(bo, *fence, true);
 517
 518 err:
 519         return r;
 520 }
 521
 522 int amdgpu_bo_restore_from_shadow(struct amdgpu_device *adev,
 523                                   struct amdgpu_ring *ring,
 524                                   struct amdgpu_bo *bo,
 525                                   struct reservation_object *resv,
 526                                   struct fence **fence,
 527                                   bool direct)
 528
 529 {
 530         struct amdgpu_bo *shadow = bo->shadow;
 531         uint64_t bo_addr, shadow_addr;
 532         int r;
 533
 534         if (!shadow)
 535                 return -EINVAL;
 536
 537         bo_addr = amdgpu_bo_gpu_offset(bo);
 538         shadow_addr = amdgpu_bo_gpu_offset(bo->shadow);
 539
 540         r = reservation_object_reserve_shared(bo->tbo.resv);
 541         if (r)
 542                 goto err;
 543
 544         r = amdgpu_copy_buffer(ring, shadow_addr, bo_addr,
 545                                amdgpu_bo_size(bo), resv, fence,
 546                                direct);
 547         if (!r)
 548                 amdgpu_bo_fence(bo, *fence, true);
 549
 550 err:
 551         return r;
 552 }
 553
 554 int amdgpu_bo_kmap(struct amdgpu_bo *bo, void **ptr)
 555 {
 556         bool is_iomem;
 557         long r;
 558
 559         if (bo->flags & AMDGPU_GEM_CREATE_NO_CPU_ACCESS)
 560                 return -EPERM;
 561
 562         if (bo->kptr) {
 563                 if (ptr) {
 564                         *ptr = bo->kptr;
 565                 }
 566                 return 0;
 567         }
 568
 569         r = reservation_object_wait_timeout_rcu(bo->tbo.resv, false, false,
 570                                                 MAX_SCHEDULE_TIMEOUT);
 571         if (r < 0)
 572                 return r;
 573
 574         r = ttm_bo_kmap(&bo->tbo, 0, bo->tbo.num_pages, &bo->kmap);
 575         if (r)
 576                 return r;
 577
 578         bo->kptr = ttm_kmap_obj_virtual(&bo->kmap, &is_iomem);
 579         if (ptr)
 580                 *ptr = bo->kptr;
 581
 582         return 0;
 583 }
 584
 585 void amdgpu_bo_kunmap(struct amdgpu_bo *bo)
 586 {
 587         if (bo->kptr == NULL)
 588                 return;
 589         bo->kptr = NULL;
 590         ttm_bo_kunmap(&bo->kmap);
 591 }
 592
 593 struct amdgpu_bo *amdgpu_bo_ref(struct amdgpu_bo *bo)
 594 {
 595         if (bo == NULL)
 596                 return NULL;
 597
 598         ttm_bo_reference(&bo->tbo);
 599         return bo;
 600 }
 601
 602 void amdgpu_bo_unref(struct amdgpu_bo **bo)
 603 {
 604         struct ttm_buffer_object *tbo;
 605
 606         if ((*bo) == NULL)
 607                 return;
 608
 609         tbo = &((*bo)->tbo);
 610         ttm_bo_unref(&tbo);
 611         if (tbo == NULL)
 612                 *bo = NULL;
 613 }
 614
 615 int amdgpu_bo_pin_restricted(struct amdgpu_bo *bo, u32 domain,
 616                              u64 min_offset, u64 max_offset,
 617                              u64 *gpu_addr)
 618 {
 619         int r, i;
 620         unsigned fpfn, lpfn;
 621
 622         if (amdgpu_ttm_tt_get_usermm(bo->tbo.ttm))
 623                 return -EPERM;
 624
 625         if (WARN_ON_ONCE(min_offset > max_offset))
 626                 return -EINVAL;
 627
 628         if (bo->pin_count) {
 629                 uint32_t mem_type = bo->tbo.mem.mem_type;
 630
 631                 if (domain != amdgpu_mem_type_to_domain(mem_type))
 632                         return -EINVAL;
 633
 634                 bo->pin_count++;
 635                 if (gpu_addr)
 636                         *gpu_addr = amdgpu_bo_gpu_offset(bo);
 637
 638                 if (max_offset != 0) {
 639                         u64 domain_start = bo->tbo.bdev->man[mem_type].gpu_offset;
 640                         WARN_ON_ONCE(max_offset <
 641                                      (amdgpu_bo_gpu_offset(bo) - domain_start));
 642                 }
 643
 644                 return 0;
 645         }
 646         amdgpu_ttm_placement_from_domain(bo, domain);
 647         for (i = 0; i < bo->placement.num_placement; i++) {
 648                 /* force to pin into visible video ram */
 649                 if ((bo->placements[i].flags & TTM_PL_FLAG_VRAM) &&
 650                     !(bo->flags & AMDGPU_GEM_CREATE_NO_CPU_ACCESS) &&
 651                     (!max_offset || max_offset >
 652                      bo->adev->mc.visible_vram_size)) {
 653                         if (WARN_ON_ONCE(min_offset >
 654                                          bo->adev->mc.visible_vram_size))
 655                                 return -EINVAL;
 656                         fpfn = min_offset >> PAGE_SHIFT;
 657                         lpfn = bo->adev->mc.visible_vram_size >> PAGE_SHIFT;
 658                 } else {
 659                         fpfn = min_offset >> PAGE_SHIFT;
 660                         lpfn = max_offset >> PAGE_SHIFT;
 661                 }
 662                 if (fpfn > bo->placements[i].fpfn)
 663                         bo->placements[i].fpfn = fpfn;
 664                 if (!bo->placements[i].lpfn ||
 665                     (lpfn && lpfn < bo->placements[i].lpfn))
 666                         bo->placements[i].lpfn = lpfn;
 667                 bo->placements[i].flags |= TTM_PL_FLAG_NO_EVICT;
 668         }
 669
 670         r = ttm_bo_validate(&bo->tbo, &bo->placement, false, false);
 671         if (unlikely(r)) {
 672                 dev_err(bo->adev->dev, "%p pin failed\n", bo);
 673                 goto error;
 674         }
 675         r = amdgpu_ttm_bind(&bo->tbo, &bo->tbo.mem);
 676         if (unlikely(r)) {
 677                 dev_err(bo->adev->dev, "%p bind failed\n", bo);
 678                 goto error;
 679         }
 680
 681         bo->pin_count = 1;
 682         if (gpu_addr != NULL)
 683                 *gpu_addr = amdgpu_bo_gpu_offset(bo);
 684         if (domain == AMDGPU_GEM_DOMAIN_VRAM) {
 685                 bo->adev->vram_pin_size += amdgpu_bo_size(bo);
 686                 if (bo->flags & AMDGPU_GEM_CREATE_NO_CPU_ACCESS)
 687                         bo->adev->invisible_pin_size += amdgpu_bo_size(bo);
 688         } else if (domain == AMDGPU_GEM_DOMAIN_GTT) {
 689                 bo->adev->gart_pin_size += amdgpu_bo_size(bo);
 690         }
 691
 692 error:
 693         return r;
 694 }
 695
 696 int amdgpu_bo_pin(struct amdgpu_bo *bo, u32 domain, u64 *gpu_addr)
 697 {
 698         return amdgpu_bo_pin_restricted(bo, domain, 0, 0, gpu_addr);
 699 }
 700
 701 int amdgpu_bo_unpin(struct amdgpu_bo *bo)
 702 {
 703         int r, i;
 704
 705         if (!bo->pin_count) {
 706                 dev_warn(bo->adev->dev, "%p unpin not necessary\n", bo);
 707                 return 0;
 708         }
 709         bo->pin_count--;
 710         if (bo->pin_count)
 711                 return 0;
 712         for (i = 0; i < bo->placement.num_placement; i++) {
 713                 bo->placements[i].lpfn = 0;
 714                 bo->placements[i].flags &= ~TTM_PL_FLAG_NO_EVICT;
 715         }
 716         r = ttm_bo_validate(&bo->tbo, &bo->placement, false, false);
 717         if (unlikely(r)) {
 718                 dev_err(bo->adev->dev, "%p validate failed for unpin\n", bo);
 719                 goto error;
 720         }
 721
 722         if (bo->tbo.mem.mem_type == TTM_PL_VRAM) {
 723                 bo->adev->vram_pin_size -= amdgpu_bo_size(bo);
 724                 if (bo->flags & AMDGPU_GEM_CREATE_NO_CPU_ACCESS)
 725                         bo->adev->invisible_pin_size -= amdgpu_bo_size(bo);
 726         } else if (bo->tbo.mem.mem_type == TTM_PL_TT) {
 727                 bo->adev->gart_pin_size -= amdgpu_bo_size(bo);
 728         }
 729
 730 error:
 731         return r;
 732 }
 733
 734 int amdgpu_bo_evict_vram(struct amdgpu_device *adev)
 735 {
 736         /* late 2.6.33 fix IGP hibernate - we need pm ops to do this correct */
 737         if (0 && (adev->flags & AMD_IS_APU)) {
 738                 /* Useless to evict on IGP chips */
 739                 return 0;
 740         }
 741         return ttm_bo_evict_mm(&adev->mman.bdev, TTM_PL_VRAM);
 742 }
 743
 744 static const char *amdgpu_vram_names[] = {
 745         "UNKNOWN",
 746         "GDDR1",
 747         "DDR2",
 748         "GDDR3",
 749         "GDDR4",
 750         "GDDR5",
 751         "HBM",
 752         "DDR3"
 753 };
 754
 755 int amdgpu_bo_init(struct amdgpu_device *adev)
 756 {
 757         /* reserve PAT memory space to WC for VRAM */
 758         arch_io_reserve_memtype_wc(adev->mc.aper_base,
 759                                    adev->mc.aper_size);
 760
 761         /* Add an MTRR for the VRAM */
 762         adev->mc.vram_mtrr = arch_phys_wc_add(adev->mc.aper_base,
 763                                               adev->mc.aper_size);
 764         DRM_INFO("Detected VRAM RAM=%lluM, BAR=%lluM\n",
 765                 adev->mc.mc_vram_size >> 20,
 766                 (unsigned long long)adev->mc.aper_size >> 20);
 767         DRM_INFO("RAM width %dbits %s\n",
 768                  adev->mc.vram_width, amdgpu_vram_names[adev->mc.vram_type]);
 769         return amdgpu_ttm_init(adev);
 770 }
 771
 772 void amdgpu_bo_fini(struct amdgpu_device *adev)
 773 {
 774         amdgpu_ttm_fini(adev);
 775         arch_phys_wc_del(adev->mc.vram_mtrr);
 776         arch_io_free_memtype_wc(adev->mc.aper_base, adev->mc.aper_size);
 777 }
 778
 779 int amdgpu_bo_fbdev_mmap(struct amdgpu_bo *bo,
 780                              struct vm_area_struct *vma)
 781 {
 782         return ttm_fbdev_mmap(vma, &bo->tbo);
 783 }
 784
 785 int amdgpu_bo_set_tiling_flags(struct amdgpu_bo *bo, u64 tiling_flags)
 786 {
 787         if (AMDGPU_TILING_GET(tiling_flags, TILE_SPLIT) > 6)
 788                 return -EINVAL;
 789
 790         bo->tiling_flags = tiling_flags;
 791         return 0;
 792 }
 793
 794 void amdgpu_bo_get_tiling_flags(struct amdgpu_bo *bo, u64 *tiling_flags)
 795 {
 796         lockdep_assert_held(&bo->tbo.resv->lock.base);
 797
 798         if (tiling_flags)
 799                 *tiling_flags = bo->tiling_flags;
 800 }
 801
 802 int amdgpu_bo_set_metadata (struct amdgpu_bo *bo, void *metadata,
 803                             uint32_t metadata_size, uint64_t flags)
 804 {
 805         void *buffer;
 806
 807         if (!metadata_size) {
 808                 if (bo->metadata_size) {
 809                         kfree(bo->metadata);
 810                         bo->metadata = NULL;
 811                         bo->metadata_size = 0;
 812                 }
 813                 return 0;
 814         }
 815
 816         if (metadata == NULL)
 817                 return -EINVAL;
 818
 819         buffer = kmemdup(metadata, metadata_size, GFP_KERNEL);
 820         if (buffer == NULL)
 821                 return -ENOMEM;
 822
 823         kfree(bo->metadata);
 824         bo->metadata_flags = flags;
 825         bo->metadata = buffer;
 826         bo->metadata_size = metadata_size;
 827
 828         return 0;
 829 }
 830
 831 int amdgpu_bo_get_metadata(struct amdgpu_bo *bo, void *buffer,
 832                            size_t buffer_size, uint32_t *metadata_size,
 833                            uint64_t *flags)
 834 {
 835         if (!buffer && !metadata_size)
 836                 return -EINVAL;
 837
 838         if (buffer) {
 839                 if (buffer_size < bo->metadata_size)
 840                         return -EINVAL;
 841
 842                 if (bo->metadata_size)
 843                         memcpy(buffer, bo->metadata, bo->metadata_size);
 844         }
 845
 846         if (metadata_size)
 847                 *metadata_size = bo->metadata_size;
 848         if (flags)
 849                 *flags = bo->metadata_flags;
 850
 851         return 0;
 852 }
 853
 854 void amdgpu_bo_move_notify(struct ttm_buffer_object *bo,
 855                            struct ttm_mem_reg *new_mem)
 856 {
 857         struct amdgpu_bo *abo;
 858         struct ttm_mem_reg *old_mem = &bo->mem;
 859
 860         if (!amdgpu_ttm_bo_is_amdgpu_bo(bo))
 861                 return;
 862
 863         abo = container_of(bo, struct amdgpu_bo, tbo);
 864         amdgpu_vm_bo_invalidate(abo->adev, abo);
 865
 866         /* update statistics */
 867         if (!new_mem)
 868                 return;
 869
 870         /* move_notify is called before move happens */
 871         amdgpu_update_memory_usage(abo->adev, &bo->mem, new_mem);
 872
 873         trace_amdgpu_ttm_bo_move(abo, new_mem->mem_type, old_mem->mem_type);
 874 }
 875
 876 int amdgpu_bo_fault_reserve_notify(struct ttm_buffer_object *bo)
 877 {
 878         struct amdgpu_device *adev;
 879         struct amdgpu_bo *abo;
 880         unsigned long offset, size, lpfn;
 881         int i, r;
 882
 883         if (!amdgpu_ttm_bo_is_amdgpu_bo(bo))
 884                 return 0;
 885
 886         abo = container_of(bo, struct amdgpu_bo, tbo);
 887         adev = abo->adev;
 888         if (bo->mem.mem_type != TTM_PL_VRAM)
 889                 return 0;
 890
 891         size = bo->mem.num_pages << PAGE_SHIFT;
 892         offset = bo->mem.start << PAGE_SHIFT;
 893         if ((offset + size) <= adev->mc.visible_vram_size)
 894                 return 0;
 895
 896         /* Can't move a pinned BO to visible VRAM */
 897         if (abo->pin_count > 0)
 898                 return -EINVAL;
 899
 900         /* hurrah the memory is not visible ! */
 901         amdgpu_ttm_placement_from_domain(abo, AMDGPU_GEM_DOMAIN_VRAM);
 902         lpfn =  adev->mc.visible_vram_size >> PAGE_SHIFT;
 903         for (i = 0; i < abo->placement.num_placement; i++) {
 904                 /* Force into visible VRAM */
 905                 if ((abo->placements[i].flags & TTM_PL_FLAG_VRAM) &&
 906                     (!abo->placements[i].lpfn ||
 907                      abo->placements[i].lpfn > lpfn))
 908                         abo->placements[i].lpfn = lpfn;
 909         }
 910         r = ttm_bo_validate(bo, &abo->placement, false, false);
 911         if (unlikely(r == -ENOMEM)) {
 912                 amdgpu_ttm_placement_from_domain(abo, AMDGPU_GEM_DOMAIN_GTT);
 913                 return ttm_bo_validate(bo, &abo->placement, false, false);
 914         } else if (unlikely(r != 0)) {
 915                 return r;
 916         }
 917
 918         offset = bo->mem.start << PAGE_SHIFT;
 919         /* this should never happen */
 920         if ((offset + size) > adev->mc.visible_vram_size)
 921                 return -EINVAL;
 922
 923         return 0;
 924 }
 925
 926 /**
 927  * amdgpu_bo_fence - add fence to buffer object
 928  *
 929  * @bo: buffer object in question
 930  * @fence: fence to add
 931  * @shared: true if fence should be added shared
 932  *
 933  */
 934 void amdgpu_bo_fence(struct amdgpu_bo *bo, struct fence *fence,
 935                      bool shared)
 936 {
 937         struct reservation_object *resv = bo->tbo.resv;
 938
 939         if (shared)
 940                 reservation_object_add_shared_fence(resv, fence);
 941         else
 942                 reservation_object_add_excl_fence(resv, fence);
 943 }
 944
 945 /**
 946  * amdgpu_bo_gpu_offset - return GPU offset of bo
 947  * @bo: amdgpu object for which we query the offset
 948  *
 949  * Returns current GPU offset of the object.
 950  *
 951  * Note: object should either be pinned or reserved when calling this
 952  * function, it might be useful to add check for this for debugging.
 953  */
 954 u64 amdgpu_bo_gpu_offset(struct amdgpu_bo *bo)
 955 {
 956         WARN_ON_ONCE(bo->tbo.mem.mem_type == TTM_PL_SYSTEM);
 957         WARN_ON_ONCE(bo->tbo.mem.mem_type == TTM_PL_TT &&
 958                      !amdgpu_ttm_is_bound(bo->tbo.ttm));
 959         WARN_ON_ONCE(!ww_mutex_is_locked(&bo->tbo.resv->lock) &&
 960                      !bo->pin_count);
 961         WARN_ON_ONCE(bo->tbo.mem.start == AMDGPU_BO_INVALID_OFFSET);
 962
 963         return bo->tbo.offset;
 964 }