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