Linux Audio

Check our new training course

Loading...
v5.9
   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 <linux/dma-fence-array.h>
  29#include <linux/interval_tree_generic.h>
  30#include <linux/idr.h>
 
  31
  32#include <drm/amdgpu_drm.h>
 
  33#include "amdgpu.h"
  34#include "amdgpu_trace.h"
  35#include "amdgpu_amdkfd.h"
  36#include "amdgpu_gmc.h"
  37#include "amdgpu_xgmi.h"
 
 
 
  38
  39/**
  40 * DOC: GPUVM
  41 *
  42 * GPUVM is similar to the legacy gart on older asics, however
  43 * rather than there being a single global gart table
  44 * for the entire GPU, there are multiple VM page tables active
  45 * at any given time.  The VM page tables can contain a mix
  46 * vram pages and system memory pages and system memory pages
 
  47 * can be mapped as snooped (cached system pages) or unsnooped
  48 * (uncached system pages).
  49 * Each VM has an ID associated with it and there is a page table
  50 * associated with each VMID.  When execting a command buffer,
  51 * the kernel tells the the ring what VMID to use for that command
 
  52 * buffer.  VMIDs are allocated dynamically as commands are submitted.
  53 * The userspace drivers maintain their own address space and the kernel
  54 * sets up their pages tables accordingly when they submit their
  55 * command buffers and a VMID is assigned.
  56 * Cayman/Trinity support up to 8 active VMs at any given time;
  57 * SI supports 16.
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
  58 */
  59
  60#define START(node) ((node)->start)
  61#define LAST(node) ((node)->last)
  62
  63INTERVAL_TREE_DEFINE(struct amdgpu_bo_va_mapping, rb, uint64_t, __subtree_last,
  64		     START, LAST, static, amdgpu_vm_it)
  65
  66#undef START
  67#undef LAST
  68
  69/**
  70 * struct amdgpu_prt_cb - Helper to disable partial resident texture feature from a fence callback
  71 */
  72struct amdgpu_prt_cb {
  73
  74	/**
  75	 * @adev: amdgpu device
  76	 */
  77	struct amdgpu_device *adev;
  78
  79	/**
  80	 * @cb: callback
  81	 */
  82	struct dma_fence_cb cb;
  83};
  84
  85/*
  86 * vm eviction_lock can be taken in MMU notifiers. Make sure no reclaim-FS
  87 * happens while holding this lock anywhere to prevent deadlocks when
  88 * an MMU notifier runs in reclaim-FS context.
  89 */
  90static inline void amdgpu_vm_eviction_lock(struct amdgpu_vm *vm)
  91{
  92	mutex_lock(&vm->eviction_lock);
  93	vm->saved_flags = memalloc_nofs_save();
  94}
  95
  96static inline int amdgpu_vm_eviction_trylock(struct amdgpu_vm *vm)
  97{
  98	if (mutex_trylock(&vm->eviction_lock)) {
  99		vm->saved_flags = memalloc_nofs_save();
 100		return 1;
 101	}
 102	return 0;
 103}
 104
 105static inline void amdgpu_vm_eviction_unlock(struct amdgpu_vm *vm)
 106{
 107	memalloc_nofs_restore(vm->saved_flags);
 108	mutex_unlock(&vm->eviction_lock);
 109}
 110
 111/**
 112 * amdgpu_vm_level_shift - return the addr shift for each level
 113 *
 114 * @adev: amdgpu_device pointer
 115 * @level: VMPT level
 
 
 
 
 116 *
 117 * Returns:
 118 * The number of bits the pfn needs to be right shifted for a level.
 119 */
 120static unsigned amdgpu_vm_level_shift(struct amdgpu_device *adev,
 121				      unsigned level)
 122{
 123	switch (level) {
 124	case AMDGPU_VM_PDB2:
 125	case AMDGPU_VM_PDB1:
 126	case AMDGPU_VM_PDB0:
 127		return 9 * (AMDGPU_VM_PDB0 - level) +
 128			adev->vm_manager.block_size;
 129	case AMDGPU_VM_PTB:
 130		return 0;
 131	default:
 132		return ~0;
 133	}
 134}
 135
 136/**
 137 * amdgpu_vm_num_entries - return the number of entries in a PD/PT
 138 *
 139 * @adev: amdgpu_device pointer
 140 * @level: VMPT level
 141 *
 142 * Returns:
 143 * The number of entries in a page directory or page table.
 144 */
 145static unsigned amdgpu_vm_num_entries(struct amdgpu_device *adev,
 146				      unsigned level)
 147{
 148	unsigned shift = amdgpu_vm_level_shift(adev,
 149					       adev->vm_manager.root_level);
 150
 151	if (level == adev->vm_manager.root_level)
 152		/* For the root directory */
 153		return round_up(adev->vm_manager.max_pfn, 1ULL << shift)
 154			>> shift;
 155	else if (level != AMDGPU_VM_PTB)
 156		/* Everything in between */
 157		return 512;
 158	else
 159		/* For the page tables on the leaves */
 160		return AMDGPU_VM_PTE_COUNT(adev);
 161}
 162
 163/**
 164 * amdgpu_vm_num_ats_entries - return the number of ATS entries in the root PD
 165 *
 166 * @adev: amdgpu_device pointer
 167 *
 168 * Returns:
 169 * The number of entries in the root page directory which needs the ATS setting.
 170 */
 171static unsigned amdgpu_vm_num_ats_entries(struct amdgpu_device *adev)
 172{
 173	unsigned shift;
 174
 175	shift = amdgpu_vm_level_shift(adev, adev->vm_manager.root_level);
 176	return AMDGPU_GMC_HOLE_START >> (shift + AMDGPU_GPU_PAGE_SHIFT);
 177}
 178
 179/**
 180 * amdgpu_vm_entries_mask - the mask to get the entry number of a PD/PT
 181 *
 182 * @adev: amdgpu_device pointer
 183 * @level: VMPT level
 184 *
 185 * Returns:
 186 * The mask to extract the entry number of a PD/PT from an address.
 187 */
 188static uint32_t amdgpu_vm_entries_mask(struct amdgpu_device *adev,
 189				       unsigned int level)
 190{
 191	if (level <= adev->vm_manager.root_level)
 192		return 0xffffffff;
 193	else if (level != AMDGPU_VM_PTB)
 194		return 0x1ff;
 195	else
 196		return AMDGPU_VM_PTE_COUNT(adev) - 1;
 197}
 198
 199/**
 200 * amdgpu_vm_bo_size - returns the size of the BOs in bytes
 201 *
 202 * @adev: amdgpu_device pointer
 203 * @level: VMPT level
 204 *
 205 * Returns:
 206 * The size of the BO for a page directory or page table in bytes.
 207 */
 208static unsigned amdgpu_vm_bo_size(struct amdgpu_device *adev, unsigned level)
 209{
 210	return AMDGPU_GPU_PAGE_ALIGN(amdgpu_vm_num_entries(adev, level) * 8);
 211}
 212
 213/**
 214 * amdgpu_vm_bo_evicted - vm_bo is evicted
 215 *
 216 * @vm_bo: vm_bo which is evicted
 217 *
 218 * State for PDs/PTs and per VM BOs which are not at the location they should
 219 * be.
 220 */
 221static void amdgpu_vm_bo_evicted(struct amdgpu_vm_bo_base *vm_bo)
 222{
 223	struct amdgpu_vm *vm = vm_bo->vm;
 224	struct amdgpu_bo *bo = vm_bo->bo;
 225
 226	vm_bo->moved = true;
 
 227	if (bo->tbo.type == ttm_bo_type_kernel)
 228		list_move(&vm_bo->vm_status, &vm->evicted);
 229	else
 230		list_move_tail(&vm_bo->vm_status, &vm->evicted);
 
 231}
 232/**
 233 * amdgpu_vm_bo_moved - vm_bo is moved
 234 *
 235 * @vm_bo: vm_bo which is moved
 236 *
 237 * State for per VM BOs which are moved, but that change is not yet reflected
 238 * in the page tables.
 239 */
 240static void amdgpu_vm_bo_moved(struct amdgpu_vm_bo_base *vm_bo)
 241{
 
 242	list_move(&vm_bo->vm_status, &vm_bo->vm->moved);
 
 243}
 244
 245/**
 246 * amdgpu_vm_bo_idle - vm_bo is idle
 247 *
 248 * @vm_bo: vm_bo which is now idle
 249 *
 250 * State for PDs/PTs and per VM BOs which have gone through the state machine
 251 * and are now idle.
 252 */
 253static void amdgpu_vm_bo_idle(struct amdgpu_vm_bo_base *vm_bo)
 254{
 
 255	list_move(&vm_bo->vm_status, &vm_bo->vm->idle);
 
 256	vm_bo->moved = false;
 257}
 258
 259/**
 260 * amdgpu_vm_bo_invalidated - vm_bo is invalidated
 261 *
 262 * @vm_bo: vm_bo which is now invalidated
 263 *
 264 * State for normal BOs which are invalidated and that change not yet reflected
 265 * in the PTs.
 266 */
 267static void amdgpu_vm_bo_invalidated(struct amdgpu_vm_bo_base *vm_bo)
 268{
 269	spin_lock(&vm_bo->vm->invalidated_lock);
 270	list_move(&vm_bo->vm_status, &vm_bo->vm->invalidated);
 271	spin_unlock(&vm_bo->vm->invalidated_lock);
 272}
 273
 274/**
 275 * amdgpu_vm_bo_relocated - vm_bo is reloacted
 276 *
 277 * @vm_bo: vm_bo which is relocated
 278 *
 279 * State for PDs/PTs which needs to update their parent PD.
 280 * For the root PD, just move to idle state.
 281 */
 282static void amdgpu_vm_bo_relocated(struct amdgpu_vm_bo_base *vm_bo)
 283{
 284	if (vm_bo->bo->parent)
 
 285		list_move(&vm_bo->vm_status, &vm_bo->vm->relocated);
 286	else
 
 287		amdgpu_vm_bo_idle(vm_bo);
 
 288}
 289
 290/**
 291 * amdgpu_vm_bo_done - vm_bo is done
 292 *
 293 * @vm_bo: vm_bo which is now done
 294 *
 295 * State for normal BOs which are invalidated and that change has been updated
 296 * in the PTs.
 297 */
 298static void amdgpu_vm_bo_done(struct amdgpu_vm_bo_base *vm_bo)
 299{
 300	spin_lock(&vm_bo->vm->invalidated_lock);
 301	list_del_init(&vm_bo->vm_status);
 302	spin_unlock(&vm_bo->vm->invalidated_lock);
 303}
 304
 305/**
 306 * amdgpu_vm_bo_base_init - Adds bo to the list of bos associated with the vm
 307 *
 308 * @base: base structure for tracking BO usage in a VM
 309 * @vm: vm to which bo is to be added
 310 * @bo: amdgpu buffer object
 311 *
 312 * Initialize a bo_va_base structure and add it to the appropriate lists
 313 *
 314 */
 315static void amdgpu_vm_bo_base_init(struct amdgpu_vm_bo_base *base,
 316				   struct amdgpu_vm *vm,
 317				   struct amdgpu_bo *bo)
 318{
 319	base->vm = vm;
 320	base->bo = bo;
 321	base->next = NULL;
 322	INIT_LIST_HEAD(&base->vm_status);
 323
 324	if (!bo)
 325		return;
 326	base->next = bo->vm_bo;
 327	bo->vm_bo = base;
 328
 329	if (bo->tbo.base.resv != vm->root.base.bo->tbo.base.resv)
 330		return;
 331
 332	vm->bulk_moveable = false;
 
 
 333	if (bo->tbo.type == ttm_bo_type_kernel && bo->parent)
 334		amdgpu_vm_bo_relocated(base);
 335	else
 336		amdgpu_vm_bo_idle(base);
 337
 338	if (bo->preferred_domains &
 339	    amdgpu_mem_type_to_domain(bo->tbo.mem.mem_type))
 340		return;
 341
 342	/*
 343	 * we checked all the prerequisites, but it looks like this per vm bo
 344	 * is currently evicted. add the bo to the evicted list to make sure it
 345	 * is validated on next vm use to avoid fault.
 346	 * */
 347	amdgpu_vm_bo_evicted(base);
 348}
 349
 350/**
 351 * amdgpu_vm_pt_parent - get the parent page directory
 352 *
 353 * @pt: child page table
 354 *
 355 * Helper to get the parent entry for the child page table. NULL if we are at
 356 * the root page directory.
 357 */
 358static struct amdgpu_vm_pt *amdgpu_vm_pt_parent(struct amdgpu_vm_pt *pt)
 359{
 360	struct amdgpu_bo *parent = pt->base.bo->parent;
 361
 362	if (!parent)
 363		return NULL;
 364
 365	return container_of(parent->vm_bo, struct amdgpu_vm_pt, base);
 366}
 367
 368/*
 369 * amdgpu_vm_pt_cursor - state for for_each_amdgpu_vm_pt
 370 */
 371struct amdgpu_vm_pt_cursor {
 372	uint64_t pfn;
 373	struct amdgpu_vm_pt *parent;
 374	struct amdgpu_vm_pt *entry;
 375	unsigned level;
 376};
 377
 378/**
 379 * amdgpu_vm_pt_start - start PD/PT walk
 380 *
 381 * @adev: amdgpu_device pointer
 382 * @vm: amdgpu_vm structure
 383 * @start: start address of the walk
 384 * @cursor: state to initialize
 385 *
 386 * Initialize a amdgpu_vm_pt_cursor to start a walk.
 387 */
 388static void amdgpu_vm_pt_start(struct amdgpu_device *adev,
 389			       struct amdgpu_vm *vm, uint64_t start,
 390			       struct amdgpu_vm_pt_cursor *cursor)
 391{
 392	cursor->pfn = start;
 393	cursor->parent = NULL;
 394	cursor->entry = &vm->root;
 395	cursor->level = adev->vm_manager.root_level;
 396}
 397
 398/**
 399 * amdgpu_vm_pt_descendant - go to child node
 400 *
 401 * @adev: amdgpu_device pointer
 402 * @cursor: current state
 403 *
 404 * Walk to the child node of the current node.
 405 * Returns:
 406 * True if the walk was possible, false otherwise.
 407 */
 408static bool amdgpu_vm_pt_descendant(struct amdgpu_device *adev,
 409				    struct amdgpu_vm_pt_cursor *cursor)
 410{
 411	unsigned mask, shift, idx;
 412
 413	if (!cursor->entry->entries)
 414		return false;
 415
 416	BUG_ON(!cursor->entry->base.bo);
 417	mask = amdgpu_vm_entries_mask(adev, cursor->level);
 418	shift = amdgpu_vm_level_shift(adev, cursor->level);
 419
 420	++cursor->level;
 421	idx = (cursor->pfn >> shift) & mask;
 422	cursor->parent = cursor->entry;
 423	cursor->entry = &cursor->entry->entries[idx];
 424	return true;
 425}
 426
 427/**
 428 * amdgpu_vm_pt_sibling - go to sibling node
 429 *
 430 * @adev: amdgpu_device pointer
 431 * @cursor: current state
 432 *
 433 * Walk to the sibling node of the current node.
 434 * Returns:
 435 * True if the walk was possible, false otherwise.
 436 */
 437static bool amdgpu_vm_pt_sibling(struct amdgpu_device *adev,
 438				 struct amdgpu_vm_pt_cursor *cursor)
 439{
 440	unsigned shift, num_entries;
 441
 442	/* Root doesn't have a sibling */
 443	if (!cursor->parent)
 444		return false;
 445
 446	/* Go to our parents and see if we got a sibling */
 447	shift = amdgpu_vm_level_shift(adev, cursor->level - 1);
 448	num_entries = amdgpu_vm_num_entries(adev, cursor->level - 1);
 449
 450	if (cursor->entry == &cursor->parent->entries[num_entries - 1])
 451		return false;
 452
 453	cursor->pfn += 1ULL << shift;
 454	cursor->pfn &= ~((1ULL << shift) - 1);
 455	++cursor->entry;
 456	return true;
 457}
 458
 459/**
 460 * amdgpu_vm_pt_ancestor - go to parent node
 461 *
 462 * @cursor: current state
 463 *
 464 * Walk to the parent node of the current node.
 465 * Returns:
 466 * True if the walk was possible, false otherwise.
 467 */
 468static bool amdgpu_vm_pt_ancestor(struct amdgpu_vm_pt_cursor *cursor)
 469{
 470	if (!cursor->parent)
 471		return false;
 472
 473	--cursor->level;
 474	cursor->entry = cursor->parent;
 475	cursor->parent = amdgpu_vm_pt_parent(cursor->parent);
 476	return true;
 477}
 478
 479/**
 480 * amdgpu_vm_pt_next - get next PD/PT in hieratchy
 481 *
 482 * @adev: amdgpu_device pointer
 483 * @cursor: current state
 484 *
 485 * Walk the PD/PT tree to the next node.
 486 */
 487static void amdgpu_vm_pt_next(struct amdgpu_device *adev,
 488			      struct amdgpu_vm_pt_cursor *cursor)
 489{
 490	/* First try a newborn child */
 491	if (amdgpu_vm_pt_descendant(adev, cursor))
 492		return;
 493
 494	/* If that didn't worked try to find a sibling */
 495	while (!amdgpu_vm_pt_sibling(adev, cursor)) {
 496		/* No sibling, go to our parents and grandparents */
 497		if (!amdgpu_vm_pt_ancestor(cursor)) {
 498			cursor->pfn = ~0ll;
 499			return;
 500		}
 501	}
 502}
 503
 504/**
 505 * amdgpu_vm_pt_first_dfs - start a deep first search
 506 *
 507 * @adev: amdgpu_device structure
 508 * @vm: amdgpu_vm structure
 509 * @start: optional cursor to start with
 510 * @cursor: state to initialize
 511 *
 512 * Starts a deep first traversal of the PD/PT tree.
 513 */
 514static void amdgpu_vm_pt_first_dfs(struct amdgpu_device *adev,
 515				   struct amdgpu_vm *vm,
 516				   struct amdgpu_vm_pt_cursor *start,
 517				   struct amdgpu_vm_pt_cursor *cursor)
 518{
 519	if (start)
 520		*cursor = *start;
 521	else
 522		amdgpu_vm_pt_start(adev, vm, 0, cursor);
 523	while (amdgpu_vm_pt_descendant(adev, cursor));
 524}
 525
 526/**
 527 * amdgpu_vm_pt_continue_dfs - check if the deep first search should continue
 528 *
 529 * @start: starting point for the search
 530 * @entry: current entry
 531 *
 532 * Returns:
 533 * True when the search should continue, false otherwise.
 534 */
 535static bool amdgpu_vm_pt_continue_dfs(struct amdgpu_vm_pt_cursor *start,
 536				      struct amdgpu_vm_pt *entry)
 537{
 538	return entry && (!start || entry != start->entry);
 539}
 540
 541/**
 542 * amdgpu_vm_pt_next_dfs - get the next node for a deep first search
 543 *
 544 * @adev: amdgpu_device structure
 545 * @cursor: current state
 546 *
 547 * Move the cursor to the next node in a deep first search.
 548 */
 549static void amdgpu_vm_pt_next_dfs(struct amdgpu_device *adev,
 550				  struct amdgpu_vm_pt_cursor *cursor)
 551{
 552	if (!cursor->entry)
 553		return;
 554
 555	if (!cursor->parent)
 556		cursor->entry = NULL;
 557	else if (amdgpu_vm_pt_sibling(adev, cursor))
 558		while (amdgpu_vm_pt_descendant(adev, cursor));
 559	else
 560		amdgpu_vm_pt_ancestor(cursor);
 561}
 562
 563/*
 564 * for_each_amdgpu_vm_pt_dfs_safe - safe deep first search of all PDs/PTs
 565 */
 566#define for_each_amdgpu_vm_pt_dfs_safe(adev, vm, start, cursor, entry)		\
 567	for (amdgpu_vm_pt_first_dfs((adev), (vm), (start), &(cursor)),		\
 568	     (entry) = (cursor).entry, amdgpu_vm_pt_next_dfs((adev), &(cursor));\
 569	     amdgpu_vm_pt_continue_dfs((start), (entry));			\
 570	     (entry) = (cursor).entry, amdgpu_vm_pt_next_dfs((adev), &(cursor)))
 571
 572/**
 573 * amdgpu_vm_get_pd_bo - add the VM PD to a validation list
 574 *
 575 * @vm: vm providing the BOs
 576 * @validated: head of validation list
 577 * @entry: entry to add
 578 *
 579 * Add the page directory to the list of BOs to
 580 * validate for command submission.
 581 */
 582void amdgpu_vm_get_pd_bo(struct amdgpu_vm *vm,
 583			 struct list_head *validated,
 584			 struct amdgpu_bo_list_entry *entry)
 585{
 586	entry->priority = 0;
 587	entry->tv.bo = &vm->root.base.bo->tbo;
 588	/* Two for VM updates, one for TTM and one for the CS job */
 589	entry->tv.num_shared = 4;
 590	entry->user_pages = NULL;
 591	list_add(&entry->tv.head, validated);
 592}
 593
 594/**
 595 * amdgpu_vm_del_from_lru_notify - update bulk_moveable flag
 596 *
 597 * @bo: BO which was removed from the LRU
 598 *
 599 * Make sure the bulk_moveable flag is updated when a BO is removed from the
 600 * LRU.
 601 */
 602void amdgpu_vm_del_from_lru_notify(struct ttm_buffer_object *bo)
 603{
 604	struct amdgpu_bo *abo;
 605	struct amdgpu_vm_bo_base *bo_base;
 606
 607	if (!amdgpu_bo_is_amdgpu_bo(bo))
 608		return;
 609
 610	if (bo->mem.placement & TTM_PL_FLAG_NO_EVICT)
 611		return;
 612
 613	abo = ttm_to_amdgpu_bo(bo);
 614	if (!abo->parent)
 615		return;
 616	for (bo_base = abo->vm_bo; bo_base; bo_base = bo_base->next) {
 617		struct amdgpu_vm *vm = bo_base->vm;
 618
 619		if (abo->tbo.base.resv == vm->root.base.bo->tbo.base.resv)
 620			vm->bulk_moveable = false;
 621	}
 622
 623}
 624/**
 625 * amdgpu_vm_move_to_lru_tail - move all BOs to the end of LRU
 626 *
 627 * @adev: amdgpu device pointer
 628 * @vm: vm providing the BOs
 629 *
 630 * Move all BOs to the end of LRU and remember their positions to put them
 631 * together.
 632 */
 633void amdgpu_vm_move_to_lru_tail(struct amdgpu_device *adev,
 634				struct amdgpu_vm *vm)
 635{
 636	struct amdgpu_vm_bo_base *bo_base;
 637
 638	if (vm->bulk_moveable) {
 639		spin_lock(&ttm_bo_glob.lru_lock);
 640		ttm_bo_bulk_move_lru_tail(&vm->lru_bulk_move);
 641		spin_unlock(&ttm_bo_glob.lru_lock);
 642		return;
 643	}
 644
 645	memset(&vm->lru_bulk_move, 0, sizeof(vm->lru_bulk_move));
 646
 647	spin_lock(&ttm_bo_glob.lru_lock);
 648	list_for_each_entry(bo_base, &vm->idle, vm_status) {
 649		struct amdgpu_bo *bo = bo_base->bo;
 650
 651		if (!bo->parent)
 652			continue;
 653
 654		ttm_bo_move_to_lru_tail(&bo->tbo, &vm->lru_bulk_move);
 655		if (bo->shadow)
 656			ttm_bo_move_to_lru_tail(&bo->shadow->tbo,
 657						&vm->lru_bulk_move);
 658	}
 659	spin_unlock(&ttm_bo_glob.lru_lock);
 660
 661	vm->bulk_moveable = true;
 662}
 663
 664/**
 665 * amdgpu_vm_validate_pt_bos - validate the page table BOs
 666 *
 667 * @adev: amdgpu device pointer
 668 * @vm: vm providing the BOs
 669 * @validate: callback to do the validation
 670 * @param: parameter for the validation callback
 671 *
 672 * Validate the page table BOs on command submission if neccessary.
 673 *
 674 * Returns:
 675 * Validation result.
 676 */
 677int amdgpu_vm_validate_pt_bos(struct amdgpu_device *adev, struct amdgpu_vm *vm,
 678			      int (*validate)(void *p, struct amdgpu_bo *bo),
 679			      void *param)
 680{
 681	struct amdgpu_vm_bo_base *bo_base, *tmp;
 
 
 682	int r;
 683
 684	vm->bulk_moveable &= list_empty(&vm->evicted);
 
 
 
 
 
 685
 686	list_for_each_entry_safe(bo_base, tmp, &vm->evicted, vm_status) {
 687		struct amdgpu_bo *bo = bo_base->bo;
 688
 689		r = validate(param, bo);
 690		if (r)
 691			return r;
 
 
 
 
 
 692
 693		if (bo->tbo.type != ttm_bo_type_kernel) {
 694			amdgpu_vm_bo_moved(bo_base);
 695		} else {
 696			vm->update_funcs->map_table(bo);
 697			amdgpu_vm_bo_relocated(bo_base);
 698		}
 
 699	}
 
 700
 701	amdgpu_vm_eviction_lock(vm);
 702	vm->evicting = false;
 703	amdgpu_vm_eviction_unlock(vm);
 704
 705	return 0;
 706}
 707
 708/**
 709 * amdgpu_vm_ready - check VM is ready for updates
 710 *
 711 * @vm: VM to check
 712 *
 713 * Check if all VM PDs/PTs are ready for updates
 714 *
 715 * Returns:
 716 * True if eviction list is empty.
 717 */
 718bool amdgpu_vm_ready(struct amdgpu_vm *vm)
 719{
 720	return list_empty(&vm->evicted);
 721}
 722
 723/**
 724 * amdgpu_vm_clear_bo - initially clear the PDs/PTs
 725 *
 726 * @adev: amdgpu_device pointer
 727 * @vm: VM to clear BO from
 728 * @bo: BO to clear
 729 * @immediate: use an immediate update
 730 *
 731 * Root PD needs to be reserved when calling this.
 732 *
 733 * Returns:
 734 * 0 on success, errno otherwise.
 735 */
 736static int amdgpu_vm_clear_bo(struct amdgpu_device *adev,
 737			      struct amdgpu_vm *vm,
 738			      struct amdgpu_bo *bo,
 739			      bool immediate)
 740{
 741	struct ttm_operation_ctx ctx = { true, false };
 742	unsigned level = adev->vm_manager.root_level;
 743	struct amdgpu_vm_update_params params;
 744	struct amdgpu_bo *ancestor = bo;
 745	unsigned entries, ats_entries;
 746	uint64_t addr;
 747	int r;
 748
 749	/* Figure out our place in the hierarchy */
 750	if (ancestor->parent) {
 751		++level;
 752		while (ancestor->parent->parent) {
 753			++level;
 754			ancestor = ancestor->parent;
 755		}
 756	}
 757
 758	entries = amdgpu_bo_size(bo) / 8;
 759	if (!vm->pte_support_ats) {
 760		ats_entries = 0;
 761
 762	} else if (!bo->parent) {
 763		ats_entries = amdgpu_vm_num_ats_entries(adev);
 764		ats_entries = min(ats_entries, entries);
 765		entries -= ats_entries;
 766
 767	} else {
 768		struct amdgpu_vm_pt *pt;
 769
 770		pt = container_of(ancestor->vm_bo, struct amdgpu_vm_pt, base);
 771		ats_entries = amdgpu_vm_num_ats_entries(adev);
 772		if ((pt - vm->root.entries) >= ats_entries) {
 773			ats_entries = 0;
 774		} else {
 775			ats_entries = entries;
 776			entries = 0;
 777		}
 778	}
 779
 780	r = ttm_bo_validate(&bo->tbo, &bo->placement, &ctx);
 781	if (r)
 782		return r;
 783
 784	if (bo->shadow) {
 785		r = ttm_bo_validate(&bo->shadow->tbo, &bo->shadow->placement,
 786				    &ctx);
 787		if (r)
 788			return r;
 789	}
 790
 791	r = vm->update_funcs->map_table(bo);
 792	if (r)
 793		return r;
 794
 795	memset(&params, 0, sizeof(params));
 796	params.adev = adev;
 797	params.vm = vm;
 798	params.immediate = immediate;
 799
 800	r = vm->update_funcs->prepare(&params, NULL, AMDGPU_SYNC_EXPLICIT);
 801	if (r)
 802		return r;
 803
 804	addr = 0;
 805	if (ats_entries) {
 806		uint64_t value = 0, flags;
 807
 808		flags = AMDGPU_PTE_DEFAULT_ATC;
 809		if (level != AMDGPU_VM_PTB) {
 810			/* Handle leaf PDEs as PTEs */
 811			flags |= AMDGPU_PDE_PTE;
 812			amdgpu_gmc_get_vm_pde(adev, level, &value, &flags);
 813		}
 814
 815		r = vm->update_funcs->update(&params, bo, addr, 0, ats_entries,
 816					     value, flags);
 817		if (r)
 818			return r;
 819
 820		addr += ats_entries * 8;
 821	}
 822
 823	if (entries) {
 824		uint64_t value = 0, flags = 0;
 825
 826		if (adev->asic_type >= CHIP_VEGA10) {
 827			if (level != AMDGPU_VM_PTB) {
 828				/* Handle leaf PDEs as PTEs */
 829				flags |= AMDGPU_PDE_PTE;
 830				amdgpu_gmc_get_vm_pde(adev, level,
 831						      &value, &flags);
 832			} else {
 833				/* Workaround for fault priority problem on GMC9 */
 834				flags = AMDGPU_PTE_EXECUTABLE;
 835			}
 836		}
 837
 838		r = vm->update_funcs->update(&params, bo, addr, 0, entries,
 839					     value, flags);
 840		if (r)
 841			return r;
 842	}
 843
 844	return vm->update_funcs->commit(&params, NULL);
 845}
 846
 847/**
 848 * amdgpu_vm_bo_param - fill in parameters for PD/PT allocation
 849 *
 850 * @adev: amdgpu_device pointer
 851 * @vm: requesting vm
 852 * @level: the page table level
 853 * @immediate: use a immediate update
 854 * @bp: resulting BO allocation parameters
 855 */
 856static void amdgpu_vm_bo_param(struct amdgpu_device *adev, struct amdgpu_vm *vm,
 857			       int level, bool immediate,
 858			       struct amdgpu_bo_param *bp)
 859{
 860	memset(bp, 0, sizeof(*bp));
 861
 862	bp->size = amdgpu_vm_bo_size(adev, level);
 863	bp->byte_align = AMDGPU_GPU_PAGE_SIZE;
 864	bp->domain = AMDGPU_GEM_DOMAIN_VRAM;
 865	bp->domain = amdgpu_bo_get_preferred_pin_domain(adev, bp->domain);
 866	bp->flags = AMDGPU_GEM_CREATE_VRAM_CONTIGUOUS |
 867		AMDGPU_GEM_CREATE_CPU_GTT_USWC;
 868	if (vm->use_cpu_for_update)
 869		bp->flags |= AMDGPU_GEM_CREATE_CPU_ACCESS_REQUIRED;
 870	else if (!vm->root.base.bo || vm->root.base.bo->shadow)
 871		bp->flags |= AMDGPU_GEM_CREATE_SHADOW;
 872	bp->type = ttm_bo_type_kernel;
 873	bp->no_wait_gpu = immediate;
 874	if (vm->root.base.bo)
 875		bp->resv = vm->root.base.bo->tbo.base.resv;
 876}
 877
 878/**
 879 * amdgpu_vm_alloc_pts - Allocate a specific page table
 880 *
 881 * @adev: amdgpu_device pointer
 882 * @vm: VM to allocate page tables for
 883 * @cursor: Which page table to allocate
 884 * @immediate: use an immediate update
 885 *
 886 * Make sure a specific page table or directory is allocated.
 887 *
 888 * Returns:
 889 * 1 if page table needed to be allocated, 0 if page table was already
 890 * allocated, negative errno if an error occurred.
 891 */
 892static int amdgpu_vm_alloc_pts(struct amdgpu_device *adev,
 893			       struct amdgpu_vm *vm,
 894			       struct amdgpu_vm_pt_cursor *cursor,
 895			       bool immediate)
 896{
 897	struct amdgpu_vm_pt *entry = cursor->entry;
 898	struct amdgpu_bo_param bp;
 899	struct amdgpu_bo *pt;
 900	int r;
 901
 902	if (cursor->level < AMDGPU_VM_PTB && !entry->entries) {
 903		unsigned num_entries;
 904
 905		num_entries = amdgpu_vm_num_entries(adev, cursor->level);
 906		entry->entries = kvmalloc_array(num_entries,
 907						sizeof(*entry->entries),
 908						GFP_KERNEL | __GFP_ZERO);
 909		if (!entry->entries)
 910			return -ENOMEM;
 911	}
 912
 913	if (entry->base.bo)
 914		return 0;
 915
 916	amdgpu_vm_bo_param(adev, vm, cursor->level, immediate, &bp);
 917
 918	r = amdgpu_bo_create(adev, &bp, &pt);
 919	if (r)
 920		return r;
 921
 922	/* Keep a reference to the root directory to avoid
 923	 * freeing them up in the wrong order.
 924	 */
 925	pt->parent = amdgpu_bo_ref(cursor->parent->base.bo);
 926	amdgpu_vm_bo_base_init(&entry->base, vm, pt);
 927
 928	r = amdgpu_vm_clear_bo(adev, vm, pt, immediate);
 929	if (r)
 930		goto error_free_pt;
 931
 932	return 0;
 933
 934error_free_pt:
 935	amdgpu_bo_unref(&pt->shadow);
 936	amdgpu_bo_unref(&pt);
 937	return r;
 938}
 939
 940/**
 941 * amdgpu_vm_free_table - fre one PD/PT
 942 *
 943 * @entry: PDE to free
 944 */
 945static void amdgpu_vm_free_table(struct amdgpu_vm_pt *entry)
 946{
 947	if (entry->base.bo) {
 948		entry->base.bo->vm_bo = NULL;
 949		list_del(&entry->base.vm_status);
 950		amdgpu_bo_unref(&entry->base.bo->shadow);
 951		amdgpu_bo_unref(&entry->base.bo);
 952	}
 953	kvfree(entry->entries);
 954	entry->entries = NULL;
 955}
 956
 957/**
 958 * amdgpu_vm_free_pts - free PD/PT levels
 959 *
 960 * @adev: amdgpu device structure
 961 * @vm: amdgpu vm structure
 962 * @start: optional cursor where to start freeing PDs/PTs
 963 *
 964 * Free the page directory or page table level and all sub levels.
 965 */
 966static void amdgpu_vm_free_pts(struct amdgpu_device *adev,
 967			       struct amdgpu_vm *vm,
 968			       struct amdgpu_vm_pt_cursor *start)
 969{
 970	struct amdgpu_vm_pt_cursor cursor;
 971	struct amdgpu_vm_pt *entry;
 972
 973	vm->bulk_moveable = false;
 974
 975	for_each_amdgpu_vm_pt_dfs_safe(adev, vm, start, cursor, entry)
 976		amdgpu_vm_free_table(entry);
 
 977
 978	if (start)
 979		amdgpu_vm_free_table(start->entry);
 980}
 981
 982/**
 983 * amdgpu_vm_check_compute_bug - check whether asic has compute vm bug
 984 *
 985 * @adev: amdgpu_device pointer
 986 */
 987void amdgpu_vm_check_compute_bug(struct amdgpu_device *adev)
 988{
 989	const struct amdgpu_ip_block *ip_block;
 990	bool has_compute_vm_bug;
 991	struct amdgpu_ring *ring;
 992	int i;
 993
 994	has_compute_vm_bug = false;
 995
 996	ip_block = amdgpu_device_ip_get_ip_block(adev, AMD_IP_BLOCK_TYPE_GFX);
 997	if (ip_block) {
 998		/* Compute has a VM bug for GFX version < 7.
 999		   Compute has a VM bug for GFX 8 MEC firmware version < 673.*/
1000		if (ip_block->version->major <= 7)
1001			has_compute_vm_bug = true;
1002		else if (ip_block->version->major == 8)
1003			if (adev->gfx.mec_fw_version < 673)
1004				has_compute_vm_bug = true;
1005	}
1006
1007	for (i = 0; i < adev->num_rings; i++) {
1008		ring = adev->rings[i];
1009		if (ring->funcs->type == AMDGPU_RING_TYPE_COMPUTE)
1010			/* only compute rings */
1011			ring->has_compute_vm_bug = has_compute_vm_bug;
1012		else
1013			ring->has_compute_vm_bug = false;
1014	}
1015}
1016
1017/**
1018 * amdgpu_vm_need_pipeline_sync - Check if pipe sync is needed for job.
1019 *
1020 * @ring: ring on which the job will be submitted
1021 * @job: job to submit
1022 *
1023 * Returns:
1024 * True if sync is needed.
1025 */
1026bool amdgpu_vm_need_pipeline_sync(struct amdgpu_ring *ring,
1027				  struct amdgpu_job *job)
1028{
1029	struct amdgpu_device *adev = ring->adev;
1030	unsigned vmhub = ring->funcs->vmhub;
1031	struct amdgpu_vmid_mgr *id_mgr = &adev->vm_manager.id_mgr[vmhub];
1032	struct amdgpu_vmid *id;
1033	bool gds_switch_needed;
1034	bool vm_flush_needed = job->vm_needs_flush || ring->has_compute_vm_bug;
1035
1036	if (job->vmid == 0)
1037		return false;
1038	id = &id_mgr->ids[job->vmid];
1039	gds_switch_needed = ring->funcs->emit_gds_switch && (
1040		id->gds_base != job->gds_base ||
1041		id->gds_size != job->gds_size ||
1042		id->gws_base != job->gws_base ||
1043		id->gws_size != job->gws_size ||
1044		id->oa_base != job->oa_base ||
1045		id->oa_size != job->oa_size);
1046
1047	if (amdgpu_vmid_had_gpu_reset(adev, id))
1048		return true;
1049
1050	return vm_flush_needed || gds_switch_needed;
 
 
 
 
 
 
1051}
1052
1053/**
1054 * amdgpu_vm_flush - hardware flush the vm
1055 *
1056 * @ring: ring to use for flush
1057 * @job:  related job
1058 * @need_pipe_sync: is pipe sync needed
1059 *
1060 * Emit a VM flush when it is necessary.
1061 *
1062 * Returns:
1063 * 0 on success, errno otherwise.
1064 */
1065int amdgpu_vm_flush(struct amdgpu_ring *ring, struct amdgpu_job *job,
1066		    bool need_pipe_sync)
1067{
1068	struct amdgpu_device *adev = ring->adev;
1069	unsigned vmhub = ring->funcs->vmhub;
1070	struct amdgpu_vmid_mgr *id_mgr = &adev->vm_manager.id_mgr[vmhub];
1071	struct amdgpu_vmid *id = &id_mgr->ids[job->vmid];
1072	bool gds_switch_needed = ring->funcs->emit_gds_switch && (
1073		id->gds_base != job->gds_base ||
1074		id->gds_size != job->gds_size ||
1075		id->gws_base != job->gws_base ||
1076		id->gws_size != job->gws_size ||
1077		id->oa_base != job->oa_base ||
1078		id->oa_size != job->oa_size);
1079	bool vm_flush_needed = job->vm_needs_flush;
1080	struct dma_fence *fence = NULL;
1081	bool pasid_mapping_needed = false;
1082	unsigned patch_offset = 0;
1083	bool update_spm_vmid_needed = (job->vm && (job->vm->reserved_vmid[vmhub] != NULL));
1084	int r;
1085
1086	if (update_spm_vmid_needed && adev->gfx.rlc.funcs->update_spm_vmid)
1087		adev->gfx.rlc.funcs->update_spm_vmid(adev, job->vmid);
1088
1089	if (amdgpu_vmid_had_gpu_reset(adev, id)) {
1090		gds_switch_needed = true;
1091		vm_flush_needed = true;
1092		pasid_mapping_needed = true;
 
1093	}
1094
1095	mutex_lock(&id_mgr->lock);
1096	if (id->pasid != job->pasid || !id->pasid_mapping ||
1097	    !dma_fence_is_signaled(id->pasid_mapping))
1098		pasid_mapping_needed = true;
1099	mutex_unlock(&id_mgr->lock);
1100
1101	gds_switch_needed &= !!ring->funcs->emit_gds_switch;
1102	vm_flush_needed &= !!ring->funcs->emit_vm_flush  &&
1103			job->vm_pd_addr != AMDGPU_BO_INVALID_OFFSET;
1104	pasid_mapping_needed &= adev->gmc.gmc_funcs->emit_pasid_mapping &&
1105		ring->funcs->emit_wreg;
1106
1107	if (!vm_flush_needed && !gds_switch_needed && !need_pipe_sync)
1108		return 0;
1109
 
1110	if (ring->funcs->init_cond_exec)
1111		patch_offset = amdgpu_ring_init_cond_exec(ring);
1112
1113	if (need_pipe_sync)
1114		amdgpu_ring_emit_pipeline_sync(ring);
1115
1116	if (vm_flush_needed) {
1117		trace_amdgpu_vm_flush(ring, job->vmid, job->vm_pd_addr);
1118		amdgpu_ring_emit_vm_flush(ring, job->vmid, job->vm_pd_addr);
1119	}
1120
1121	if (pasid_mapping_needed)
1122		amdgpu_gmc_emit_pasid_mapping(ring, job->vmid, job->pasid);
1123
 
 
 
 
 
 
 
 
 
 
 
1124	if (vm_flush_needed || pasid_mapping_needed) {
1125		r = amdgpu_fence_emit(ring, &fence, 0);
1126		if (r)
1127			return r;
1128	}
1129
1130	if (vm_flush_needed) {
1131		mutex_lock(&id_mgr->lock);
1132		dma_fence_put(id->last_flush);
1133		id->last_flush = dma_fence_get(fence);
1134		id->current_gpu_reset_count =
1135			atomic_read(&adev->gpu_reset_counter);
1136		mutex_unlock(&id_mgr->lock);
1137	}
1138
1139	if (pasid_mapping_needed) {
1140		mutex_lock(&id_mgr->lock);
1141		id->pasid = job->pasid;
1142		dma_fence_put(id->pasid_mapping);
1143		id->pasid_mapping = dma_fence_get(fence);
1144		mutex_unlock(&id_mgr->lock);
1145	}
1146	dma_fence_put(fence);
1147
1148	if (ring->funcs->emit_gds_switch && gds_switch_needed) {
1149		id->gds_base = job->gds_base;
1150		id->gds_size = job->gds_size;
1151		id->gws_base = job->gws_base;
1152		id->gws_size = job->gws_size;
1153		id->oa_base = job->oa_base;
1154		id->oa_size = job->oa_size;
1155		amdgpu_ring_emit_gds_switch(ring, job->vmid, job->gds_base,
1156					    job->gds_size, job->gws_base,
1157					    job->gws_size, job->oa_base,
1158					    job->oa_size);
1159	}
1160
1161	if (ring->funcs->patch_cond_exec)
1162		amdgpu_ring_patch_cond_exec(ring, patch_offset);
1163
1164	/* the double SWITCH_BUFFER here *cannot* be skipped by COND_EXEC */
1165	if (ring->funcs->emit_switch_buffer) {
1166		amdgpu_ring_emit_switch_buffer(ring);
1167		amdgpu_ring_emit_switch_buffer(ring);
1168	}
 
1169	return 0;
1170}
1171
1172/**
1173 * amdgpu_vm_bo_find - find the bo_va for a specific vm & bo
1174 *
1175 * @vm: requested vm
1176 * @bo: requested buffer object
1177 *
1178 * Find @bo inside the requested vm.
1179 * Search inside the @bos vm list for the requested vm
1180 * Returns the found bo_va or NULL if none is found
1181 *
1182 * Object has to be reserved!
1183 *
1184 * Returns:
1185 * Found bo_va or NULL.
1186 */
1187struct amdgpu_bo_va *amdgpu_vm_bo_find(struct amdgpu_vm *vm,
1188				       struct amdgpu_bo *bo)
1189{
1190	struct amdgpu_vm_bo_base *base;
1191
1192	for (base = bo->vm_bo; base; base = base->next) {
1193		if (base->vm != vm)
1194			continue;
1195
1196		return container_of(base, struct amdgpu_bo_va, base);
1197	}
1198	return NULL;
1199}
1200
1201/**
1202 * amdgpu_vm_map_gart - Resolve gart mapping of addr
1203 *
1204 * @pages_addr: optional DMA address to use for lookup
1205 * @addr: the unmapped addr
1206 *
1207 * Look up the physical address of the page that the pte resolves
1208 * to.
1209 *
1210 * Returns:
1211 * The pointer for the page table entry.
1212 */
1213uint64_t amdgpu_vm_map_gart(const dma_addr_t *pages_addr, uint64_t addr)
1214{
1215	uint64_t result;
1216
1217	/* page table offset */
1218	result = pages_addr[addr >> PAGE_SHIFT];
1219
1220	/* in case cpu page size != gpu page size*/
1221	result |= addr & (~PAGE_MASK);
1222
1223	result &= 0xFFFFFFFFFFFFF000ULL;
1224
1225	return result;
1226}
1227
1228/**
1229 * amdgpu_vm_update_pde - update a single level in the hierarchy
1230 *
1231 * @params: parameters for the update
1232 * @vm: requested vm
1233 * @entry: entry to update
1234 *
1235 * Makes sure the requested entry in parent is up to date.
1236 */
1237static int amdgpu_vm_update_pde(struct amdgpu_vm_update_params *params,
1238				struct amdgpu_vm *vm,
1239				struct amdgpu_vm_pt *entry)
1240{
1241	struct amdgpu_vm_pt *parent = amdgpu_vm_pt_parent(entry);
1242	struct amdgpu_bo *bo = parent->base.bo, *pbo;
1243	uint64_t pde, pt, flags;
1244	unsigned level;
1245
1246	for (level = 0, pbo = bo->parent; pbo; ++level)
1247		pbo = pbo->parent;
1248
1249	level += params->adev->vm_manager.root_level;
1250	amdgpu_gmc_get_pde_for_bo(entry->base.bo, level, &pt, &flags);
1251	pde = (entry - parent->entries) * 8;
1252	return vm->update_funcs->update(params, bo, pde, pt, 1, 0, flags);
1253}
1254
1255/**
1256 * amdgpu_vm_invalidate_pds - mark all PDs as invalid
1257 *
1258 * @adev: amdgpu_device pointer
1259 * @vm: related vm
1260 *
1261 * Mark all PD level as invalid after an error.
1262 */
1263static void amdgpu_vm_invalidate_pds(struct amdgpu_device *adev,
1264				     struct amdgpu_vm *vm)
1265{
1266	struct amdgpu_vm_pt_cursor cursor;
1267	struct amdgpu_vm_pt *entry;
1268
1269	for_each_amdgpu_vm_pt_dfs_safe(adev, vm, NULL, cursor, entry)
1270		if (entry->base.bo && !entry->base.moved)
1271			amdgpu_vm_bo_relocated(&entry->base);
1272}
1273
1274/**
1275 * amdgpu_vm_update_pdes - make sure that all directories are valid
1276 *
1277 * @adev: amdgpu_device pointer
1278 * @vm: requested vm
1279 * @immediate: submit immediately to the paging queue
1280 *
1281 * Makes sure all directories are up to date.
1282 *
1283 * Returns:
1284 * 0 for success, error for failure.
1285 */
1286int amdgpu_vm_update_pdes(struct amdgpu_device *adev,
1287			  struct amdgpu_vm *vm, bool immediate)
1288{
1289	struct amdgpu_vm_update_params params;
1290	int r;
 
 
 
 
 
 
 
1291
1292	if (list_empty(&vm->relocated))
1293		return 0;
1294
 
 
 
1295	memset(&params, 0, sizeof(params));
1296	params.adev = adev;
1297	params.vm = vm;
1298	params.immediate = immediate;
1299
1300	r = vm->update_funcs->prepare(&params, NULL, AMDGPU_SYNC_EXPLICIT);
1301	if (r)
1302		return r;
1303
1304	while (!list_empty(&vm->relocated)) {
1305		struct amdgpu_vm_pt *entry;
1306
1307		entry = list_first_entry(&vm->relocated, struct amdgpu_vm_pt,
1308					 base.vm_status);
1309		amdgpu_vm_bo_idle(&entry->base);
1310
1311		r = amdgpu_vm_update_pde(&params, vm, entry);
1312		if (r)
1313			goto error;
1314	}
1315
1316	r = vm->update_funcs->commit(&params, &vm->last_update);
1317	if (r)
1318		goto error;
1319	return 0;
1320
1321error:
1322	amdgpu_vm_invalidate_pds(adev, vm);
1323	return r;
1324}
1325
1326/*
1327 * amdgpu_vm_update_flags - figure out flags for PTE updates
1328 *
1329 * Make sure to set the right flags for the PTEs at the desired level.
1330 */
1331static void amdgpu_vm_update_flags(struct amdgpu_vm_update_params *params,
1332				   struct amdgpu_bo *bo, unsigned level,
1333				   uint64_t pe, uint64_t addr,
1334				   unsigned count, uint32_t incr,
1335				   uint64_t flags)
1336
1337{
1338	if (level != AMDGPU_VM_PTB) {
1339		flags |= AMDGPU_PDE_PTE;
1340		amdgpu_gmc_get_vm_pde(params->adev, level, &addr, &flags);
1341
1342	} else if (params->adev->asic_type >= CHIP_VEGA10 &&
1343		   !(flags & AMDGPU_PTE_VALID) &&
1344		   !(flags & AMDGPU_PTE_PRT)) {
1345
1346		/* Workaround for fault priority problem on GMC9 */
1347		flags |= AMDGPU_PTE_EXECUTABLE;
 
 
1348	}
1349
1350	params->vm->update_funcs->update(params, bo, pe, addr, count, incr,
1351					 flags);
 
1352}
1353
1354/**
1355 * amdgpu_vm_fragment - get fragment for PTEs
 
 
1356 *
1357 * @params: see amdgpu_vm_update_params definition
1358 * @start: first PTE to handle
1359 * @end: last PTE to handle
1360 * @flags: hw mapping flags
1361 * @frag: resulting fragment size
1362 * @frag_end: end of this fragment
1363 *
1364 * Returns the first possible fragment for the start and end address.
1365 */
1366static void amdgpu_vm_fragment(struct amdgpu_vm_update_params *params,
1367			       uint64_t start, uint64_t end, uint64_t flags,
1368			       unsigned int *frag, uint64_t *frag_end)
1369{
1370	/**
1371	 * The MC L1 TLB supports variable sized pages, based on a fragment
1372	 * field in the PTE. When this field is set to a non-zero value, page
1373	 * granularity is increased from 4KB to (1 << (12 + frag)). The PTE
1374	 * flags are considered valid for all PTEs within the fragment range
1375	 * and corresponding mappings are assumed to be physically contiguous.
1376	 *
1377	 * The L1 TLB can store a single PTE for the whole fragment,
1378	 * significantly increasing the space available for translation
1379	 * caching. This leads to large improvements in throughput when the
1380	 * TLB is under pressure.
1381	 *
1382	 * The L2 TLB distributes small and large fragments into two
1383	 * asymmetric partitions. The large fragment cache is significantly
1384	 * larger. Thus, we try to use large fragments wherever possible.
1385	 * Userspace can support this by aligning virtual base address and
1386	 * allocation size to the fragment size.
1387	 *
1388	 * Starting with Vega10 the fragment size only controls the L1. The L2
1389	 * is now directly feed with small/huge/giant pages from the walker.
1390	 */
1391	unsigned max_frag;
1392
1393	if (params->adev->asic_type < CHIP_VEGA10)
1394		max_frag = params->adev->vm_manager.fragment_size;
1395	else
1396		max_frag = 31;
1397
1398	/* system pages are non continuously */
1399	if (params->pages_addr) {
1400		*frag = 0;
1401		*frag_end = end;
1402		return;
1403	}
1404
1405	/* This intentionally wraps around if no bit is set */
1406	*frag = min((unsigned)ffs(start) - 1, (unsigned)fls64(end - start) - 1);
1407	if (*frag >= max_frag) {
1408		*frag = max_frag;
1409		*frag_end = end & ~((1ULL << max_frag) - 1);
1410	} else {
1411		*frag_end = start + (1 << *frag);
1412	}
1413}
1414
1415/**
1416 * amdgpu_vm_update_ptes - make sure that page tables are valid
1417 *
1418 * @params: see amdgpu_vm_update_params definition
1419 * @start: start of GPU address range
1420 * @end: end of GPU address range
1421 * @dst: destination address to map to, the next dst inside the function
1422 * @flags: mapping flags
1423 *
1424 * Update the page tables in the range @start - @end.
1425 *
1426 * Returns:
1427 * 0 for success, -EINVAL for failure.
1428 */
1429static int amdgpu_vm_update_ptes(struct amdgpu_vm_update_params *params,
1430				 uint64_t start, uint64_t end,
1431				 uint64_t dst, uint64_t flags)
1432{
1433	struct amdgpu_device *adev = params->adev;
1434	struct amdgpu_vm_pt_cursor cursor;
1435	uint64_t frag_start = start, frag_end;
1436	unsigned int frag;
1437	int r;
1438
1439	/* figure out the initial fragment */
1440	amdgpu_vm_fragment(params, frag_start, end, flags, &frag, &frag_end);
1441
1442	/* walk over the address space and update the PTs */
1443	amdgpu_vm_pt_start(adev, params->vm, start, &cursor);
1444	while (cursor.pfn < end) {
1445		unsigned shift, parent_shift, mask;
1446		uint64_t incr, entry_end, pe_start;
1447		struct amdgpu_bo *pt;
1448
1449		if (!params->unlocked) {
1450			/* make sure that the page tables covering the
1451			 * address range are actually allocated
1452			 */
1453			r = amdgpu_vm_alloc_pts(params->adev, params->vm,
1454						&cursor, params->immediate);
1455			if (r)
1456				return r;
1457		}
1458
1459		shift = amdgpu_vm_level_shift(adev, cursor.level);
1460		parent_shift = amdgpu_vm_level_shift(adev, cursor.level - 1);
1461		if (params->unlocked) {
1462			/* Unlocked updates are only allowed on the leaves */
1463			if (amdgpu_vm_pt_descendant(adev, &cursor))
1464				continue;
1465		} else if (adev->asic_type < CHIP_VEGA10 &&
1466			   (flags & AMDGPU_PTE_VALID)) {
1467			/* No huge page support before GMC v9 */
1468			if (cursor.level != AMDGPU_VM_PTB) {
1469				if (!amdgpu_vm_pt_descendant(adev, &cursor))
1470					return -ENOENT;
1471				continue;
1472			}
1473		} else if (frag < shift) {
1474			/* We can't use this level when the fragment size is
1475			 * smaller than the address shift. Go to the next
1476			 * child entry and try again.
1477			 */
1478			if (amdgpu_vm_pt_descendant(adev, &cursor))
1479				continue;
1480		} else if (frag >= parent_shift) {
1481			/* If the fragment size is even larger than the parent
1482			 * shift we should go up one level and check it again.
1483			 */
1484			if (!amdgpu_vm_pt_ancestor(&cursor))
1485				return -EINVAL;
1486			continue;
1487		}
1488
1489		pt = cursor.entry->base.bo;
1490		if (!pt) {
1491			/* We need all PDs and PTs for mapping something, */
1492			if (flags & AMDGPU_PTE_VALID)
1493				return -ENOENT;
1494
1495			/* but unmapping something can happen at a higher
1496			 * level.
1497			 */
1498			if (!amdgpu_vm_pt_ancestor(&cursor))
1499				return -EINVAL;
1500
1501			pt = cursor.entry->base.bo;
1502			shift = parent_shift;
1503		}
1504
1505		/* Looks good so far, calculate parameters for the update */
1506		incr = (uint64_t)AMDGPU_GPU_PAGE_SIZE << shift;
1507		mask = amdgpu_vm_entries_mask(adev, cursor.level);
1508		pe_start = ((cursor.pfn >> shift) & mask) * 8;
1509		entry_end = ((uint64_t)mask + 1) << shift;
1510		entry_end += cursor.pfn & ~(entry_end - 1);
1511		entry_end = min(entry_end, end);
1512
1513		do {
1514			uint64_t upd_end = min(entry_end, frag_end);
1515			unsigned nptes = (upd_end - frag_start) >> shift;
1516
1517			/* This can happen when we set higher level PDs to
1518			 * silent to stop fault floods.
1519			 */
1520			nptes = max(nptes, 1u);
1521			amdgpu_vm_update_flags(params, pt, cursor.level,
1522					       pe_start, dst, nptes, incr,
1523					       flags | AMDGPU_PTE_FRAG(frag));
1524
1525			pe_start += nptes * 8;
1526			dst += (uint64_t)nptes * AMDGPU_GPU_PAGE_SIZE << shift;
1527
1528			frag_start = upd_end;
1529			if (frag_start >= frag_end) {
1530				/* figure out the next fragment */
1531				amdgpu_vm_fragment(params, frag_start, end,
1532						   flags, &frag, &frag_end);
1533				if (frag < shift)
1534					break;
1535			}
1536		} while (frag_start < entry_end);
1537
1538		if (amdgpu_vm_pt_descendant(adev, &cursor)) {
1539			/* Free all child entries.
1540			 * Update the tables with the flags and addresses and free up subsequent
1541			 * tables in the case of huge pages or freed up areas.
1542			 * This is the maximum you can free, because all other page tables are not
1543			 * completely covered by the range and so potentially still in use.
1544			 */
1545			while (cursor.pfn < frag_start) {
1546				amdgpu_vm_free_pts(adev, params->vm, &cursor);
1547				amdgpu_vm_pt_next(adev, &cursor);
1548			}
1549
1550		} else if (frag >= shift) {
1551			/* or just move on to the next on the same level. */
1552			amdgpu_vm_pt_next(adev, &cursor);
1553		}
1554	}
1555
1556	return 0;
1557}
1558
1559/**
1560 * amdgpu_vm_bo_update_mapping - update a mapping in the vm page table
1561 *
1562 * @adev: amdgpu_device pointer
1563 * @vm: requested vm
1564 * @immediate: immediate submission in a page fault
1565 * @unlocked: unlocked invalidation during MM callback
 
1566 * @resv: fences we need to sync to
1567 * @start: start of mapped range
1568 * @last: last mapped entry
1569 * @flags: flags for the entries
1570 * @addr: addr to set the area to
 
 
1571 * @pages_addr: DMA addresses to use for mapping
1572 * @fence: optional resulting fence
1573 *
1574 * Fill in the page table entries between @start and @last.
1575 *
1576 * Returns:
1577 * 0 for success, -EINVAL for failure.
1578 */
1579static int amdgpu_vm_bo_update_mapping(struct amdgpu_device *adev,
1580				       struct amdgpu_vm *vm, bool immediate,
1581				       bool unlocked, struct dma_resv *resv,
1582				       uint64_t start, uint64_t last,
1583				       uint64_t flags, uint64_t addr,
1584				       dma_addr_t *pages_addr,
1585				       struct dma_fence **fence)
1586{
1587	struct amdgpu_vm_update_params params;
 
 
1588	enum amdgpu_sync_mode sync_mode;
1589	int r;
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1590
1591	memset(&params, 0, sizeof(params));
1592	params.adev = adev;
1593	params.vm = vm;
1594	params.immediate = immediate;
1595	params.pages_addr = pages_addr;
1596	params.unlocked = unlocked;
1597
1598	/* Implicitly sync to command submissions in the same VM before
1599	 * unmapping. Sync to moving fences before mapping.
1600	 */
1601	if (!(flags & AMDGPU_PTE_VALID))
1602		sync_mode = AMDGPU_SYNC_EQ_OWNER;
1603	else
1604		sync_mode = AMDGPU_SYNC_EXPLICIT;
1605
1606	amdgpu_vm_eviction_lock(vm);
1607	if (vm->evicting) {
1608		r = -EBUSY;
1609		goto error_unlock;
1610	}
1611
1612	if (!unlocked && !dma_fence_is_signaled(vm->last_unlocked)) {
1613		struct dma_fence *tmp = dma_fence_get_stub();
1614
1615		amdgpu_bo_fence(vm->root.base.bo, vm->last_unlocked, true);
1616		swap(vm->last_unlocked, tmp);
1617		dma_fence_put(tmp);
1618	}
1619
1620	r = vm->update_funcs->prepare(&params, resv, sync_mode);
1621	if (r)
1622		goto error_unlock;
1623
1624	r = amdgpu_vm_update_ptes(&params, start, last + 1, addr, flags);
1625	if (r)
1626		goto error_unlock;
1627
1628	r = vm->update_funcs->commit(&params, fence);
1629
1630error_unlock:
1631	amdgpu_vm_eviction_unlock(vm);
1632	return r;
1633}
1634
1635/**
1636 * amdgpu_vm_bo_split_mapping - split a mapping into smaller chunks
1637 *
1638 * @adev: amdgpu_device pointer
1639 * @resv: fences we need to sync to
1640 * @pages_addr: DMA addresses to use for mapping
1641 * @vm: requested vm
1642 * @mapping: mapped range and flags to use for the update
1643 * @flags: HW flags for the mapping
1644 * @bo_adev: amdgpu_device pointer that bo actually been allocated
1645 * @nodes: array of drm_mm_nodes with the MC addresses
1646 * @fence: optional resulting fence
1647 *
1648 * Split the mapping into smaller chunks so that each update fits
1649 * into a SDMA IB.
1650 *
1651 * Returns:
1652 * 0 for success, -EINVAL for failure.
1653 */
1654static int amdgpu_vm_bo_split_mapping(struct amdgpu_device *adev,
1655				      struct dma_resv *resv,
1656				      dma_addr_t *pages_addr,
1657				      struct amdgpu_vm *vm,
1658				      struct amdgpu_bo_va_mapping *mapping,
1659				      uint64_t flags,
1660				      struct amdgpu_device *bo_adev,
1661				      struct drm_mm_node *nodes,
1662				      struct dma_fence **fence)
1663{
1664	unsigned min_linear_pages = 1 << adev->vm_manager.fragment_size;
1665	uint64_t pfn, start = mapping->start;
1666	int r;
1667
1668	/* normally,bo_va->flags only contians READABLE and WIRTEABLE bit go here
1669	 * but in case of something, we filter the flags in first place
1670	 */
1671	if (!(mapping->flags & AMDGPU_PTE_READABLE))
1672		flags &= ~AMDGPU_PTE_READABLE;
1673	if (!(mapping->flags & AMDGPU_PTE_WRITEABLE))
1674		flags &= ~AMDGPU_PTE_WRITEABLE;
1675
1676	/* Apply ASIC specific mapping flags */
1677	amdgpu_gmc_get_vm_pte(adev, mapping, &flags);
1678
1679	trace_amdgpu_vm_bo_update(mapping);
1680
1681	pfn = mapping->offset >> PAGE_SHIFT;
1682	if (nodes) {
1683		while (pfn >= nodes->size) {
1684			pfn -= nodes->size;
1685			++nodes;
1686		}
1687	}
1688
1689	do {
1690		dma_addr_t *dma_addr = NULL;
1691		uint64_t max_entries;
1692		uint64_t addr, last;
1693
1694		if (nodes) {
1695			addr = nodes->start << PAGE_SHIFT;
1696			max_entries = (nodes->size - pfn) *
1697				AMDGPU_GPU_PAGES_IN_CPU_PAGE;
1698		} else {
1699			addr = 0;
1700			max_entries = S64_MAX;
1701		}
1702
1703		if (pages_addr) {
1704			uint64_t count;
 
 
1705
1706			for (count = 1;
1707			     count < max_entries / AMDGPU_GPU_PAGES_IN_CPU_PAGE;
1708			     ++count) {
1709				uint64_t idx = pfn + count;
1710
1711				if (pages_addr[idx] !=
1712				    (pages_addr[idx - 1] + PAGE_SIZE))
1713					break;
1714			}
1715
1716			if (count < min_linear_pages) {
1717				addr = pfn << PAGE_SHIFT;
1718				dma_addr = pages_addr;
1719			} else {
1720				addr = pages_addr[pfn];
1721				max_entries = count *
1722					AMDGPU_GPU_PAGES_IN_CPU_PAGE;
1723			}
1724
1725		} else if (flags & (AMDGPU_PTE_VALID | AMDGPU_PTE_PRT)) {
1726			addr += bo_adev->vm_manager.vram_base_offset;
1727			addr += pfn << PAGE_SHIFT;
 
1728		}
1729
1730		last = min((uint64_t)mapping->last, start + max_entries - 1);
1731		r = amdgpu_vm_bo_update_mapping(adev, vm, false, false, resv,
1732						start, last, flags, addr,
1733						dma_addr, fence);
1734		if (r)
1735			return r;
 
 
 
 
 
 
1736
1737		pfn += (last - start + 1) / AMDGPU_GPU_PAGES_IN_CPU_PAGE;
1738		if (nodes && nodes->size == pfn) {
1739			pfn = 0;
1740			++nodes;
 
 
 
 
 
1741		}
1742		start = last + 1;
 
1743
1744	} while (unlikely(start != mapping->last + 1));
 
1745
1746	return 0;
 
 
 
1747}
1748
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1749/**
1750 * amdgpu_vm_bo_update - update all BO mappings in the vm page table
1751 *
1752 * @adev: amdgpu_device pointer
1753 * @bo_va: requested BO and VM object
1754 * @clear: if true clear the entries
1755 *
1756 * Fill in the page table entries for @bo_va.
1757 *
1758 * Returns:
1759 * 0 for success, -EINVAL for failure.
1760 */
1761int amdgpu_vm_bo_update(struct amdgpu_device *adev, struct amdgpu_bo_va *bo_va,
1762			bool clear)
1763{
1764	struct amdgpu_bo *bo = bo_va->base.bo;
1765	struct amdgpu_vm *vm = bo_va->base.vm;
1766	struct amdgpu_bo_va_mapping *mapping;
1767	dma_addr_t *pages_addr = NULL;
1768	struct ttm_mem_reg *mem;
1769	struct drm_mm_node *nodes;
1770	struct dma_fence **last_update;
 
1771	struct dma_resv *resv;
 
1772	uint64_t flags;
1773	struct amdgpu_device *bo_adev = adev;
1774	int r;
1775
1776	if (clear || !bo) {
1777		mem = NULL;
1778		nodes = NULL;
1779		resv = vm->root.base.bo->tbo.base.resv;
1780	} else {
1781		struct ttm_dma_tt *ttm;
1782
1783		mem = &bo->tbo.mem;
1784		nodes = mem->mm_node;
1785		if (mem->mem_type == TTM_PL_TT) {
1786			ttm = container_of(bo->tbo.ttm, struct ttm_dma_tt, ttm);
1787			pages_addr = ttm->dma_address;
1788		}
1789		resv = bo->tbo.base.resv;
 
 
 
 
 
 
 
 
 
 
 
 
1790	}
1791
1792	if (bo) {
 
 
1793		flags = amdgpu_ttm_tt_pte_flags(adev, bo->tbo.ttm, mem);
1794
1795		if (amdgpu_bo_encrypted(bo))
1796			flags |= AMDGPU_PTE_TMZ;
1797
1798		bo_adev = amdgpu_ttm_adev(bo->tbo.bdev);
 
1799	} else {
1800		flags = 0x0;
 
1801	}
1802
1803	if (clear || (bo && bo->tbo.base.resv ==
1804		      vm->root.base.bo->tbo.base.resv))
1805		last_update = &vm->last_update;
1806	else
1807		last_update = &bo_va->last_pt_update;
1808
1809	if (!clear && bo_va->base.moved) {
1810		bo_va->base.moved = false;
1811		list_splice_init(&bo_va->valids, &bo_va->invalids);
1812
1813	} else if (bo_va->cleared != clear) {
1814		list_splice_init(&bo_va->valids, &bo_va->invalids);
1815	}
1816
1817	list_for_each_entry(mapping, &bo_va->invalids, list) {
1818		r = amdgpu_vm_bo_split_mapping(adev, resv, pages_addr, vm,
1819					       mapping, flags, bo_adev, nodes,
1820					       last_update);
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1821		if (r)
1822			return r;
1823	}
1824
1825	/* If the BO is not in its preferred location add it back to
1826	 * the evicted list so that it gets validated again on the
1827	 * next command submission.
1828	 */
1829	if (bo && bo->tbo.base.resv == vm->root.base.bo->tbo.base.resv) {
1830		uint32_t mem_type = bo->tbo.mem.mem_type;
1831
1832		if (!(bo->preferred_domains &
1833		      amdgpu_mem_type_to_domain(mem_type)))
1834			amdgpu_vm_bo_evicted(&bo_va->base);
1835		else
1836			amdgpu_vm_bo_idle(&bo_va->base);
1837	} else {
1838		amdgpu_vm_bo_done(&bo_va->base);
1839	}
1840
1841	list_splice_init(&bo_va->invalids, &bo_va->valids);
1842	bo_va->cleared = clear;
 
1843
1844	if (trace_amdgpu_vm_bo_mapping_enabled()) {
1845		list_for_each_entry(mapping, &bo_va->valids, list)
1846			trace_amdgpu_vm_bo_mapping(mapping);
1847	}
1848
1849	return 0;
1850}
1851
1852/**
1853 * amdgpu_vm_update_prt_state - update the global PRT state
1854 *
1855 * @adev: amdgpu_device pointer
1856 */
1857static void amdgpu_vm_update_prt_state(struct amdgpu_device *adev)
1858{
1859	unsigned long flags;
1860	bool enable;
1861
1862	spin_lock_irqsave(&adev->vm_manager.prt_lock, flags);
1863	enable = !!atomic_read(&adev->vm_manager.num_prt_users);
1864	adev->gmc.gmc_funcs->set_prt(adev, enable);
1865	spin_unlock_irqrestore(&adev->vm_manager.prt_lock, flags);
1866}
1867
1868/**
1869 * amdgpu_vm_prt_get - add a PRT user
1870 *
1871 * @adev: amdgpu_device pointer
1872 */
1873static void amdgpu_vm_prt_get(struct amdgpu_device *adev)
1874{
1875	if (!adev->gmc.gmc_funcs->set_prt)
1876		return;
1877
1878	if (atomic_inc_return(&adev->vm_manager.num_prt_users) == 1)
1879		amdgpu_vm_update_prt_state(adev);
1880}
1881
1882/**
1883 * amdgpu_vm_prt_put - drop a PRT user
1884 *
1885 * @adev: amdgpu_device pointer
1886 */
1887static void amdgpu_vm_prt_put(struct amdgpu_device *adev)
1888{
1889	if (atomic_dec_return(&adev->vm_manager.num_prt_users) == 0)
1890		amdgpu_vm_update_prt_state(adev);
1891}
1892
1893/**
1894 * amdgpu_vm_prt_cb - callback for updating the PRT status
1895 *
1896 * @fence: fence for the callback
1897 * @_cb: the callback function
1898 */
1899static void amdgpu_vm_prt_cb(struct dma_fence *fence, struct dma_fence_cb *_cb)
1900{
1901	struct amdgpu_prt_cb *cb = container_of(_cb, struct amdgpu_prt_cb, cb);
1902
1903	amdgpu_vm_prt_put(cb->adev);
1904	kfree(cb);
1905}
1906
1907/**
1908 * amdgpu_vm_add_prt_cb - add callback for updating the PRT status
1909 *
1910 * @adev: amdgpu_device pointer
1911 * @fence: fence for the callback
1912 */
1913static void amdgpu_vm_add_prt_cb(struct amdgpu_device *adev,
1914				 struct dma_fence *fence)
1915{
1916	struct amdgpu_prt_cb *cb;
1917
1918	if (!adev->gmc.gmc_funcs->set_prt)
1919		return;
1920
1921	cb = kmalloc(sizeof(struct amdgpu_prt_cb), GFP_KERNEL);
1922	if (!cb) {
1923		/* Last resort when we are OOM */
1924		if (fence)
1925			dma_fence_wait(fence, false);
1926
1927		amdgpu_vm_prt_put(adev);
1928	} else {
1929		cb->adev = adev;
1930		if (!fence || dma_fence_add_callback(fence, &cb->cb,
1931						     amdgpu_vm_prt_cb))
1932			amdgpu_vm_prt_cb(fence, &cb->cb);
1933	}
1934}
1935
1936/**
1937 * amdgpu_vm_free_mapping - free a mapping
1938 *
1939 * @adev: amdgpu_device pointer
1940 * @vm: requested vm
1941 * @mapping: mapping to be freed
1942 * @fence: fence of the unmap operation
1943 *
1944 * Free a mapping and make sure we decrease the PRT usage count if applicable.
1945 */
1946static void amdgpu_vm_free_mapping(struct amdgpu_device *adev,
1947				   struct amdgpu_vm *vm,
1948				   struct amdgpu_bo_va_mapping *mapping,
1949				   struct dma_fence *fence)
1950{
1951	if (mapping->flags & AMDGPU_PTE_PRT)
1952		amdgpu_vm_add_prt_cb(adev, fence);
1953	kfree(mapping);
1954}
1955
1956/**
1957 * amdgpu_vm_prt_fini - finish all prt mappings
1958 *
1959 * @adev: amdgpu_device pointer
1960 * @vm: requested vm
1961 *
1962 * Register a cleanup callback to disable PRT support after VM dies.
1963 */
1964static void amdgpu_vm_prt_fini(struct amdgpu_device *adev, struct amdgpu_vm *vm)
1965{
1966	struct dma_resv *resv = vm->root.base.bo->tbo.base.resv;
1967	struct dma_fence *excl, **shared;
1968	unsigned i, shared_count;
1969	int r;
1970
1971	r = dma_resv_get_fences_rcu(resv, &excl,
1972					      &shared_count, &shared);
1973	if (r) {
1974		/* Not enough memory to grab the fence list, as last resort
1975		 * block for all the fences to complete.
1976		 */
1977		dma_resv_wait_timeout_rcu(resv, true, false,
1978						    MAX_SCHEDULE_TIMEOUT);
1979		return;
1980	}
1981
1982	/* Add a callback for each fence in the reservation object */
1983	amdgpu_vm_prt_get(adev);
1984	amdgpu_vm_add_prt_cb(adev, excl);
1985
1986	for (i = 0; i < shared_count; ++i) {
1987		amdgpu_vm_prt_get(adev);
1988		amdgpu_vm_add_prt_cb(adev, shared[i]);
1989	}
1990
1991	kfree(shared);
1992}
1993
1994/**
1995 * amdgpu_vm_clear_freed - clear freed BOs in the PT
1996 *
1997 * @adev: amdgpu_device pointer
1998 * @vm: requested vm
1999 * @fence: optional resulting fence (unchanged if no work needed to be done
2000 * or if an error occurred)
2001 *
2002 * Make sure all freed BOs are cleared in the PT.
2003 * PTs have to be reserved and mutex must be locked!
2004 *
2005 * Returns:
2006 * 0 for success.
2007 *
2008 */
2009int amdgpu_vm_clear_freed(struct amdgpu_device *adev,
2010			  struct amdgpu_vm *vm,
2011			  struct dma_fence **fence)
2012{
2013	struct dma_resv *resv = vm->root.base.bo->tbo.base.resv;
2014	struct amdgpu_bo_va_mapping *mapping;
2015	uint64_t init_pte_value = 0;
2016	struct dma_fence *f = NULL;
2017	int r;
2018
2019	while (!list_empty(&vm->freed)) {
2020		mapping = list_first_entry(&vm->freed,
2021			struct amdgpu_bo_va_mapping, list);
2022		list_del(&mapping->list);
2023
2024		if (vm->pte_support_ats &&
2025		    mapping->start < AMDGPU_GMC_HOLE_START)
2026			init_pte_value = AMDGPU_PTE_DEFAULT_ATC;
2027
2028		r = amdgpu_vm_bo_update_mapping(adev, vm, false, false, resv,
2029						mapping->start, mapping->last,
2030						init_pte_value, 0, NULL, &f);
 
2031		amdgpu_vm_free_mapping(adev, vm, mapping, f);
2032		if (r) {
2033			dma_fence_put(f);
2034			return r;
2035		}
2036	}
2037
2038	if (fence && f) {
2039		dma_fence_put(*fence);
2040		*fence = f;
2041	} else {
2042		dma_fence_put(f);
2043	}
2044
2045	return 0;
2046
2047}
2048
2049/**
2050 * amdgpu_vm_handle_moved - handle moved BOs in the PT
2051 *
2052 * @adev: amdgpu_device pointer
2053 * @vm: requested vm
2054 *
2055 * Make sure all BOs which are moved are updated in the PTs.
2056 *
2057 * Returns:
2058 * 0 for success.
2059 *
2060 * PTs have to be reserved!
2061 */
2062int amdgpu_vm_handle_moved(struct amdgpu_device *adev,
2063			   struct amdgpu_vm *vm)
2064{
2065	struct amdgpu_bo_va *bo_va, *tmp;
2066	struct dma_resv *resv;
2067	bool clear;
2068	int r;
2069
2070	list_for_each_entry_safe(bo_va, tmp, &vm->moved, base.vm_status) {
 
 
 
 
 
2071		/* Per VM BOs never need to bo cleared in the page tables */
2072		r = amdgpu_vm_bo_update(adev, bo_va, false);
2073		if (r)
2074			return r;
 
2075	}
2076
2077	spin_lock(&vm->invalidated_lock);
2078	while (!list_empty(&vm->invalidated)) {
2079		bo_va = list_first_entry(&vm->invalidated, struct amdgpu_bo_va,
2080					 base.vm_status);
2081		resv = bo_va->base.bo->tbo.base.resv;
2082		spin_unlock(&vm->invalidated_lock);
2083
2084		/* Try to reserve the BO to avoid clearing its ptes */
2085		if (!amdgpu_vm_debug && dma_resv_trylock(resv))
2086			clear = false;
2087		/* Somebody else is using the BO right now */
2088		else
2089			clear = true;
2090
2091		r = amdgpu_vm_bo_update(adev, bo_va, clear);
2092		if (r)
2093			return r;
2094
2095		if (!clear)
2096			dma_resv_unlock(resv);
2097		spin_lock(&vm->invalidated_lock);
2098	}
2099	spin_unlock(&vm->invalidated_lock);
2100
2101	return 0;
2102}
2103
2104/**
2105 * amdgpu_vm_bo_add - add a bo to a specific vm
2106 *
2107 * @adev: amdgpu_device pointer
2108 * @vm: requested vm
2109 * @bo: amdgpu buffer object
2110 *
2111 * Add @bo into the requested vm.
2112 * Add @bo to the list of bos associated with the vm
2113 *
2114 * Returns:
2115 * Newly added bo_va or NULL for failure
2116 *
2117 * Object has to be reserved!
2118 */
2119struct amdgpu_bo_va *amdgpu_vm_bo_add(struct amdgpu_device *adev,
2120				      struct amdgpu_vm *vm,
2121				      struct amdgpu_bo *bo)
2122{
2123	struct amdgpu_bo_va *bo_va;
2124
2125	bo_va = kzalloc(sizeof(struct amdgpu_bo_va), GFP_KERNEL);
2126	if (bo_va == NULL) {
2127		return NULL;
2128	}
2129	amdgpu_vm_bo_base_init(&bo_va->base, vm, bo);
2130
2131	bo_va->ref_count = 1;
2132	INIT_LIST_HEAD(&bo_va->valids);
2133	INIT_LIST_HEAD(&bo_va->invalids);
2134
2135	if (bo && amdgpu_xgmi_same_hive(adev, amdgpu_ttm_adev(bo->tbo.bdev)) &&
2136	    (bo->preferred_domains & AMDGPU_GEM_DOMAIN_VRAM)) {
 
 
 
2137		bo_va->is_xgmi = true;
2138		/* Power up XGMI if it can be potentially used */
2139		amdgpu_xgmi_set_pstate(adev, AMDGPU_XGMI_PSTATE_MAX_VEGA20);
2140	}
2141
2142	return bo_va;
2143}
2144
2145
2146/**
2147 * amdgpu_vm_bo_insert_mapping - insert a new mapping
2148 *
2149 * @adev: amdgpu_device pointer
2150 * @bo_va: bo_va to store the address
2151 * @mapping: the mapping to insert
2152 *
2153 * Insert a new mapping into all structures.
2154 */
2155static void amdgpu_vm_bo_insert_map(struct amdgpu_device *adev,
2156				    struct amdgpu_bo_va *bo_va,
2157				    struct amdgpu_bo_va_mapping *mapping)
2158{
2159	struct amdgpu_vm *vm = bo_va->base.vm;
2160	struct amdgpu_bo *bo = bo_va->base.bo;
2161
2162	mapping->bo_va = bo_va;
2163	list_add(&mapping->list, &bo_va->invalids);
2164	amdgpu_vm_it_insert(mapping, &vm->va);
2165
2166	if (mapping->flags & AMDGPU_PTE_PRT)
2167		amdgpu_vm_prt_get(adev);
2168
2169	if (bo && bo->tbo.base.resv == vm->root.base.bo->tbo.base.resv &&
2170	    !bo_va->base.moved) {
2171		list_move(&bo_va->base.vm_status, &vm->moved);
2172	}
2173	trace_amdgpu_vm_bo_map(bo_va, mapping);
2174}
2175
2176/**
2177 * amdgpu_vm_bo_map - map bo inside a vm
2178 *
2179 * @adev: amdgpu_device pointer
2180 * @bo_va: bo_va to store the address
2181 * @saddr: where to map the BO
2182 * @offset: requested offset in the BO
2183 * @size: BO size in bytes
2184 * @flags: attributes of pages (read/write/valid/etc.)
2185 *
2186 * Add a mapping of the BO at the specefied addr into the VM.
2187 *
2188 * Returns:
2189 * 0 for success, error for failure.
2190 *
2191 * Object has to be reserved and unreserved outside!
2192 */
2193int amdgpu_vm_bo_map(struct amdgpu_device *adev,
2194		     struct amdgpu_bo_va *bo_va,
2195		     uint64_t saddr, uint64_t offset,
2196		     uint64_t size, uint64_t flags)
2197{
2198	struct amdgpu_bo_va_mapping *mapping, *tmp;
2199	struct amdgpu_bo *bo = bo_va->base.bo;
2200	struct amdgpu_vm *vm = bo_va->base.vm;
2201	uint64_t eaddr;
2202
2203	/* validate the parameters */
2204	if (saddr & AMDGPU_GPU_PAGE_MASK || offset & AMDGPU_GPU_PAGE_MASK ||
2205	    size == 0 || size & AMDGPU_GPU_PAGE_MASK)
2206		return -EINVAL;
2207
2208	/* make sure object fit at this offset */
2209	eaddr = saddr + size - 1;
2210	if (saddr >= eaddr ||
2211	    (bo && offset + size > amdgpu_bo_size(bo)) ||
2212	    (eaddr >= adev->vm_manager.max_pfn << AMDGPU_GPU_PAGE_SHIFT))
2213		return -EINVAL;
2214
2215	saddr /= AMDGPU_GPU_PAGE_SIZE;
2216	eaddr /= AMDGPU_GPU_PAGE_SIZE;
2217
2218	tmp = amdgpu_vm_it_iter_first(&vm->va, saddr, eaddr);
2219	if (tmp) {
2220		/* bo and tmp overlap, invalid addr */
2221		dev_err(adev->dev, "bo %p va 0x%010Lx-0x%010Lx conflict with "
2222			"0x%010Lx-0x%010Lx\n", bo, saddr, eaddr,
2223			tmp->start, tmp->last + 1);
2224		return -EINVAL;
2225	}
2226
2227	mapping = kmalloc(sizeof(*mapping), GFP_KERNEL);
2228	if (!mapping)
2229		return -ENOMEM;
2230
2231	mapping->start = saddr;
2232	mapping->last = eaddr;
2233	mapping->offset = offset;
2234	mapping->flags = flags;
2235
2236	amdgpu_vm_bo_insert_map(adev, bo_va, mapping);
2237
2238	return 0;
2239}
2240
2241/**
2242 * amdgpu_vm_bo_replace_map - map bo inside a vm, replacing existing mappings
2243 *
2244 * @adev: amdgpu_device pointer
2245 * @bo_va: bo_va to store the address
2246 * @saddr: where to map the BO
2247 * @offset: requested offset in the BO
2248 * @size: BO size in bytes
2249 * @flags: attributes of pages (read/write/valid/etc.)
2250 *
2251 * Add a mapping of the BO at the specefied addr into the VM. Replace existing
2252 * mappings as we do so.
2253 *
2254 * Returns:
2255 * 0 for success, error for failure.
2256 *
2257 * Object has to be reserved and unreserved outside!
2258 */
2259int amdgpu_vm_bo_replace_map(struct amdgpu_device *adev,
2260			     struct amdgpu_bo_va *bo_va,
2261			     uint64_t saddr, uint64_t offset,
2262			     uint64_t size, uint64_t flags)
2263{
2264	struct amdgpu_bo_va_mapping *mapping;
2265	struct amdgpu_bo *bo = bo_va->base.bo;
2266	uint64_t eaddr;
2267	int r;
2268
2269	/* validate the parameters */
2270	if (saddr & AMDGPU_GPU_PAGE_MASK || offset & AMDGPU_GPU_PAGE_MASK ||
2271	    size == 0 || size & AMDGPU_GPU_PAGE_MASK)
2272		return -EINVAL;
2273
2274	/* make sure object fit at this offset */
2275	eaddr = saddr + size - 1;
2276	if (saddr >= eaddr ||
2277	    (bo && offset + size > amdgpu_bo_size(bo)) ||
2278	    (eaddr >= adev->vm_manager.max_pfn << AMDGPU_GPU_PAGE_SHIFT))
2279		return -EINVAL;
2280
2281	/* Allocate all the needed memory */
2282	mapping = kmalloc(sizeof(*mapping), GFP_KERNEL);
2283	if (!mapping)
2284		return -ENOMEM;
2285
2286	r = amdgpu_vm_bo_clear_mappings(adev, bo_va->base.vm, saddr, size);
2287	if (r) {
2288		kfree(mapping);
2289		return r;
2290	}
2291
2292	saddr /= AMDGPU_GPU_PAGE_SIZE;
2293	eaddr /= AMDGPU_GPU_PAGE_SIZE;
2294
2295	mapping->start = saddr;
2296	mapping->last = eaddr;
2297	mapping->offset = offset;
2298	mapping->flags = flags;
2299
2300	amdgpu_vm_bo_insert_map(adev, bo_va, mapping);
2301
2302	return 0;
2303}
2304
2305/**
2306 * amdgpu_vm_bo_unmap - remove bo mapping from vm
2307 *
2308 * @adev: amdgpu_device pointer
2309 * @bo_va: bo_va to remove the address from
2310 * @saddr: where to the BO is mapped
2311 *
2312 * Remove a mapping of the BO at the specefied addr from the VM.
2313 *
2314 * Returns:
2315 * 0 for success, error for failure.
2316 *
2317 * Object has to be reserved and unreserved outside!
2318 */
2319int amdgpu_vm_bo_unmap(struct amdgpu_device *adev,
2320		       struct amdgpu_bo_va *bo_va,
2321		       uint64_t saddr)
2322{
2323	struct amdgpu_bo_va_mapping *mapping;
2324	struct amdgpu_vm *vm = bo_va->base.vm;
2325	bool valid = true;
2326
2327	saddr /= AMDGPU_GPU_PAGE_SIZE;
2328
2329	list_for_each_entry(mapping, &bo_va->valids, list) {
2330		if (mapping->start == saddr)
2331			break;
2332	}
2333
2334	if (&mapping->list == &bo_va->valids) {
2335		valid = false;
2336
2337		list_for_each_entry(mapping, &bo_va->invalids, list) {
2338			if (mapping->start == saddr)
2339				break;
2340		}
2341
2342		if (&mapping->list == &bo_va->invalids)
2343			return -ENOENT;
2344	}
2345
2346	list_del(&mapping->list);
2347	amdgpu_vm_it_remove(mapping, &vm->va);
2348	mapping->bo_va = NULL;
2349	trace_amdgpu_vm_bo_unmap(bo_va, mapping);
2350
2351	if (valid)
2352		list_add(&mapping->list, &vm->freed);
2353	else
2354		amdgpu_vm_free_mapping(adev, vm, mapping,
2355				       bo_va->last_pt_update);
2356
2357	return 0;
2358}
2359
2360/**
2361 * amdgpu_vm_bo_clear_mappings - remove all mappings in a specific range
2362 *
2363 * @adev: amdgpu_device pointer
2364 * @vm: VM structure to use
2365 * @saddr: start of the range
2366 * @size: size of the range
2367 *
2368 * Remove all mappings in a range, split them as appropriate.
2369 *
2370 * Returns:
2371 * 0 for success, error for failure.
2372 */
2373int amdgpu_vm_bo_clear_mappings(struct amdgpu_device *adev,
2374				struct amdgpu_vm *vm,
2375				uint64_t saddr, uint64_t size)
2376{
2377	struct amdgpu_bo_va_mapping *before, *after, *tmp, *next;
2378	LIST_HEAD(removed);
2379	uint64_t eaddr;
2380
2381	eaddr = saddr + size - 1;
2382	saddr /= AMDGPU_GPU_PAGE_SIZE;
2383	eaddr /= AMDGPU_GPU_PAGE_SIZE;
2384
2385	/* Allocate all the needed memory */
2386	before = kzalloc(sizeof(*before), GFP_KERNEL);
2387	if (!before)
2388		return -ENOMEM;
2389	INIT_LIST_HEAD(&before->list);
2390
2391	after = kzalloc(sizeof(*after), GFP_KERNEL);
2392	if (!after) {
2393		kfree(before);
2394		return -ENOMEM;
2395	}
2396	INIT_LIST_HEAD(&after->list);
2397
2398	/* Now gather all removed mappings */
2399	tmp = amdgpu_vm_it_iter_first(&vm->va, saddr, eaddr);
2400	while (tmp) {
2401		/* Remember mapping split at the start */
2402		if (tmp->start < saddr) {
2403			before->start = tmp->start;
2404			before->last = saddr - 1;
2405			before->offset = tmp->offset;
2406			before->flags = tmp->flags;
2407			before->bo_va = tmp->bo_va;
2408			list_add(&before->list, &tmp->bo_va->invalids);
2409		}
2410
2411		/* Remember mapping split at the end */
2412		if (tmp->last > eaddr) {
2413			after->start = eaddr + 1;
2414			after->last = tmp->last;
2415			after->offset = tmp->offset;
2416			after->offset += after->start - tmp->start;
2417			after->flags = tmp->flags;
2418			after->bo_va = tmp->bo_va;
2419			list_add(&after->list, &tmp->bo_va->invalids);
2420		}
2421
2422		list_del(&tmp->list);
2423		list_add(&tmp->list, &removed);
2424
2425		tmp = amdgpu_vm_it_iter_next(tmp, saddr, eaddr);
2426	}
2427
2428	/* And free them up */
2429	list_for_each_entry_safe(tmp, next, &removed, list) {
2430		amdgpu_vm_it_remove(tmp, &vm->va);
2431		list_del(&tmp->list);
2432
2433		if (tmp->start < saddr)
2434		    tmp->start = saddr;
2435		if (tmp->last > eaddr)
2436		    tmp->last = eaddr;
2437
2438		tmp->bo_va = NULL;
2439		list_add(&tmp->list, &vm->freed);
2440		trace_amdgpu_vm_bo_unmap(NULL, tmp);
2441	}
2442
2443	/* Insert partial mapping before the range */
2444	if (!list_empty(&before->list)) {
2445		amdgpu_vm_it_insert(before, &vm->va);
2446		if (before->flags & AMDGPU_PTE_PRT)
2447			amdgpu_vm_prt_get(adev);
2448	} else {
2449		kfree(before);
2450	}
2451
2452	/* Insert partial mapping after the range */
2453	if (!list_empty(&after->list)) {
2454		amdgpu_vm_it_insert(after, &vm->va);
2455		if (after->flags & AMDGPU_PTE_PRT)
2456			amdgpu_vm_prt_get(adev);
2457	} else {
2458		kfree(after);
2459	}
2460
2461	return 0;
2462}
2463
2464/**
2465 * amdgpu_vm_bo_lookup_mapping - find mapping by address
2466 *
2467 * @vm: the requested VM
2468 * @addr: the address
2469 *
2470 * Find a mapping by it's address.
2471 *
2472 * Returns:
2473 * The amdgpu_bo_va_mapping matching for addr or NULL
2474 *
2475 */
2476struct amdgpu_bo_va_mapping *amdgpu_vm_bo_lookup_mapping(struct amdgpu_vm *vm,
2477							 uint64_t addr)
2478{
2479	return amdgpu_vm_it_iter_first(&vm->va, addr, addr);
2480}
2481
2482/**
2483 * amdgpu_vm_bo_trace_cs - trace all reserved mappings
2484 *
2485 * @vm: the requested vm
2486 * @ticket: CS ticket
2487 *
2488 * Trace all mappings of BOs reserved during a command submission.
2489 */
2490void amdgpu_vm_bo_trace_cs(struct amdgpu_vm *vm, struct ww_acquire_ctx *ticket)
2491{
2492	struct amdgpu_bo_va_mapping *mapping;
2493
2494	if (!trace_amdgpu_vm_bo_cs_enabled())
2495		return;
2496
2497	for (mapping = amdgpu_vm_it_iter_first(&vm->va, 0, U64_MAX); mapping;
2498	     mapping = amdgpu_vm_it_iter_next(mapping, 0, U64_MAX)) {
2499		if (mapping->bo_va && mapping->bo_va->base.bo) {
2500			struct amdgpu_bo *bo;
2501
2502			bo = mapping->bo_va->base.bo;
2503			if (dma_resv_locking_ctx(bo->tbo.base.resv) !=
2504			    ticket)
2505				continue;
2506		}
2507
2508		trace_amdgpu_vm_bo_cs(mapping);
2509	}
2510}
2511
2512/**
2513 * amdgpu_vm_bo_rmv - remove a bo to a specific vm
2514 *
2515 * @adev: amdgpu_device pointer
2516 * @bo_va: requested bo_va
2517 *
2518 * Remove @bo_va->bo from the requested vm.
2519 *
2520 * Object have to be reserved!
2521 */
2522void amdgpu_vm_bo_rmv(struct amdgpu_device *adev,
2523		      struct amdgpu_bo_va *bo_va)
2524{
2525	struct amdgpu_bo_va_mapping *mapping, *next;
2526	struct amdgpu_bo *bo = bo_va->base.bo;
2527	struct amdgpu_vm *vm = bo_va->base.vm;
2528	struct amdgpu_vm_bo_base **base;
2529
 
 
2530	if (bo) {
2531		if (bo->tbo.base.resv == vm->root.base.bo->tbo.base.resv)
2532			vm->bulk_moveable = false;
 
2533
2534		for (base = &bo_va->base.bo->vm_bo; *base;
2535		     base = &(*base)->next) {
2536			if (*base != &bo_va->base)
2537				continue;
2538
2539			*base = bo_va->base.next;
2540			break;
2541		}
2542	}
2543
2544	spin_lock(&vm->invalidated_lock);
2545	list_del(&bo_va->base.vm_status);
2546	spin_unlock(&vm->invalidated_lock);
2547
2548	list_for_each_entry_safe(mapping, next, &bo_va->valids, list) {
2549		list_del(&mapping->list);
2550		amdgpu_vm_it_remove(mapping, &vm->va);
2551		mapping->bo_va = NULL;
2552		trace_amdgpu_vm_bo_unmap(bo_va, mapping);
2553		list_add(&mapping->list, &vm->freed);
2554	}
2555	list_for_each_entry_safe(mapping, next, &bo_va->invalids, list) {
2556		list_del(&mapping->list);
2557		amdgpu_vm_it_remove(mapping, &vm->va);
2558		amdgpu_vm_free_mapping(adev, vm, mapping,
2559				       bo_va->last_pt_update);
2560	}
2561
2562	dma_fence_put(bo_va->last_pt_update);
2563
2564	if (bo && bo_va->is_xgmi)
2565		amdgpu_xgmi_set_pstate(adev, AMDGPU_XGMI_PSTATE_MIN);
2566
2567	kfree(bo_va);
2568}
2569
2570/**
2571 * amdgpu_vm_evictable - check if we can evict a VM
2572 *
2573 * @bo: A page table of the VM.
2574 *
2575 * Check if it is possible to evict a VM.
2576 */
2577bool amdgpu_vm_evictable(struct amdgpu_bo *bo)
2578{
2579	struct amdgpu_vm_bo_base *bo_base = bo->vm_bo;
2580
2581	/* Page tables of a destroyed VM can go away immediately */
2582	if (!bo_base || !bo_base->vm)
2583		return true;
2584
2585	/* Don't evict VM page tables while they are busy */
2586	if (!dma_resv_test_signaled_rcu(bo->tbo.base.resv, true))
2587		return false;
2588
2589	/* Try to block ongoing updates */
2590	if (!amdgpu_vm_eviction_trylock(bo_base->vm))
2591		return false;
2592
2593	/* Don't evict VM page tables while they are updated */
2594	if (!dma_fence_is_signaled(bo_base->vm->last_unlocked)) {
2595		amdgpu_vm_eviction_unlock(bo_base->vm);
2596		return false;
2597	}
2598
2599	bo_base->vm->evicting = true;
2600	amdgpu_vm_eviction_unlock(bo_base->vm);
2601	return true;
2602}
2603
2604/**
2605 * amdgpu_vm_bo_invalidate - mark the bo as invalid
2606 *
2607 * @adev: amdgpu_device pointer
2608 * @bo: amdgpu buffer object
2609 * @evicted: is the BO evicted
2610 *
2611 * Mark @bo as invalid.
2612 */
2613void amdgpu_vm_bo_invalidate(struct amdgpu_device *adev,
2614			     struct amdgpu_bo *bo, bool evicted)
2615{
2616	struct amdgpu_vm_bo_base *bo_base;
2617
2618	/* shadow bo doesn't have bo base, its validation needs its parent */
2619	if (bo->parent && bo->parent->shadow == bo)
2620		bo = bo->parent;
2621
2622	for (bo_base = bo->vm_bo; bo_base; bo_base = bo_base->next) {
2623		struct amdgpu_vm *vm = bo_base->vm;
2624
2625		if (evicted && bo->tbo.base.resv == vm->root.base.bo->tbo.base.resv) {
2626			amdgpu_vm_bo_evicted(bo_base);
2627			continue;
2628		}
2629
2630		if (bo_base->moved)
2631			continue;
2632		bo_base->moved = true;
2633
2634		if (bo->tbo.type == ttm_bo_type_kernel)
2635			amdgpu_vm_bo_relocated(bo_base);
2636		else if (bo->tbo.base.resv == vm->root.base.bo->tbo.base.resv)
2637			amdgpu_vm_bo_moved(bo_base);
2638		else
2639			amdgpu_vm_bo_invalidated(bo_base);
2640	}
2641}
2642
2643/**
2644 * amdgpu_vm_get_block_size - calculate VM page table size as power of two
2645 *
2646 * @vm_size: VM size
2647 *
2648 * Returns:
2649 * VM page table as power of two
2650 */
2651static uint32_t amdgpu_vm_get_block_size(uint64_t vm_size)
2652{
2653	/* Total bits covered by PD + PTs */
2654	unsigned bits = ilog2(vm_size) + 18;
2655
2656	/* Make sure the PD is 4K in size up to 8GB address space.
2657	   Above that split equal between PD and PTs */
2658	if (vm_size <= 8)
2659		return (bits - 9);
2660	else
2661		return ((bits + 3) / 2);
2662}
2663
2664/**
2665 * amdgpu_vm_adjust_size - adjust vm size, block size and fragment size
2666 *
2667 * @adev: amdgpu_device pointer
2668 * @min_vm_size: the minimum vm size in GB if it's set auto
2669 * @fragment_size_default: Default PTE fragment size
2670 * @max_level: max VMPT level
2671 * @max_bits: max address space size in bits
2672 *
2673 */
2674void amdgpu_vm_adjust_size(struct amdgpu_device *adev, uint32_t min_vm_size,
2675			   uint32_t fragment_size_default, unsigned max_level,
2676			   unsigned max_bits)
2677{
2678	unsigned int max_size = 1 << (max_bits - 30);
2679	unsigned int vm_size;
2680	uint64_t tmp;
2681
2682	/* adjust vm size first */
2683	if (amdgpu_vm_size != -1) {
2684		vm_size = amdgpu_vm_size;
2685		if (vm_size > max_size) {
2686			dev_warn(adev->dev, "VM size (%d) too large, max is %u GB\n",
2687				 amdgpu_vm_size, max_size);
2688			vm_size = max_size;
2689		}
2690	} else {
2691		struct sysinfo si;
2692		unsigned int phys_ram_gb;
2693
2694		/* Optimal VM size depends on the amount of physical
2695		 * RAM available. Underlying requirements and
2696		 * assumptions:
2697		 *
2698		 *  - Need to map system memory and VRAM from all GPUs
2699		 *     - VRAM from other GPUs not known here
2700		 *     - Assume VRAM <= system memory
2701		 *  - On GFX8 and older, VM space can be segmented for
2702		 *    different MTYPEs
2703		 *  - Need to allow room for fragmentation, guard pages etc.
2704		 *
2705		 * This adds up to a rough guess of system memory x3.
2706		 * Round up to power of two to maximize the available
2707		 * VM size with the given page table size.
2708		 */
2709		si_meminfo(&si);
2710		phys_ram_gb = ((uint64_t)si.totalram * si.mem_unit +
2711			       (1 << 30) - 1) >> 30;
2712		vm_size = roundup_pow_of_two(
2713			min(max(phys_ram_gb * 3, min_vm_size), max_size));
2714	}
2715
2716	adev->vm_manager.max_pfn = (uint64_t)vm_size << 18;
2717
2718	tmp = roundup_pow_of_two(adev->vm_manager.max_pfn);
2719	if (amdgpu_vm_block_size != -1)
2720		tmp >>= amdgpu_vm_block_size - 9;
2721	tmp = DIV_ROUND_UP(fls64(tmp) - 1, 9) - 1;
2722	adev->vm_manager.num_level = min(max_level, (unsigned)tmp);
2723	switch (adev->vm_manager.num_level) {
2724	case 3:
2725		adev->vm_manager.root_level = AMDGPU_VM_PDB2;
2726		break;
2727	case 2:
2728		adev->vm_manager.root_level = AMDGPU_VM_PDB1;
2729		break;
2730	case 1:
2731		adev->vm_manager.root_level = AMDGPU_VM_PDB0;
2732		break;
2733	default:
2734		dev_err(adev->dev, "VMPT only supports 2~4+1 levels\n");
2735	}
2736	/* block size depends on vm size and hw setup*/
2737	if (amdgpu_vm_block_size != -1)
2738		adev->vm_manager.block_size =
2739			min((unsigned)amdgpu_vm_block_size, max_bits
2740			    - AMDGPU_GPU_PAGE_SHIFT
2741			    - 9 * adev->vm_manager.num_level);
2742	else if (adev->vm_manager.num_level > 1)
2743		adev->vm_manager.block_size = 9;
2744	else
2745		adev->vm_manager.block_size = amdgpu_vm_get_block_size(tmp);
2746
2747	if (amdgpu_vm_fragment_size == -1)
2748		adev->vm_manager.fragment_size = fragment_size_default;
2749	else
2750		adev->vm_manager.fragment_size = amdgpu_vm_fragment_size;
2751
2752	DRM_INFO("vm size is %u GB, %u levels, block size is %u-bit, fragment size is %u-bit\n",
2753		 vm_size, adev->vm_manager.num_level + 1,
2754		 adev->vm_manager.block_size,
2755		 adev->vm_manager.fragment_size);
2756}
2757
2758/**
2759 * amdgpu_vm_wait_idle - wait for the VM to become idle
2760 *
2761 * @vm: VM object to wait for
2762 * @timeout: timeout to wait for VM to become idle
2763 */
2764long amdgpu_vm_wait_idle(struct amdgpu_vm *vm, long timeout)
2765{
2766	timeout = dma_resv_wait_timeout_rcu(vm->root.base.bo->tbo.base.resv,
2767					    true, true, timeout);
 
2768	if (timeout <= 0)
2769		return timeout;
2770
2771	return dma_fence_wait_timeout(vm->last_unlocked, true, timeout);
2772}
2773
2774/**
2775 * amdgpu_vm_init - initialize a vm instance
2776 *
2777 * @adev: amdgpu_device pointer
2778 * @vm: requested vm
2779 * @vm_context: Indicates if it GFX or Compute context
2780 * @pasid: Process address space identifier
2781 *
2782 * Init @vm fields.
2783 *
2784 * Returns:
2785 * 0 for success, error for failure.
2786 */
2787int amdgpu_vm_init(struct amdgpu_device *adev, struct amdgpu_vm *vm,
2788		   int vm_context, unsigned int pasid)
2789{
2790	struct amdgpu_bo_param bp;
2791	struct amdgpu_bo *root;
2792	int r, i;
2793
2794	vm->va = RB_ROOT_CACHED;
2795	for (i = 0; i < AMDGPU_MAX_VMHUBS; i++)
2796		vm->reserved_vmid[i] = NULL;
2797	INIT_LIST_HEAD(&vm->evicted);
2798	INIT_LIST_HEAD(&vm->relocated);
2799	INIT_LIST_HEAD(&vm->moved);
2800	INIT_LIST_HEAD(&vm->idle);
2801	INIT_LIST_HEAD(&vm->invalidated);
2802	spin_lock_init(&vm->invalidated_lock);
2803	INIT_LIST_HEAD(&vm->freed);
2804
 
 
2805
2806	/* create scheduler entities for page table updates */
2807	r = drm_sched_entity_init(&vm->immediate, DRM_SCHED_PRIORITY_NORMAL,
2808				  adev->vm_manager.vm_pte_scheds,
2809				  adev->vm_manager.vm_pte_num_scheds, NULL);
2810	if (r)
2811		return r;
2812
2813	r = drm_sched_entity_init(&vm->delayed, DRM_SCHED_PRIORITY_NORMAL,
2814				  adev->vm_manager.vm_pte_scheds,
2815				  adev->vm_manager.vm_pte_num_scheds, NULL);
2816	if (r)
2817		goto error_free_immediate;
2818
2819	vm->pte_support_ats = false;
2820	vm->is_compute_context = false;
2821
2822	if (vm_context == AMDGPU_VM_CONTEXT_COMPUTE) {
2823		vm->use_cpu_for_update = !!(adev->vm_manager.vm_update_mode &
2824						AMDGPU_VM_USE_CPU_FOR_COMPUTE);
2825
2826		if (adev->asic_type == CHIP_RAVEN)
2827			vm->pte_support_ats = true;
2828	} else {
2829		vm->use_cpu_for_update = !!(adev->vm_manager.vm_update_mode &
2830						AMDGPU_VM_USE_CPU_FOR_GFX);
2831	}
2832	DRM_DEBUG_DRIVER("VM update mode is %s\n",
2833			 vm->use_cpu_for_update ? "CPU" : "SDMA");
2834	WARN_ONCE((vm->use_cpu_for_update &&
2835		   !amdgpu_gmc_vram_full_visible(&adev->gmc)),
2836		  "CPU update of VM recommended only for large BAR system\n");
2837
2838	if (vm->use_cpu_for_update)
2839		vm->update_funcs = &amdgpu_vm_cpu_funcs;
2840	else
2841		vm->update_funcs = &amdgpu_vm_sdma_funcs;
2842	vm->last_update = NULL;
2843	vm->last_unlocked = dma_fence_get_stub();
 
2844
2845	mutex_init(&vm->eviction_lock);
2846	vm->evicting = false;
2847
2848	amdgpu_vm_bo_param(adev, vm, adev->vm_manager.root_level, false, &bp);
2849	if (vm_context == AMDGPU_VM_CONTEXT_COMPUTE)
2850		bp.flags &= ~AMDGPU_GEM_CREATE_SHADOW;
2851	r = amdgpu_bo_create(adev, &bp, &root);
2852	if (r)
2853		goto error_free_delayed;
2854
2855	r = amdgpu_bo_reserve(root, true);
2856	if (r)
2857		goto error_free_root;
2858
2859	r = dma_resv_reserve_shared(root->tbo.base.resv, 1);
2860	if (r)
2861		goto error_unreserve;
2862
2863	amdgpu_vm_bo_base_init(&vm->root.base, vm, root);
2864
2865	r = amdgpu_vm_clear_bo(adev, vm, root, false);
2866	if (r)
2867		goto error_unreserve;
2868
2869	amdgpu_bo_unreserve(vm->root.base.bo);
2870
2871	if (pasid) {
2872		unsigned long flags;
2873
2874		spin_lock_irqsave(&adev->vm_manager.pasid_lock, flags);
2875		r = idr_alloc(&adev->vm_manager.pasid_idr, vm, pasid, pasid + 1,
2876			      GFP_ATOMIC);
2877		spin_unlock_irqrestore(&adev->vm_manager.pasid_lock, flags);
2878		if (r < 0)
2879			goto error_free_root;
2880
2881		vm->pasid = pasid;
2882	}
2883
2884	INIT_KFIFO(vm->faults);
2885
2886	return 0;
2887
2888error_unreserve:
2889	amdgpu_bo_unreserve(vm->root.base.bo);
2890
2891error_free_root:
2892	amdgpu_bo_unref(&vm->root.base.bo->shadow);
2893	amdgpu_bo_unref(&vm->root.base.bo);
2894	vm->root.base.bo = NULL;
2895
2896error_free_delayed:
 
2897	dma_fence_put(vm->last_unlocked);
2898	drm_sched_entity_destroy(&vm->delayed);
2899
2900error_free_immediate:
2901	drm_sched_entity_destroy(&vm->immediate);
2902
2903	return r;
2904}
2905
2906/**
2907 * amdgpu_vm_check_clean_reserved - check if a VM is clean
2908 *
2909 * @adev: amdgpu_device pointer
2910 * @vm: the VM to check
2911 *
2912 * check all entries of the root PD, if any subsequent PDs are allocated,
2913 * it means there are page table creating and filling, and is no a clean
2914 * VM
2915 *
2916 * Returns:
2917 *	0 if this VM is clean
2918 */
2919static int amdgpu_vm_check_clean_reserved(struct amdgpu_device *adev,
2920	struct amdgpu_vm *vm)
2921{
2922	enum amdgpu_vm_level root = adev->vm_manager.root_level;
2923	unsigned int entries = amdgpu_vm_num_entries(adev, root);
2924	unsigned int i = 0;
2925
2926	if (!(vm->root.entries))
2927		return 0;
2928
2929	for (i = 0; i < entries; i++) {
2930		if (vm->root.entries[i].base.bo)
2931			return -EINVAL;
2932	}
2933
2934	return 0;
2935}
2936
2937/**
2938 * amdgpu_vm_make_compute - Turn a GFX VM into a compute VM
2939 *
2940 * @adev: amdgpu_device pointer
2941 * @vm: requested vm
2942 * @pasid: pasid to use
2943 *
2944 * This only works on GFX VMs that don't have any BOs added and no
2945 * page tables allocated yet.
2946 *
2947 * Changes the following VM parameters:
2948 * - use_cpu_for_update
2949 * - pte_supports_ats
2950 * - pasid (old PASID is released, because compute manages its own PASIDs)
2951 *
2952 * Reinitializes the page directory to reflect the changed ATS
2953 * setting.
2954 *
2955 * Returns:
2956 * 0 for success, -errno for errors.
2957 */
2958int amdgpu_vm_make_compute(struct amdgpu_device *adev, struct amdgpu_vm *vm,
2959			   unsigned int pasid)
2960{
2961	bool pte_support_ats = (adev->asic_type == CHIP_RAVEN);
2962	int r;
2963
2964	r = amdgpu_bo_reserve(vm->root.base.bo, true);
2965	if (r)
2966		return r;
2967
2968	/* Sanity checks */
2969	r = amdgpu_vm_check_clean_reserved(adev, vm);
2970	if (r)
2971		goto unreserve_bo;
2972
2973	if (pasid) {
2974		unsigned long flags;
2975
2976		spin_lock_irqsave(&adev->vm_manager.pasid_lock, flags);
2977		r = idr_alloc(&adev->vm_manager.pasid_idr, vm, pasid, pasid + 1,
2978			      GFP_ATOMIC);
2979		spin_unlock_irqrestore(&adev->vm_manager.pasid_lock, flags);
2980
2981		if (r == -ENOSPC)
2982			goto unreserve_bo;
2983		r = 0;
2984	}
2985
2986	/* Check if PD needs to be reinitialized and do it before
2987	 * changing any other state, in case it fails.
2988	 */
2989	if (pte_support_ats != vm->pte_support_ats) {
2990		vm->pte_support_ats = pte_support_ats;
2991		r = amdgpu_vm_clear_bo(adev, vm, vm->root.base.bo, false);
 
2992		if (r)
2993			goto free_idr;
2994	}
2995
2996	/* Update VM state */
2997	vm->use_cpu_for_update = !!(adev->vm_manager.vm_update_mode &
2998				    AMDGPU_VM_USE_CPU_FOR_COMPUTE);
2999	DRM_DEBUG_DRIVER("VM update mode is %s\n",
3000			 vm->use_cpu_for_update ? "CPU" : "SDMA");
3001	WARN_ONCE((vm->use_cpu_for_update &&
3002		   !amdgpu_gmc_vram_full_visible(&adev->gmc)),
3003		  "CPU update of VM recommended only for large BAR system\n");
3004
3005	if (vm->use_cpu_for_update) {
3006		/* Sync with last SDMA update/clear before switching to CPU */
3007		r = amdgpu_bo_sync_wait(vm->root.base.bo,
3008					AMDGPU_FENCE_OWNER_UNDEFINED, true);
3009		if (r)
3010			goto free_idr;
3011
3012		vm->update_funcs = &amdgpu_vm_cpu_funcs;
3013	} else {
3014		vm->update_funcs = &amdgpu_vm_sdma_funcs;
3015	}
 
 
 
 
 
 
 
 
3016	dma_fence_put(vm->last_update);
3017	vm->last_update = NULL;
3018	vm->is_compute_context = true;
3019
3020	if (vm->pasid) {
3021		unsigned long flags;
3022
3023		spin_lock_irqsave(&adev->vm_manager.pasid_lock, flags);
3024		idr_remove(&adev->vm_manager.pasid_idr, vm->pasid);
3025		spin_unlock_irqrestore(&adev->vm_manager.pasid_lock, flags);
3026
3027		/* Free the original amdgpu allocated pasid
3028		 * Will be replaced with kfd allocated pasid
3029		 */
3030		amdgpu_pasid_free(vm->pasid);
3031		vm->pasid = 0;
3032	}
3033
3034	/* Free the shadow bo for compute VM */
3035	amdgpu_bo_unref(&vm->root.base.bo->shadow);
3036
3037	if (pasid)
3038		vm->pasid = pasid;
3039
3040	goto unreserve_bo;
3041
3042free_idr:
3043	if (pasid) {
3044		unsigned long flags;
3045
3046		spin_lock_irqsave(&adev->vm_manager.pasid_lock, flags);
3047		idr_remove(&adev->vm_manager.pasid_idr, pasid);
3048		spin_unlock_irqrestore(&adev->vm_manager.pasid_lock, flags);
3049	}
3050unreserve_bo:
3051	amdgpu_bo_unreserve(vm->root.base.bo);
3052	return r;
3053}
3054
3055/**
3056 * amdgpu_vm_release_compute - release a compute vm
3057 * @adev: amdgpu_device pointer
3058 * @vm: a vm turned into compute vm by calling amdgpu_vm_make_compute
3059 *
3060 * This is a correspondant of amdgpu_vm_make_compute. It decouples compute
3061 * pasid from vm. Compute should stop use of vm after this call.
3062 */
3063void amdgpu_vm_release_compute(struct amdgpu_device *adev, struct amdgpu_vm *vm)
3064{
3065	if (vm->pasid) {
3066		unsigned long flags;
3067
3068		spin_lock_irqsave(&adev->vm_manager.pasid_lock, flags);
3069		idr_remove(&adev->vm_manager.pasid_idr, vm->pasid);
3070		spin_unlock_irqrestore(&adev->vm_manager.pasid_lock, flags);
3071	}
3072	vm->pasid = 0;
3073	vm->is_compute_context = false;
3074}
3075
3076/**
3077 * amdgpu_vm_fini - tear down a vm instance
3078 *
3079 * @adev: amdgpu_device pointer
3080 * @vm: requested vm
3081 *
3082 * Tear down @vm.
3083 * Unbind the VM and remove all bos from the vm bo list
3084 */
3085void amdgpu_vm_fini(struct amdgpu_device *adev, struct amdgpu_vm *vm)
3086{
3087	struct amdgpu_bo_va_mapping *mapping, *tmp;
3088	bool prt_fini_needed = !!adev->gmc.gmc_funcs->set_prt;
3089	struct amdgpu_bo *root;
 
3090	int i;
3091
3092	amdgpu_amdkfd_gpuvm_destroy_cb(adev, vm);
3093
3094	root = amdgpu_bo_ref(vm->root.base.bo);
3095	amdgpu_bo_reserve(root, true);
3096	if (vm->pasid) {
3097		unsigned long flags;
3098
3099		spin_lock_irqsave(&adev->vm_manager.pasid_lock, flags);
3100		idr_remove(&adev->vm_manager.pasid_idr, vm->pasid);
3101		spin_unlock_irqrestore(&adev->vm_manager.pasid_lock, flags);
3102		vm->pasid = 0;
3103	}
3104
 
 
 
3105	dma_fence_wait(vm->last_unlocked, false);
3106	dma_fence_put(vm->last_unlocked);
 
 
 
 
 
3107
3108	list_for_each_entry_safe(mapping, tmp, &vm->freed, list) {
3109		if (mapping->flags & AMDGPU_PTE_PRT && prt_fini_needed) {
3110			amdgpu_vm_prt_fini(adev, vm);
3111			prt_fini_needed = false;
3112		}
3113
3114		list_del(&mapping->list);
3115		amdgpu_vm_free_mapping(adev, vm, mapping, NULL);
3116	}
3117
3118	amdgpu_vm_free_pts(adev, vm, NULL);
3119	amdgpu_bo_unreserve(root);
3120	amdgpu_bo_unref(&root);
3121	WARN_ON(vm->root.base.bo);
3122
3123	drm_sched_entity_destroy(&vm->immediate);
3124	drm_sched_entity_destroy(&vm->delayed);
3125
3126	if (!RB_EMPTY_ROOT(&vm->va.rb_root)) {
3127		dev_err(adev->dev, "still active bo inside vm\n");
3128	}
3129	rbtree_postorder_for_each_entry_safe(mapping, tmp,
3130					     &vm->va.rb_root, rb) {
3131		/* Don't remove the mapping here, we don't want to trigger a
3132		 * rebalance and the tree is about to be destroyed anyway.
3133		 */
3134		list_del(&mapping->list);
3135		kfree(mapping);
3136	}
3137
3138	dma_fence_put(vm->last_update);
3139	for (i = 0; i < AMDGPU_MAX_VMHUBS; i++)
3140		amdgpu_vmid_free_reserved(adev, vm, i);
3141}
3142
3143/**
3144 * amdgpu_vm_manager_init - init the VM manager
3145 *
3146 * @adev: amdgpu_device pointer
3147 *
3148 * Initialize the VM manager structures
3149 */
3150void amdgpu_vm_manager_init(struct amdgpu_device *adev)
3151{
3152	unsigned i;
3153
 
 
 
 
 
 
3154	amdgpu_vmid_mgr_init(adev);
3155
3156	adev->vm_manager.fence_context =
3157		dma_fence_context_alloc(AMDGPU_MAX_RINGS);
3158	for (i = 0; i < AMDGPU_MAX_RINGS; ++i)
3159		adev->vm_manager.seqno[i] = 0;
3160
3161	spin_lock_init(&adev->vm_manager.prt_lock);
3162	atomic_set(&adev->vm_manager.num_prt_users, 0);
3163
3164	/* If not overridden by the user, by default, only in large BAR systems
3165	 * Compute VM tables will be updated by CPU
3166	 */
3167#ifdef CONFIG_X86_64
3168	if (amdgpu_vm_update_mode == -1) {
3169		if (amdgpu_gmc_vram_full_visible(&adev->gmc))
 
 
 
 
3170			adev->vm_manager.vm_update_mode =
3171				AMDGPU_VM_USE_CPU_FOR_COMPUTE;
3172		else
3173			adev->vm_manager.vm_update_mode = 0;
3174	} else
3175		adev->vm_manager.vm_update_mode = amdgpu_vm_update_mode;
3176#else
3177	adev->vm_manager.vm_update_mode = 0;
3178#endif
3179
3180	idr_init(&adev->vm_manager.pasid_idr);
3181	spin_lock_init(&adev->vm_manager.pasid_lock);
3182}
3183
3184/**
3185 * amdgpu_vm_manager_fini - cleanup VM manager
3186 *
3187 * @adev: amdgpu_device pointer
3188 *
3189 * Cleanup the VM manager and free resources.
3190 */
3191void amdgpu_vm_manager_fini(struct amdgpu_device *adev)
3192{
3193	WARN_ON(!idr_is_empty(&adev->vm_manager.pasid_idr));
3194	idr_destroy(&adev->vm_manager.pasid_idr);
3195
3196	amdgpu_vmid_mgr_fini(adev);
3197}
3198
3199/**
3200 * amdgpu_vm_ioctl - Manages VMID reservation for vm hubs.
3201 *
3202 * @dev: drm device pointer
3203 * @data: drm_amdgpu_vm
3204 * @filp: drm file pointer
3205 *
3206 * Returns:
3207 * 0 for success, -errno for errors.
3208 */
3209int amdgpu_vm_ioctl(struct drm_device *dev, void *data, struct drm_file *filp)
3210{
3211	union drm_amdgpu_vm *args = data;
3212	struct amdgpu_device *adev = dev->dev_private;
3213	struct amdgpu_fpriv *fpriv = filp->driver_priv;
3214	long timeout = msecs_to_jiffies(2000);
3215	int r;
3216
3217	switch (args->in.op) {
3218	case AMDGPU_VM_OP_RESERVE_VMID:
3219		/* We only have requirement to reserve vmid from gfxhub */
3220		r = amdgpu_vmid_alloc_reserved(adev, &fpriv->vm,
3221					       AMDGPU_GFXHUB_0);
3222		if (r)
3223			return r;
3224		break;
3225	case AMDGPU_VM_OP_UNRESERVE_VMID:
3226		if (amdgpu_sriov_runtime(adev))
3227			timeout = 8 * timeout;
3228
3229		/* Wait vm idle to make sure the vmid set in SPM_VMID is
3230		 * not referenced anymore.
3231		 */
3232		r = amdgpu_bo_reserve(fpriv->vm.root.base.bo, true);
3233		if (r)
3234			return r;
3235
3236		r = amdgpu_vm_wait_idle(&fpriv->vm, timeout);
3237		if (r < 0)
3238			return r;
3239
3240		amdgpu_bo_unreserve(fpriv->vm.root.base.bo);
3241		amdgpu_vmid_free_reserved(adev, &fpriv->vm, AMDGPU_GFXHUB_0);
3242		break;
3243	default:
3244		return -EINVAL;
3245	}
3246
3247	return 0;
3248}
3249
3250/**
3251 * amdgpu_vm_get_task_info - Extracts task info for a PASID.
3252 *
3253 * @adev: drm device pointer
3254 * @pasid: PASID identifier for VM
3255 * @task_info: task_info to fill.
3256 */
3257void amdgpu_vm_get_task_info(struct amdgpu_device *adev, unsigned int pasid,
3258			 struct amdgpu_task_info *task_info)
3259{
3260	struct amdgpu_vm *vm;
3261	unsigned long flags;
3262
3263	spin_lock_irqsave(&adev->vm_manager.pasid_lock, flags);
3264
3265	vm = idr_find(&adev->vm_manager.pasid_idr, pasid);
3266	if (vm)
3267		*task_info = vm->task_info;
3268
3269	spin_unlock_irqrestore(&adev->vm_manager.pasid_lock, flags);
3270}
3271
3272/**
3273 * amdgpu_vm_set_task_info - Sets VMs task info.
3274 *
3275 * @vm: vm for which to set the info
3276 */
3277void amdgpu_vm_set_task_info(struct amdgpu_vm *vm)
3278{
3279	if (vm->task_info.pid)
3280		return;
3281
3282	vm->task_info.pid = current->pid;
3283	get_task_comm(vm->task_info.task_name, current);
3284
3285	if (current->group_leader->mm != current->mm)
3286		return;
3287
3288	vm->task_info.tgid = current->group_leader->pid;
3289	get_task_comm(vm->task_info.process_name, current->group_leader);
3290}
3291
3292/**
3293 * amdgpu_vm_handle_fault - graceful handling of VM faults.
3294 * @adev: amdgpu device pointer
3295 * @pasid: PASID of the VM
3296 * @addr: Address of the fault
 
3297 *
3298 * Try to gracefully handle a VM fault. Return true if the fault was handled and
3299 * shouldn't be reported any more.
3300 */
3301bool amdgpu_vm_handle_fault(struct amdgpu_device *adev, unsigned int pasid,
3302			    uint64_t addr)
3303{
 
3304	struct amdgpu_bo *root;
 
3305	uint64_t value, flags;
3306	struct amdgpu_vm *vm;
3307	long r;
3308
3309	spin_lock(&adev->vm_manager.pasid_lock);
3310	vm = idr_find(&adev->vm_manager.pasid_idr, pasid);
3311	if (vm)
3312		root = amdgpu_bo_ref(vm->root.base.bo);
3313	else
 
3314		root = NULL;
3315	spin_unlock(&adev->vm_manager.pasid_lock);
 
3316
3317	if (!root)
3318		return false;
3319
 
 
 
 
 
 
 
 
3320	r = amdgpu_bo_reserve(root, true);
3321	if (r)
3322		goto error_unref;
3323
3324	/* Double check that the VM still exists */
3325	spin_lock(&adev->vm_manager.pasid_lock);
3326	vm = idr_find(&adev->vm_manager.pasid_idr, pasid);
3327	if (vm && vm->root.base.bo != root)
3328		vm = NULL;
3329	spin_unlock(&adev->vm_manager.pasid_lock);
3330	if (!vm)
3331		goto error_unlock;
3332
3333	addr /= AMDGPU_GPU_PAGE_SIZE;
3334	flags = AMDGPU_PTE_VALID | AMDGPU_PTE_SNOOPED |
3335		AMDGPU_PTE_SYSTEM;
3336
3337	if (vm->is_compute_context) {
3338		/* Intentionally setting invalid PTE flag
3339		 * combination to force a no-retry-fault
3340		 */
3341		flags = AMDGPU_PTE_EXECUTABLE | AMDGPU_PDE_PTE |
3342			AMDGPU_PTE_TF;
3343		value = 0;
3344
3345	} else if (amdgpu_vm_fault_stop == AMDGPU_VM_FAULT_STOP_NEVER) {
3346		/* Redirect the access to the dummy page */
3347		value = adev->dummy_page_addr;
3348		flags |= AMDGPU_PTE_EXECUTABLE | AMDGPU_PTE_READABLE |
3349			AMDGPU_PTE_WRITEABLE;
3350
3351	} else {
3352		/* Let the hw retry silently on the PTE */
3353		value = 0;
3354	}
3355
3356	r = amdgpu_vm_bo_update_mapping(adev, vm, true, false, NULL, addr,
3357					addr + 1, flags, value, NULL, NULL);
 
 
 
 
 
 
3358	if (r)
3359		goto error_unlock;
3360
3361	r = amdgpu_vm_update_pdes(adev, vm, true);
3362
3363error_unlock:
3364	amdgpu_bo_unreserve(root);
3365	if (r < 0)
3366		DRM_ERROR("Can't handle page fault (%ld)\n", r);
3367
3368error_unref:
3369	amdgpu_bo_unref(&root);
3370
3371	return false;
3372}
v6.2
   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
  29#include <linux/dma-fence-array.h>
  30#include <linux/interval_tree_generic.h>
  31#include <linux/idr.h>
  32#include <linux/dma-buf.h>
  33
  34#include <drm/amdgpu_drm.h>
  35#include <drm/drm_drv.h>
  36#include "amdgpu.h"
  37#include "amdgpu_trace.h"
  38#include "amdgpu_amdkfd.h"
  39#include "amdgpu_gmc.h"
  40#include "amdgpu_xgmi.h"
  41#include "amdgpu_dma_buf.h"
  42#include "amdgpu_res_cursor.h"
  43#include "kfd_svm.h"
  44
  45/**
  46 * DOC: GPUVM
  47 *
  48 * GPUVM is the MMU functionality provided on the GPU.
  49 * GPUVM is similar to the legacy GART on older asics, however
  50 * rather than there being a single global GART table
  51 * for the entire GPU, there can be multiple GPUVM page tables active
  52 * at any given time.  The GPUVM page tables can contain a mix
  53 * VRAM pages and system pages (both memory and MMIO) and system pages
  54 * can be mapped as snooped (cached system pages) or unsnooped
  55 * (uncached system pages).
  56 *
  57 * Each active GPUVM has an ID associated with it and there is a page table
  58 * linked with each VMID.  When executing a command buffer,
  59 * the kernel tells the engine what VMID to use for that command
  60 * buffer.  VMIDs are allocated dynamically as commands are submitted.
  61 * The userspace drivers maintain their own address space and the kernel
  62 * sets up their pages tables accordingly when they submit their
  63 * command buffers and a VMID is assigned.
  64 * The hardware supports up to 16 active GPUVMs at any given time.
  65 *
  66 * Each GPUVM is represented by a 1-2 or 1-5 level page table, depending
  67 * on the ASIC family.  GPUVM supports RWX attributes on each page as well
  68 * as other features such as encryption and caching attributes.
  69 *
  70 * VMID 0 is special.  It is the GPUVM used for the kernel driver.  In
  71 * addition to an aperture managed by a page table, VMID 0 also has
  72 * several other apertures.  There is an aperture for direct access to VRAM
  73 * and there is a legacy AGP aperture which just forwards accesses directly
  74 * to the matching system physical addresses (or IOVAs when an IOMMU is
  75 * present).  These apertures provide direct access to these memories without
  76 * incurring the overhead of a page table.  VMID 0 is used by the kernel
  77 * driver for tasks like memory management.
  78 *
  79 * GPU clients (i.e., engines on the GPU) use GPUVM VMIDs to access memory.
  80 * For user applications, each application can have their own unique GPUVM
  81 * address space.  The application manages the address space and the kernel
  82 * driver manages the GPUVM page tables for each process.  If an GPU client
  83 * accesses an invalid page, it will generate a GPU page fault, similar to
  84 * accessing an invalid page on a CPU.
  85 */
  86
  87#define START(node) ((node)->start)
  88#define LAST(node) ((node)->last)
  89
  90INTERVAL_TREE_DEFINE(struct amdgpu_bo_va_mapping, rb, uint64_t, __subtree_last,
  91		     START, LAST, static, amdgpu_vm_it)
  92
  93#undef START
  94#undef LAST
  95
  96/**
  97 * struct amdgpu_prt_cb - Helper to disable partial resident texture feature from a fence callback
  98 */
  99struct amdgpu_prt_cb {
 100
 101	/**
 102	 * @adev: amdgpu device
 103	 */
 104	struct amdgpu_device *adev;
 105
 106	/**
 107	 * @cb: callback
 108	 */
 109	struct dma_fence_cb cb;
 110};
 111
 112/**
 113 * struct amdgpu_vm_tlb_seq_cb - Helper to increment the TLB flush sequence
 
 
 114 */
 115struct amdgpu_vm_tlb_seq_cb {
 116	/**
 117	 * @vm: pointer to the amdgpu_vm structure to set the fence sequence on
 118	 */
 119	struct amdgpu_vm *vm;
 
 
 
 
 
 
 
 
 
 120
 121	/**
 122	 * @cb: callback
 123	 */
 124	struct dma_fence_cb cb;
 125};
 126
 127/**
 128 * amdgpu_vm_set_pasid - manage pasid and vm ptr mapping
 129 *
 130 * @adev: amdgpu_device pointer
 131 * @vm: amdgpu_vm pointer
 132 * @pasid: the pasid the VM is using on this GPU
 133 *
 134 * Set the pasid this VM is using on this GPU, can also be used to remove the
 135 * pasid by passing in zero.
 136 *
 
 
 137 */
 138int amdgpu_vm_set_pasid(struct amdgpu_device *adev, struct amdgpu_vm *vm,
 139			u32 pasid)
 140{
 141	int r;
 142
 143	if (vm->pasid == pasid)
 
 
 
 
 144		return 0;
 
 
 
 
 145
 146	if (vm->pasid) {
 147		r = xa_err(xa_erase_irq(&adev->vm_manager.pasids, vm->pasid));
 148		if (r < 0)
 149			return r;
 
 
 
 
 
 
 
 
 
 
 150
 151		vm->pasid = 0;
 152	}
 
 
 
 
 
 
 
 
 
 153
 154	if (pasid) {
 155		r = xa_err(xa_store_irq(&adev->vm_manager.pasids, pasid, vm,
 156					GFP_KERNEL));
 157		if (r < 0)
 158			return r;
 
 
 
 
 
 
 159
 160		vm->pasid = pasid;
 161	}
 
 162
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 163
 164	return 0;
 
 
 
 
 
 
 
 
 
 
 
 165}
 166
 167/**
 168 * amdgpu_vm_bo_evicted - vm_bo is evicted
 169 *
 170 * @vm_bo: vm_bo which is evicted
 171 *
 172 * State for PDs/PTs and per VM BOs which are not at the location they should
 173 * be.
 174 */
 175static void amdgpu_vm_bo_evicted(struct amdgpu_vm_bo_base *vm_bo)
 176{
 177	struct amdgpu_vm *vm = vm_bo->vm;
 178	struct amdgpu_bo *bo = vm_bo->bo;
 179
 180	vm_bo->moved = true;
 181	spin_lock(&vm_bo->vm->status_lock);
 182	if (bo->tbo.type == ttm_bo_type_kernel)
 183		list_move(&vm_bo->vm_status, &vm->evicted);
 184	else
 185		list_move_tail(&vm_bo->vm_status, &vm->evicted);
 186	spin_unlock(&vm_bo->vm->status_lock);
 187}
 188/**
 189 * amdgpu_vm_bo_moved - vm_bo is moved
 190 *
 191 * @vm_bo: vm_bo which is moved
 192 *
 193 * State for per VM BOs which are moved, but that change is not yet reflected
 194 * in the page tables.
 195 */
 196static void amdgpu_vm_bo_moved(struct amdgpu_vm_bo_base *vm_bo)
 197{
 198	spin_lock(&vm_bo->vm->status_lock);
 199	list_move(&vm_bo->vm_status, &vm_bo->vm->moved);
 200	spin_unlock(&vm_bo->vm->status_lock);
 201}
 202
 203/**
 204 * amdgpu_vm_bo_idle - vm_bo is idle
 205 *
 206 * @vm_bo: vm_bo which is now idle
 207 *
 208 * State for PDs/PTs and per VM BOs which have gone through the state machine
 209 * and are now idle.
 210 */
 211static void amdgpu_vm_bo_idle(struct amdgpu_vm_bo_base *vm_bo)
 212{
 213	spin_lock(&vm_bo->vm->status_lock);
 214	list_move(&vm_bo->vm_status, &vm_bo->vm->idle);
 215	spin_unlock(&vm_bo->vm->status_lock);
 216	vm_bo->moved = false;
 217}
 218
 219/**
 220 * amdgpu_vm_bo_invalidated - vm_bo is invalidated
 221 *
 222 * @vm_bo: vm_bo which is now invalidated
 223 *
 224 * State for normal BOs which are invalidated and that change not yet reflected
 225 * in the PTs.
 226 */
 227static void amdgpu_vm_bo_invalidated(struct amdgpu_vm_bo_base *vm_bo)
 228{
 229	spin_lock(&vm_bo->vm->status_lock);
 230	list_move(&vm_bo->vm_status, &vm_bo->vm->invalidated);
 231	spin_unlock(&vm_bo->vm->status_lock);
 232}
 233
 234/**
 235 * amdgpu_vm_bo_relocated - vm_bo is reloacted
 236 *
 237 * @vm_bo: vm_bo which is relocated
 238 *
 239 * State for PDs/PTs which needs to update their parent PD.
 240 * For the root PD, just move to idle state.
 241 */
 242static void amdgpu_vm_bo_relocated(struct amdgpu_vm_bo_base *vm_bo)
 243{
 244	if (vm_bo->bo->parent) {
 245		spin_lock(&vm_bo->vm->status_lock);
 246		list_move(&vm_bo->vm_status, &vm_bo->vm->relocated);
 247		spin_unlock(&vm_bo->vm->status_lock);
 248	} else {
 249		amdgpu_vm_bo_idle(vm_bo);
 250	}
 251}
 252
 253/**
 254 * amdgpu_vm_bo_done - vm_bo is done
 255 *
 256 * @vm_bo: vm_bo which is now done
 257 *
 258 * State for normal BOs which are invalidated and that change has been updated
 259 * in the PTs.
 260 */
 261static void amdgpu_vm_bo_done(struct amdgpu_vm_bo_base *vm_bo)
 262{
 263	spin_lock(&vm_bo->vm->status_lock);
 264	list_move(&vm_bo->vm_status, &vm_bo->vm->done);
 265	spin_unlock(&vm_bo->vm->status_lock);
 266}
 267
 268/**
 269 * amdgpu_vm_bo_base_init - Adds bo to the list of bos associated with the vm
 270 *
 271 * @base: base structure for tracking BO usage in a VM
 272 * @vm: vm to which bo is to be added
 273 * @bo: amdgpu buffer object
 274 *
 275 * Initialize a bo_va_base structure and add it to the appropriate lists
 276 *
 277 */
 278void amdgpu_vm_bo_base_init(struct amdgpu_vm_bo_base *base,
 279			    struct amdgpu_vm *vm, struct amdgpu_bo *bo)
 
 280{
 281	base->vm = vm;
 282	base->bo = bo;
 283	base->next = NULL;
 284	INIT_LIST_HEAD(&base->vm_status);
 285
 286	if (!bo)
 287		return;
 288	base->next = bo->vm_bo;
 289	bo->vm_bo = base;
 290
 291	if (bo->tbo.base.resv != vm->root.bo->tbo.base.resv)
 292		return;
 293
 294	dma_resv_assert_held(vm->root.bo->tbo.base.resv);
 295
 296	ttm_bo_set_bulk_move(&bo->tbo, &vm->lru_bulk_move);
 297	if (bo->tbo.type == ttm_bo_type_kernel && bo->parent)
 298		amdgpu_vm_bo_relocated(base);
 299	else
 300		amdgpu_vm_bo_idle(base);
 301
 302	if (bo->preferred_domains &
 303	    amdgpu_mem_type_to_domain(bo->tbo.resource->mem_type))
 304		return;
 305
 306	/*
 307	 * we checked all the prerequisites, but it looks like this per vm bo
 308	 * is currently evicted. add the bo to the evicted list to make sure it
 309	 * is validated on next vm use to avoid fault.
 310	 * */
 311	amdgpu_vm_bo_evicted(base);
 312}
 313
 314/**
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 315 * amdgpu_vm_get_pd_bo - add the VM PD to a validation list
 316 *
 317 * @vm: vm providing the BOs
 318 * @validated: head of validation list
 319 * @entry: entry to add
 320 *
 321 * Add the page directory to the list of BOs to
 322 * validate for command submission.
 323 */
 324void amdgpu_vm_get_pd_bo(struct amdgpu_vm *vm,
 325			 struct list_head *validated,
 326			 struct amdgpu_bo_list_entry *entry)
 327{
 328	entry->priority = 0;
 329	entry->tv.bo = &vm->root.bo->tbo;
 330	/* Two for VM updates, one for TTM and one for the CS job */
 331	entry->tv.num_shared = 4;
 332	entry->user_pages = NULL;
 333	list_add(&entry->tv.head, validated);
 334}
 335
 336/**
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 337 * amdgpu_vm_move_to_lru_tail - move all BOs to the end of LRU
 338 *
 339 * @adev: amdgpu device pointer
 340 * @vm: vm providing the BOs
 341 *
 342 * Move all BOs to the end of LRU and remember their positions to put them
 343 * together.
 344 */
 345void amdgpu_vm_move_to_lru_tail(struct amdgpu_device *adev,
 346				struct amdgpu_vm *vm)
 347{
 348	spin_lock(&adev->mman.bdev.lru_lock);
 349	ttm_lru_bulk_move_tail(&vm->lru_bulk_move);
 350	spin_unlock(&adev->mman.bdev.lru_lock);
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 351}
 352
 353/**
 354 * amdgpu_vm_validate_pt_bos - validate the page table BOs
 355 *
 356 * @adev: amdgpu device pointer
 357 * @vm: vm providing the BOs
 358 * @validate: callback to do the validation
 359 * @param: parameter for the validation callback
 360 *
 361 * Validate the page table BOs on command submission if neccessary.
 362 *
 363 * Returns:
 364 * Validation result.
 365 */
 366int amdgpu_vm_validate_pt_bos(struct amdgpu_device *adev, struct amdgpu_vm *vm,
 367			      int (*validate)(void *p, struct amdgpu_bo *bo),
 368			      void *param)
 369{
 370	struct amdgpu_vm_bo_base *bo_base;
 371	struct amdgpu_bo *shadow;
 372	struct amdgpu_bo *bo;
 373	int r;
 374
 375	spin_lock(&vm->status_lock);
 376	while (!list_empty(&vm->evicted)) {
 377		bo_base = list_first_entry(&vm->evicted,
 378					   struct amdgpu_vm_bo_base,
 379					   vm_status);
 380		spin_unlock(&vm->status_lock);
 381
 382		bo = bo_base->bo;
 383		shadow = amdgpu_bo_shadowed(bo);
 384
 385		r = validate(param, bo);
 386		if (r)
 387			return r;
 388		if (shadow) {
 389			r = validate(param, shadow);
 390			if (r)
 391				return r;
 392		}
 393
 394		if (bo->tbo.type != ttm_bo_type_kernel) {
 395			amdgpu_vm_bo_moved(bo_base);
 396		} else {
 397			vm->update_funcs->map_table(to_amdgpu_bo_vm(bo));
 398			amdgpu_vm_bo_relocated(bo_base);
 399		}
 400		spin_lock(&vm->status_lock);
 401	}
 402	spin_unlock(&vm->status_lock);
 403
 404	amdgpu_vm_eviction_lock(vm);
 405	vm->evicting = false;
 406	amdgpu_vm_eviction_unlock(vm);
 407
 408	return 0;
 409}
 410
 411/**
 412 * amdgpu_vm_ready - check VM is ready for updates
 413 *
 414 * @vm: VM to check
 415 *
 416 * Check if all VM PDs/PTs are ready for updates
 417 *
 418 * Returns:
 419 * True if VM is not evicting.
 420 */
 421bool amdgpu_vm_ready(struct amdgpu_vm *vm)
 422{
 423	bool empty;
 424	bool ret;
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 425
 426	amdgpu_vm_eviction_lock(vm);
 427	ret = !vm->evicting;
 428	amdgpu_vm_eviction_unlock(vm);
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 429
 430	spin_lock(&vm->status_lock);
 431	empty = list_empty(&vm->evicted);
 432	spin_unlock(&vm->status_lock);
 433
 434	return ret && empty;
 
 435}
 436
 437/**
 438 * amdgpu_vm_check_compute_bug - check whether asic has compute vm bug
 439 *
 440 * @adev: amdgpu_device pointer
 441 */
 442void amdgpu_vm_check_compute_bug(struct amdgpu_device *adev)
 443{
 444	const struct amdgpu_ip_block *ip_block;
 445	bool has_compute_vm_bug;
 446	struct amdgpu_ring *ring;
 447	int i;
 448
 449	has_compute_vm_bug = false;
 450
 451	ip_block = amdgpu_device_ip_get_ip_block(adev, AMD_IP_BLOCK_TYPE_GFX);
 452	if (ip_block) {
 453		/* Compute has a VM bug for GFX version < 7.
 454		   Compute has a VM bug for GFX 8 MEC firmware version < 673.*/
 455		if (ip_block->version->major <= 7)
 456			has_compute_vm_bug = true;
 457		else if (ip_block->version->major == 8)
 458			if (adev->gfx.mec_fw_version < 673)
 459				has_compute_vm_bug = true;
 460	}
 461
 462	for (i = 0; i < adev->num_rings; i++) {
 463		ring = adev->rings[i];
 464		if (ring->funcs->type == AMDGPU_RING_TYPE_COMPUTE)
 465			/* only compute rings */
 466			ring->has_compute_vm_bug = has_compute_vm_bug;
 467		else
 468			ring->has_compute_vm_bug = false;
 469	}
 470}
 471
 472/**
 473 * amdgpu_vm_need_pipeline_sync - Check if pipe sync is needed for job.
 474 *
 475 * @ring: ring on which the job will be submitted
 476 * @job: job to submit
 477 *
 478 * Returns:
 479 * True if sync is needed.
 480 */
 481bool amdgpu_vm_need_pipeline_sync(struct amdgpu_ring *ring,
 482				  struct amdgpu_job *job)
 483{
 484	struct amdgpu_device *adev = ring->adev;
 485	unsigned vmhub = ring->funcs->vmhub;
 486	struct amdgpu_vmid_mgr *id_mgr = &adev->vm_manager.id_mgr[vmhub];
 
 
 
 487
 488	if (job->vmid == 0)
 489		return false;
 
 
 
 
 
 
 
 
 490
 491	if (job->vm_needs_flush || ring->has_compute_vm_bug)
 492		return true;
 493
 494	if (ring->funcs->emit_gds_switch && job->gds_switch_needed)
 495		return true;
 496
 497	if (amdgpu_vmid_had_gpu_reset(adev, &id_mgr->ids[job->vmid]))
 498		return true;
 499
 500	return false;
 501}
 502
 503/**
 504 * amdgpu_vm_flush - hardware flush the vm
 505 *
 506 * @ring: ring to use for flush
 507 * @job:  related job
 508 * @need_pipe_sync: is pipe sync needed
 509 *
 510 * Emit a VM flush when it is necessary.
 511 *
 512 * Returns:
 513 * 0 on success, errno otherwise.
 514 */
 515int amdgpu_vm_flush(struct amdgpu_ring *ring, struct amdgpu_job *job,
 516		    bool need_pipe_sync)
 517{
 518	struct amdgpu_device *adev = ring->adev;
 519	unsigned vmhub = ring->funcs->vmhub;
 520	struct amdgpu_vmid_mgr *id_mgr = &adev->vm_manager.id_mgr[vmhub];
 521	struct amdgpu_vmid *id = &id_mgr->ids[job->vmid];
 522	bool spm_update_needed = job->spm_update_needed;
 523	bool gds_switch_needed = ring->funcs->emit_gds_switch &&
 524		job->gds_switch_needed;
 
 
 
 
 525	bool vm_flush_needed = job->vm_needs_flush;
 526	struct dma_fence *fence = NULL;
 527	bool pasid_mapping_needed = false;
 528	unsigned patch_offset = 0;
 
 529	int r;
 530
 
 
 
 531	if (amdgpu_vmid_had_gpu_reset(adev, id)) {
 532		gds_switch_needed = true;
 533		vm_flush_needed = true;
 534		pasid_mapping_needed = true;
 535		spm_update_needed = true;
 536	}
 537
 538	mutex_lock(&id_mgr->lock);
 539	if (id->pasid != job->pasid || !id->pasid_mapping ||
 540	    !dma_fence_is_signaled(id->pasid_mapping))
 541		pasid_mapping_needed = true;
 542	mutex_unlock(&id_mgr->lock);
 543
 544	gds_switch_needed &= !!ring->funcs->emit_gds_switch;
 545	vm_flush_needed &= !!ring->funcs->emit_vm_flush  &&
 546			job->vm_pd_addr != AMDGPU_BO_INVALID_OFFSET;
 547	pasid_mapping_needed &= adev->gmc.gmc_funcs->emit_pasid_mapping &&
 548		ring->funcs->emit_wreg;
 549
 550	if (!vm_flush_needed && !gds_switch_needed && !need_pipe_sync)
 551		return 0;
 552
 553	amdgpu_ring_ib_begin(ring);
 554	if (ring->funcs->init_cond_exec)
 555		patch_offset = amdgpu_ring_init_cond_exec(ring);
 556
 557	if (need_pipe_sync)
 558		amdgpu_ring_emit_pipeline_sync(ring);
 559
 560	if (vm_flush_needed) {
 561		trace_amdgpu_vm_flush(ring, job->vmid, job->vm_pd_addr);
 562		amdgpu_ring_emit_vm_flush(ring, job->vmid, job->vm_pd_addr);
 563	}
 564
 565	if (pasid_mapping_needed)
 566		amdgpu_gmc_emit_pasid_mapping(ring, job->vmid, job->pasid);
 567
 568	if (spm_update_needed && adev->gfx.rlc.funcs->update_spm_vmid)
 569		adev->gfx.rlc.funcs->update_spm_vmid(adev, job->vmid);
 570
 571	if (!ring->is_mes_queue && ring->funcs->emit_gds_switch &&
 572	    gds_switch_needed) {
 573		amdgpu_ring_emit_gds_switch(ring, job->vmid, job->gds_base,
 574					    job->gds_size, job->gws_base,
 575					    job->gws_size, job->oa_base,
 576					    job->oa_size);
 577	}
 578
 579	if (vm_flush_needed || pasid_mapping_needed) {
 580		r = amdgpu_fence_emit(ring, &fence, NULL, 0);
 581		if (r)
 582			return r;
 583	}
 584
 585	if (vm_flush_needed) {
 586		mutex_lock(&id_mgr->lock);
 587		dma_fence_put(id->last_flush);
 588		id->last_flush = dma_fence_get(fence);
 589		id->current_gpu_reset_count =
 590			atomic_read(&adev->gpu_reset_counter);
 591		mutex_unlock(&id_mgr->lock);
 592	}
 593
 594	if (pasid_mapping_needed) {
 595		mutex_lock(&id_mgr->lock);
 596		id->pasid = job->pasid;
 597		dma_fence_put(id->pasid_mapping);
 598		id->pasid_mapping = dma_fence_get(fence);
 599		mutex_unlock(&id_mgr->lock);
 600	}
 601	dma_fence_put(fence);
 602
 
 
 
 
 
 
 
 
 
 
 
 
 
 603	if (ring->funcs->patch_cond_exec)
 604		amdgpu_ring_patch_cond_exec(ring, patch_offset);
 605
 606	/* the double SWITCH_BUFFER here *cannot* be skipped by COND_EXEC */
 607	if (ring->funcs->emit_switch_buffer) {
 608		amdgpu_ring_emit_switch_buffer(ring);
 609		amdgpu_ring_emit_switch_buffer(ring);
 610	}
 611	amdgpu_ring_ib_end(ring);
 612	return 0;
 613}
 614
 615/**
 616 * amdgpu_vm_bo_find - find the bo_va for a specific vm & bo
 617 *
 618 * @vm: requested vm
 619 * @bo: requested buffer object
 620 *
 621 * Find @bo inside the requested vm.
 622 * Search inside the @bos vm list for the requested vm
 623 * Returns the found bo_va or NULL if none is found
 624 *
 625 * Object has to be reserved!
 626 *
 627 * Returns:
 628 * Found bo_va or NULL.
 629 */
 630struct amdgpu_bo_va *amdgpu_vm_bo_find(struct amdgpu_vm *vm,
 631				       struct amdgpu_bo *bo)
 632{
 633	struct amdgpu_vm_bo_base *base;
 634
 635	for (base = bo->vm_bo; base; base = base->next) {
 636		if (base->vm != vm)
 637			continue;
 638
 639		return container_of(base, struct amdgpu_bo_va, base);
 640	}
 641	return NULL;
 642}
 643
 644/**
 645 * amdgpu_vm_map_gart - Resolve gart mapping of addr
 646 *
 647 * @pages_addr: optional DMA address to use for lookup
 648 * @addr: the unmapped addr
 649 *
 650 * Look up the physical address of the page that the pte resolves
 651 * to.
 652 *
 653 * Returns:
 654 * The pointer for the page table entry.
 655 */
 656uint64_t amdgpu_vm_map_gart(const dma_addr_t *pages_addr, uint64_t addr)
 657{
 658	uint64_t result;
 659
 660	/* page table offset */
 661	result = pages_addr[addr >> PAGE_SHIFT];
 662
 663	/* in case cpu page size != gpu page size*/
 664	result |= addr & (~PAGE_MASK);
 665
 666	result &= 0xFFFFFFFFFFFFF000ULL;
 667
 668	return result;
 669}
 670
 671/**
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 672 * amdgpu_vm_update_pdes - make sure that all directories are valid
 673 *
 674 * @adev: amdgpu_device pointer
 675 * @vm: requested vm
 676 * @immediate: submit immediately to the paging queue
 677 *
 678 * Makes sure all directories are up to date.
 679 *
 680 * Returns:
 681 * 0 for success, error for failure.
 682 */
 683int amdgpu_vm_update_pdes(struct amdgpu_device *adev,
 684			  struct amdgpu_vm *vm, bool immediate)
 685{
 686	struct amdgpu_vm_update_params params;
 687	struct amdgpu_vm_bo_base *entry;
 688	bool flush_tlb_needed = false;
 689	LIST_HEAD(relocated);
 690	int r, idx;
 691
 692	spin_lock(&vm->status_lock);
 693	list_splice_init(&vm->relocated, &relocated);
 694	spin_unlock(&vm->status_lock);
 695
 696	if (list_empty(&relocated))
 697		return 0;
 698
 699	if (!drm_dev_enter(adev_to_drm(adev), &idx))
 700		return -ENODEV;
 701
 702	memset(&params, 0, sizeof(params));
 703	params.adev = adev;
 704	params.vm = vm;
 705	params.immediate = immediate;
 706
 707	r = vm->update_funcs->prepare(&params, NULL, AMDGPU_SYNC_EXPLICIT);
 708	if (r)
 709		goto error;
 
 
 
 710
 711	list_for_each_entry(entry, &relocated, vm_status) {
 712		/* vm_flush_needed after updating moved PDEs */
 713		flush_tlb_needed |= entry->moved;
 714
 715		r = amdgpu_vm_pde_update(&params, entry);
 716		if (r)
 717			goto error;
 718	}
 719
 720	r = vm->update_funcs->commit(&params, &vm->last_update);
 721	if (r)
 722		goto error;
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 723
 724	if (flush_tlb_needed)
 725		atomic64_inc(&vm->tlb_seq);
 
 726
 727	while (!list_empty(&relocated)) {
 728		entry = list_first_entry(&relocated, struct amdgpu_vm_bo_base,
 729					 vm_status);
 730		amdgpu_vm_bo_idle(entry);
 731	}
 732
 733error:
 734	drm_dev_exit(idx);
 735	return r;
 736}
 737
 738/**
 739 * amdgpu_vm_tlb_seq_cb - make sure to increment tlb sequence
 740 * @fence: unused
 741 * @cb: the callback structure
 742 *
 743 * Increments the tlb sequence to make sure that future CS execute a VM flush.
 
 
 
 
 
 
 
 744 */
 745static void amdgpu_vm_tlb_seq_cb(struct dma_fence *fence,
 746				 struct dma_fence_cb *cb)
 
 747{
 748	struct amdgpu_vm_tlb_seq_cb *tlb_cb;
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 749
 750	tlb_cb = container_of(cb, typeof(*tlb_cb), cb);
 751	atomic64_inc(&tlb_cb->vm->tlb_seq);
 752	kfree(tlb_cb);
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 753}
 754
 755/**
 756 * amdgpu_vm_update_range - update a range in the vm page table
 757 *
 758 * @adev: amdgpu_device pointer to use for commands
 759 * @vm: the VM to update the range
 760 * @immediate: immediate submission in a page fault
 761 * @unlocked: unlocked invalidation during MM callback
 762 * @flush_tlb: trigger tlb invalidation after update completed
 763 * @resv: fences we need to sync to
 764 * @start: start of mapped range
 765 * @last: last mapped entry
 766 * @flags: flags for the entries
 767 * @offset: offset into nodes and pages_addr
 768 * @vram_base: base for vram mappings
 769 * @res: ttm_resource to map
 770 * @pages_addr: DMA addresses to use for mapping
 771 * @fence: optional resulting fence
 772 *
 773 * Fill in the page table entries between @start and @last.
 774 *
 775 * Returns:
 776 * 0 for success, negative erro code for failure.
 777 */
 778int amdgpu_vm_update_range(struct amdgpu_device *adev, struct amdgpu_vm *vm,
 779			   bool immediate, bool unlocked, bool flush_tlb,
 780			   struct dma_resv *resv, uint64_t start, uint64_t last,
 781			   uint64_t flags, uint64_t offset, uint64_t vram_base,
 782			   struct ttm_resource *res, dma_addr_t *pages_addr,
 783			   struct dma_fence **fence)
 
 784{
 785	struct amdgpu_vm_update_params params;
 786	struct amdgpu_vm_tlb_seq_cb *tlb_cb;
 787	struct amdgpu_res_cursor cursor;
 788	enum amdgpu_sync_mode sync_mode;
 789	int r, idx;
 790
 791	if (!drm_dev_enter(adev_to_drm(adev), &idx))
 792		return -ENODEV;
 793
 794	tlb_cb = kmalloc(sizeof(*tlb_cb), GFP_KERNEL);
 795	if (!tlb_cb) {
 796		r = -ENOMEM;
 797		goto error_unlock;
 798	}
 799
 800	/* Vega20+XGMI where PTEs get inadvertently cached in L2 texture cache,
 801	 * heavy-weight flush TLB unconditionally.
 802	 */
 803	flush_tlb |= adev->gmc.xgmi.num_physical_nodes &&
 804		     adev->ip_versions[GC_HWIP][0] == IP_VERSION(9, 4, 0);
 805
 806	/*
 807	 * On GFX8 and older any 8 PTE block with a valid bit set enters the TLB
 808	 */
 809	flush_tlb |= adev->ip_versions[GC_HWIP][0] < IP_VERSION(9, 0, 0);
 810
 811	memset(&params, 0, sizeof(params));
 812	params.adev = adev;
 813	params.vm = vm;
 814	params.immediate = immediate;
 815	params.pages_addr = pages_addr;
 816	params.unlocked = unlocked;
 817
 818	/* Implicitly sync to command submissions in the same VM before
 819	 * unmapping. Sync to moving fences before mapping.
 820	 */
 821	if (!(flags & AMDGPU_PTE_VALID))
 822		sync_mode = AMDGPU_SYNC_EQ_OWNER;
 823	else
 824		sync_mode = AMDGPU_SYNC_EXPLICIT;
 825
 826	amdgpu_vm_eviction_lock(vm);
 827	if (vm->evicting) {
 828		r = -EBUSY;
 829		goto error_free;
 830	}
 831
 832	if (!unlocked && !dma_fence_is_signaled(vm->last_unlocked)) {
 833		struct dma_fence *tmp = dma_fence_get_stub();
 834
 835		amdgpu_bo_fence(vm->root.bo, vm->last_unlocked, true);
 836		swap(vm->last_unlocked, tmp);
 837		dma_fence_put(tmp);
 838	}
 839
 840	r = vm->update_funcs->prepare(&params, resv, sync_mode);
 841	if (r)
 842		goto error_free;
 843
 844	amdgpu_res_first(pages_addr ? NULL : res, offset,
 845			 (last - start + 1) * AMDGPU_GPU_PAGE_SIZE, &cursor);
 846	while (cursor.remaining) {
 847		uint64_t tmp, num_entries, addr;
 
 848
 849		num_entries = cursor.size >> AMDGPU_GPU_PAGE_SHIFT;
 850		if (pages_addr) {
 851			bool contiguous = true;
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 852
 853			if (num_entries > AMDGPU_GPU_PAGES_IN_CPU_PAGE) {
 854				uint64_t pfn = cursor.start >> PAGE_SHIFT;
 855				uint64_t count;
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 856
 857				contiguous = pages_addr[pfn + 1] ==
 858					pages_addr[pfn] + PAGE_SIZE;
 
 
 
 
 
 
 
 
 
 
 
 859
 860				tmp = num_entries /
 861					AMDGPU_GPU_PAGES_IN_CPU_PAGE;
 862				for (count = 2; count < tmp; ++count) {
 863					uint64_t idx = pfn + count;
 864
 865					if (contiguous != (pages_addr[idx] ==
 866					    pages_addr[idx - 1] + PAGE_SIZE))
 867						break;
 868				}
 869				num_entries = count *
 870					AMDGPU_GPU_PAGES_IN_CPU_PAGE;
 
 
 871			}
 872
 873			if (!contiguous) {
 874				addr = cursor.start;
 875				params.pages_addr = pages_addr;
 876			} else {
 877				addr = pages_addr[cursor.start >> PAGE_SHIFT];
 878				params.pages_addr = NULL;
 
 879			}
 880
 881		} else if (flags & (AMDGPU_PTE_VALID | AMDGPU_PTE_PRT)) {
 882			addr = vram_base + cursor.start;
 883		} else {
 884			addr = 0;
 885		}
 886
 887		tmp = start + num_entries;
 888		r = amdgpu_vm_ptes_update(&params, start, tmp, addr, flags);
 
 
 889		if (r)
 890			goto error_free;
 891
 892		amdgpu_res_next(&cursor, num_entries * AMDGPU_GPU_PAGE_SIZE);
 893		start = tmp;
 894	}
 895
 896	r = vm->update_funcs->commit(&params, fence);
 897
 898	if (flush_tlb || params.table_freed) {
 899		tlb_cb->vm = vm;
 900		if (fence && *fence &&
 901		    !dma_fence_add_callback(*fence, &tlb_cb->cb,
 902					   amdgpu_vm_tlb_seq_cb)) {
 903			dma_fence_put(vm->last_tlb_flush);
 904			vm->last_tlb_flush = dma_fence_get(*fence);
 905		} else {
 906			amdgpu_vm_tlb_seq_cb(NULL, &tlb_cb->cb);
 907		}
 908		tlb_cb = NULL;
 909	}
 910
 911error_free:
 912	kfree(tlb_cb);
 913
 914error_unlock:
 915	amdgpu_vm_eviction_unlock(vm);
 916	drm_dev_exit(idx);
 917	return r;
 918}
 919
 920void amdgpu_vm_get_memory(struct amdgpu_vm *vm, uint64_t *vram_mem,
 921				uint64_t *gtt_mem, uint64_t *cpu_mem)
 922{
 923	struct amdgpu_bo_va *bo_va, *tmp;
 924
 925	spin_lock(&vm->status_lock);
 926	list_for_each_entry_safe(bo_va, tmp, &vm->idle, base.vm_status) {
 927		if (!bo_va->base.bo)
 928			continue;
 929		amdgpu_bo_get_memory(bo_va->base.bo, vram_mem,
 930				gtt_mem, cpu_mem);
 931	}
 932	list_for_each_entry_safe(bo_va, tmp, &vm->evicted, base.vm_status) {
 933		if (!bo_va->base.bo)
 934			continue;
 935		amdgpu_bo_get_memory(bo_va->base.bo, vram_mem,
 936				gtt_mem, cpu_mem);
 937	}
 938	list_for_each_entry_safe(bo_va, tmp, &vm->relocated, base.vm_status) {
 939		if (!bo_va->base.bo)
 940			continue;
 941		amdgpu_bo_get_memory(bo_va->base.bo, vram_mem,
 942				gtt_mem, cpu_mem);
 943	}
 944	list_for_each_entry_safe(bo_va, tmp, &vm->moved, base.vm_status) {
 945		if (!bo_va->base.bo)
 946			continue;
 947		amdgpu_bo_get_memory(bo_va->base.bo, vram_mem,
 948				gtt_mem, cpu_mem);
 949	}
 950	list_for_each_entry_safe(bo_va, tmp, &vm->invalidated, base.vm_status) {
 951		if (!bo_va->base.bo)
 952			continue;
 953		amdgpu_bo_get_memory(bo_va->base.bo, vram_mem,
 954				gtt_mem, cpu_mem);
 955	}
 956	list_for_each_entry_safe(bo_va, tmp, &vm->done, base.vm_status) {
 957		if (!bo_va->base.bo)
 958			continue;
 959		amdgpu_bo_get_memory(bo_va->base.bo, vram_mem,
 960				gtt_mem, cpu_mem);
 961	}
 962	spin_unlock(&vm->status_lock);
 963}
 964/**
 965 * amdgpu_vm_bo_update - update all BO mappings in the vm page table
 966 *
 967 * @adev: amdgpu_device pointer
 968 * @bo_va: requested BO and VM object
 969 * @clear: if true clear the entries
 970 *
 971 * Fill in the page table entries for @bo_va.
 972 *
 973 * Returns:
 974 * 0 for success, -EINVAL for failure.
 975 */
 976int amdgpu_vm_bo_update(struct amdgpu_device *adev, struct amdgpu_bo_va *bo_va,
 977			bool clear)
 978{
 979	struct amdgpu_bo *bo = bo_va->base.bo;
 980	struct amdgpu_vm *vm = bo_va->base.vm;
 981	struct amdgpu_bo_va_mapping *mapping;
 982	dma_addr_t *pages_addr = NULL;
 983	struct ttm_resource *mem;
 
 984	struct dma_fence **last_update;
 985	bool flush_tlb = clear;
 986	struct dma_resv *resv;
 987	uint64_t vram_base;
 988	uint64_t flags;
 
 989	int r;
 990
 991	if (clear || !bo) {
 992		mem = NULL;
 993		resv = vm->root.bo->tbo.base.resv;
 
 994	} else {
 995		struct drm_gem_object *obj = &bo->tbo.base;
 996
 
 
 
 
 
 
 997		resv = bo->tbo.base.resv;
 998		if (obj->import_attach && bo_va->is_xgmi) {
 999			struct dma_buf *dma_buf = obj->import_attach->dmabuf;
1000			struct drm_gem_object *gobj = dma_buf->priv;
1001			struct amdgpu_bo *abo = gem_to_amdgpu_bo(gobj);
1002
1003			if (abo->tbo.resource->mem_type == TTM_PL_VRAM)
1004				bo = gem_to_amdgpu_bo(gobj);
1005		}
1006		mem = bo->tbo.resource;
1007		if (mem->mem_type == TTM_PL_TT ||
1008		    mem->mem_type == AMDGPU_PL_PREEMPT)
1009			pages_addr = bo->tbo.ttm->dma_address;
1010	}
1011
1012	if (bo) {
1013		struct amdgpu_device *bo_adev;
1014
1015		flags = amdgpu_ttm_tt_pte_flags(adev, bo->tbo.ttm, mem);
1016
1017		if (amdgpu_bo_encrypted(bo))
1018			flags |= AMDGPU_PTE_TMZ;
1019
1020		bo_adev = amdgpu_ttm_adev(bo->tbo.bdev);
1021		vram_base = bo_adev->vm_manager.vram_base_offset;
1022	} else {
1023		flags = 0x0;
1024		vram_base = 0;
1025	}
1026
1027	if (clear || (bo && bo->tbo.base.resv ==
1028		      vm->root.bo->tbo.base.resv))
1029		last_update = &vm->last_update;
1030	else
1031		last_update = &bo_va->last_pt_update;
1032
1033	if (!clear && bo_va->base.moved) {
1034		flush_tlb = true;
1035		list_splice_init(&bo_va->valids, &bo_va->invalids);
1036
1037	} else if (bo_va->cleared != clear) {
1038		list_splice_init(&bo_va->valids, &bo_va->invalids);
1039	}
1040
1041	list_for_each_entry(mapping, &bo_va->invalids, list) {
1042		uint64_t update_flags = flags;
1043
1044		/* normally,bo_va->flags only contians READABLE and WIRTEABLE bit go here
1045		 * but in case of something, we filter the flags in first place
1046		 */
1047		if (!(mapping->flags & AMDGPU_PTE_READABLE))
1048			update_flags &= ~AMDGPU_PTE_READABLE;
1049		if (!(mapping->flags & AMDGPU_PTE_WRITEABLE))
1050			update_flags &= ~AMDGPU_PTE_WRITEABLE;
1051
1052		/* Apply ASIC specific mapping flags */
1053		amdgpu_gmc_get_vm_pte(adev, mapping, &update_flags);
1054
1055		trace_amdgpu_vm_bo_update(mapping);
1056
1057		r = amdgpu_vm_update_range(adev, vm, false, false, flush_tlb,
1058					   resv, mapping->start, mapping->last,
1059					   update_flags, mapping->offset,
1060					   vram_base, mem, pages_addr,
1061					   last_update);
1062		if (r)
1063			return r;
1064	}
1065
1066	/* If the BO is not in its preferred location add it back to
1067	 * the evicted list so that it gets validated again on the
1068	 * next command submission.
1069	 */
1070	if (bo && bo->tbo.base.resv == vm->root.bo->tbo.base.resv) {
1071		uint32_t mem_type = bo->tbo.resource->mem_type;
1072
1073		if (!(bo->preferred_domains &
1074		      amdgpu_mem_type_to_domain(mem_type)))
1075			amdgpu_vm_bo_evicted(&bo_va->base);
1076		else
1077			amdgpu_vm_bo_idle(&bo_va->base);
1078	} else {
1079		amdgpu_vm_bo_done(&bo_va->base);
1080	}
1081
1082	list_splice_init(&bo_va->invalids, &bo_va->valids);
1083	bo_va->cleared = clear;
1084	bo_va->base.moved = false;
1085
1086	if (trace_amdgpu_vm_bo_mapping_enabled()) {
1087		list_for_each_entry(mapping, &bo_va->valids, list)
1088			trace_amdgpu_vm_bo_mapping(mapping);
1089	}
1090
1091	return 0;
1092}
1093
1094/**
1095 * amdgpu_vm_update_prt_state - update the global PRT state
1096 *
1097 * @adev: amdgpu_device pointer
1098 */
1099static void amdgpu_vm_update_prt_state(struct amdgpu_device *adev)
1100{
1101	unsigned long flags;
1102	bool enable;
1103
1104	spin_lock_irqsave(&adev->vm_manager.prt_lock, flags);
1105	enable = !!atomic_read(&adev->vm_manager.num_prt_users);
1106	adev->gmc.gmc_funcs->set_prt(adev, enable);
1107	spin_unlock_irqrestore(&adev->vm_manager.prt_lock, flags);
1108}
1109
1110/**
1111 * amdgpu_vm_prt_get - add a PRT user
1112 *
1113 * @adev: amdgpu_device pointer
1114 */
1115static void amdgpu_vm_prt_get(struct amdgpu_device *adev)
1116{
1117	if (!adev->gmc.gmc_funcs->set_prt)
1118		return;
1119
1120	if (atomic_inc_return(&adev->vm_manager.num_prt_users) == 1)
1121		amdgpu_vm_update_prt_state(adev);
1122}
1123
1124/**
1125 * amdgpu_vm_prt_put - drop a PRT user
1126 *
1127 * @adev: amdgpu_device pointer
1128 */
1129static void amdgpu_vm_prt_put(struct amdgpu_device *adev)
1130{
1131	if (atomic_dec_return(&adev->vm_manager.num_prt_users) == 0)
1132		amdgpu_vm_update_prt_state(adev);
1133}
1134
1135/**
1136 * amdgpu_vm_prt_cb - callback for updating the PRT status
1137 *
1138 * @fence: fence for the callback
1139 * @_cb: the callback function
1140 */
1141static void amdgpu_vm_prt_cb(struct dma_fence *fence, struct dma_fence_cb *_cb)
1142{
1143	struct amdgpu_prt_cb *cb = container_of(_cb, struct amdgpu_prt_cb, cb);
1144
1145	amdgpu_vm_prt_put(cb->adev);
1146	kfree(cb);
1147}
1148
1149/**
1150 * amdgpu_vm_add_prt_cb - add callback for updating the PRT status
1151 *
1152 * @adev: amdgpu_device pointer
1153 * @fence: fence for the callback
1154 */
1155static void amdgpu_vm_add_prt_cb(struct amdgpu_device *adev,
1156				 struct dma_fence *fence)
1157{
1158	struct amdgpu_prt_cb *cb;
1159
1160	if (!adev->gmc.gmc_funcs->set_prt)
1161		return;
1162
1163	cb = kmalloc(sizeof(struct amdgpu_prt_cb), GFP_KERNEL);
1164	if (!cb) {
1165		/* Last resort when we are OOM */
1166		if (fence)
1167			dma_fence_wait(fence, false);
1168
1169		amdgpu_vm_prt_put(adev);
1170	} else {
1171		cb->adev = adev;
1172		if (!fence || dma_fence_add_callback(fence, &cb->cb,
1173						     amdgpu_vm_prt_cb))
1174			amdgpu_vm_prt_cb(fence, &cb->cb);
1175	}
1176}
1177
1178/**
1179 * amdgpu_vm_free_mapping - free a mapping
1180 *
1181 * @adev: amdgpu_device pointer
1182 * @vm: requested vm
1183 * @mapping: mapping to be freed
1184 * @fence: fence of the unmap operation
1185 *
1186 * Free a mapping and make sure we decrease the PRT usage count if applicable.
1187 */
1188static void amdgpu_vm_free_mapping(struct amdgpu_device *adev,
1189				   struct amdgpu_vm *vm,
1190				   struct amdgpu_bo_va_mapping *mapping,
1191				   struct dma_fence *fence)
1192{
1193	if (mapping->flags & AMDGPU_PTE_PRT)
1194		amdgpu_vm_add_prt_cb(adev, fence);
1195	kfree(mapping);
1196}
1197
1198/**
1199 * amdgpu_vm_prt_fini - finish all prt mappings
1200 *
1201 * @adev: amdgpu_device pointer
1202 * @vm: requested vm
1203 *
1204 * Register a cleanup callback to disable PRT support after VM dies.
1205 */
1206static void amdgpu_vm_prt_fini(struct amdgpu_device *adev, struct amdgpu_vm *vm)
1207{
1208	struct dma_resv *resv = vm->root.bo->tbo.base.resv;
1209	struct dma_resv_iter cursor;
1210	struct dma_fence *fence;
 
 
 
 
 
 
 
 
 
 
 
 
1211
1212	dma_resv_for_each_fence(&cursor, resv, DMA_RESV_USAGE_BOOKKEEP, fence) {
1213		/* Add a callback for each fence in the reservation object */
 
 
 
1214		amdgpu_vm_prt_get(adev);
1215		amdgpu_vm_add_prt_cb(adev, fence);
1216	}
 
 
1217}
1218
1219/**
1220 * amdgpu_vm_clear_freed - clear freed BOs in the PT
1221 *
1222 * @adev: amdgpu_device pointer
1223 * @vm: requested vm
1224 * @fence: optional resulting fence (unchanged if no work needed to be done
1225 * or if an error occurred)
1226 *
1227 * Make sure all freed BOs are cleared in the PT.
1228 * PTs have to be reserved and mutex must be locked!
1229 *
1230 * Returns:
1231 * 0 for success.
1232 *
1233 */
1234int amdgpu_vm_clear_freed(struct amdgpu_device *adev,
1235			  struct amdgpu_vm *vm,
1236			  struct dma_fence **fence)
1237{
1238	struct dma_resv *resv = vm->root.bo->tbo.base.resv;
1239	struct amdgpu_bo_va_mapping *mapping;
1240	uint64_t init_pte_value = 0;
1241	struct dma_fence *f = NULL;
1242	int r;
1243
1244	while (!list_empty(&vm->freed)) {
1245		mapping = list_first_entry(&vm->freed,
1246			struct amdgpu_bo_va_mapping, list);
1247		list_del(&mapping->list);
1248
1249		if (vm->pte_support_ats &&
1250		    mapping->start < AMDGPU_GMC_HOLE_START)
1251			init_pte_value = AMDGPU_PTE_DEFAULT_ATC;
1252
1253		r = amdgpu_vm_update_range(adev, vm, false, false, true, resv,
1254					   mapping->start, mapping->last,
1255					   init_pte_value, 0, 0, NULL, NULL,
1256					   &f);
1257		amdgpu_vm_free_mapping(adev, vm, mapping, f);
1258		if (r) {
1259			dma_fence_put(f);
1260			return r;
1261		}
1262	}
1263
1264	if (fence && f) {
1265		dma_fence_put(*fence);
1266		*fence = f;
1267	} else {
1268		dma_fence_put(f);
1269	}
1270
1271	return 0;
1272
1273}
1274
1275/**
1276 * amdgpu_vm_handle_moved - handle moved BOs in the PT
1277 *
1278 * @adev: amdgpu_device pointer
1279 * @vm: requested vm
1280 *
1281 * Make sure all BOs which are moved are updated in the PTs.
1282 *
1283 * Returns:
1284 * 0 for success.
1285 *
1286 * PTs have to be reserved!
1287 */
1288int amdgpu_vm_handle_moved(struct amdgpu_device *adev,
1289			   struct amdgpu_vm *vm)
1290{
1291	struct amdgpu_bo_va *bo_va;
1292	struct dma_resv *resv;
1293	bool clear;
1294	int r;
1295
1296	spin_lock(&vm->status_lock);
1297	while (!list_empty(&vm->moved)) {
1298		bo_va = list_first_entry(&vm->moved, struct amdgpu_bo_va,
1299					 base.vm_status);
1300		spin_unlock(&vm->status_lock);
1301
1302		/* Per VM BOs never need to bo cleared in the page tables */
1303		r = amdgpu_vm_bo_update(adev, bo_va, false);
1304		if (r)
1305			return r;
1306		spin_lock(&vm->status_lock);
1307	}
1308
 
1309	while (!list_empty(&vm->invalidated)) {
1310		bo_va = list_first_entry(&vm->invalidated, struct amdgpu_bo_va,
1311					 base.vm_status);
1312		resv = bo_va->base.bo->tbo.base.resv;
1313		spin_unlock(&vm->status_lock);
1314
1315		/* Try to reserve the BO to avoid clearing its ptes */
1316		if (!amdgpu_vm_debug && dma_resv_trylock(resv))
1317			clear = false;
1318		/* Somebody else is using the BO right now */
1319		else
1320			clear = true;
1321
1322		r = amdgpu_vm_bo_update(adev, bo_va, clear);
1323		if (r)
1324			return r;
1325
1326		if (!clear)
1327			dma_resv_unlock(resv);
1328		spin_lock(&vm->status_lock);
1329	}
1330	spin_unlock(&vm->status_lock);
1331
1332	return 0;
1333}
1334
1335/**
1336 * amdgpu_vm_bo_add - add a bo to a specific vm
1337 *
1338 * @adev: amdgpu_device pointer
1339 * @vm: requested vm
1340 * @bo: amdgpu buffer object
1341 *
1342 * Add @bo into the requested vm.
1343 * Add @bo to the list of bos associated with the vm
1344 *
1345 * Returns:
1346 * Newly added bo_va or NULL for failure
1347 *
1348 * Object has to be reserved!
1349 */
1350struct amdgpu_bo_va *amdgpu_vm_bo_add(struct amdgpu_device *adev,
1351				      struct amdgpu_vm *vm,
1352				      struct amdgpu_bo *bo)
1353{
1354	struct amdgpu_bo_va *bo_va;
1355
1356	bo_va = kzalloc(sizeof(struct amdgpu_bo_va), GFP_KERNEL);
1357	if (bo_va == NULL) {
1358		return NULL;
1359	}
1360	amdgpu_vm_bo_base_init(&bo_va->base, vm, bo);
1361
1362	bo_va->ref_count = 1;
1363	INIT_LIST_HEAD(&bo_va->valids);
1364	INIT_LIST_HEAD(&bo_va->invalids);
1365
1366	if (!bo)
1367		return bo_va;
1368
1369	dma_resv_assert_held(bo->tbo.base.resv);
1370	if (amdgpu_dmabuf_is_xgmi_accessible(adev, bo)) {
1371		bo_va->is_xgmi = true;
1372		/* Power up XGMI if it can be potentially used */
1373		amdgpu_xgmi_set_pstate(adev, AMDGPU_XGMI_PSTATE_MAX_VEGA20);
1374	}
1375
1376	return bo_va;
1377}
1378
1379
1380/**
1381 * amdgpu_vm_bo_insert_map - insert a new mapping
1382 *
1383 * @adev: amdgpu_device pointer
1384 * @bo_va: bo_va to store the address
1385 * @mapping: the mapping to insert
1386 *
1387 * Insert a new mapping into all structures.
1388 */
1389static void amdgpu_vm_bo_insert_map(struct amdgpu_device *adev,
1390				    struct amdgpu_bo_va *bo_va,
1391				    struct amdgpu_bo_va_mapping *mapping)
1392{
1393	struct amdgpu_vm *vm = bo_va->base.vm;
1394	struct amdgpu_bo *bo = bo_va->base.bo;
1395
1396	mapping->bo_va = bo_va;
1397	list_add(&mapping->list, &bo_va->invalids);
1398	amdgpu_vm_it_insert(mapping, &vm->va);
1399
1400	if (mapping->flags & AMDGPU_PTE_PRT)
1401		amdgpu_vm_prt_get(adev);
1402
1403	if (bo && bo->tbo.base.resv == vm->root.bo->tbo.base.resv &&
1404	    !bo_va->base.moved) {
1405		amdgpu_vm_bo_moved(&bo_va->base);
1406	}
1407	trace_amdgpu_vm_bo_map(bo_va, mapping);
1408}
1409
1410/**
1411 * amdgpu_vm_bo_map - map bo inside a vm
1412 *
1413 * @adev: amdgpu_device pointer
1414 * @bo_va: bo_va to store the address
1415 * @saddr: where to map the BO
1416 * @offset: requested offset in the BO
1417 * @size: BO size in bytes
1418 * @flags: attributes of pages (read/write/valid/etc.)
1419 *
1420 * Add a mapping of the BO at the specefied addr into the VM.
1421 *
1422 * Returns:
1423 * 0 for success, error for failure.
1424 *
1425 * Object has to be reserved and unreserved outside!
1426 */
1427int amdgpu_vm_bo_map(struct amdgpu_device *adev,
1428		     struct amdgpu_bo_va *bo_va,
1429		     uint64_t saddr, uint64_t offset,
1430		     uint64_t size, uint64_t flags)
1431{
1432	struct amdgpu_bo_va_mapping *mapping, *tmp;
1433	struct amdgpu_bo *bo = bo_va->base.bo;
1434	struct amdgpu_vm *vm = bo_va->base.vm;
1435	uint64_t eaddr;
1436
1437	/* validate the parameters */
1438	if (saddr & ~PAGE_MASK || offset & ~PAGE_MASK ||
1439	    size == 0 || size & ~PAGE_MASK)
1440		return -EINVAL;
1441
1442	/* make sure object fit at this offset */
1443	eaddr = saddr + size - 1;
1444	if (saddr >= eaddr ||
1445	    (bo && offset + size > amdgpu_bo_size(bo)) ||
1446	    (eaddr >= adev->vm_manager.max_pfn << AMDGPU_GPU_PAGE_SHIFT))
1447		return -EINVAL;
1448
1449	saddr /= AMDGPU_GPU_PAGE_SIZE;
1450	eaddr /= AMDGPU_GPU_PAGE_SIZE;
1451
1452	tmp = amdgpu_vm_it_iter_first(&vm->va, saddr, eaddr);
1453	if (tmp) {
1454		/* bo and tmp overlap, invalid addr */
1455		dev_err(adev->dev, "bo %p va 0x%010Lx-0x%010Lx conflict with "
1456			"0x%010Lx-0x%010Lx\n", bo, saddr, eaddr,
1457			tmp->start, tmp->last + 1);
1458		return -EINVAL;
1459	}
1460
1461	mapping = kmalloc(sizeof(*mapping), GFP_KERNEL);
1462	if (!mapping)
1463		return -ENOMEM;
1464
1465	mapping->start = saddr;
1466	mapping->last = eaddr;
1467	mapping->offset = offset;
1468	mapping->flags = flags;
1469
1470	amdgpu_vm_bo_insert_map(adev, bo_va, mapping);
1471
1472	return 0;
1473}
1474
1475/**
1476 * amdgpu_vm_bo_replace_map - map bo inside a vm, replacing existing mappings
1477 *
1478 * @adev: amdgpu_device pointer
1479 * @bo_va: bo_va to store the address
1480 * @saddr: where to map the BO
1481 * @offset: requested offset in the BO
1482 * @size: BO size in bytes
1483 * @flags: attributes of pages (read/write/valid/etc.)
1484 *
1485 * Add a mapping of the BO at the specefied addr into the VM. Replace existing
1486 * mappings as we do so.
1487 *
1488 * Returns:
1489 * 0 for success, error for failure.
1490 *
1491 * Object has to be reserved and unreserved outside!
1492 */
1493int amdgpu_vm_bo_replace_map(struct amdgpu_device *adev,
1494			     struct amdgpu_bo_va *bo_va,
1495			     uint64_t saddr, uint64_t offset,
1496			     uint64_t size, uint64_t flags)
1497{
1498	struct amdgpu_bo_va_mapping *mapping;
1499	struct amdgpu_bo *bo = bo_va->base.bo;
1500	uint64_t eaddr;
1501	int r;
1502
1503	/* validate the parameters */
1504	if (saddr & ~PAGE_MASK || offset & ~PAGE_MASK ||
1505	    size == 0 || size & ~PAGE_MASK)
1506		return -EINVAL;
1507
1508	/* make sure object fit at this offset */
1509	eaddr = saddr + size - 1;
1510	if (saddr >= eaddr ||
1511	    (bo && offset + size > amdgpu_bo_size(bo)) ||
1512	    (eaddr >= adev->vm_manager.max_pfn << AMDGPU_GPU_PAGE_SHIFT))
1513		return -EINVAL;
1514
1515	/* Allocate all the needed memory */
1516	mapping = kmalloc(sizeof(*mapping), GFP_KERNEL);
1517	if (!mapping)
1518		return -ENOMEM;
1519
1520	r = amdgpu_vm_bo_clear_mappings(adev, bo_va->base.vm, saddr, size);
1521	if (r) {
1522		kfree(mapping);
1523		return r;
1524	}
1525
1526	saddr /= AMDGPU_GPU_PAGE_SIZE;
1527	eaddr /= AMDGPU_GPU_PAGE_SIZE;
1528
1529	mapping->start = saddr;
1530	mapping->last = eaddr;
1531	mapping->offset = offset;
1532	mapping->flags = flags;
1533
1534	amdgpu_vm_bo_insert_map(adev, bo_va, mapping);
1535
1536	return 0;
1537}
1538
1539/**
1540 * amdgpu_vm_bo_unmap - remove bo mapping from vm
1541 *
1542 * @adev: amdgpu_device pointer
1543 * @bo_va: bo_va to remove the address from
1544 * @saddr: where to the BO is mapped
1545 *
1546 * Remove a mapping of the BO at the specefied addr from the VM.
1547 *
1548 * Returns:
1549 * 0 for success, error for failure.
1550 *
1551 * Object has to be reserved and unreserved outside!
1552 */
1553int amdgpu_vm_bo_unmap(struct amdgpu_device *adev,
1554		       struct amdgpu_bo_va *bo_va,
1555		       uint64_t saddr)
1556{
1557	struct amdgpu_bo_va_mapping *mapping;
1558	struct amdgpu_vm *vm = bo_va->base.vm;
1559	bool valid = true;
1560
1561	saddr /= AMDGPU_GPU_PAGE_SIZE;
1562
1563	list_for_each_entry(mapping, &bo_va->valids, list) {
1564		if (mapping->start == saddr)
1565			break;
1566	}
1567
1568	if (&mapping->list == &bo_va->valids) {
1569		valid = false;
1570
1571		list_for_each_entry(mapping, &bo_va->invalids, list) {
1572			if (mapping->start == saddr)
1573				break;
1574		}
1575
1576		if (&mapping->list == &bo_va->invalids)
1577			return -ENOENT;
1578	}
1579
1580	list_del(&mapping->list);
1581	amdgpu_vm_it_remove(mapping, &vm->va);
1582	mapping->bo_va = NULL;
1583	trace_amdgpu_vm_bo_unmap(bo_va, mapping);
1584
1585	if (valid)
1586		list_add(&mapping->list, &vm->freed);
1587	else
1588		amdgpu_vm_free_mapping(adev, vm, mapping,
1589				       bo_va->last_pt_update);
1590
1591	return 0;
1592}
1593
1594/**
1595 * amdgpu_vm_bo_clear_mappings - remove all mappings in a specific range
1596 *
1597 * @adev: amdgpu_device pointer
1598 * @vm: VM structure to use
1599 * @saddr: start of the range
1600 * @size: size of the range
1601 *
1602 * Remove all mappings in a range, split them as appropriate.
1603 *
1604 * Returns:
1605 * 0 for success, error for failure.
1606 */
1607int amdgpu_vm_bo_clear_mappings(struct amdgpu_device *adev,
1608				struct amdgpu_vm *vm,
1609				uint64_t saddr, uint64_t size)
1610{
1611	struct amdgpu_bo_va_mapping *before, *after, *tmp, *next;
1612	LIST_HEAD(removed);
1613	uint64_t eaddr;
1614
1615	eaddr = saddr + size - 1;
1616	saddr /= AMDGPU_GPU_PAGE_SIZE;
1617	eaddr /= AMDGPU_GPU_PAGE_SIZE;
1618
1619	/* Allocate all the needed memory */
1620	before = kzalloc(sizeof(*before), GFP_KERNEL);
1621	if (!before)
1622		return -ENOMEM;
1623	INIT_LIST_HEAD(&before->list);
1624
1625	after = kzalloc(sizeof(*after), GFP_KERNEL);
1626	if (!after) {
1627		kfree(before);
1628		return -ENOMEM;
1629	}
1630	INIT_LIST_HEAD(&after->list);
1631
1632	/* Now gather all removed mappings */
1633	tmp = amdgpu_vm_it_iter_first(&vm->va, saddr, eaddr);
1634	while (tmp) {
1635		/* Remember mapping split at the start */
1636		if (tmp->start < saddr) {
1637			before->start = tmp->start;
1638			before->last = saddr - 1;
1639			before->offset = tmp->offset;
1640			before->flags = tmp->flags;
1641			before->bo_va = tmp->bo_va;
1642			list_add(&before->list, &tmp->bo_va->invalids);
1643		}
1644
1645		/* Remember mapping split at the end */
1646		if (tmp->last > eaddr) {
1647			after->start = eaddr + 1;
1648			after->last = tmp->last;
1649			after->offset = tmp->offset;
1650			after->offset += (after->start - tmp->start) << PAGE_SHIFT;
1651			after->flags = tmp->flags;
1652			after->bo_va = tmp->bo_va;
1653			list_add(&after->list, &tmp->bo_va->invalids);
1654		}
1655
1656		list_del(&tmp->list);
1657		list_add(&tmp->list, &removed);
1658
1659		tmp = amdgpu_vm_it_iter_next(tmp, saddr, eaddr);
1660	}
1661
1662	/* And free them up */
1663	list_for_each_entry_safe(tmp, next, &removed, list) {
1664		amdgpu_vm_it_remove(tmp, &vm->va);
1665		list_del(&tmp->list);
1666
1667		if (tmp->start < saddr)
1668		    tmp->start = saddr;
1669		if (tmp->last > eaddr)
1670		    tmp->last = eaddr;
1671
1672		tmp->bo_va = NULL;
1673		list_add(&tmp->list, &vm->freed);
1674		trace_amdgpu_vm_bo_unmap(NULL, tmp);
1675	}
1676
1677	/* Insert partial mapping before the range */
1678	if (!list_empty(&before->list)) {
1679		amdgpu_vm_it_insert(before, &vm->va);
1680		if (before->flags & AMDGPU_PTE_PRT)
1681			amdgpu_vm_prt_get(adev);
1682	} else {
1683		kfree(before);
1684	}
1685
1686	/* Insert partial mapping after the range */
1687	if (!list_empty(&after->list)) {
1688		amdgpu_vm_it_insert(after, &vm->va);
1689		if (after->flags & AMDGPU_PTE_PRT)
1690			amdgpu_vm_prt_get(adev);
1691	} else {
1692		kfree(after);
1693	}
1694
1695	return 0;
1696}
1697
1698/**
1699 * amdgpu_vm_bo_lookup_mapping - find mapping by address
1700 *
1701 * @vm: the requested VM
1702 * @addr: the address
1703 *
1704 * Find a mapping by it's address.
1705 *
1706 * Returns:
1707 * The amdgpu_bo_va_mapping matching for addr or NULL
1708 *
1709 */
1710struct amdgpu_bo_va_mapping *amdgpu_vm_bo_lookup_mapping(struct amdgpu_vm *vm,
1711							 uint64_t addr)
1712{
1713	return amdgpu_vm_it_iter_first(&vm->va, addr, addr);
1714}
1715
1716/**
1717 * amdgpu_vm_bo_trace_cs - trace all reserved mappings
1718 *
1719 * @vm: the requested vm
1720 * @ticket: CS ticket
1721 *
1722 * Trace all mappings of BOs reserved during a command submission.
1723 */
1724void amdgpu_vm_bo_trace_cs(struct amdgpu_vm *vm, struct ww_acquire_ctx *ticket)
1725{
1726	struct amdgpu_bo_va_mapping *mapping;
1727
1728	if (!trace_amdgpu_vm_bo_cs_enabled())
1729		return;
1730
1731	for (mapping = amdgpu_vm_it_iter_first(&vm->va, 0, U64_MAX); mapping;
1732	     mapping = amdgpu_vm_it_iter_next(mapping, 0, U64_MAX)) {
1733		if (mapping->bo_va && mapping->bo_va->base.bo) {
1734			struct amdgpu_bo *bo;
1735
1736			bo = mapping->bo_va->base.bo;
1737			if (dma_resv_locking_ctx(bo->tbo.base.resv) !=
1738			    ticket)
1739				continue;
1740		}
1741
1742		trace_amdgpu_vm_bo_cs(mapping);
1743	}
1744}
1745
1746/**
1747 * amdgpu_vm_bo_del - remove a bo from a specific vm
1748 *
1749 * @adev: amdgpu_device pointer
1750 * @bo_va: requested bo_va
1751 *
1752 * Remove @bo_va->bo from the requested vm.
1753 *
1754 * Object have to be reserved!
1755 */
1756void amdgpu_vm_bo_del(struct amdgpu_device *adev,
1757		      struct amdgpu_bo_va *bo_va)
1758{
1759	struct amdgpu_bo_va_mapping *mapping, *next;
1760	struct amdgpu_bo *bo = bo_va->base.bo;
1761	struct amdgpu_vm *vm = bo_va->base.vm;
1762	struct amdgpu_vm_bo_base **base;
1763
1764	dma_resv_assert_held(vm->root.bo->tbo.base.resv);
1765
1766	if (bo) {
1767		dma_resv_assert_held(bo->tbo.base.resv);
1768		if (bo->tbo.base.resv == vm->root.bo->tbo.base.resv)
1769			ttm_bo_set_bulk_move(&bo->tbo, NULL);
1770
1771		for (base = &bo_va->base.bo->vm_bo; *base;
1772		     base = &(*base)->next) {
1773			if (*base != &bo_va->base)
1774				continue;
1775
1776			*base = bo_va->base.next;
1777			break;
1778		}
1779	}
1780
1781	spin_lock(&vm->status_lock);
1782	list_del(&bo_va->base.vm_status);
1783	spin_unlock(&vm->status_lock);
1784
1785	list_for_each_entry_safe(mapping, next, &bo_va->valids, list) {
1786		list_del(&mapping->list);
1787		amdgpu_vm_it_remove(mapping, &vm->va);
1788		mapping->bo_va = NULL;
1789		trace_amdgpu_vm_bo_unmap(bo_va, mapping);
1790		list_add(&mapping->list, &vm->freed);
1791	}
1792	list_for_each_entry_safe(mapping, next, &bo_va->invalids, list) {
1793		list_del(&mapping->list);
1794		amdgpu_vm_it_remove(mapping, &vm->va);
1795		amdgpu_vm_free_mapping(adev, vm, mapping,
1796				       bo_va->last_pt_update);
1797	}
1798
1799	dma_fence_put(bo_va->last_pt_update);
1800
1801	if (bo && bo_va->is_xgmi)
1802		amdgpu_xgmi_set_pstate(adev, AMDGPU_XGMI_PSTATE_MIN);
1803
1804	kfree(bo_va);
1805}
1806
1807/**
1808 * amdgpu_vm_evictable - check if we can evict a VM
1809 *
1810 * @bo: A page table of the VM.
1811 *
1812 * Check if it is possible to evict a VM.
1813 */
1814bool amdgpu_vm_evictable(struct amdgpu_bo *bo)
1815{
1816	struct amdgpu_vm_bo_base *bo_base = bo->vm_bo;
1817
1818	/* Page tables of a destroyed VM can go away immediately */
1819	if (!bo_base || !bo_base->vm)
1820		return true;
1821
1822	/* Don't evict VM page tables while they are busy */
1823	if (!dma_resv_test_signaled(bo->tbo.base.resv, DMA_RESV_USAGE_BOOKKEEP))
1824		return false;
1825
1826	/* Try to block ongoing updates */
1827	if (!amdgpu_vm_eviction_trylock(bo_base->vm))
1828		return false;
1829
1830	/* Don't evict VM page tables while they are updated */
1831	if (!dma_fence_is_signaled(bo_base->vm->last_unlocked)) {
1832		amdgpu_vm_eviction_unlock(bo_base->vm);
1833		return false;
1834	}
1835
1836	bo_base->vm->evicting = true;
1837	amdgpu_vm_eviction_unlock(bo_base->vm);
1838	return true;
1839}
1840
1841/**
1842 * amdgpu_vm_bo_invalidate - mark the bo as invalid
1843 *
1844 * @adev: amdgpu_device pointer
1845 * @bo: amdgpu buffer object
1846 * @evicted: is the BO evicted
1847 *
1848 * Mark @bo as invalid.
1849 */
1850void amdgpu_vm_bo_invalidate(struct amdgpu_device *adev,
1851			     struct amdgpu_bo *bo, bool evicted)
1852{
1853	struct amdgpu_vm_bo_base *bo_base;
1854
1855	/* shadow bo doesn't have bo base, its validation needs its parent */
1856	if (bo->parent && (amdgpu_bo_shadowed(bo->parent) == bo))
1857		bo = bo->parent;
1858
1859	for (bo_base = bo->vm_bo; bo_base; bo_base = bo_base->next) {
1860		struct amdgpu_vm *vm = bo_base->vm;
1861
1862		if (evicted && bo->tbo.base.resv == vm->root.bo->tbo.base.resv) {
1863			amdgpu_vm_bo_evicted(bo_base);
1864			continue;
1865		}
1866
1867		if (bo_base->moved)
1868			continue;
1869		bo_base->moved = true;
1870
1871		if (bo->tbo.type == ttm_bo_type_kernel)
1872			amdgpu_vm_bo_relocated(bo_base);
1873		else if (bo->tbo.base.resv == vm->root.bo->tbo.base.resv)
1874			amdgpu_vm_bo_moved(bo_base);
1875		else
1876			amdgpu_vm_bo_invalidated(bo_base);
1877	}
1878}
1879
1880/**
1881 * amdgpu_vm_get_block_size - calculate VM page table size as power of two
1882 *
1883 * @vm_size: VM size
1884 *
1885 * Returns:
1886 * VM page table as power of two
1887 */
1888static uint32_t amdgpu_vm_get_block_size(uint64_t vm_size)
1889{
1890	/* Total bits covered by PD + PTs */
1891	unsigned bits = ilog2(vm_size) + 18;
1892
1893	/* Make sure the PD is 4K in size up to 8GB address space.
1894	   Above that split equal between PD and PTs */
1895	if (vm_size <= 8)
1896		return (bits - 9);
1897	else
1898		return ((bits + 3) / 2);
1899}
1900
1901/**
1902 * amdgpu_vm_adjust_size - adjust vm size, block size and fragment size
1903 *
1904 * @adev: amdgpu_device pointer
1905 * @min_vm_size: the minimum vm size in GB if it's set auto
1906 * @fragment_size_default: Default PTE fragment size
1907 * @max_level: max VMPT level
1908 * @max_bits: max address space size in bits
1909 *
1910 */
1911void amdgpu_vm_adjust_size(struct amdgpu_device *adev, uint32_t min_vm_size,
1912			   uint32_t fragment_size_default, unsigned max_level,
1913			   unsigned max_bits)
1914{
1915	unsigned int max_size = 1 << (max_bits - 30);
1916	unsigned int vm_size;
1917	uint64_t tmp;
1918
1919	/* adjust vm size first */
1920	if (amdgpu_vm_size != -1) {
1921		vm_size = amdgpu_vm_size;
1922		if (vm_size > max_size) {
1923			dev_warn(adev->dev, "VM size (%d) too large, max is %u GB\n",
1924				 amdgpu_vm_size, max_size);
1925			vm_size = max_size;
1926		}
1927	} else {
1928		struct sysinfo si;
1929		unsigned int phys_ram_gb;
1930
1931		/* Optimal VM size depends on the amount of physical
1932		 * RAM available. Underlying requirements and
1933		 * assumptions:
1934		 *
1935		 *  - Need to map system memory and VRAM from all GPUs
1936		 *     - VRAM from other GPUs not known here
1937		 *     - Assume VRAM <= system memory
1938		 *  - On GFX8 and older, VM space can be segmented for
1939		 *    different MTYPEs
1940		 *  - Need to allow room for fragmentation, guard pages etc.
1941		 *
1942		 * This adds up to a rough guess of system memory x3.
1943		 * Round up to power of two to maximize the available
1944		 * VM size with the given page table size.
1945		 */
1946		si_meminfo(&si);
1947		phys_ram_gb = ((uint64_t)si.totalram * si.mem_unit +
1948			       (1 << 30) - 1) >> 30;
1949		vm_size = roundup_pow_of_two(
1950			min(max(phys_ram_gb * 3, min_vm_size), max_size));
1951	}
1952
1953	adev->vm_manager.max_pfn = (uint64_t)vm_size << 18;
1954
1955	tmp = roundup_pow_of_two(adev->vm_manager.max_pfn);
1956	if (amdgpu_vm_block_size != -1)
1957		tmp >>= amdgpu_vm_block_size - 9;
1958	tmp = DIV_ROUND_UP(fls64(tmp) - 1, 9) - 1;
1959	adev->vm_manager.num_level = min(max_level, (unsigned)tmp);
1960	switch (adev->vm_manager.num_level) {
1961	case 3:
1962		adev->vm_manager.root_level = AMDGPU_VM_PDB2;
1963		break;
1964	case 2:
1965		adev->vm_manager.root_level = AMDGPU_VM_PDB1;
1966		break;
1967	case 1:
1968		adev->vm_manager.root_level = AMDGPU_VM_PDB0;
1969		break;
1970	default:
1971		dev_err(adev->dev, "VMPT only supports 2~4+1 levels\n");
1972	}
1973	/* block size depends on vm size and hw setup*/
1974	if (amdgpu_vm_block_size != -1)
1975		adev->vm_manager.block_size =
1976			min((unsigned)amdgpu_vm_block_size, max_bits
1977			    - AMDGPU_GPU_PAGE_SHIFT
1978			    - 9 * adev->vm_manager.num_level);
1979	else if (adev->vm_manager.num_level > 1)
1980		adev->vm_manager.block_size = 9;
1981	else
1982		adev->vm_manager.block_size = amdgpu_vm_get_block_size(tmp);
1983
1984	if (amdgpu_vm_fragment_size == -1)
1985		adev->vm_manager.fragment_size = fragment_size_default;
1986	else
1987		adev->vm_manager.fragment_size = amdgpu_vm_fragment_size;
1988
1989	DRM_INFO("vm size is %u GB, %u levels, block size is %u-bit, fragment size is %u-bit\n",
1990		 vm_size, adev->vm_manager.num_level + 1,
1991		 adev->vm_manager.block_size,
1992		 adev->vm_manager.fragment_size);
1993}
1994
1995/**
1996 * amdgpu_vm_wait_idle - wait for the VM to become idle
1997 *
1998 * @vm: VM object to wait for
1999 * @timeout: timeout to wait for VM to become idle
2000 */
2001long amdgpu_vm_wait_idle(struct amdgpu_vm *vm, long timeout)
2002{
2003	timeout = dma_resv_wait_timeout(vm->root.bo->tbo.base.resv,
2004					DMA_RESV_USAGE_BOOKKEEP,
2005					true, timeout);
2006	if (timeout <= 0)
2007		return timeout;
2008
2009	return dma_fence_wait_timeout(vm->last_unlocked, true, timeout);
2010}
2011
2012/**
2013 * amdgpu_vm_init - initialize a vm instance
2014 *
2015 * @adev: amdgpu_device pointer
2016 * @vm: requested vm
 
 
2017 *
2018 * Init @vm fields.
2019 *
2020 * Returns:
2021 * 0 for success, error for failure.
2022 */
2023int amdgpu_vm_init(struct amdgpu_device *adev, struct amdgpu_vm *vm)
 
