brintos

brintos / linux-shallow public Read only

0
0
Text · 26.3 KiB · f78a043 Raw
991 lines · c
1// SPDX-License-Identifier: GPL-2.0 OR MIT2/*3 * Copyright 2022 Advanced Micro Devices, Inc.4 *5 * Permission is hereby granted, free of charge, to any person obtaining a6 * copy of this software and associated documentation files (the "Software"),7 * to deal in the Software without restriction, including without limitation8 * the rights to use, copy, modify, merge, publish, distribute, sublicense,9 * and/or sell copies of the Software, and to permit persons to whom the10 * Software is furnished to do so, subject to the following conditions:11 *12 * The above copyright notice and this permission notice shall be included in13 * all copies or substantial portions of the Software.14 *15 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR16 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,17 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL18 * THE COPYRIGHT HOLDER(S) OR AUTHOR(S) BE LIABLE FOR ANY CLAIM, DAMAGES OR19 * 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 OR21 * 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_pt32 */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 level42 *43 * @adev: amdgpu_device pointer44 * @level: VMPT level45 *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/PT67 *68 * @adev: amdgpu_device pointer69 * @level: VMPT level70 *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/PT94 *95 * @adev: amdgpu_device pointer96 * @level: VMPT level97 *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	else109		return AMDGPU_VM_PTE_COUNT(adev) - 1;110}111 112/**113 * amdgpu_vm_pt_size - returns the size of the page table in bytes114 *115 * @adev: amdgpu_device pointer116 * @level: VMPT level117 *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 directory129 *130 * @pt: child page table131 *132 * Helper to get the parent entry for the child page table. NULL if we are at133 * 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 walk148 *149 * @adev: amdgpu_device pointer150 * @vm: amdgpu_vm structure151 * @start: start address of the walk152 * @cursor: state to initialize153 *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 node168 *169 * @adev: amdgpu_device pointer170 * @cursor: current state171 *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 node197 *198 * @adev: amdgpu_device pointer199 * @cursor: current state200 *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 node232 *233 * @cursor: current state234 *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 hieratchy252 *253 * @adev: amdgpu_device pointer254 * @cursor: current state255 *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 search277 *278 * @adev: amdgpu_device structure279 * @vm: amdgpu_vm structure280 * @start: optional cursor to start with281 * @cursor: state to initialize282 *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	else293		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 continue301 *302 * @start: starting point for the search303 * @entry: current entry304 *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 search316 *317 * @adev: amdgpu_device structure318 * @cursor: current state319 *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	else334		amdgpu_vm_pt_ancestor(cursor);335}336 337/*338 * for_each_amdgpu_vm_pt_dfs_safe - safe deep first search of all PDs/PTs339 */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/PTs348 *349 * @adev: amdgpu_device pointer350 * @vm: VM to clear BO from351 * @vmbo: BO to clear352 * @immediate: use an immediate update353 *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/PT430 *431 * @adev: amdgpu_device pointer432 * @vm: requesting vm433 * @level: the page table level434 * @immediate: use a immediate update435 * @vmbo: pointer to the buffer object pointer436 * @xcp_id: GPU partition id437 */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	else453		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	else462		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 table481 *482 * @adev: amdgpu_device pointer483 * @vm: VM to allocate page tables for484 * @cursor: Which page table to allocate485 * @immediate: use an immediate update486 *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 already491 * 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 avoid514	 * 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/PT532 *533 * @entry: PDE to free534 */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 levels572 *573 * @adev: amdgpu device structure574 * @params: see amdgpu_vm_update_params definition575 *576 * Free the page directory objects saved in the flush list577 */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 list602 *603 * @params: parameters for the update604 * @cursor: first PT entry to start DF search from, non NULL605 *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 PD627 * @adev: amdgpu device structure628 * @vm: amdgpu vm structure629 *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 hierarchy645 *646 * @params: parameters for the update647 * @entry: entry to update648 *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 flags676 *677 * @adev: amdgpu_device pointer678 * @flags: pointer to PTE flags679 *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 TF687	 * 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 updates697 *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 combination723	 * with TF enabled. The AMDGPU_VM_NORETRY_FLAGS flag combination724	 * does not work when TF is enabled. So, replace them with725	 * AMDGPU_VM_NORETRY_FLAGS_TF flag combination which works for726	 * 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 different732	 * NUMA nodes. Only do this for contiguous ranges that can be assumed733	 * 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 PTEs746 *747 * @params: see amdgpu_vm_update_params definition748 * @start: first PTE to handle749 * @end: last PTE to handle750 * @flags: hw mapping flags751 * @frag: resulting fragment size752 * @frag_end: end of this fragment753 *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 fragment762	 * field in the PTE. When this field is set to a non-zero value, page763	 * granularity is increased from 4KB to (1 << (12 + frag)). The PTE764	 * flags are considered valid for all PTEs within the fragment range765	 * 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 translation769	 * caching. This leads to large improvements in throughput when the770	 * TLB is under pressure.771	 *772	 * The L2 TLB distributes small and large fragments into two773	 * asymmetric partitions. The large fragment cache is significantly774	 * larger. Thus, we try to use large fragments wherever possible.775	 * Userspace can support this by aligning virtual base address and776	 * allocation size to the fragment size.777	 *778	 * Starting with Vega10 the fragment size only controls the L1. The L2779	 * 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	else786		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 valid807 *808 * @params: see amdgpu_vm_update_params definition809 * @start: start of GPU address range810 * @end: end of GPU address range811 * @dst: destination address to map to, the next dst inside the function812 * @flags: mapping flags813 *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 the842			 * address range are actually allocated843			 */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 is866			 * smaller than the address shift. Go to the next867			 * 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 parent873			 * 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 higher887			 * 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 to913			 * 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 subsequent942			 * 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 not944			 * 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 accessible966 * @adev: amdgpu device structure967 * @vm: amdgpu vm structure968 *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}991