Linux Audio

Check our new training course

Loading...
v6.13.7
  1// SPDX-License-Identifier: GPL-2.0 OR MIT
  2/*
  3 * Copyright 2022 Advanced Micro Devices, Inc.
  4 *
  5 * Permission is hereby granted, free of charge, to any person obtaining a
  6 * copy of this software and associated documentation files (the "Software"),
  7 * to deal in the Software without restriction, including without limitation
  8 * the rights to use, copy, modify, merge, publish, distribute, sublicense,
  9 * and/or sell copies of the Software, and to permit persons to whom the
 10 * Software is furnished to do so, subject to the following conditions:
 11 *
 12 * The above copyright notice and this permission notice shall be included in
 13 * all copies or substantial portions of the Software.
 14 *
 15 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 16 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 17 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
 18 * THE COPYRIGHT HOLDER(S) OR AUTHOR(S) BE LIABLE FOR ANY CLAIM, DAMAGES OR
 19 * OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
 20 * ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
 21 * OTHER DEALINGS IN THE SOFTWARE.
 22 */
 23
 24#include <drm/drm_drv.h>
 25
 26#include "amdgpu.h"
 27#include "amdgpu_trace.h"
 28#include "amdgpu_vm.h"
 29
 30/*
 31 * amdgpu_vm_pt_cursor - state for for_each_amdgpu_vm_pt
 32 */
 33struct amdgpu_vm_pt_cursor {
 34	uint64_t pfn;
 35	struct amdgpu_vm_bo_base *parent;
 36	struct amdgpu_vm_bo_base *entry;
 37	unsigned int level;
 38};
 39
 40/**
 41 * amdgpu_vm_pt_level_shift - return the addr shift for each level
 42 *
 43 * @adev: amdgpu_device pointer
 44 * @level: VMPT level
 45 *
 46 * Returns:
 47 * The number of bits the pfn needs to be right shifted for a level.
 48 */
 49static unsigned int amdgpu_vm_pt_level_shift(struct amdgpu_device *adev,
 50					     unsigned int level)
 51{
 52	switch (level) {
 53	case AMDGPU_VM_PDB2:
 54	case AMDGPU_VM_PDB1:
 55	case AMDGPU_VM_PDB0:
 56		return 9 * (AMDGPU_VM_PDB0 - level) +
 57			adev->vm_manager.block_size;
 58	case AMDGPU_VM_PTB:
 59		return 0;
 60	default:
 61		return ~0;
 62	}
 63}
 64
 65/**
 66 * amdgpu_vm_pt_num_entries - return the number of entries in a PD/PT
 67 *
 68 * @adev: amdgpu_device pointer
 69 * @level: VMPT level
 70 *
 71 * Returns:
 72 * The number of entries in a page directory or page table.
 73 */
 74static unsigned int amdgpu_vm_pt_num_entries(struct amdgpu_device *adev,
 75					     unsigned int level)
 76{
 77	unsigned int shift;
 78
 79	shift = amdgpu_vm_pt_level_shift(adev, adev->vm_manager.root_level);
 80	if (level == adev->vm_manager.root_level)
 81		/* For the root directory */
 82		return round_up(adev->vm_manager.max_pfn, 1ULL << shift)
 83			>> shift;
 84	else if (level != AMDGPU_VM_PTB)
 85		/* Everything in between */
 86		return 512;
 87
 88	/* For the page tables on the leaves */
 89	return AMDGPU_VM_PTE_COUNT(adev);
 90}
 91
 92/**
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 93 * amdgpu_vm_pt_entries_mask - the mask to get the entry number of a PD/PT
 94 *
 95 * @adev: amdgpu_device pointer
 96 * @level: VMPT level
 97 *
 98 * Returns:
 99 * The mask to extract the entry number of a PD/PT from an address.
100 */
101static uint32_t amdgpu_vm_pt_entries_mask(struct amdgpu_device *adev,
102					  unsigned int level)
103{
104	if (level <= adev->vm_manager.root_level)
105		return 0xffffffff;
106	else if (level != AMDGPU_VM_PTB)
107		return 0x1ff;
108	else
109		return AMDGPU_VM_PTE_COUNT(adev) - 1;
110}
111
112/**
113 * amdgpu_vm_pt_size - returns the size of the page table in bytes
114 *
115 * @adev: amdgpu_device pointer
116 * @level: VMPT level
117 *
118 * Returns:
119 * The size of the BO for a page directory or page table in bytes.
120 */
121static unsigned int amdgpu_vm_pt_size(struct amdgpu_device *adev,
122				      unsigned int level)
123{
124	return AMDGPU_GPU_PAGE_ALIGN(amdgpu_vm_pt_num_entries(adev, level) * 8);
125}
126
127/**
128 * amdgpu_vm_pt_parent - get the parent page directory
129 *
130 * @pt: child page table
131 *
132 * Helper to get the parent entry for the child page table. NULL if we are at
133 * the root page directory.
134 */
135static struct amdgpu_vm_bo_base *
136amdgpu_vm_pt_parent(struct amdgpu_vm_bo_base *pt)
137{
138	struct amdgpu_bo *parent = pt->bo->parent;
139
140	if (!parent)
141		return NULL;
142
143	return parent->vm_bo;
144}
145
146/**
147 * amdgpu_vm_pt_start - start PD/PT walk
148 *
149 * @adev: amdgpu_device pointer
150 * @vm: amdgpu_vm structure
151 * @start: start address of the walk
152 * @cursor: state to initialize
153 *
154 * Initialize a amdgpu_vm_pt_cursor to start a walk.
155 */
156static void amdgpu_vm_pt_start(struct amdgpu_device *adev,
157			       struct amdgpu_vm *vm, uint64_t start,
158			       struct amdgpu_vm_pt_cursor *cursor)
159{
160	cursor->pfn = start;
161	cursor->parent = NULL;
162	cursor->entry = &vm->root;
163	cursor->level = adev->vm_manager.root_level;
164}
165
166/**
167 * amdgpu_vm_pt_descendant - go to child node
168 *
169 * @adev: amdgpu_device pointer
170 * @cursor: current state
171 *
172 * Walk to the child node of the current node.
173 * Returns:
174 * True if the walk was possible, false otherwise.
175 */
176static bool amdgpu_vm_pt_descendant(struct amdgpu_device *adev,
177				    struct amdgpu_vm_pt_cursor *cursor)
178{
179	unsigned int mask, shift, idx;
180
181	if ((cursor->level == AMDGPU_VM_PTB) || !cursor->entry ||
182	    !cursor->entry->bo)
183		return false;
184
185	mask = amdgpu_vm_pt_entries_mask(adev, cursor->level);
186	shift = amdgpu_vm_pt_level_shift(adev, cursor->level);
187
188	++cursor->level;
189	idx = (cursor->pfn >> shift) & mask;
190	cursor->parent = cursor->entry;
191	cursor->entry = &to_amdgpu_bo_vm(cursor->entry->bo)->entries[idx];
192	return true;
193}
194
195/**
196 * amdgpu_vm_pt_sibling - go to sibling node
197 *
198 * @adev: amdgpu_device pointer
199 * @cursor: current state
200 *
201 * Walk to the sibling node of the current node.
202 * Returns:
203 * True if the walk was possible, false otherwise.
204 */
205static bool amdgpu_vm_pt_sibling(struct amdgpu_device *adev,
206				 struct amdgpu_vm_pt_cursor *cursor)
207{
208
209	unsigned int shift, num_entries;
210	struct amdgpu_bo_vm *parent;
211
212	/* Root doesn't have a sibling */
213	if (!cursor->parent)
214		return false;
215
216	/* Go to our parents and see if we got a sibling */
217	shift = amdgpu_vm_pt_level_shift(adev, cursor->level - 1);
218	num_entries = amdgpu_vm_pt_num_entries(adev, cursor->level - 1);
219	parent = to_amdgpu_bo_vm(cursor->parent->bo);
220
221	if (cursor->entry == &parent->entries[num_entries - 1])
222		return false;
223
224	cursor->pfn += 1ULL << shift;
225	cursor->pfn &= ~((1ULL << shift) - 1);
226	++cursor->entry;
227	return true;
228}
229
230/**
231 * amdgpu_vm_pt_ancestor - go to parent node
232 *
233 * @cursor: current state
234 *
235 * Walk to the parent node of the current node.
236 * Returns:
237 * True if the walk was possible, false otherwise.
238 */
239static bool amdgpu_vm_pt_ancestor(struct amdgpu_vm_pt_cursor *cursor)
240{
241	if (!cursor->parent)
242		return false;
243
244	--cursor->level;
245	cursor->entry = cursor->parent;
246	cursor->parent = amdgpu_vm_pt_parent(cursor->parent);
247	return true;
248}
249
250/**
251 * amdgpu_vm_pt_next - get next PD/PT in hieratchy
252 *
253 * @adev: amdgpu_device pointer
254 * @cursor: current state
255 *
256 * Walk the PD/PT tree to the next node.
257 */
258static void amdgpu_vm_pt_next(struct amdgpu_device *adev,
259			      struct amdgpu_vm_pt_cursor *cursor)
260{
261	/* First try a newborn child */
262	if (amdgpu_vm_pt_descendant(adev, cursor))
263		return;
264
265	/* If that didn't worked try to find a sibling */
266	while (!amdgpu_vm_pt_sibling(adev, cursor)) {
267		/* No sibling, go to our parents and grandparents */
268		if (!amdgpu_vm_pt_ancestor(cursor)) {
269			cursor->pfn = ~0ll;
270			return;
271		}
272	}
273}
274
275/**
276 * amdgpu_vm_pt_first_dfs - start a deep first search
277 *
278 * @adev: amdgpu_device structure
279 * @vm: amdgpu_vm structure
280 * @start: optional cursor to start with
281 * @cursor: state to initialize
282 *
283 * Starts a deep first traversal of the PD/PT tree.
284 */
285static void amdgpu_vm_pt_first_dfs(struct amdgpu_device *adev,
286				   struct amdgpu_vm *vm,
287				   struct amdgpu_vm_pt_cursor *start,
288				   struct amdgpu_vm_pt_cursor *cursor)
289{
290	if (start)
291		*cursor = *start;
292	else
293		amdgpu_vm_pt_start(adev, vm, 0, cursor);
294
295	while (amdgpu_vm_pt_descendant(adev, cursor))
296		;
297}
298
299/**
300 * amdgpu_vm_pt_continue_dfs - check if the deep first search should continue
301 *
302 * @start: starting point for the search
303 * @entry: current entry
304 *
305 * Returns:
306 * True when the search should continue, false otherwise.
307 */
308static bool amdgpu_vm_pt_continue_dfs(struct amdgpu_vm_pt_cursor *start,
309				      struct amdgpu_vm_bo_base *entry)
310{
311	return entry && (!start || entry != start->entry);
312}
313
314/**
315 * amdgpu_vm_pt_next_dfs - get the next node for a deep first search
316 *
317 * @adev: amdgpu_device structure
318 * @cursor: current state
319 *
320 * Move the cursor to the next node in a deep first search.
321 */
322static void amdgpu_vm_pt_next_dfs(struct amdgpu_device *adev,
323				  struct amdgpu_vm_pt_cursor *cursor)
324{
325	if (!cursor->entry)
326		return;
327
328	if (!cursor->parent)
329		cursor->entry = NULL;
330	else if (amdgpu_vm_pt_sibling(adev, cursor))
331		while (amdgpu_vm_pt_descendant(adev, cursor))
332			;
333	else
334		amdgpu_vm_pt_ancestor(cursor);
335}
336
337/*
338 * for_each_amdgpu_vm_pt_dfs_safe - safe deep first search of all PDs/PTs
339 */
340#define for_each_amdgpu_vm_pt_dfs_safe(adev, vm, start, cursor, entry)		\
341	for (amdgpu_vm_pt_first_dfs((adev), (vm), (start), &(cursor)),		\
342	     (entry) = (cursor).entry, amdgpu_vm_pt_next_dfs((adev), &(cursor));\
343	     amdgpu_vm_pt_continue_dfs((start), (entry));			\
344	     (entry) = (cursor).entry, amdgpu_vm_pt_next_dfs((adev), &(cursor)))
345
346/**
347 * amdgpu_vm_pt_clear - initially clear the PDs/PTs
348 *
349 * @adev: amdgpu_device pointer
350 * @vm: VM to clear BO from
351 * @vmbo: BO to clear
352 * @immediate: use an immediate update
353 *
354 * Root PD needs to be reserved when calling this.
355 *
356 * Returns:
357 * 0 on success, errno otherwise.
358 */
359int amdgpu_vm_pt_clear(struct amdgpu_device *adev, struct amdgpu_vm *vm,
360		       struct amdgpu_bo_vm *vmbo, bool immediate)
361{
362	unsigned int level = adev->vm_manager.root_level;
363	struct ttm_operation_ctx ctx = { true, false };
364	struct amdgpu_vm_update_params params;
365	struct amdgpu_bo *ancestor = &vmbo->bo;
366	unsigned int entries;
367	struct amdgpu_bo *bo = &vmbo->bo;
368	uint64_t addr;
369	int r, idx;
370
371	/* Figure out our place in the hierarchy */
372	if (ancestor->parent) {
373		++level;
374		while (ancestor->parent->parent) {
375			++level;
376			ancestor = ancestor->parent;
377		}
378	}
379
380	entries = amdgpu_bo_size(bo) / 8;
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
381
382	r = ttm_bo_validate(&bo->tbo, &bo->placement, &ctx);
383	if (r)
384		return r;
385
 
 
 
 
 
 
 
 
386	if (!drm_dev_enter(adev_to_drm(adev), &idx))
387		return -ENODEV;
388
389	r = vm->update_funcs->map_table(vmbo);
390	if (r)
391		goto exit;
392
393	memset(&params, 0, sizeof(params));
394	params.adev = adev;
395	params.vm = vm;
396	params.immediate = immediate;
397
398	r = vm->update_funcs->prepare(&params, NULL);
399	if (r)
400		goto exit;
401
402	addr = 0;
 
 
403
404	uint64_t value = 0, flags = 0;
405	if (adev->asic_type >= CHIP_VEGA10) {
406		if (level != AMDGPU_VM_PTB) {
407			/* Handle leaf PDEs as PTEs */
408			flags |= AMDGPU_PDE_PTE_FLAG(adev);
409			amdgpu_gmc_get_vm_pde(adev, level,
410					      &value, &flags);
411		} else {
412			/* Workaround for fault priority problem on GMC9 */
413			flags = AMDGPU_PTE_EXECUTABLE;
414		}
 
 
 
 
 
 
 
415	}
416
417	r = vm->update_funcs->update(&params, vmbo, addr, 0, entries,
418				     value, flags);
419	if (r)
420		goto exit;
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
421
422	r = vm->update_funcs->commit(&params, NULL);
423exit:
424	drm_dev_exit(idx);
425	return r;
426}
427
428/**
429 * amdgpu_vm_pt_create - create bo for PD/PT
430 *
431 * @adev: amdgpu_device pointer
432 * @vm: requesting vm
433 * @level: the page table level
434 * @immediate: use a immediate update
435 * @vmbo: pointer to the buffer object pointer
436 * @xcp_id: GPU partition id
437 */
438int amdgpu_vm_pt_create(struct amdgpu_device *adev, struct amdgpu_vm *vm,
439			int level, bool immediate, struct amdgpu_bo_vm **vmbo,
440			int32_t xcp_id)
441{
442	struct amdgpu_bo_param bp;
 
 
443	unsigned int num_entries;
 
444
445	memset(&bp, 0, sizeof(bp));
446
447	bp.size = amdgpu_vm_pt_size(adev, level);
448	bp.byte_align = AMDGPU_GPU_PAGE_SIZE;
449
450	if (!adev->gmc.is_app_apu)
451		bp.domain = AMDGPU_GEM_DOMAIN_VRAM;
452	else
453		bp.domain = AMDGPU_GEM_DOMAIN_GTT;
454
455	bp.domain = amdgpu_bo_get_preferred_domain(adev, bp.domain);
456	bp.flags = AMDGPU_GEM_CREATE_VRAM_CONTIGUOUS |
457		AMDGPU_GEM_CREATE_CPU_GTT_USWC;
458
459	if (level < AMDGPU_VM_PTB)
460		num_entries = amdgpu_vm_pt_num_entries(adev, level);
461	else
462		num_entries = 0;
463
464	bp.bo_ptr_size = struct_size((*vmbo), entries, num_entries);
465
466	if (vm->use_cpu_for_update)
467		bp.flags |= AMDGPU_GEM_CREATE_CPU_ACCESS_REQUIRED;
468
469	bp.type = ttm_bo_type_kernel;
470	bp.no_wait_gpu = immediate;
471	bp.xcp_id_plus1 = xcp_id + 1;
472
473	if (vm->root.bo)
474		bp.resv = vm->root.bo->tbo.base.resv;
475
476	return amdgpu_bo_create_vm(adev, &bp, vmbo);
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
477}
478
479/**
480 * amdgpu_vm_pt_alloc - Allocate a specific page table
481 *
482 * @adev: amdgpu_device pointer
483 * @vm: VM to allocate page tables for
484 * @cursor: Which page table to allocate
485 * @immediate: use an immediate update
486 *
487 * Make sure a specific page table or directory is allocated.
488 *
489 * Returns:
490 * 1 if page table needed to be allocated, 0 if page table was already
491 * allocated, negative errno if an error occurred.
492 */
493static int amdgpu_vm_pt_alloc(struct amdgpu_device *adev,
494			      struct amdgpu_vm *vm,
495			      struct amdgpu_vm_pt_cursor *cursor,
496			      bool immediate)
497{
498	struct amdgpu_vm_bo_base *entry = cursor->entry;
499	struct amdgpu_bo *pt_bo;
500	struct amdgpu_bo_vm *pt;
501	int r;
502
503	if (entry->bo)
504		return 0;
505
506	amdgpu_vm_eviction_unlock(vm);
507	r = amdgpu_vm_pt_create(adev, vm, cursor->level, immediate, &pt,
508				vm->root.bo->xcp_id);
509	amdgpu_vm_eviction_lock(vm);
510	if (r)
511		return r;
512
513	/* Keep a reference to the root directory to avoid
514	 * freeing them up in the wrong order.
515	 */
516	pt_bo = &pt->bo;
517	pt_bo->parent = amdgpu_bo_ref(cursor->parent->bo);
518	amdgpu_vm_bo_base_init(entry, vm, pt_bo);
519	r = amdgpu_vm_pt_clear(adev, vm, pt, immediate);
520	if (r)
521		goto error_free_pt;
522
523	return 0;
524
525error_free_pt:
 
526	amdgpu_bo_unref(&pt_bo);
527	return r;
528}
529
530/**
531 * amdgpu_vm_pt_free - free one PD/PT
532 *
533 * @entry: PDE to free
534 */
535static void amdgpu_vm_pt_free(struct amdgpu_vm_bo_base *entry)
536{
 
 
537	if (!entry->bo)
538		return;
539
540	entry->bo->vm_bo = NULL;
 
 
 
 
 
541	ttm_bo_set_bulk_move(&entry->bo->tbo, NULL);
542
543	spin_lock(&entry->vm->status_lock);
544	list_del(&entry->vm_status);
545	spin_unlock(&entry->vm->status_lock);
546	amdgpu_bo_unref(&entry->bo);
547}
548
549void amdgpu_vm_pt_free_work(struct work_struct *work)
550{
551	struct amdgpu_vm_bo_base *entry, *next;
552	struct amdgpu_vm *vm;
553	LIST_HEAD(pt_freed);
554
555	vm = container_of(work, struct amdgpu_vm, pt_free_work);
556
557	spin_lock(&vm->status_lock);
558	list_splice_init(&vm->pt_freed, &pt_freed);
559	spin_unlock(&vm->status_lock);
560
561	/* flush_work in amdgpu_vm_fini ensure vm->root.bo is valid. */
562	amdgpu_bo_reserve(vm->root.bo, true);
563
564	list_for_each_entry_safe(entry, next, &pt_freed, vm_status)
565		amdgpu_vm_pt_free(entry);
566
567	amdgpu_bo_unreserve(vm->root.bo);
568}
569
570/**
571 * amdgpu_vm_pt_free_list - free PD/PT levels
572 *
573 * @adev: amdgpu device structure
574 * @params: see amdgpu_vm_update_params definition
 
 
575 *
576 * Free the page directory objects saved in the flush list
577 */
578void amdgpu_vm_pt_free_list(struct amdgpu_device *adev,
579			    struct amdgpu_vm_update_params *params)
 
 
580{
581	struct amdgpu_vm_bo_base *entry, *next;
582	struct amdgpu_vm *vm = params->vm;
583	bool unlocked = params->unlocked;
584
585	if (list_empty(&params->tlb_flush_waitlist))
586		return;
587
588	if (unlocked) {
589		spin_lock(&vm->status_lock);
590		list_splice_init(&params->tlb_flush_waitlist, &vm->pt_freed);
 
 
 
 
591		spin_unlock(&vm->status_lock);
592		schedule_work(&vm->pt_free_work);
593		return;
594	}
595
596	list_for_each_entry_safe(entry, next, &params->tlb_flush_waitlist, vm_status)
597		amdgpu_vm_pt_free(entry);
598}
599
600/**
601 * amdgpu_vm_pt_add_list - add PD/PT level to the flush list
602 *
603 * @params: parameters for the update
604 * @cursor: first PT entry to start DF search from, non NULL
605 *
606 * This list will be freed after TLB flush.
607 */
608static void amdgpu_vm_pt_add_list(struct amdgpu_vm_update_params *params,
609				  struct amdgpu_vm_pt_cursor *cursor)
610{
611	struct amdgpu_vm_pt_cursor seek;
612	struct amdgpu_vm_bo_base *entry;
613
614	spin_lock(&params->vm->status_lock);
615	for_each_amdgpu_vm_pt_dfs_safe(params->adev, params->vm, cursor, seek, entry) {
616		if (entry && entry->bo)
617			list_move(&entry->vm_status, &params->tlb_flush_waitlist);
618	}
619
620	/* enter start node now */
621	list_move(&cursor->entry->vm_status, &params->tlb_flush_waitlist);
622	spin_unlock(&params->vm->status_lock);
623}
624
625/**
626 * amdgpu_vm_pt_free_root - free root PD
627 * @adev: amdgpu device structure
628 * @vm: amdgpu vm structure
629 *
630 * Free the root page directory and everything below it.
631 */
632void amdgpu_vm_pt_free_root(struct amdgpu_device *adev, struct amdgpu_vm *vm)
633{
634	struct amdgpu_vm_pt_cursor cursor;
635	struct amdgpu_vm_bo_base *entry;
636
637	for_each_amdgpu_vm_pt_dfs_safe(adev, vm, NULL, cursor, entry) {
638		if (entry)
639			amdgpu_vm_pt_free(entry);
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
640	}
 
641}
642
643/**
644 * amdgpu_vm_pde_update - update a single level in the hierarchy
645 *
646 * @params: parameters for the update
647 * @entry: entry to update
648 *
649 * Makes sure the requested entry in parent is up to date.
650 */
651int amdgpu_vm_pde_update(struct amdgpu_vm_update_params *params,
652			 struct amdgpu_vm_bo_base *entry)
653{
654	struct amdgpu_vm_bo_base *parent = amdgpu_vm_pt_parent(entry);
655	struct amdgpu_bo *bo, *pbo;
656	struct amdgpu_vm *vm = params->vm;
657	uint64_t pde, pt, flags;
658	unsigned int level;
659
660	if (WARN_ON(!parent))
661		return -EINVAL;
662
663	bo = parent->bo;
664	for (level = 0, pbo = bo->parent; pbo; ++level)
665		pbo = pbo->parent;
666
667	level += params->adev->vm_manager.root_level;
668	amdgpu_gmc_get_pde_for_bo(entry->bo, level, &pt, &flags);
669	pde = (entry - to_amdgpu_bo_vm(parent->bo)->entries) * 8;
670	return vm->update_funcs->update(params, to_amdgpu_bo_vm(bo), pde, pt,
671					1, 0, flags);
672}
673
674/**
675 * amdgpu_vm_pte_update_noretry_flags - Update PTE no-retry flags
676 *
677 * @adev: amdgpu_device pointer
678 * @flags: pointer to PTE flags
679 *
680 * Update PTE no-retry flags when TF is enabled.
681 */
682static void amdgpu_vm_pte_update_noretry_flags(struct amdgpu_device *adev,
683						uint64_t *flags)
684{
685	/*
686	 * Update no-retry flags with the corresponding TF
687	 * no-retry combination.
688	 */
689	if ((*flags & AMDGPU_VM_NORETRY_FLAGS) == AMDGPU_VM_NORETRY_FLAGS) {
690		*flags &= ~AMDGPU_VM_NORETRY_FLAGS;
691		*flags |= adev->gmc.noretry_flags;
692	}
693}
694
695/*
696 * amdgpu_vm_pte_update_flags - figure out flags for PTE updates
697 *
698 * Make sure to set the right flags for the PTEs at the desired level.
699 */
700static void amdgpu_vm_pte_update_flags(struct amdgpu_vm_update_params *params,
701				       struct amdgpu_bo_vm *pt,
702				       unsigned int level,
703				       uint64_t pe, uint64_t addr,
704				       unsigned int count, uint32_t incr,
705				       uint64_t flags)
706{
707	struct amdgpu_device *adev = params->adev;
708
709	if (level != AMDGPU_VM_PTB) {
710		flags |= AMDGPU_PDE_PTE_FLAG(params->adev);
711		amdgpu_gmc_get_vm_pde(adev, level, &addr, &flags);
712
713	} else if (adev->asic_type >= CHIP_VEGA10 &&
714		   !(flags & AMDGPU_PTE_VALID) &&
715		   !(flags & AMDGPU_PTE_PRT_FLAG(params->adev))) {
716
717		/* Workaround for fault priority problem on GMC9 */
718		flags |= AMDGPU_PTE_EXECUTABLE;
719	}
720
721	/*
722	 * Update no-retry flags to use the no-retry flag combination
723	 * with TF enabled. The AMDGPU_VM_NORETRY_FLAGS flag combination
724	 * does not work when TF is enabled. So, replace them with
725	 * AMDGPU_VM_NORETRY_FLAGS_TF flag combination which works for
726	 * all cases.
727	 */
728	if (level == AMDGPU_VM_PTB)
729		amdgpu_vm_pte_update_noretry_flags(adev, &flags);
730
731	/* APUs mapping system memory may need different MTYPEs on different
732	 * NUMA nodes. Only do this for contiguous ranges that can be assumed
733	 * to be on the same NUMA node.
734	 */
735	if ((flags & AMDGPU_PTE_SYSTEM) && (adev->flags & AMD_IS_APU) &&
736	    adev->gmc.gmc_funcs->override_vm_pte_flags &&
737	    num_possible_nodes() > 1 && !params->pages_addr && params->allow_override)
738		amdgpu_gmc_override_vm_pte_flags(adev, params->vm, addr, &flags);
739
740	params->vm->update_funcs->update(params, pt, pe, addr, count, incr,
741					 flags);
742}
743
744/**
745 * amdgpu_vm_pte_fragment - get fragment for PTEs
746 *
747 * @params: see amdgpu_vm_update_params definition
748 * @start: first PTE to handle
749 * @end: last PTE to handle
750 * @flags: hw mapping flags
751 * @frag: resulting fragment size
752 * @frag_end: end of this fragment
753 *
754 * Returns the first possible fragment for the start and end address.
755 */
756static void amdgpu_vm_pte_fragment(struct amdgpu_vm_update_params *params,
757				   uint64_t start, uint64_t end, uint64_t flags,
758				   unsigned int *frag, uint64_t *frag_end)
759{
760	/**
761	 * The MC L1 TLB supports variable sized pages, based on a fragment
762	 * field in the PTE. When this field is set to a non-zero value, page
763	 * granularity is increased from 4KB to (1 << (12 + frag)). The PTE
764	 * flags are considered valid for all PTEs within the fragment range
765	 * and corresponding mappings are assumed to be physically contiguous.
766	 *
767	 * The L1 TLB can store a single PTE for the whole fragment,
768	 * significantly increasing the space available for translation
769	 * caching. This leads to large improvements in throughput when the
770	 * TLB is under pressure.
771	 *
772	 * The L2 TLB distributes small and large fragments into two
773	 * asymmetric partitions. The large fragment cache is significantly
774	 * larger. Thus, we try to use large fragments wherever possible.
775	 * Userspace can support this by aligning virtual base address and
776	 * allocation size to the fragment size.
777	 *
778	 * Starting with Vega10 the fragment size only controls the L1. The L2
779	 * is now directly feed with small/huge/giant pages from the walker.
780	 */
781	unsigned int max_frag;
782
783	if (params->adev->asic_type < CHIP_VEGA10)
784		max_frag = params->adev->vm_manager.fragment_size;
785	else
786		max_frag = 31;
787
788	/* system pages are non continuously */
789	if (params->pages_addr) {
790		*frag = 0;
791		*frag_end = end;
792		return;
793	}
794
795	/* This intentionally wraps around if no bit is set */
796	*frag = min_t(unsigned int, ffs(start) - 1, fls64(end - start) - 1);
797	if (*frag >= max_frag) {
798		*frag = max_frag;
799		*frag_end = end & ~((1ULL << max_frag) - 1);
800	} else {
801		*frag_end = start + (1 << *frag);
802	}
803}
804
805/**
806 * amdgpu_vm_ptes_update - make sure that page tables are valid
807 *
808 * @params: see amdgpu_vm_update_params definition
809 * @start: start of GPU address range
810 * @end: end of GPU address range
811 * @dst: destination address to map to, the next dst inside the function
812 * @flags: mapping flags
813 *
814 * Update the page tables in the range @start - @end.
815 *
816 * Returns:
817 * 0 for success, -EINVAL for failure.
818 */
819int amdgpu_vm_ptes_update(struct amdgpu_vm_update_params *params,
820			  uint64_t start, uint64_t end,
821			  uint64_t dst, uint64_t flags)
822{
823	struct amdgpu_device *adev = params->adev;
824	struct amdgpu_vm_pt_cursor cursor;
825	uint64_t frag_start = start, frag_end;
826	unsigned int frag;
827	int r;
828
829	/* figure out the initial fragment */
830	amdgpu_vm_pte_fragment(params, frag_start, end, flags, &frag,
831			       &frag_end);
832
833	/* walk over the address space and update the PTs */
834	amdgpu_vm_pt_start(adev, params->vm, start, &cursor);
835	while (cursor.pfn < end) {
836		unsigned int shift, parent_shift, mask;
837		uint64_t incr, entry_end, pe_start;
838		struct amdgpu_bo *pt;
839
840		if (!params->unlocked) {
841			/* make sure that the page tables covering the
842			 * address range are actually allocated
843			 */
844			r = amdgpu_vm_pt_alloc(params->adev, params->vm,
845					       &cursor, params->immediate);
846			if (r)
847				return r;
848		}
849
850		shift = amdgpu_vm_pt_level_shift(adev, cursor.level);
851		parent_shift = amdgpu_vm_pt_level_shift(adev, cursor.level - 1);
852		if (params->unlocked) {
853			/* Unlocked updates are only allowed on the leaves */
854			if (amdgpu_vm_pt_descendant(adev, &cursor))
855				continue;
856		} else if (adev->asic_type < CHIP_VEGA10 &&
857			   (flags & AMDGPU_PTE_VALID)) {
858			/* No huge page support before GMC v9 */
859			if (cursor.level != AMDGPU_VM_PTB) {
860				if (!amdgpu_vm_pt_descendant(adev, &cursor))
861					return -ENOENT;
862				continue;
863			}
864		} else if (frag < shift) {
865			/* We can't use this level when the fragment size is
866			 * smaller than the address shift. Go to the next
867			 * child entry and try again.
868			 */
869			if (amdgpu_vm_pt_descendant(adev, &cursor))
870				continue;
871		} else if (frag >= parent_shift) {
872			/* If the fragment size is even larger than the parent
873			 * shift we should go up one level and check it again.
874			 */
875			if (!amdgpu_vm_pt_ancestor(&cursor))
876				return -EINVAL;
877			continue;
878		}
879
880		pt = cursor.entry->bo;
881		if (!pt) {
882			/* We need all PDs and PTs for mapping something, */
883			if (flags & AMDGPU_PTE_VALID)
884				return -ENOENT;
885
886			/* but unmapping something can happen at a higher
887			 * level.
888			 */
889			if (!amdgpu_vm_pt_ancestor(&cursor))
890				return -EINVAL;
891
892			pt = cursor.entry->bo;
893			shift = parent_shift;
894			frag_end = max(frag_end, ALIGN(frag_start + 1,
895				   1ULL << shift));
896		}
897
898		/* Looks good so far, calculate parameters for the update */
899		incr = (uint64_t)AMDGPU_GPU_PAGE_SIZE << shift;
900		mask = amdgpu_vm_pt_entries_mask(adev, cursor.level);
901		pe_start = ((cursor.pfn >> shift) & mask) * 8;
902		entry_end = ((uint64_t)mask + 1) << shift;
903		entry_end += cursor.pfn & ~(entry_end - 1);
904		entry_end = min(entry_end, end);
905
906		do {
907			struct amdgpu_vm *vm = params->vm;
908			uint64_t upd_end = min(entry_end, frag_end);
909			unsigned int nptes = (upd_end - frag_start) >> shift;
910			uint64_t upd_flags = flags | AMDGPU_PTE_FRAG(frag);
911
912			/* This can happen when we set higher level PDs to
913			 * silent to stop fault floods.
914			 */
915			nptes = max(nptes, 1u);
916
917			trace_amdgpu_vm_update_ptes(params, frag_start, upd_end,
918						    min(nptes, 32u), dst, incr,
919						    upd_flags,
920						    vm->task_info ? vm->task_info->tgid : 0,
921						    vm->immediate.fence_context);
922			amdgpu_vm_pte_update_flags(params, to_amdgpu_bo_vm(pt),
923						   cursor.level, pe_start, dst,
924						   nptes, incr, upd_flags);
925
926			pe_start += nptes * 8;
927			dst += nptes * incr;
928
929			frag_start = upd_end;
930			if (frag_start >= frag_end) {
931				/* figure out the next fragment */
932				amdgpu_vm_pte_fragment(params, frag_start, end,
933						       flags, &frag, &frag_end);
934				if (frag < shift)
935					break;
936			}
937		} while (frag_start < entry_end);
938
939		if (amdgpu_vm_pt_descendant(adev, &cursor)) {
940			/* Free all child entries.
941			 * Update the tables with the flags and addresses and free up subsequent
942			 * tables in the case of huge pages or freed up areas.
943			 * This is the maximum you can free, because all other page tables are not
944			 * completely covered by the range and so potentially still in use.
945			 */
946			while (cursor.pfn < frag_start) {
947				/* Make sure previous mapping is freed */
948				if (cursor.entry->bo) {
949					params->needs_flush = true;
950					amdgpu_vm_pt_add_list(params, &cursor);
 
 
951				}
952				amdgpu_vm_pt_next(adev, &cursor);
953			}
954
955		} else if (frag >= shift) {
956			/* or just move on to the next on the same level. */
957			amdgpu_vm_pt_next(adev, &cursor);
958		}
959	}
960
961	return 0;
962}
963
964/**
965 * amdgpu_vm_pt_map_tables - have bo of root PD cpu accessible
966 * @adev: amdgpu device structure
967 * @vm: amdgpu vm structure
968 *
969 * make root page directory and everything below it cpu accessible.
970 */
971int amdgpu_vm_pt_map_tables(struct amdgpu_device *adev, struct amdgpu_vm *vm)
972{
973	struct amdgpu_vm_pt_cursor cursor;
974	struct amdgpu_vm_bo_base *entry;
975
976	for_each_amdgpu_vm_pt_dfs_safe(adev, vm, NULL, cursor, entry) {
977
978		struct amdgpu_bo_vm *bo;
979		int r;
980
981		if (entry->bo) {
982			bo = to_amdgpu_bo_vm(entry->bo);
983			r = vm->update_funcs->map_table(bo);
984			if (r)
985				return r;
986		}
987	}
988
989	return 0;
990}
v6.8
   1// SPDX-License-Identifier: GPL-2.0 OR MIT
   2/*
   3 * Copyright 2022 Advanced Micro Devices, Inc.
   4 *
   5 * Permission is hereby granted, free of charge, to any person obtaining a
   6 * copy of this software and associated documentation files (the "Software"),
   7 * to deal in the Software without restriction, including without limitation
   8 * the rights to use, copy, modify, merge, publish, distribute, sublicense,
   9 * and/or sell copies of the Software, and to permit persons to whom the
  10 * Software is furnished to do so, subject to the following conditions:
  11 *
  12 * The above copyright notice and this permission notice shall be included in
  13 * all copies or substantial portions of the Software.
  14 *
  15 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  16 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  17 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
  18 * THE COPYRIGHT HOLDER(S) OR AUTHOR(S) BE LIABLE FOR ANY CLAIM, DAMAGES OR
  19 * OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
  20 * ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
  21 * OTHER DEALINGS IN THE SOFTWARE.
  22 */
  23
  24#include <drm/drm_drv.h>
  25
  26#include "amdgpu.h"
  27#include "amdgpu_trace.h"
  28#include "amdgpu_vm.h"
  29
  30/*
  31 * amdgpu_vm_pt_cursor - state for for_each_amdgpu_vm_pt
  32 */
  33struct amdgpu_vm_pt_cursor {
  34	uint64_t pfn;
  35	struct amdgpu_vm_bo_base *parent;
  36	struct amdgpu_vm_bo_base *entry;
  37	unsigned int level;
  38};
  39
  40/**
  41 * amdgpu_vm_pt_level_shift - return the addr shift for each level
  42 *
  43 * @adev: amdgpu_device pointer
  44 * @level: VMPT level
  45 *
  46 * Returns:
  47 * The number of bits the pfn needs to be right shifted for a level.
  48 */
  49static unsigned int amdgpu_vm_pt_level_shift(struct amdgpu_device *adev,
  50					     unsigned int level)
  51{
  52	switch (level) {
  53	case AMDGPU_VM_PDB2:
  54	case AMDGPU_VM_PDB1:
  55	case AMDGPU_VM_PDB0:
  56		return 9 * (AMDGPU_VM_PDB0 - level) +
  57			adev->vm_manager.block_size;
  58	case AMDGPU_VM_PTB:
  59		return 0;
  60	default:
  61		return ~0;
  62	}
  63}
  64
  65/**
  66 * amdgpu_vm_pt_num_entries - return the number of entries in a PD/PT
  67 *
  68 * @adev: amdgpu_device pointer
  69 * @level: VMPT level
  70 *
  71 * Returns:
  72 * The number of entries in a page directory or page table.
  73 */
  74static unsigned int amdgpu_vm_pt_num_entries(struct amdgpu_device *adev,
  75					     unsigned int level)
  76{
  77	unsigned int shift;
  78
  79	shift = amdgpu_vm_pt_level_shift(adev, adev->vm_manager.root_level);
  80	if (level == adev->vm_manager.root_level)
  81		/* For the root directory */
  82		return round_up(adev->vm_manager.max_pfn, 1ULL << shift)
  83			>> shift;
  84	else if (level != AMDGPU_VM_PTB)
  85		/* Everything in between */
  86		return 512;
  87
  88	/* For the page tables on the leaves */
  89	return AMDGPU_VM_PTE_COUNT(adev);
  90}
  91
  92/**
  93 * amdgpu_vm_pt_num_ats_entries - return the number of ATS entries in the root PD
  94 *
  95 * @adev: amdgpu_device pointer
  96 *
  97 * Returns:
  98 * The number of entries in the root page directory which needs the ATS setting.
  99 */
 100static unsigned int amdgpu_vm_pt_num_ats_entries(struct amdgpu_device *adev)
 101{
 102	unsigned int shift;
 103
 104	shift = amdgpu_vm_pt_level_shift(adev, adev->vm_manager.root_level);
 105	return AMDGPU_GMC_HOLE_START >> (shift + AMDGPU_GPU_PAGE_SHIFT);
 106}
 107
 108/**
 109 * amdgpu_vm_pt_entries_mask - the mask to get the entry number of a PD/PT
 110 *
 111 * @adev: amdgpu_device pointer
 112 * @level: VMPT level
 113 *
 114 * Returns:
 115 * The mask to extract the entry number of a PD/PT from an address.
 116 */
 117static uint32_t amdgpu_vm_pt_entries_mask(struct amdgpu_device *adev,
 118					  unsigned int level)
 119{
 120	if (level <= adev->vm_manager.root_level)
 121		return 0xffffffff;
 122	else if (level != AMDGPU_VM_PTB)
 123		return 0x1ff;
 124	else
 125		return AMDGPU_VM_PTE_COUNT(adev) - 1;
 126}
 127
 128/**
 129 * amdgpu_vm_pt_size - returns the size of the page table in bytes
 130 *
 131 * @adev: amdgpu_device pointer
 132 * @level: VMPT level
 133 *
 134 * Returns:
 135 * The size of the BO for a page directory or page table in bytes.
 136 */
 137static unsigned int amdgpu_vm_pt_size(struct amdgpu_device *adev,
 138				      unsigned int level)
 139{
 140	return AMDGPU_GPU_PAGE_ALIGN(amdgpu_vm_pt_num_entries(adev, level) * 8);
 141}
 142
 143/**
 144 * amdgpu_vm_pt_parent - get the parent page directory
 145 *
 146 * @pt: child page table
 147 *
 148 * Helper to get the parent entry for the child page table. NULL if we are at
 149 * the root page directory.
 150 */
 151static struct amdgpu_vm_bo_base *
 152amdgpu_vm_pt_parent(struct amdgpu_vm_bo_base *pt)
 153{
 154	struct amdgpu_bo *parent = pt->bo->parent;
 155
 156	if (!parent)
 157		return NULL;
 158
 159	return parent->vm_bo;
 160}
 161
 162/**
 163 * amdgpu_vm_pt_start - start PD/PT walk
 164 *
 165 * @adev: amdgpu_device pointer
 166 * @vm: amdgpu_vm structure
 167 * @start: start address of the walk
 168 * @cursor: state to initialize
 169 *
 170 * Initialize a amdgpu_vm_pt_cursor to start a walk.
 171 */
 172static void amdgpu_vm_pt_start(struct amdgpu_device *adev,
 173			       struct amdgpu_vm *vm, uint64_t start,
 174			       struct amdgpu_vm_pt_cursor *cursor)
 175{
 176	cursor->pfn = start;
 177	cursor->parent = NULL;
 178	cursor->entry = &vm->root;
 179	cursor->level = adev->vm_manager.root_level;
 180}
 181
 182/**
 183 * amdgpu_vm_pt_descendant - go to child node
 184 *
 185 * @adev: amdgpu_device pointer
 186 * @cursor: current state
 187 *
 188 * Walk to the child node of the current node.
 189 * Returns:
 190 * True if the walk was possible, false otherwise.
 191 */
 192static bool amdgpu_vm_pt_descendant(struct amdgpu_device *adev,
 193				    struct amdgpu_vm_pt_cursor *cursor)
 194{
 195	unsigned int mask, shift, idx;
 196
 197	if ((cursor->level == AMDGPU_VM_PTB) || !cursor->entry ||
 198	    !cursor->entry->bo)
 199		return false;
 200
 201	mask = amdgpu_vm_pt_entries_mask(adev, cursor->level);
 202	shift = amdgpu_vm_pt_level_shift(adev, cursor->level);
 203
 204	++cursor->level;
 205	idx = (cursor->pfn >> shift) & mask;
 206	cursor->parent = cursor->entry;
 207	cursor->entry = &to_amdgpu_bo_vm(cursor->entry->bo)->entries[idx];
 208	return true;
 209}
 210
 211/**
 212 * amdgpu_vm_pt_sibling - go to sibling node
 213 *
 214 * @adev: amdgpu_device pointer
 215 * @cursor: current state
 216 *
 217 * Walk to the sibling node of the current node.
 218 * Returns:
 219 * True if the walk was possible, false otherwise.
 220 */
 221static bool amdgpu_vm_pt_sibling(struct amdgpu_device *adev,
 222				 struct amdgpu_vm_pt_cursor *cursor)
 223{
 224
 225	unsigned int shift, num_entries;
 226	struct amdgpu_bo_vm *parent;
 227
 228	/* Root doesn't have a sibling */
 229	if (!cursor->parent)
 230		return false;
 231
 232	/* Go to our parents and see if we got a sibling */
 233	shift = amdgpu_vm_pt_level_shift(adev, cursor->level - 1);
 234	num_entries = amdgpu_vm_pt_num_entries(adev, cursor->level - 1);
 235	parent = to_amdgpu_bo_vm(cursor->parent->bo);
 236
 237	if (cursor->entry == &parent->entries[num_entries - 1])
 238		return false;
 239
 240	cursor->pfn += 1ULL << shift;
 241	cursor->pfn &= ~((1ULL << shift) - 1);
 242	++cursor->entry;
 243	return true;
 244}
 245
 246/**
 247 * amdgpu_vm_pt_ancestor - go to parent node
 248 *
 249 * @cursor: current state
 250 *
 251 * Walk to the parent node of the current node.
 252 * Returns:
 253 * True if the walk was possible, false otherwise.
 254 */
 255static bool amdgpu_vm_pt_ancestor(struct amdgpu_vm_pt_cursor *cursor)
 256{
 257	if (!cursor->parent)
 258		return false;
 259
 260	--cursor->level;
 261	cursor->entry = cursor->parent;
 262	cursor->parent = amdgpu_vm_pt_parent(cursor->parent);
 263	return true;
 264}
 265
 266/**
 267 * amdgpu_vm_pt_next - get next PD/PT in hieratchy
 268 *
 269 * @adev: amdgpu_device pointer
 270 * @cursor: current state
 271 *
 272 * Walk the PD/PT tree to the next node.
 273 */
 274static void amdgpu_vm_pt_next(struct amdgpu_device *adev,
 275			      struct amdgpu_vm_pt_cursor *cursor)
 276{
 277	/* First try a newborn child */
 278	if (amdgpu_vm_pt_descendant(adev, cursor))
 279		return;
 280
 281	/* If that didn't worked try to find a sibling */
 282	while (!amdgpu_vm_pt_sibling(adev, cursor)) {
 283		/* No sibling, go to our parents and grandparents */
 284		if (!amdgpu_vm_pt_ancestor(cursor)) {
 285			cursor->pfn = ~0ll;
 286			return;
 287		}
 288	}
 289}
 290
 291/**
 292 * amdgpu_vm_pt_first_dfs - start a deep first search
 293 *
 294 * @adev: amdgpu_device structure
 295 * @vm: amdgpu_vm structure
 296 * @start: optional cursor to start with
 297 * @cursor: state to initialize
 298 *
 299 * Starts a deep first traversal of the PD/PT tree.
 300 */
 301static void amdgpu_vm_pt_first_dfs(struct amdgpu_device *adev,
 302				   struct amdgpu_vm *vm,
 303				   struct amdgpu_vm_pt_cursor *start,
 304				   struct amdgpu_vm_pt_cursor *cursor)
 305{
 306	if (start)
 307		*cursor = *start;
 308	else
 309		amdgpu_vm_pt_start(adev, vm, 0, cursor);
 310
 311	while (amdgpu_vm_pt_descendant(adev, cursor))
 312		;
 313}
 314
 315/**
 316 * amdgpu_vm_pt_continue_dfs - check if the deep first search should continue
 317 *
 318 * @start: starting point for the search
 319 * @entry: current entry
 320 *
 321 * Returns:
 322 * True when the search should continue, false otherwise.
 323 */
 324static bool amdgpu_vm_pt_continue_dfs(struct amdgpu_vm_pt_cursor *start,
 325				      struct amdgpu_vm_bo_base *entry)
 326{
 327	return entry && (!start || entry != start->entry);
 328}
 329
 330/**
 331 * amdgpu_vm_pt_next_dfs - get the next node for a deep first search
 332 *
 333 * @adev: amdgpu_device structure
 334 * @cursor: current state
 335 *
 336 * Move the cursor to the next node in a deep first search.
 337 */
 338static void amdgpu_vm_pt_next_dfs(struct amdgpu_device *adev,
 339				  struct amdgpu_vm_pt_cursor *cursor)
 340{
 341	if (!cursor->entry)
 342		return;
 343
 344	if (!cursor->parent)
 345		cursor->entry = NULL;
 346	else if (amdgpu_vm_pt_sibling(adev, cursor))
 347		while (amdgpu_vm_pt_descendant(adev, cursor))
 348			;
 349	else
 350		amdgpu_vm_pt_ancestor(cursor);
 351}
 352
 353/*
 354 * for_each_amdgpu_vm_pt_dfs_safe - safe deep first search of all PDs/PTs
 355 */
 356#define for_each_amdgpu_vm_pt_dfs_safe(adev, vm, start, cursor, entry)		\
 357	for (amdgpu_vm_pt_first_dfs((adev), (vm), (start), &(cursor)),		\
 358	     (entry) = (cursor).entry, amdgpu_vm_pt_next_dfs((adev), &(cursor));\
 359	     amdgpu_vm_pt_continue_dfs((start), (entry));			\
 360	     (entry) = (cursor).entry, amdgpu_vm_pt_next_dfs((adev), &(cursor)))
 361
 362/**
 363 * amdgpu_vm_pt_clear - initially clear the PDs/PTs
 364 *
 365 * @adev: amdgpu_device pointer
 366 * @vm: VM to clear BO from
 367 * @vmbo: BO to clear
 368 * @immediate: use an immediate update
 369 *
 370 * Root PD needs to be reserved when calling this.
 371 *
 372 * Returns:
 373 * 0 on success, errno otherwise.
 374 */
 375int amdgpu_vm_pt_clear(struct amdgpu_device *adev, struct amdgpu_vm *vm,
 376		       struct amdgpu_bo_vm *vmbo, bool immediate)
 377{
 378	unsigned int level = adev->vm_manager.root_level;
 379	struct ttm_operation_ctx ctx = { true, false };
 380	struct amdgpu_vm_update_params params;
 381	struct amdgpu_bo *ancestor = &vmbo->bo;
 382	unsigned int entries, ats_entries;
 383	struct amdgpu_bo *bo = &vmbo->bo;
 384	uint64_t addr;
 385	int r, idx;
 386
 387	/* Figure out our place in the hierarchy */
 388	if (ancestor->parent) {
 389		++level;
 390		while (ancestor->parent->parent) {
 391			++level;
 392			ancestor = ancestor->parent;
 393		}
 394	}
 395
 396	entries = amdgpu_bo_size(bo) / 8;
 397	if (!vm->pte_support_ats) {
 398		ats_entries = 0;
 399
 400	} else if (!bo->parent) {
 401		ats_entries = amdgpu_vm_pt_num_ats_entries(adev);
 402		ats_entries = min(ats_entries, entries);
 403		entries -= ats_entries;
 404
 405	} else {
 406		struct amdgpu_vm_bo_base *pt;
 407
 408		pt = ancestor->vm_bo;
 409		ats_entries = amdgpu_vm_pt_num_ats_entries(adev);
 410		if ((pt - to_amdgpu_bo_vm(vm->root.bo)->entries) >=
 411		    ats_entries) {
 412			ats_entries = 0;
 413		} else {
 414			ats_entries = entries;
 415			entries = 0;
 416		}
 417	}
 418
 419	r = ttm_bo_validate(&bo->tbo, &bo->placement, &ctx);
 420	if (r)
 421		return r;
 422
 423	if (vmbo->shadow) {
 424		struct amdgpu_bo *shadow = vmbo->shadow;
 425
 426		r = ttm_bo_validate(&shadow->tbo, &shadow->placement, &ctx);
 427		if (r)
 428			return r;
 429	}
 430
 431	if (!drm_dev_enter(adev_to_drm(adev), &idx))
 432		return -ENODEV;
 433
 434	r = vm->update_funcs->map_table(vmbo);
 435	if (r)
 436		goto exit;
 437
 438	memset(&params, 0, sizeof(params));
 439	params.adev = adev;
 440	params.vm = vm;
 441	params.immediate = immediate;
 442
 443	r = vm->update_funcs->prepare(&params, NULL, AMDGPU_SYNC_EXPLICIT);
 444	if (r)
 445		goto exit;
 446
 447	addr = 0;
 448	if (ats_entries) {
 449		uint64_t value = 0, flags;
 450
 451		flags = AMDGPU_PTE_DEFAULT_ATC;
 
 452		if (level != AMDGPU_VM_PTB) {
 453			/* Handle leaf PDEs as PTEs */
 454			flags |= AMDGPU_PDE_PTE;
 455			amdgpu_gmc_get_vm_pde(adev, level, &value, &flags);
 
 
 
 
 456		}
 457
 458		r = vm->update_funcs->update(&params, vmbo, addr, 0,
 459					     ats_entries, value, flags);
 460		if (r)
 461			goto exit;
 462
 463		addr += ats_entries * 8;
 464	}
 465
 466	if (entries) {
 467		uint64_t value = 0, flags = 0;
 468
 469		if (adev->asic_type >= CHIP_VEGA10) {
 470			if (level != AMDGPU_VM_PTB) {
 471				/* Handle leaf PDEs as PTEs */
 472				flags |= AMDGPU_PDE_PTE;
 473				amdgpu_gmc_get_vm_pde(adev, level,
 474						      &value, &flags);
 475			} else {
 476				/* Workaround for fault priority problem on GMC9 */
 477				flags = AMDGPU_PTE_EXECUTABLE;
 478			}
 479		}
 480
 481		r = vm->update_funcs->update(&params, vmbo, addr, 0, entries,
 482					     value, flags);
 483		if (r)
 484			goto exit;
 485	}
 486
 487	r = vm->update_funcs->commit(&params, NULL);
 488exit:
 489	drm_dev_exit(idx);
 490	return r;
 491}
 492
 493/**
 494 * amdgpu_vm_pt_create - create bo for PD/PT
 495 *
 496 * @adev: amdgpu_device pointer
 497 * @vm: requesting vm
 498 * @level: the page table level
 499 * @immediate: use a immediate update
 500 * @vmbo: pointer to the buffer object pointer
 501 * @xcp_id: GPU partition id
 502 */
 503int amdgpu_vm_pt_create(struct amdgpu_device *adev, struct amdgpu_vm *vm,
 504			int level, bool immediate, struct amdgpu_bo_vm **vmbo,
 505			int32_t xcp_id)
 506{
 507	struct amdgpu_bo_param bp;
 508	struct amdgpu_bo *bo;
 509	struct dma_resv *resv;
 510	unsigned int num_entries;
 511	int r;
 512
 513	memset(&bp, 0, sizeof(bp));
 514
 515	bp.size = amdgpu_vm_pt_size(adev, level);
 516	bp.byte_align = AMDGPU_GPU_PAGE_SIZE;
 517
 518	if (!adev->gmc.is_app_apu)
 519		bp.domain = AMDGPU_GEM_DOMAIN_VRAM;
 520	else
 521		bp.domain = AMDGPU_GEM_DOMAIN_GTT;
 522
 523	bp.domain = amdgpu_bo_get_preferred_domain(adev, bp.domain);
 524	bp.flags = AMDGPU_GEM_CREATE_VRAM_CONTIGUOUS |
 525		AMDGPU_GEM_CREATE_CPU_GTT_USWC;
 526
 527	if (level < AMDGPU_VM_PTB)
 528		num_entries = amdgpu_vm_pt_num_entries(adev, level);
 529	else
 530		num_entries = 0;
 531
 532	bp.bo_ptr_size = struct_size((*vmbo), entries, num_entries);
 533
 534	if (vm->use_cpu_for_update)
 535		bp.flags |= AMDGPU_GEM_CREATE_CPU_ACCESS_REQUIRED;
 536
 537	bp.type = ttm_bo_type_kernel;
 538	bp.no_wait_gpu = immediate;
 539	bp.xcp_id_plus1 = xcp_id + 1;
 540
 541	if (vm->root.bo)
 542		bp.resv = vm->root.bo->tbo.base.resv;
 543
 544	r = amdgpu_bo_create_vm(adev, &bp, vmbo);
 545	if (r)
 546		return r;
 547
 548	bo = &(*vmbo)->bo;
 549	if (vm->is_compute_context || (adev->flags & AMD_IS_APU)) {
 550		(*vmbo)->shadow = NULL;
 551		return 0;
 552	}
 553
 554	if (!bp.resv)
 555		WARN_ON(dma_resv_lock(bo->tbo.base.resv,
 556				      NULL));
 557	resv = bp.resv;
 558	memset(&bp, 0, sizeof(bp));
 559	bp.size = amdgpu_vm_pt_size(adev, level);
 560	bp.domain = AMDGPU_GEM_DOMAIN_GTT;
 561	bp.flags = AMDGPU_GEM_CREATE_CPU_GTT_USWC;
 562	bp.type = ttm_bo_type_kernel;
 563	bp.resv = bo->tbo.base.resv;
 564	bp.bo_ptr_size = sizeof(struct amdgpu_bo);
 565	bp.xcp_id_plus1 = xcp_id + 1;
 566
 567	r = amdgpu_bo_create(adev, &bp, &(*vmbo)->shadow);
 568
 569	if (!resv)
 570		dma_resv_unlock(bo->tbo.base.resv);
 571
 572	if (r) {
 573		amdgpu_bo_unref(&bo);
 574		return r;
 575	}
 576
 577	amdgpu_bo_add_to_shadow_list(*vmbo);
 578
 579	return 0;
 580}
 581
 582/**
 583 * amdgpu_vm_pt_alloc - Allocate a specific page table
 584 *
 585 * @adev: amdgpu_device pointer
 586 * @vm: VM to allocate page tables for
 587 * @cursor: Which page table to allocate
 588 * @immediate: use an immediate update
 589 *
 590 * Make sure a specific page table or directory is allocated.
 591 *
 592 * Returns:
 593 * 1 if page table needed to be allocated, 0 if page table was already
 594 * allocated, negative errno if an error occurred.
 595 */
 596static int amdgpu_vm_pt_alloc(struct amdgpu_device *adev,
 597			      struct amdgpu_vm *vm,
 598			      struct amdgpu_vm_pt_cursor *cursor,
 599			      bool immediate)
 600{
 601	struct amdgpu_vm_bo_base *entry = cursor->entry;
 602	struct amdgpu_bo *pt_bo;
 603	struct amdgpu_bo_vm *pt;
 604	int r;
 605
 606	if (entry->bo)
 607		return 0;
 608
 609	amdgpu_vm_eviction_unlock(vm);
 610	r = amdgpu_vm_pt_create(adev, vm, cursor->level, immediate, &pt,
 611				vm->root.bo->xcp_id);
 612	amdgpu_vm_eviction_lock(vm);
 613	if (r)
 614		return r;
 615
 616	/* Keep a reference to the root directory to avoid
 617	 * freeing them up in the wrong order.
 618	 */
 619	pt_bo = &pt->bo;
 620	pt_bo->parent = amdgpu_bo_ref(cursor->parent->bo);
 621	amdgpu_vm_bo_base_init(entry, vm, pt_bo);
 622	r = amdgpu_vm_pt_clear(adev, vm, pt, immediate);
 623	if (r)
 624		goto error_free_pt;
 625
 626	return 0;
 627
 628error_free_pt:
 629	amdgpu_bo_unref(&pt->shadow);
 630	amdgpu_bo_unref(&pt_bo);
 631	return r;
 632}
 633
 634/**
 635 * amdgpu_vm_pt_free - free one PD/PT
 636 *
 637 * @entry: PDE to free
 638 */
 639static void amdgpu_vm_pt_free(struct amdgpu_vm_bo_base *entry)
 640{
 641	struct amdgpu_bo *shadow;
 642
 643	if (!entry->bo)
 644		return;
 645
 646	entry->bo->vm_bo = NULL;
 647	shadow = amdgpu_bo_shadowed(entry->bo);
 648	if (shadow) {
 649		ttm_bo_set_bulk_move(&shadow->tbo, NULL);
 650		amdgpu_bo_unref(&shadow);
 651	}
 652	ttm_bo_set_bulk_move(&entry->bo->tbo, NULL);
 653
 654	spin_lock(&entry->vm->status_lock);
 655	list_del(&entry->vm_status);
 656	spin_unlock(&entry->vm->status_lock);
 657	amdgpu_bo_unref(&entry->bo);
 658}
 659
 660void amdgpu_vm_pt_free_work(struct work_struct *work)
 661{
 662	struct amdgpu_vm_bo_base *entry, *next;
 663	struct amdgpu_vm *vm;
 664	LIST_HEAD(pt_freed);
 665
 666	vm = container_of(work, struct amdgpu_vm, pt_free_work);
 667
 668	spin_lock(&vm->status_lock);
 669	list_splice_init(&vm->pt_freed, &pt_freed);
 670	spin_unlock(&vm->status_lock);
 671
 672	/* flush_work in amdgpu_vm_fini ensure vm->root.bo is valid. */
 673	amdgpu_bo_reserve(vm->root.bo, true);
 674
 675	list_for_each_entry_safe(entry, next, &pt_freed, vm_status)
 676		amdgpu_vm_pt_free(entry);
 677
 678	amdgpu_bo_unreserve(vm->root.bo);
 679}
 680
 681/**
 682 * amdgpu_vm_pt_free_dfs - free PD/PT levels
 683 *
 684 * @adev: amdgpu device structure
 685 * @vm: amdgpu vm structure
 686 * @start: optional cursor where to start freeing PDs/PTs
 687 * @unlocked: vm resv unlock status
 688 *
 689 * Free the page directory or page table level and all sub levels.
 690 */
 691static void amdgpu_vm_pt_free_dfs(struct amdgpu_device *adev,
 692				  struct amdgpu_vm *vm,
 693				  struct amdgpu_vm_pt_cursor *start,
 694				  bool unlocked)
 695{
 696	struct amdgpu_vm_pt_cursor cursor;
 697	struct amdgpu_vm_bo_base *entry;
 
 
 
 
 698
 699	if (unlocked) {
 700		spin_lock(&vm->status_lock);
 701		for_each_amdgpu_vm_pt_dfs_safe(adev, vm, start, cursor, entry)
 702			list_move(&entry->vm_status, &vm->pt_freed);
 703
 704		if (start)
 705			list_move(&start->entry->vm_status, &vm->pt_freed);
 706		spin_unlock(&vm->status_lock);
 707		schedule_work(&vm->pt_free_work);
 708		return;
 709	}
 710
 711	for_each_amdgpu_vm_pt_dfs_safe(adev, vm, start, cursor, entry)
 712		amdgpu_vm_pt_free(entry);
 
 713
 714	if (start)
 715		amdgpu_vm_pt_free(start->entry);
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 716}
 717
 718/**
 719 * amdgpu_vm_pt_free_root - free root PD
 720 * @adev: amdgpu device structure
 721 * @vm: amdgpu vm structure
 722 *
 723 * Free the root page directory and everything below it.
 724 */
 725void amdgpu_vm_pt_free_root(struct amdgpu_device *adev, struct amdgpu_vm *vm)
 726{
 727	amdgpu_vm_pt_free_dfs(adev, vm, NULL, false);
 728}
 729
 730/**
 731 * amdgpu_vm_pt_is_root_clean - check if a root PD is clean
 732 *
 733 * @adev: amdgpu_device pointer
 734 * @vm: the VM to check
 735 *
 736 * Check all entries of the root PD, if any subsequent PDs are allocated,
 737 * it means there are page table creating and filling, and is no a clean
 738 * VM
 739 *
 740 * Returns:
 741 *	0 if this VM is clean
 742 */
 743bool amdgpu_vm_pt_is_root_clean(struct amdgpu_device *adev,
 744				struct amdgpu_vm *vm)
 745{
 746	enum amdgpu_vm_level root = adev->vm_manager.root_level;
 747	unsigned int entries = amdgpu_vm_pt_num_entries(adev, root);
 748	unsigned int i = 0;
 749
 750	for (i = 0; i < entries; i++) {
 751		if (to_amdgpu_bo_vm(vm->root.bo)->entries[i].bo)
 752			return false;
 753	}
 754	return true;
 755}
 756
 757/**
 758 * amdgpu_vm_pde_update - update a single level in the hierarchy
 759 *
 760 * @params: parameters for the update
 761 * @entry: entry to update
 762 *
 763 * Makes sure the requested entry in parent is up to date.
 764 */
 765int amdgpu_vm_pde_update(struct amdgpu_vm_update_params *params,
 766			 struct amdgpu_vm_bo_base *entry)
 767{
 768	struct amdgpu_vm_bo_base *parent = amdgpu_vm_pt_parent(entry);
 769	struct amdgpu_bo *bo = parent->bo, *pbo;
 770	struct amdgpu_vm *vm = params->vm;
 771	uint64_t pde, pt, flags;
 772	unsigned int level;
 773
 
 
 
 
 774	for (level = 0, pbo = bo->parent; pbo; ++level)
 775		pbo = pbo->parent;
 776
 777	level += params->adev->vm_manager.root_level;
 778	amdgpu_gmc_get_pde_for_bo(entry->bo, level, &pt, &flags);
 779	pde = (entry - to_amdgpu_bo_vm(parent->bo)->entries) * 8;
 780	return vm->update_funcs->update(params, to_amdgpu_bo_vm(bo), pde, pt,
 781					1, 0, flags);
 782}
 783
 784/**
 785 * amdgpu_vm_pte_update_noretry_flags - Update PTE no-retry flags
 786 *
 787 * @adev: amdgpu_device pointer
 788 * @flags: pointer to PTE flags
 789 *
 790 * Update PTE no-retry flags when TF is enabled.
 791 */
 792static void amdgpu_vm_pte_update_noretry_flags(struct amdgpu_device *adev,
 793						uint64_t *flags)
 794{
 795	/*
 796	 * Update no-retry flags with the corresponding TF
 797	 * no-retry combination.
 798	 */
 799	if ((*flags & AMDGPU_VM_NORETRY_FLAGS) == AMDGPU_VM_NORETRY_FLAGS) {
 800		*flags &= ~AMDGPU_VM_NORETRY_FLAGS;
 801		*flags |= adev->gmc.noretry_flags;
 802	}
 803}
 804
 805/*
 806 * amdgpu_vm_pte_update_flags - figure out flags for PTE updates
 807 *
 808 * Make sure to set the right flags for the PTEs at the desired level.
 809 */
 810static void amdgpu_vm_pte_update_flags(struct amdgpu_vm_update_params *params,
 811				       struct amdgpu_bo_vm *pt,
 812				       unsigned int level,
 813				       uint64_t pe, uint64_t addr,
 814				       unsigned int count, uint32_t incr,
 815				       uint64_t flags)
 816{
 817	struct amdgpu_device *adev = params->adev;
 818
 819	if (level != AMDGPU_VM_PTB) {
 820		flags |= AMDGPU_PDE_PTE;
 821		amdgpu_gmc_get_vm_pde(adev, level, &addr, &flags);
 822
 823	} else if (adev->asic_type >= CHIP_VEGA10 &&
 824		   !(flags & AMDGPU_PTE_VALID) &&
 825		   !(flags & AMDGPU_PTE_PRT)) {
 826
 827		/* Workaround for fault priority problem on GMC9 */
 828		flags |= AMDGPU_PTE_EXECUTABLE;
 829	}
 830
 831	/*
 832	 * Update no-retry flags to use the no-retry flag combination
 833	 * with TF enabled. The AMDGPU_VM_NORETRY_FLAGS flag combination
 834	 * does not work when TF is enabled. So, replace them with
 835	 * AMDGPU_VM_NORETRY_FLAGS_TF flag combination which works for
 836	 * all cases.
 837	 */
 838	if (level == AMDGPU_VM_PTB)
 839		amdgpu_vm_pte_update_noretry_flags(adev, &flags);
 840
 841	/* APUs mapping system memory may need different MTYPEs on different
 842	 * NUMA nodes. Only do this for contiguous ranges that can be assumed
 843	 * to be on the same NUMA node.
 844	 */
 845	if ((flags & AMDGPU_PTE_SYSTEM) && (adev->flags & AMD_IS_APU) &&
 846	    adev->gmc.gmc_funcs->override_vm_pte_flags &&
 847	    num_possible_nodes() > 1 && !params->pages_addr && params->allow_override)
 848		amdgpu_gmc_override_vm_pte_flags(adev, params->vm, addr, &flags);
 849
 850	params->vm->update_funcs->update(params, pt, pe, addr, count, incr,
 851					 flags);
 852}
 853
 854/**
 855 * amdgpu_vm_pte_fragment - get fragment for PTEs
 856 *
 857 * @params: see amdgpu_vm_update_params definition
 858 * @start: first PTE to handle
 859 * @end: last PTE to handle
 860 * @flags: hw mapping flags
 861 * @frag: resulting fragment size
 862 * @frag_end: end of this fragment
 863 *
 864 * Returns the first possible fragment for the start and end address.
 865 */
 866static void amdgpu_vm_pte_fragment(struct amdgpu_vm_update_params *params,
 867				   uint64_t start, uint64_t end, uint64_t flags,
 868				   unsigned int *frag, uint64_t *frag_end)
 869{
 870	/**
 871	 * The MC L1 TLB supports variable sized pages, based on a fragment
 872	 * field in the PTE. When this field is set to a non-zero value, page
 873	 * granularity is increased from 4KB to (1 << (12 + frag)). The PTE
 874	 * flags are considered valid for all PTEs within the fragment range
 875	 * and corresponding mappings are assumed to be physically contiguous.
 876	 *
 877	 * The L1 TLB can store a single PTE for the whole fragment,
 878	 * significantly increasing the space available for translation
 879	 * caching. This leads to large improvements in throughput when the
 880	 * TLB is under pressure.
 881	 *
 882	 * The L2 TLB distributes small and large fragments into two
 883	 * asymmetric partitions. The large fragment cache is significantly
 884	 * larger. Thus, we try to use large fragments wherever possible.
 885	 * Userspace can support this by aligning virtual base address and
 886	 * allocation size to the fragment size.
 887	 *
 888	 * Starting with Vega10 the fragment size only controls the L1. The L2
 889	 * is now directly feed with small/huge/giant pages from the walker.
 890	 */
 891	unsigned int max_frag;
 892
 893	if (params->adev->asic_type < CHIP_VEGA10)
 894		max_frag = params->adev->vm_manager.fragment_size;
 895	else
 896		max_frag = 31;
 897
 898	/* system pages are non continuously */
 899	if (params->pages_addr) {
 900		*frag = 0;
 901		*frag_end = end;
 902		return;
 903	}
 904
 905	/* This intentionally wraps around if no bit is set */
 906	*frag = min_t(unsigned int, ffs(start) - 1, fls64(end - start) - 1);
 907	if (*frag >= max_frag) {
 908		*frag = max_frag;
 909		*frag_end = end & ~((1ULL << max_frag) - 1);
 910	} else {
 911		*frag_end = start + (1 << *frag);
 912	}
 913}
 914
 915/**
 916 * amdgpu_vm_ptes_update - make sure that page tables are valid
 917 *
 918 * @params: see amdgpu_vm_update_params definition
 919 * @start: start of GPU address range
 920 * @end: end of GPU address range
 921 * @dst: destination address to map to, the next dst inside the function
 922 * @flags: mapping flags
 923 *
 924 * Update the page tables in the range @start - @end.
 925 *
 926 * Returns:
 927 * 0 for success, -EINVAL for failure.
 928 */
 929int amdgpu_vm_ptes_update(struct amdgpu_vm_update_params *params,
 930			  uint64_t start, uint64_t end,
 931			  uint64_t dst, uint64_t flags)
 932{
 933	struct amdgpu_device *adev = params->adev;
 934	struct amdgpu_vm_pt_cursor cursor;
 935	uint64_t frag_start = start, frag_end;
 936	unsigned int frag;
 937	int r;
 938
 939	/* figure out the initial fragment */
 940	amdgpu_vm_pte_fragment(params, frag_start, end, flags, &frag,
 941			       &frag_end);
 942
 943	/* walk over the address space and update the PTs */
 944	amdgpu_vm_pt_start(adev, params->vm, start, &cursor);
 945	while (cursor.pfn < end) {
 946		unsigned int shift, parent_shift, mask;
 947		uint64_t incr, entry_end, pe_start;
 948		struct amdgpu_bo *pt;
 949
 950		if (!params->unlocked) {
 951			/* make sure that the page tables covering the
 952			 * address range are actually allocated
 953			 */
 954			r = amdgpu_vm_pt_alloc(params->adev, params->vm,
 955					       &cursor, params->immediate);
 956			if (r)
 957				return r;
 958		}
 959
 960		shift = amdgpu_vm_pt_level_shift(adev, cursor.level);
 961		parent_shift = amdgpu_vm_pt_level_shift(adev, cursor.level - 1);
 962		if (params->unlocked) {
 963			/* Unlocked updates are only allowed on the leaves */
 964			if (amdgpu_vm_pt_descendant(adev, &cursor))
 965				continue;
 966		} else if (adev->asic_type < CHIP_VEGA10 &&
 967			   (flags & AMDGPU_PTE_VALID)) {
 968			/* No huge page support before GMC v9 */
 969			if (cursor.level != AMDGPU_VM_PTB) {
 970				if (!amdgpu_vm_pt_descendant(adev, &cursor))
 971					return -ENOENT;
 972				continue;
 973			}
 974		} else if (frag < shift) {
 975			/* We can't use this level when the fragment size is
 976			 * smaller than the address shift. Go to the next
 977			 * child entry and try again.
 978			 */
 979			if (amdgpu_vm_pt_descendant(adev, &cursor))
 980				continue;
 981		} else if (frag >= parent_shift) {
 982			/* If the fragment size is even larger than the parent
 983			 * shift we should go up one level and check it again.
 984			 */
 985			if (!amdgpu_vm_pt_ancestor(&cursor))
 986				return -EINVAL;
 987			continue;
 988		}
 989
 990		pt = cursor.entry->bo;
 991		if (!pt) {
 992			/* We need all PDs and PTs for mapping something, */
 993			if (flags & AMDGPU_PTE_VALID)
 994				return -ENOENT;
 995
 996			/* but unmapping something can happen at a higher
 997			 * level.
 998			 */
 999			if (!amdgpu_vm_pt_ancestor(&cursor))
1000				return -EINVAL;
1001
1002			pt = cursor.entry->bo;
1003			shift = parent_shift;
1004			frag_end = max(frag_end, ALIGN(frag_start + 1,
1005				   1ULL << shift));
1006		}
1007
1008		/* Looks good so far, calculate parameters for the update */
1009		incr = (uint64_t)AMDGPU_GPU_PAGE_SIZE << shift;
1010		mask = amdgpu_vm_pt_entries_mask(adev, cursor.level);
1011		pe_start = ((cursor.pfn >> shift) & mask) * 8;
1012		entry_end = ((uint64_t)mask + 1) << shift;
1013		entry_end += cursor.pfn & ~(entry_end - 1);
1014		entry_end = min(entry_end, end);
1015
1016		do {
1017			struct amdgpu_vm *vm = params->vm;
1018			uint64_t upd_end = min(entry_end, frag_end);
1019			unsigned int nptes = (upd_end - frag_start) >> shift;
1020			uint64_t upd_flags = flags | AMDGPU_PTE_FRAG(frag);
1021
1022			/* This can happen when we set higher level PDs to
1023			 * silent to stop fault floods.
1024			 */
1025			nptes = max(nptes, 1u);
1026
1027			trace_amdgpu_vm_update_ptes(params, frag_start, upd_end,
1028						    min(nptes, 32u), dst, incr,
1029						    upd_flags,
1030						    vm->task_info.tgid,
1031						    vm->immediate.fence_context);
1032			amdgpu_vm_pte_update_flags(params, to_amdgpu_bo_vm(pt),
1033						   cursor.level, pe_start, dst,
1034						   nptes, incr, upd_flags);
1035
1036			pe_start += nptes * 8;
1037			dst += nptes * incr;
1038
1039			frag_start = upd_end;
1040			if (frag_start >= frag_end) {
1041				/* figure out the next fragment */
1042				amdgpu_vm_pte_fragment(params, frag_start, end,
1043						       flags, &frag, &frag_end);
1044				if (frag < shift)
1045					break;
1046			}
1047		} while (frag_start < entry_end);
1048
1049		if (amdgpu_vm_pt_descendant(adev, &cursor)) {
1050			/* Free all child entries.
1051			 * Update the tables with the flags and addresses and free up subsequent
1052			 * tables in the case of huge pages or freed up areas.
1053			 * This is the maximum you can free, because all other page tables are not
1054			 * completely covered by the range and so potentially still in use.
1055			 */
1056			while (cursor.pfn < frag_start) {
1057				/* Make sure previous mapping is freed */
1058				if (cursor.entry->bo) {
1059					params->table_freed = true;
1060					amdgpu_vm_pt_free_dfs(adev, params->vm,
1061							      &cursor,
1062							      params->unlocked);
1063				}
1064				amdgpu_vm_pt_next(adev, &cursor);
1065			}
1066
1067		} else if (frag >= shift) {
1068			/* or just move on to the next on the same level. */
1069			amdgpu_vm_pt_next(adev, &cursor);
1070		}
1071	}
1072
1073	return 0;
1074}
1075
1076/**
1077 * amdgpu_vm_pt_map_tables - have bo of root PD cpu accessible
1078 * @adev: amdgpu device structure
1079 * @vm: amdgpu vm structure
1080 *
1081 * make root page directory and everything below it cpu accessible.
1082 */
1083int amdgpu_vm_pt_map_tables(struct amdgpu_device *adev, struct amdgpu_vm *vm)
1084{
1085	struct amdgpu_vm_pt_cursor cursor;
1086	struct amdgpu_vm_bo_base *entry;
1087
1088	for_each_amdgpu_vm_pt_dfs_safe(adev, vm, NULL, cursor, entry) {
1089
1090		struct amdgpu_bo_vm *bo;
1091		int r;
1092
1093		if (entry->bo) {
1094			bo = to_amdgpu_bo_vm(entry->bo);
1095			r = vm->update_funcs->map_table(bo);
1096			if (r)
1097				return r;
1098		}
1099	}
1100
1101	return 0;
1102}