2024{
2025	struct amdgpu_bo *root_bo;
2026	struct amdgpu_bo_vm *root;
2027	int r, i;
2028
2029	vm->va = RB_ROOT_CACHED;
2030	for (i = 0; i < AMDGPU_MAX_VMHUBS; i++)
2031		vm->reserved_vmid[i] = NULL;
2032	INIT_LIST_HEAD(&vm->evicted);
2033	INIT_LIST_HEAD(&vm->relocated);
2034	INIT_LIST_HEAD(&vm->moved);
2035	INIT_LIST_HEAD(&vm->idle);
2036	INIT_LIST_HEAD(&vm->invalidated);
2037	spin_lock_init(&vm->status_lock);
2038	INIT_LIST_HEAD(&vm->freed);
2039	INIT_LIST_HEAD(&vm->done);
2040	INIT_LIST_HEAD(&vm->pt_freed);
2041	INIT_WORK(&vm->pt_free_work, amdgpu_vm_pt_free_work);
2042
2043	/* create scheduler entities for page table updates */
2044	r = drm_sched_entity_init(&vm->immediate, DRM_SCHED_PRIORITY_NORMAL,
2045				  adev->vm_manager.vm_pte_scheds,
2046				  adev->vm_manager.vm_pte_num_scheds, NULL);
2047	if (r)
2048		return r;
2049
2050	r = drm_sched_entity_init(&vm->delayed, DRM_SCHED_PRIORITY_NORMAL,
2051				  adev->vm_manager.vm_pte_scheds,
2052				  adev->vm_manager.vm_pte_num_scheds, NULL);
2053	if (r)
2054		goto error_free_immediate;
2055
2056	vm->pte_support_ats = false;
2057	vm->is_compute_context = false;
2058
2059	vm->use_cpu_for_update = !!(adev->vm_manager.vm_update_mode &
2060				    AMDGPU_VM_USE_CPU_FOR_GFX);
 
2061
 
 
 
 
 
 
2062	DRM_DEBUG_DRIVER("VM update mode is %s\n",
2063			 vm->use_cpu_for_update ? "CPU" : "SDMA");
2064	WARN_ONCE((vm->use_cpu_for_update &&
2065		   !amdgpu_gmc_vram_full_visible(&adev->gmc)),
2066		  "CPU update of VM recommended only for large BAR system\n");
2067
2068	if (vm->use_cpu_for_update)
2069		vm->update_funcs = &amdgpu_vm_cpu_funcs;
2070	else
2071		vm->update_funcs = &amdgpu_vm_sdma_funcs;
2072	vm->last_update = NULL;
2073	vm->last_unlocked = dma_fence_get_stub();
2074	vm->last_tlb_flush = dma_fence_get_stub();
2075
2076	mutex_init(&vm->eviction_lock);
2077	vm->evicting = false;
2078
2079	r = amdgpu_vm_pt_create(adev, vm, adev->vm_manager.root_level,
2080				false, &root);
 
 
2081	if (r)
2082		goto error_free_delayed;
2083	root_bo = &root->bo;
2084	r = amdgpu_bo_reserve(root_bo, true);
2085	if (r)
2086		goto error_free_root;
2087
2088	r = dma_resv_reserve_fences(root_bo->tbo.base.resv, 1);
2089	if (r)
2090		goto error_unreserve;
2091
2092	amdgpu_vm_bo_base_init(&vm->root, vm, root_bo);
2093
2094	r = amdgpu_vm_pt_clear(adev, vm, root, false);
2095	if (r)
2096		goto error_unreserve;
2097
2098	amdgpu_bo_unreserve(vm->root.bo);
 
 
 
 
 
 
 
 
 
 
 
 
 
2099
2100	INIT_KFIFO(vm->faults);
2101
2102	return 0;
2103
2104error_unreserve:
2105	amdgpu_bo_unreserve(vm->root.bo);
2106
2107error_free_root:
2108	amdgpu_bo_unref(&root->shadow);
2109	amdgpu_bo_unref(&root_bo);
2110	vm->root.bo = NULL;
2111
2112error_free_delayed:
2113	dma_fence_put(vm->last_tlb_flush);
2114	dma_fence_put(vm->last_unlocked);
2115	drm_sched_entity_destroy(&vm->delayed);
2116
2117error_free_immediate:
2118	drm_sched_entity_destroy(&vm->immediate);
2119
2120	return r;
2121}
2122
2123/**
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
2124 * amdgpu_vm_make_compute - Turn a GFX VM into a compute VM
2125 *
2126 * @adev: amdgpu_device pointer
2127 * @vm: requested vm
 
2128 *
2129 * This only works on GFX VMs that don't have any BOs added and no
2130 * page tables allocated yet.
2131 *
2132 * Changes the following VM parameters:
2133 * - use_cpu_for_update
2134 * - pte_supports_ats
 
2135 *
2136 * Reinitializes the page directory to reflect the changed ATS
2137 * setting.
2138 *
2139 * Returns:
2140 * 0 for success, -errno for errors.
2141 */
2142int amdgpu_vm_make_compute(struct amdgpu_device *adev, struct amdgpu_vm *vm)
 
2143{
2144	bool pte_support_ats = (adev->asic_type == CHIP_RAVEN);
2145	int r;
2146
2147	r = amdgpu_bo_reserve(vm->root.bo, true);
2148	if (r)
2149		return r;
2150
2151	/* Sanity checks */
2152	if (!amdgpu_vm_pt_is_root_clean(adev, vm)) {
2153		r = -EINVAL;
2154		goto unreserve_bo;
 
 
 
 
 
 
 
 
 
 
 
 
2155	}
2156
2157	/* Check if PD needs to be reinitialized and do it before
2158	 * changing any other state, in case it fails.
2159	 */
2160	if (pte_support_ats != vm->pte_support_ats) {
2161		vm->pte_support_ats = pte_support_ats;
2162		r = amdgpu_vm_pt_clear(adev, vm, to_amdgpu_bo_vm(vm->root.bo),
2163				       false);
2164		if (r)
2165			goto unreserve_bo;
2166	}
2167
2168	/* Update VM state */
2169	vm->use_cpu_for_update = !!(adev->vm_manager.vm_update_mode &
2170				    AMDGPU_VM_USE_CPU_FOR_COMPUTE);
2171	DRM_DEBUG_DRIVER("VM update mode is %s\n",
2172			 vm->use_cpu_for_update ? "CPU" : "SDMA");
2173	WARN_ONCE((vm->use_cpu_for_update &&
2174		   !amdgpu_gmc_vram_full_visible(&adev->gmc)),
2175		  "CPU update of VM recommended only for large BAR system\n");
2176
2177	if (vm->use_cpu_for_update) {
2178		/* Sync with last SDMA update/clear before switching to CPU */
2179		r = amdgpu_bo_sync_wait(vm->root.bo,
2180					AMDGPU_FENCE_OWNER_UNDEFINED, true);
2181		if (r)
2182			goto unreserve_bo;
2183
2184		vm->update_funcs = &amdgpu_vm_cpu_funcs;
2185	} else {
2186		vm->update_funcs = &amdgpu_vm_sdma_funcs;
2187	}
2188	/*
2189	 * Make sure root PD gets mapped. As vm_update_mode could be changed
2190	 * when turning a GFX VM into a compute VM.
2191	 */
2192	r = vm->update_funcs->map_table(to_amdgpu_bo_vm(vm->root.bo));
2193	if (r)
2194		goto unreserve_bo;
2195
2196	dma_fence_put(vm->last_update);
2197	vm->last_update = NULL;
2198	vm->is_compute_context = true;
2199
 
 
 
 
 
 
 
 
 
 
 
 
 
 
2200	/* Free the shadow bo for compute VM */
2201	amdgpu_bo_unref(&to_amdgpu_bo_vm(vm->root.bo)->shadow);
 
 
 
2202
2203	goto unreserve_bo;
2204
 
 
 
 
 
 
 
 
2205unreserve_bo:
2206	amdgpu_bo_unreserve(vm->root.bo);
2207	return r;
2208}
2209
2210/**
2211 * amdgpu_vm_release_compute - release a compute vm
2212 * @adev: amdgpu_device pointer
2213 * @vm: a vm turned into compute vm by calling amdgpu_vm_make_compute
2214 *
2215 * This is a correspondant of amdgpu_vm_make_compute. It decouples compute
2216 * pasid from vm. Compute should stop use of vm after this call.
2217 */
2218void amdgpu_vm_release_compute(struct amdgpu_device *adev, struct amdgpu_vm *vm)
2219{
2220	amdgpu_vm_set_pasid(adev, vm, 0);
 
 
 
 
 
 
 
2221	vm->is_compute_context = false;
2222}
2223
2224/**
2225 * amdgpu_vm_fini - tear down a vm instance
2226 *
2227 * @adev: amdgpu_device pointer
2228 * @vm: requested vm
2229 *
2230 * Tear down @vm.
2231 * Unbind the VM and remove all bos from the vm bo list
2232 */
2233void amdgpu_vm_fini(struct amdgpu_device *adev, struct amdgpu_vm *vm)
2234{
2235	struct amdgpu_bo_va_mapping *mapping, *tmp;
2236	bool prt_fini_needed = !!adev->gmc.gmc_funcs->set_prt;
2237	struct amdgpu_bo *root;
2238	unsigned long flags;
2239	int i;
2240
2241	amdgpu_amdkfd_gpuvm_destroy_cb(adev, vm);
2242
2243	flush_work(&vm->pt_free_work);
 
 
 
 
 
 
 
 
 
2244
2245	root = amdgpu_bo_ref(vm->root.bo);
2246	amdgpu_bo_reserve(root, true);
2247	amdgpu_vm_set_pasid(adev, vm, 0);
2248	dma_fence_wait(vm->last_unlocked, false);
2249	dma_fence_put(vm->last_unlocked);
2250	dma_fence_wait(vm->last_tlb_flush, false);
2251	/* Make sure that all fence callbacks have completed */
2252	spin_lock_irqsave(vm->last_tlb_flush->lock, flags);
2253	spin_unlock_irqrestore(vm->last_tlb_flush->lock, flags);
2254	dma_fence_put(vm->last_tlb_flush);
2255
2256	list_for_each_entry_safe(mapping, tmp, &vm->freed, list) {
2257		if (mapping->flags & AMDGPU_PTE_PRT && prt_fini_needed) {
2258			amdgpu_vm_prt_fini(adev, vm);
2259			prt_fini_needed = false;
2260		}
2261
2262		list_del(&mapping->list);
2263		amdgpu_vm_free_mapping(adev, vm, mapping, NULL);
2264	}
2265
2266	amdgpu_vm_pt_free_root(adev, vm);
2267	amdgpu_bo_unreserve(root);
2268	amdgpu_bo_unref(&root);
2269	WARN_ON(vm->root.bo);
2270
2271	drm_sched_entity_destroy(&vm->immediate);
2272	drm_sched_entity_destroy(&vm->delayed);
2273
2274	if (!RB_EMPTY_ROOT(&vm->va.rb_root)) {
2275		dev_err(adev->dev, "still active bo inside vm\n");
2276	}
2277	rbtree_postorder_for_each_entry_safe(mapping, tmp,
2278					     &vm->va.rb_root, rb) {
2279		/* Don't remove the mapping here, we don't want to trigger a
2280		 * rebalance and the tree is about to be destroyed anyway.
2281		 */
2282		list_del(&mapping->list);
2283		kfree(mapping);
2284	}
2285
2286	dma_fence_put(vm->last_update);
2287	for (i = 0; i < AMDGPU_MAX_VMHUBS; i++)
2288		amdgpu_vmid_free_reserved(adev, vm, i);
2289}
2290
2291/**
2292 * amdgpu_vm_manager_init - init the VM manager
2293 *
2294 * @adev: amdgpu_device pointer
2295 *
2296 * Initialize the VM manager structures
2297 */
2298void amdgpu_vm_manager_init(struct amdgpu_device *adev)
2299{
2300	unsigned i;
2301
2302	/* Concurrent flushes are only possible starting with Vega10 and
2303	 * are broken on Navi10 and Navi14.
2304	 */
2305	adev->vm_manager.concurrent_flush = !(adev->asic_type < CHIP_VEGA10 ||
2306					      adev->asic_type == CHIP_NAVI10 ||
2307					      adev->asic_type == CHIP_NAVI14);
2308	amdgpu_vmid_mgr_init(adev);
2309
2310	adev->vm_manager.fence_context =
2311		dma_fence_context_alloc(AMDGPU_MAX_RINGS);
2312	for (i = 0; i < AMDGPU_MAX_RINGS; ++i)
2313		adev->vm_manager.seqno[i] = 0;
2314
2315	spin_lock_init(&adev->vm_manager.prt_lock);
2316	atomic_set(&adev->vm_manager.num_prt_users, 0);
2317
2318	/* If not overridden by the user, by default, only in large BAR systems
2319	 * Compute VM tables will be updated by CPU
2320	 */
2321#ifdef CONFIG_X86_64
2322	if (amdgpu_vm_update_mode == -1) {
2323		/* For asic with VF MMIO access protection
2324		 * avoid using CPU for VM table updates
2325		 */
2326		if (amdgpu_gmc_vram_full_visible(&adev->gmc) &&
2327		    !amdgpu_sriov_vf_mmio_access_protection(adev))
2328			adev->vm_manager.vm_update_mode =
2329				AMDGPU_VM_USE_CPU_FOR_COMPUTE;
2330		else
2331			adev->vm_manager.vm_update_mode = 0;
2332	} else
2333		adev->vm_manager.vm_update_mode = amdgpu_vm_update_mode;
2334#else
2335	adev->vm_manager.vm_update_mode = 0;
2336#endif
2337
2338	xa_init_flags(&adev->vm_manager.pasids, XA_FLAGS_LOCK_IRQ);
 
2339}
2340
2341/**
2342 * amdgpu_vm_manager_fini - cleanup VM manager
2343 *
2344 * @adev: amdgpu_device pointer
2345 *
2346 * Cleanup the VM manager and free resources.
2347 */
2348void amdgpu_vm_manager_fini(struct amdgpu_device *adev)
2349{
2350	WARN_ON(!xa_empty(&adev->vm_manager.pasids));
2351	xa_destroy(&adev->vm_manager.pasids);
2352
2353	amdgpu_vmid_mgr_fini(adev);
2354}
2355
2356/**
2357 * amdgpu_vm_ioctl - Manages VMID reservation for vm hubs.
2358 *
2359 * @dev: drm device pointer
2360 * @data: drm_amdgpu_vm
2361 * @filp: drm file pointer
2362 *
2363 * Returns:
2364 * 0 for success, -errno for errors.
2365 */
2366int amdgpu_vm_ioctl(struct drm_device *dev, void *data, struct drm_file *filp)
2367{
2368	union drm_amdgpu_vm *args = data;
2369	struct amdgpu_device *adev = drm_to_adev(dev);
2370	struct amdgpu_fpriv *fpriv = filp->driver_priv;
 
2371	int r;
2372
2373	switch (args->in.op) {
2374	case AMDGPU_VM_OP_RESERVE_VMID:
2375		/* We only have requirement to reserve vmid from gfxhub */
2376		r = amdgpu_vmid_alloc_reserved(adev, &fpriv->vm,
2377					       AMDGPU_GFXHUB_0);
2378		if (r)
2379			return r;
2380		break;
2381	case AMDGPU_VM_OP_UNRESERVE_VMID:
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
2382		amdgpu_vmid_free_reserved(adev, &fpriv->vm, AMDGPU_GFXHUB_0);
2383		break;
2384	default:
2385		return -EINVAL;
2386	}
2387
2388	return 0;
2389}
2390
2391/**
2392 * amdgpu_vm_get_task_info - Extracts task info for a PASID.
2393 *
2394 * @adev: drm device pointer
2395 * @pasid: PASID identifier for VM
2396 * @task_info: task_info to fill.
2397 */
2398void amdgpu_vm_get_task_info(struct amdgpu_device *adev, u32 pasid,
2399			 struct amdgpu_task_info *task_info)
2400{
2401	struct amdgpu_vm *vm;
2402	unsigned long flags;
2403
2404	xa_lock_irqsave(&adev->vm_manager.pasids, flags);
2405
2406	vm = xa_load(&adev->vm_manager.pasids, pasid);
2407	if (vm)
2408		*task_info = vm->task_info;
2409
2410	xa_unlock_irqrestore(&adev->vm_manager.pasids, flags);
2411}
2412
2413/**
2414 * amdgpu_vm_set_task_info - Sets VMs task info.
2415 *
2416 * @vm: vm for which to set the info
2417 */
2418void amdgpu_vm_set_task_info(struct amdgpu_vm *vm)
2419{
2420	if (vm->task_info.pid)
2421		return;
2422
2423	vm->task_info.pid = current->pid;
2424	get_task_comm(vm->task_info.task_name, current);
2425
2426	if (current->group_leader->mm != current->mm)
2427		return;
2428
2429	vm->task_info.tgid = current->group_leader->pid;
2430	get_task_comm(vm->task_info.process_name, current->group_leader);
2431}
2432
2433/**
2434 * amdgpu_vm_handle_fault - graceful handling of VM faults.
2435 * @adev: amdgpu device pointer
2436 * @pasid: PASID of the VM
2437 * @addr: Address of the fault
2438 * @write_fault: true is write fault, false is read fault
2439 *
2440 * Try to gracefully handle a VM fault. Return true if the fault was handled and
2441 * shouldn't be reported any more.
2442 */
2443bool amdgpu_vm_handle_fault(struct amdgpu_device *adev, u32 pasid,
2444			    uint64_t addr, bool write_fault)
2445{
2446	bool is_compute_context = false;
2447	struct amdgpu_bo *root;
2448	unsigned long irqflags;
2449	uint64_t value, flags;
2450	struct amdgpu_vm *vm;
2451	int r;
2452
2453	xa_lock_irqsave(&adev->vm_manager.pasids, irqflags);
2454	vm = xa_load(&adev->vm_manager.pasids, pasid);
2455	if (vm) {
2456		root = amdgpu_bo_ref(vm->root.bo);
2457		is_compute_context = vm->is_compute_context;
2458	} else {
2459		root = NULL;
2460	}
2461	xa_unlock_irqrestore(&adev->vm_manager.pasids, irqflags);
2462
2463	if (!root)
2464		return false;
2465
2466	addr /= AMDGPU_GPU_PAGE_SIZE;
2467
2468	if (is_compute_context &&
2469	    !svm_range_restore_pages(adev, pasid, addr, write_fault)) {
2470		amdgpu_bo_unref(&root);
2471		return true;
2472	}
2473
2474	r = amdgpu_bo_reserve(root, true);
2475	if (r)
2476		goto error_unref;
2477
2478	/* Double check that the VM still exists */
2479	xa_lock_irqsave(&adev->vm_manager.pasids, irqflags);
2480	vm = xa_load(&adev->vm_manager.pasids, pasid);
2481	if (vm && vm->root.bo != root)
2482		vm = NULL;
2483	xa_unlock_irqrestore(&adev->vm_manager.pasids, irqflags);
2484	if (!vm)
2485		goto error_unlock;
2486
 
2487	flags = AMDGPU_PTE_VALID | AMDGPU_PTE_SNOOPED |
2488		AMDGPU_PTE_SYSTEM;
2489
2490	if (is_compute_context) {
2491		/* Intentionally setting invalid PTE flag
2492		 * combination to force a no-retry-fault
2493		 */
2494		flags = AMDGPU_PTE_SNOOPED | AMDGPU_PTE_PRT;
 
2495		value = 0;
 
2496	} else if (amdgpu_vm_fault_stop == AMDGPU_VM_FAULT_STOP_NEVER) {
2497		/* Redirect the access to the dummy page */
2498		value = adev->dummy_page_addr;
2499		flags |= AMDGPU_PTE_EXECUTABLE | AMDGPU_PTE_READABLE |
2500			AMDGPU_PTE_WRITEABLE;
2501
2502	} else {
2503		/* Let the hw retry silently on the PTE */
2504		value = 0;
2505	}
2506
2507	r = dma_resv_reserve_fences(root->tbo.base.resv, 1);
2508	if (r) {
2509		pr_debug("failed %d to reserve fence slot\n", r);
2510		goto error_unlock;
2511	}
2512
2513	r = amdgpu_vm_update_range(adev, vm, true, false, false, NULL, addr,
2514				   addr, flags, value, 0, NULL, NULL, NULL);
2515	if (r)
2516		goto error_unlock;
2517
2518	r = amdgpu_vm_update_pdes(adev, vm, true);
2519
2520error_unlock:
2521	amdgpu_bo_unreserve(root);
2522	if (r < 0)
2523		DRM_ERROR("Can't handle page fault (%d)\n", r);
2524
2525error_unref:
2526	amdgpu_bo_unref(&root);
2527
2528	return false;
2529}
2530
2531#if defined(CONFIG_DEBUG_FS)
2532/**
2533 * amdgpu_debugfs_vm_bo_info  - print BO info for the VM
2534 *
2535 * @vm: Requested VM for printing BO info
2536 * @m: debugfs file
2537 *
2538 * Print BO information in debugfs file for the VM
2539 */
2540void amdgpu_debugfs_vm_bo_info(struct amdgpu_vm *vm, struct seq_file *m)
2541{
2542	struct amdgpu_bo_va *bo_va, *tmp;
2543	u64 total_idle = 0;
2544	u64 total_evicted = 0;
2545	u64 total_relocated = 0;
2546	u64 total_moved = 0;
2547	u64 total_invalidated = 0;
2548	u64 total_done = 0;
2549	unsigned int total_idle_objs = 0;
2550	unsigned int total_evicted_objs = 0;
2551	unsigned int total_relocated_objs = 0;
2552	unsigned int total_moved_objs = 0;
2553	unsigned int total_invalidated_objs = 0;
2554	unsigned int total_done_objs = 0;
2555	unsigned int id = 0;
2556
2557	spin_lock(&vm->status_lock);
2558	seq_puts(m, "\tIdle BOs:\n");
2559	list_for_each_entry_safe(bo_va, tmp, &vm->idle, base.vm_status) {
2560		if (!bo_va->base.bo)
2561			continue;
2562		total_idle += amdgpu_bo_print_info(id++, bo_va->base.bo, m);
2563	}
2564	total_idle_objs = id;
2565	id = 0;
2566
2567	seq_puts(m, "\tEvicted BOs:\n");
2568	list_for_each_entry_safe(bo_va, tmp, &vm->evicted, base.vm_status) {
2569		if (!bo_va->base.bo)
2570			continue;
2571		total_evicted += amdgpu_bo_print_info(id++, bo_va->base.bo, m);
2572	}
2573	total_evicted_objs = id;
2574	id = 0;
2575
2576	seq_puts(m, "\tRelocated BOs:\n");
2577	list_for_each_entry_safe(bo_va, tmp, &vm->relocated, base.vm_status) {
2578		if (!bo_va->base.bo)
2579			continue;
2580		total_relocated += amdgpu_bo_print_info(id++, bo_va->base.bo, m);
2581	}
2582	total_relocated_objs = id;
2583	id = 0;
2584
2585	seq_puts(m, "\tMoved BOs:\n");
2586	list_for_each_entry_safe(bo_va, tmp, &vm->moved, base.vm_status) {
2587		if (!bo_va->base.bo)
2588			continue;
2589		total_moved += amdgpu_bo_print_info(id++, bo_va->base.bo, m);
2590	}
2591	total_moved_objs = id;
2592	id = 0;
2593
2594	seq_puts(m, "\tInvalidated BOs:\n");
2595	list_for_each_entry_safe(bo_va, tmp, &vm->invalidated, base.vm_status) {
2596		if (!bo_va->base.bo)
2597			continue;
2598		total_invalidated += amdgpu_bo_print_info(id++,	bo_va->base.bo, m);
2599	}
2600	total_invalidated_objs = id;
2601	id = 0;
2602
2603	seq_puts(m, "\tDone BOs:\n");
2604	list_for_each_entry_safe(bo_va, tmp, &vm->done, base.vm_status) {
2605		if (!bo_va->base.bo)
2606			continue;
2607		total_done += amdgpu_bo_print_info(id++, bo_va->base.bo, m);
2608	}
2609	spin_unlock(&vm->status_lock);
2610	total_done_objs = id;
2611
2612	seq_printf(m, "\tTotal idle size:        %12lld\tobjs:\t%d\n", total_idle,
2613		   total_idle_objs);
2614	seq_printf(m, "\tTotal evicted size:     %12lld\tobjs:\t%d\n", total_evicted,
2615		   total_evicted_objs);
2616	seq_printf(m, "\tTotal relocated size:   %12lld\tobjs:\t%d\n", total_relocated,
2617		   total_relocated_objs);
2618	seq_printf(m, "\tTotal moved size:       %12lld\tobjs:\t%d\n", total_moved,
2619		   total_moved_objs);
2620	seq_printf(m, "\tTotal invalidated size: %12lld\tobjs:\t%d\n", total_invalidated,
2621		   total_invalidated_objs);
2622	seq_printf(m, "\tTotal done size:        %12lld\tobjs:\t%d\n", total_done,
2623		   total_done_objs);
2624}
2625#endif