amdgpu_vm.c 38.6 KB
Newer Older
A
Alex Deucher 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52
/*
 * Copyright 2008 Advanced Micro Devices, Inc.
 * Copyright 2008 Red Hat Inc.
 * Copyright 2009 Jerome Glisse.
 *
 * Permission is hereby granted, free of charge, to any person obtaining a
 * copy of this software and associated documentation files (the "Software"),
 * to deal in the Software without restriction, including without limitation
 * the rights to use, copy, modify, merge, publish, distribute, sublicense,
 * and/or sell copies of the Software, and to permit persons to whom the
 * Software is furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
 * THE COPYRIGHT HOLDER(S) OR AUTHOR(S) BE LIABLE FOR ANY CLAIM, DAMAGES OR
 * OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
 * ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
 * OTHER DEALINGS IN THE SOFTWARE.
 *
 * Authors: Dave Airlie
 *          Alex Deucher
 *          Jerome Glisse
 */
#include <drm/drmP.h>
#include <drm/amdgpu_drm.h>
#include "amdgpu.h"
#include "amdgpu_trace.h"

/*
 * GPUVM
 * GPUVM is similar to the legacy gart on older asics, however
 * rather than there being a single global gart table
 * for the entire GPU, there are multiple VM page tables active
 * at any given time.  The VM page tables can contain a mix
 * vram pages and system memory pages and system memory pages
 * can be mapped as snooped (cached system pages) or unsnooped
 * (uncached system pages).
 * Each VM has an ID associated with it and there is a page table
 * associated with each VMID.  When execting a command buffer,
 * the kernel tells the the ring what VMID to use for that command
 * buffer.  VMIDs are allocated dynamically as commands are submitted.
 * The userspace drivers maintain their own address space and the kernel
 * sets up their pages tables accordingly when they submit their
 * command buffers and a VMID is assigned.
 * Cayman/Trinity support up to 8 active VMs at any given time;
 * SI supports 16.
 */

53 54 55
/* Special value that no flush is necessary */
#define AMDGPU_VM_NO_FLUSH (~0ll)

A
Alex Deucher 已提交
56 57 58 59 60
/**
 * amdgpu_vm_num_pde - return the number of page directory entries
 *
 * @adev: amdgpu_device pointer
 *
61
 * Calculate the number of page directory entries.
A
Alex Deucher 已提交
62 63 64 65 66 67 68 69 70 71 72
 */
static unsigned amdgpu_vm_num_pdes(struct amdgpu_device *adev)
{
	return adev->vm_manager.max_pfn >> amdgpu_vm_block_size;
}

/**
 * amdgpu_vm_directory_size - returns the size of the page directory in bytes
 *
 * @adev: amdgpu_device pointer
 *
73
 * Calculate the size of the page directory in bytes.
A
Alex Deucher 已提交
74 75 76 77 78 79 80
 */
static unsigned amdgpu_vm_directory_size(struct amdgpu_device *adev)
{
	return AMDGPU_GPU_PAGE_ALIGN(amdgpu_vm_num_pdes(adev) * 8);
}

/**
81
 * amdgpu_vm_get_pd_bo - add the VM PD to a validation list
A
Alex Deucher 已提交
82 83
 *
 * @vm: vm providing the BOs
84
 * @validated: head of validation list
85
 * @entry: entry to add
A
Alex Deucher 已提交
86 87
 *
 * Add the page directory to the list of BOs to
88
 * validate for command submission.
A
Alex Deucher 已提交
89
 */
90 91 92
void amdgpu_vm_get_pd_bo(struct amdgpu_vm *vm,
			 struct list_head *validated,
			 struct amdgpu_bo_list_entry *entry)
A
Alex Deucher 已提交
93
{
94 95 96 97
	entry->robj = vm->page_directory;
	entry->priority = 0;
	entry->tv.bo = &vm->page_directory->tbo;
	entry->tv.shared = true;
98
	entry->user_pages = NULL;
99 100
	list_add(&entry->tv.head, validated);
}
A
Alex Deucher 已提交
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
A
Alex Deucher 已提交
107
 *
108 109
 * Add the page directory to the BO duplicates list
 * for command submission.
A
Alex Deucher 已提交
110
 */
111
void amdgpu_vm_get_pt_bos(struct amdgpu_vm *vm, struct list_head *duplicates)
A
Alex Deucher 已提交
112
{
113
	unsigned i;
A
Alex Deucher 已提交
114 115

	/* add the vm page table to the list */
116 117 118 119
	for (i = 0; i <= vm->max_pde_used; ++i) {
		struct amdgpu_bo_list_entry *entry = &vm->page_tables[i].entry;

		if (!entry->robj)
A
Alex Deucher 已提交
120 121
			continue;

122
		list_add(&entry->tv.head, duplicates);
A
Alex Deucher 已提交
123
	}
124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150

}

/**
 * amdgpu_vm_move_pt_bos_in_lru - move the PT BOs to the LRU tail
 *
 * @adev: amdgpu device instance
 * @vm: vm providing the BOs
 *
 * Move the PT BOs to the tail of the LRU.
 */
void amdgpu_vm_move_pt_bos_in_lru(struct amdgpu_device *adev,
				  struct amdgpu_vm *vm)
{
	struct ttm_bo_global *glob = adev->mman.bdev.glob;
	unsigned i;

	spin_lock(&glob->lru_lock);
	for (i = 0; i <= vm->max_pde_used; ++i) {
		struct amdgpu_bo_list_entry *entry = &vm->page_tables[i].entry;

		if (!entry->robj)
			continue;

		ttm_bo_move_to_lru_tail(&entry->robj->tbo);
	}
	spin_unlock(&glob->lru_lock);
A
Alex Deucher 已提交
151 152 153 154 155 156
}

/**
 * amdgpu_vm_grab_id - allocate the next free VMID
 *
 * @vm: vm to allocate id for
157 158
 * @ring: ring we want to submit job to
 * @sync: sync object where we add dependencies
159
 * @fence: fence protecting ID from reuse
A
Alex Deucher 已提交
160
 *
161
 * Allocate an id for the vm, adding fences to the sync obj as necessary.
A
Alex Deucher 已提交
162
 */
163
int amdgpu_vm_grab_id(struct amdgpu_vm *vm, struct amdgpu_ring *ring,
164 165
		      struct amdgpu_sync *sync, struct fence *fence,
		      unsigned *vm_id, uint64_t *vm_pd_addr)
A
Alex Deucher 已提交
166
{
167
	uint64_t pd_addr = amdgpu_bo_gpu_offset(vm->page_directory);
A
Alex Deucher 已提交
168
	struct amdgpu_device *adev = ring->adev;
169
	struct fence *updates = sync->last_vm_update;
170 171
	struct amdgpu_vm_id *id;
	unsigned i = ring->idx;
172
	int r;
A
Alex Deucher 已提交
173

174 175
	mutex_lock(&adev->vm_manager.lock);

176 177 178
	/* Check if we can use a VMID already assigned to this VM */
	do {
		struct fence *flushed;
179

180 181 182
		id = vm->ids[i++];
		if (i == AMDGPU_MAX_RINGS)
			i = 0;
183

184 185 186 187
		/* Check all the prerequisites to using this VMID */
		if (!id)
			continue;

188
		if (atomic_long_read(&id->owner) != vm->client_id)
189 190 191 192 193
			continue;

		if (pd_addr != id->pd_gpu_addr)
			continue;

194
		if (id->last_user != ring &&
195 196 197 198 199 200
		    (!id->last_flush || !fence_is_signaled(id->last_flush)))
			continue;

		flushed  = id->flushed_updates;
		if (updates && (!flushed || fence_is_later(updates, flushed)))
			continue;
201

202
		/* Good we can use this VMID */
203
		if (id->last_user == ring) {
204 205
			r = amdgpu_sync_fence(ring->adev, sync,
					      id->first);
206 207
			if (r)
				goto error;
208 209 210 211 212 213
		}

		/* And remember this submission as user of the VMID */
		r = amdgpu_sync_fence(ring->adev, &id->active, fence);
		if (r)
			goto error;
214

215 216
		list_move_tail(&id->list, &adev->vm_manager.ids_lru);
		vm->ids[ring->idx] = id;
A
Alex Deucher 已提交
217

218 219 220
		*vm_id = id - adev->vm_manager.ids;
		*vm_pd_addr = AMDGPU_VM_NO_FLUSH;
		trace_amdgpu_vm_grab_id(vm, ring->idx, *vm_id, *vm_pd_addr);
A
Alex Deucher 已提交
221

222 223 224 225
		mutex_unlock(&adev->vm_manager.lock);
		return 0;

	} while (i != ring->idx);
A
Alex Deucher 已提交
226

227 228 229
	id = list_first_entry(&adev->vm_manager.ids_lru,
			      struct amdgpu_vm_id,
			      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 236

		list_for_each_entry_safe(id, tmp, &adev->vm_manager.ids_lru,
					 list) {
237
			if (amdgpu_sync_is_idle(&id->active)) {
238 239
				list_move(&id->list, head);
				head = &id->list;
240 241
			}
		}
242 243 244
		id = list_first_entry(&adev->vm_manager.ids_lru,
				      struct amdgpu_vm_id,
				      list);
245 246
	}

247 248 249
	r = amdgpu_sync_cycle_fences(sync, &id->active, fence);
	if (r)
		goto error;
250

251 252
	fence_put(id->first);
	id->first = fence_get(fence);
253

254 255 256
	fence_put(id->last_flush);
	id->last_flush = NULL;

257 258
	fence_put(id->flushed_updates);
	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, vm->client_id);
265
	vm->ids[ring->idx] = id;
A
Alex Deucher 已提交
266

267 268 269 270 271
	*vm_id = id - adev->vm_manager.ids;
	*vm_pd_addr = pd_addr;
	trace_amdgpu_vm_grab_id(vm, ring->idx, *vm_id, *vm_pd_addr);

error:
272
	mutex_unlock(&adev->vm_manager.lock);
273
	return r;
A
Alex Deucher 已提交
274 275 276 277 278 279
}

/**
 * amdgpu_vm_flush - hardware flush the vm
 *
 * @ring: ring to use for flush
280
 * @vm_id: vmid number to use
281
 * @pd_addr: address of the page directory
A
Alex Deucher 已提交
282
 *
283
 * Emit a VM flush when it is necessary.
A
Alex Deucher 已提交
284
 */
285 286 287 288 289
int amdgpu_vm_flush(struct amdgpu_ring *ring,
		    unsigned vm_id, uint64_t pd_addr,
		    uint32_t gds_base, uint32_t gds_size,
		    uint32_t gws_base, uint32_t gws_size,
		    uint32_t oa_base, uint32_t oa_size)
A
Alex Deucher 已提交
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 295 296 297 298 299
		id->gds_base != gds_base ||
		id->gds_size != gds_size ||
		id->gws_base != gws_base ||
		id->gws_size != gws_size ||
		id->oa_base != oa_base ||
		id->oa_size != oa_size);
300
	int r;
301 302 303 304

	if (ring->funcs->emit_pipeline_sync && (
	    pd_addr != AMDGPU_VM_NO_FLUSH || gds_switch_needed))
		amdgpu_ring_emit_pipeline_sync(ring);
305

306 307
	if (ring->funcs->emit_vm_flush &&
	    pd_addr != AMDGPU_VM_NO_FLUSH) {
308 309
		struct fence *fence;

310 311
		trace_amdgpu_vm_flush(pd_addr, ring->idx, vm_id);
		amdgpu_ring_emit_vm_flush(ring, vm_id, pd_addr);
312 313

		mutex_lock(&adev->vm_manager.lock);
314 315 316 317 318 319 320 321 322
		if ((id->pd_gpu_addr == pd_addr) && (id->last_user == ring)) {
			r = amdgpu_fence_emit(ring, &fence);
			if (r) {
				mutex_unlock(&adev->vm_manager.lock);
				return r;
			}
			fence_put(id->last_flush);
			id->last_flush = fence;
		}
323
		mutex_unlock(&adev->vm_manager.lock);
A
Alex Deucher 已提交
324
	}
325

326
	if (gds_switch_needed) {
327 328 329 330 331 332
		id->gds_base = gds_base;
		id->gds_size = gds_size;
		id->gws_base = gws_base;
		id->gws_size = gws_size;
		id->oa_base = oa_base;
		id->oa_size = oa_size;
333 334 335 336
		amdgpu_ring_emit_gds_switch(ring, vm_id,
					    gds_base, gds_size,
					    gws_base, gws_size,
					    oa_base, oa_size);
337
	}
338 339

	return 0;
340 341 342 343 344 345 346 347 348 349 350 351
}

/**
 * amdgpu_vm_reset_id - reset VMID to zero
 *
 * @adev: amdgpu device structure
 * @vm_id: vmid number to use
 *
 * Reset saved GDW, GWS and OA to force switch on next flush.
 */
void amdgpu_vm_reset_id(struct amdgpu_device *adev, unsigned vm_id)
{
352 353 354 355 356 357 358 359
	struct amdgpu_vm_id *id = &adev->vm_manager.ids[vm_id];

	id->gds_base = 0;
	id->gds_size = 0;
	id->gws_base = 0;
	id->gws_size = 0;
	id->oa_base = 0;
	id->oa_size = 0;
A
Alex Deucher 已提交
360 361 362 363 364 365 366 367
}

/**
 * amdgpu_vm_bo_find - find the bo_va for a specific vm & bo
 *
 * @vm: requested vm
 * @bo: requested buffer object
 *
368
 * Find @bo inside the requested vm.
A
Alex Deucher 已提交
369 370 371 372 373 374 375 376 377 378 379 380 381 382 383 384 385 386 387 388 389 390
 * Search inside the @bos vm list for the requested vm
 * Returns the found bo_va or NULL if none is found
 *
 * Object has to be reserved!
 */
struct amdgpu_bo_va *amdgpu_vm_bo_find(struct amdgpu_vm *vm,
				       struct amdgpu_bo *bo)
{
	struct amdgpu_bo_va *bo_va;

	list_for_each_entry(bo_va, &bo->va, bo_list) {
		if (bo_va->vm == vm) {
			return bo_va;
		}
	}
	return NULL;
}

/**
 * amdgpu_vm_update_pages - helper to call the right asic function
 *
 * @adev: amdgpu_device pointer
391 392
 * @src: address where to copy page table entries from
 * @pages_addr: DMA addresses to use for mapping
A
Alex Deucher 已提交
393 394 395 396 397 398 399 400 401 402 403
 * @ib: indirect buffer to fill with commands
 * @pe: addr of the page entry
 * @addr: dst addr to write into pe
 * @count: number of page entries to update
 * @incr: increase next addr by incr bytes
 * @flags: hw access flags
 *
 * Traces the parameters and calls the right asic functions
 * to setup the page table using the DMA.
 */
static void amdgpu_vm_update_pages(struct amdgpu_device *adev,
404 405
				   uint64_t src,
				   dma_addr_t *pages_addr,
A
Alex Deucher 已提交
406 407 408
				   struct amdgpu_ib *ib,
				   uint64_t pe, uint64_t addr,
				   unsigned count, uint32_t incr,
409
				   uint32_t flags)
A
Alex Deucher 已提交
410 411 412
{
	trace_amdgpu_vm_set_page(pe, addr, count, incr, flags);

413 414
	if (src) {
		src += (addr >> 12) * 8;
A
Alex Deucher 已提交
415 416
		amdgpu_vm_copy_pte(adev, ib, pe, src, count);

417
	} else if (pages_addr) {
418 419 420 421 422 423
		amdgpu_vm_write_pte(adev, ib, pages_addr, pe, addr,
				    count, incr, flags);

	} else if (count < 3) {
		amdgpu_vm_write_pte(adev, ib, NULL, pe, addr,
				    count, incr, flags);
A
Alex Deucher 已提交
424 425 426 427 428 429 430 431 432 433 434 435

	} else {
		amdgpu_vm_set_pte_pde(adev, ib, pe, addr,
				      count, incr, flags);
	}
}

/**
 * amdgpu_vm_clear_bo - initially clear the page dir/table
 *
 * @adev: amdgpu_device pointer
 * @bo: bo to clear
436 437
 *
 * need to reserve bo first before calling it.
A
Alex Deucher 已提交
438 439
 */
static int amdgpu_vm_clear_bo(struct amdgpu_device *adev,
440
			      struct amdgpu_vm *vm,
A
Alex Deucher 已提交
441 442
			      struct amdgpu_bo *bo)
{
443
	struct amdgpu_ring *ring;
444
	struct fence *fence = NULL;
445
	struct amdgpu_job *job;
A
Alex Deucher 已提交
446 447 448 449
	unsigned entries;
	uint64_t addr;
	int r;

450 451
	ring = container_of(vm->entity.sched, struct amdgpu_ring, sched);

M
monk.liu 已提交
452 453 454 455
	r = reservation_object_reserve_shared(bo->tbo.resv);
	if (r)
		return r;

A
Alex Deucher 已提交
456 457
	r = ttm_bo_validate(&bo->tbo, &bo->placement, true, false);
	if (r)
458
		goto error;
A
Alex Deucher 已提交
459 460 461 462

	addr = amdgpu_bo_gpu_offset(bo);
	entries = amdgpu_bo_size(bo) / 8;

463 464
	r = amdgpu_job_alloc_with_ib(adev, 64, &job);
	if (r)
465
		goto error;
A
Alex Deucher 已提交
466

467
	amdgpu_vm_update_pages(adev, 0, NULL, &job->ibs[0], addr, 0, entries,
468 469 470 471
			       0, 0);
	amdgpu_ring_pad_ib(ring, &job->ibs[0]);

	WARN_ON(job->ibs[0].length_dw > 64);
472 473
	r = amdgpu_job_submit(job, ring, &vm->entity,
			      AMDGPU_FENCE_OWNER_VM, &fence);
A
Alex Deucher 已提交
474 475 476
	if (r)
		goto error_free;

477
	amdgpu_bo_fence(bo, fence, true);
478
	fence_put(fence);
479
	return 0;
480

A
Alex Deucher 已提交
481
error_free:
482
	amdgpu_job_free(job);
A
Alex Deucher 已提交
483

484
error:
A
Alex Deucher 已提交
485 486 487 488
	return r;
}

/**
489
 * amdgpu_vm_map_gart - Resolve gart mapping of addr
A
Alex Deucher 已提交
490
 *
491
 * @pages_addr: optional DMA address to use for lookup
A
Alex Deucher 已提交
492 493 494
 * @addr: the unmapped addr
 *
 * Look up the physical address of the page that the pte resolves
495
 * to and return the pointer for the page table entry.
A
Alex Deucher 已提交
496
 */
497
uint64_t amdgpu_vm_map_gart(const dma_addr_t *pages_addr, uint64_t addr)
A
Alex Deucher 已提交
498 499 500
{
	uint64_t result;

501 502 503 504 505 506 507 508 509 510 511
	if (pages_addr) {
		/* page table offset */
		result = pages_addr[addr >> PAGE_SHIFT];

		/* in case cpu page size != gpu page size*/
		result |= addr & (~PAGE_MASK);

	} else {
		/* No mapping required */
		result = addr;
	}
A
Alex Deucher 已提交
512

513
	result &= 0xFFFFFFFFFFFFF000ULL;
A
Alex Deucher 已提交
514 515 516 517 518 519 520 521 522 523 524 525 526

	return result;
}

/**
 * amdgpu_vm_update_pdes - make sure that page directory is valid
 *
 * @adev: amdgpu_device pointer
 * @vm: requested vm
 * @start: start of GPU address range
 * @end: end of GPU address range
 *
 * Allocates new page tables if necessary
527
 * and updates the page directory.
A
Alex Deucher 已提交
528 529 530 531 532
 * Returns 0 for success, error for failure.
 */
int amdgpu_vm_update_page_directory(struct amdgpu_device *adev,
				    struct amdgpu_vm *vm)
{
533
	struct amdgpu_ring *ring;
A
Alex Deucher 已提交
534 535 536 537 538
	struct amdgpu_bo *pd = vm->page_directory;
	uint64_t pd_addr = amdgpu_bo_gpu_offset(pd);
	uint32_t incr = AMDGPU_VM_PTE_COUNT * 8;
	uint64_t last_pde = ~0, last_pt = ~0;
	unsigned count = 0, pt_idx, ndw;
539
	struct amdgpu_job *job;
C
Chunming Zhou 已提交
540
	struct amdgpu_ib *ib;
541
	struct fence *fence = NULL;
C
Chunming Zhou 已提交
542

A
Alex Deucher 已提交
543 544
	int r;

545 546
	ring = container_of(vm->entity.sched, struct amdgpu_ring, sched);

A
Alex Deucher 已提交
547 548 549 550 551 552
	/* padding, etc. */
	ndw = 64;

	/* assume the worst case */
	ndw += vm->max_pde_used * 6;

553 554
	r = amdgpu_job_alloc_with_ib(adev, ndw * 4, &job);
	if (r)
A
Alex Deucher 已提交
555
		return r;
556 557

	ib = &job->ibs[0];
A
Alex Deucher 已提交
558 559 560

	/* walk over the address space and update the page directory */
	for (pt_idx = 0; pt_idx <= vm->max_pde_used; ++pt_idx) {
561
		struct amdgpu_bo *bo = vm->page_tables[pt_idx].entry.robj;
A
Alex Deucher 已提交
562 563 564 565 566 567 568 569 570 571 572 573 574 575 576
		uint64_t pde, pt;

		if (bo == NULL)
			continue;

		pt = amdgpu_bo_gpu_offset(bo);
		if (vm->page_tables[pt_idx].addr == pt)
			continue;
		vm->page_tables[pt_idx].addr = pt;

		pde = pd_addr + pt_idx * 8;
		if (((last_pde + 8 * count) != pde) ||
		    ((last_pt + incr * count) != pt)) {

			if (count) {
577
				amdgpu_vm_update_pages(adev, 0, NULL, ib,
578 579 580
						       last_pde, last_pt,
						       count, incr,
						       AMDGPU_PTE_VALID);
A
Alex Deucher 已提交
581 582 583 584 585 586 587 588 589 590 591
			}

			count = 1;
			last_pde = pde;
			last_pt = pt;
		} else {
			++count;
		}
	}

	if (count)
592
		amdgpu_vm_update_pages(adev, 0, NULL, ib, last_pde, last_pt,
593
				       count, incr, AMDGPU_PTE_VALID);
A
Alex Deucher 已提交
594

C
Chunming Zhou 已提交
595
	if (ib->length_dw != 0) {
596
		amdgpu_ring_pad_ib(ring, ib);
597 598
		amdgpu_sync_resv(adev, &job->sync, pd->tbo.resv,
				 AMDGPU_FENCE_OWNER_VM);
C
Chunming Zhou 已提交
599
		WARN_ON(ib->length_dw > ndw);
600 601
		r = amdgpu_job_submit(job, ring, &vm->entity,
				      AMDGPU_FENCE_OWNER_VM, &fence);
602 603
		if (r)
			goto error_free;
604

605
		amdgpu_bo_fence(pd, fence, true);
606 607
		fence_put(vm->page_directory_fence);
		vm->page_directory_fence = fence_get(fence);
608
		fence_put(fence);
C
Chunming Zhou 已提交
609

610 611
	} else {
		amdgpu_job_free(job);
C
Chunming Zhou 已提交
612
	}
A
Alex Deucher 已提交
613 614

	return 0;
C
Chunming Zhou 已提交
615 616

error_free:
617
	amdgpu_job_free(job);
618
	return r;
A
Alex Deucher 已提交
619 620 621 622 623 624
}

/**
 * amdgpu_vm_frag_ptes - add fragment information to PTEs
 *
 * @adev: amdgpu_device pointer
625 626
 * @src: address where to copy page table entries from
 * @pages_addr: DMA addresses to use for mapping
A
Alex Deucher 已提交
627 628 629 630 631 632 633
 * @ib: IB for the update
 * @pe_start: first PTE to handle
 * @pe_end: last PTE to handle
 * @addr: addr those PTEs should point to
 * @flags: hw mapping flags
 */
static void amdgpu_vm_frag_ptes(struct amdgpu_device *adev,
634 635
				uint64_t src,
				dma_addr_t *pages_addr,
A
Alex Deucher 已提交
636 637
				struct amdgpu_ib *ib,
				uint64_t pe_start, uint64_t pe_end,
638
				uint64_t addr, uint32_t flags)
A
Alex Deucher 已提交
639 640 641 642 643 644 645 646 647 648 649 650 651 652 653 654 655 656 657 658 659 660 661 662 663 664 665 666 667
{
	/**
	 * The MC L1 TLB supports variable sized pages, based on a fragment
	 * field in the PTE. When this field is set to a non-zero value, page
	 * granularity is increased from 4KB to (1 << (12 + frag)). The PTE
	 * flags are considered valid for all PTEs within the fragment range
	 * and corresponding mappings are assumed to be physically contiguous.
	 *
	 * The L1 TLB can store a single PTE for the whole fragment,
	 * significantly increasing the space available for translation
	 * caching. This leads to large improvements in throughput when the
	 * TLB is under pressure.
	 *
	 * The L2 TLB distributes small and large fragments into two
	 * asymmetric partitions. The large fragment cache is significantly
	 * larger. Thus, we try to use large fragments wherever possible.
	 * Userspace can support this by aligning virtual base address and
	 * allocation size to the fragment size.
	 */

	/* SI and newer are optimized for 64KB */
	uint64_t frag_flags = AMDGPU_PTE_FRAG_64KB;
	uint64_t frag_align = 0x80;

	uint64_t frag_start = ALIGN(pe_start, frag_align);
	uint64_t frag_end = pe_end & ~(frag_align - 1);

	unsigned count;

668 669 670 671
	/* Abort early if there isn't anything to do */
	if (pe_start == pe_end)
		return;

A
Alex Deucher 已提交
672
	/* system pages are non continuously */
673 674
	if (src || pages_addr || !(flags & AMDGPU_PTE_VALID) ||
	    (frag_start >= frag_end)) {
A
Alex Deucher 已提交
675 676

		count = (pe_end - pe_start) / 8;
677
		amdgpu_vm_update_pages(adev, src, pages_addr, ib, pe_start,
678 679
				       addr, count, AMDGPU_GPU_PAGE_SIZE,
				       flags);
A
Alex Deucher 已提交
680 681 682 683 684 685
		return;
	}

	/* handle the 4K area at the beginning */
	if (pe_start != frag_start) {
		count = (frag_start - pe_start) / 8;
686
		amdgpu_vm_update_pages(adev, 0, NULL, ib, pe_start, addr,
687
				       count, AMDGPU_GPU_PAGE_SIZE, flags);
A
Alex Deucher 已提交
688 689 690 691 692
		addr += AMDGPU_GPU_PAGE_SIZE * count;
	}

	/* handle the area in the middle */
	count = (frag_end - frag_start) / 8;
693
	amdgpu_vm_update_pages(adev, 0, NULL, ib, frag_start, addr, count,
694
			       AMDGPU_GPU_PAGE_SIZE, flags | frag_flags);
A
Alex Deucher 已提交
695 696 697 698 699

	/* handle the 4K area at the end */
	if (frag_end != pe_end) {
		addr += AMDGPU_GPU_PAGE_SIZE * count;
		count = (pe_end - frag_end) / 8;
700
		amdgpu_vm_update_pages(adev, 0, NULL, ib, frag_end, addr,
701
				       count, AMDGPU_GPU_PAGE_SIZE, flags);
A
Alex Deucher 已提交
702 703 704 705 706 707 708
	}
}

/**
 * amdgpu_vm_update_ptes - make sure that page tables are valid
 *
 * @adev: amdgpu_device pointer
709 710
 * @src: address where to copy page table entries from
 * @pages_addr: DMA addresses to use for mapping
A
Alex Deucher 已提交
711 712 713 714 715 716
 * @vm: requested vm
 * @start: start of GPU address range
 * @end: end of GPU address range
 * @dst: destination address to map to
 * @flags: mapping flags
 *
717
 * Update the page tables in the range @start - @end.
A
Alex Deucher 已提交
718
 */
719
static void amdgpu_vm_update_ptes(struct amdgpu_device *adev,
720 721
				  uint64_t src,
				  dma_addr_t *pages_addr,
722 723 724 725
				  struct amdgpu_vm *vm,
				  struct amdgpu_ib *ib,
				  uint64_t start, uint64_t end,
				  uint64_t dst, uint32_t flags)
A
Alex Deucher 已提交
726
{
727 728 729
	const uint64_t mask = AMDGPU_VM_PTE_COUNT - 1;

	uint64_t last_pe_start = ~0, last_pe_end = ~0, last_dst = ~0;
A
Alex Deucher 已提交
730 731 732 733 734
	uint64_t addr;

	/* walk over the address space and update the page tables */
	for (addr = start; addr < end; ) {
		uint64_t pt_idx = addr >> amdgpu_vm_block_size;
735
		struct amdgpu_bo *pt = vm->page_tables[pt_idx].entry.robj;
A
Alex Deucher 已提交
736
		unsigned nptes;
737
		uint64_t pe_start;
A
Alex Deucher 已提交
738 739 740 741 742 743

		if ((addr & ~mask) == (end & ~mask))
			nptes = end - addr;
		else
			nptes = AMDGPU_VM_PTE_COUNT - (addr & mask);

744 745
		pe_start = amdgpu_bo_gpu_offset(pt);
		pe_start += (addr & mask) * 8;
A
Alex Deucher 已提交
746

747
		if (last_pe_end != pe_start) {
A
Alex Deucher 已提交
748

749
			amdgpu_vm_frag_ptes(adev, src, pages_addr, ib,
750 751
					    last_pe_start, last_pe_end,
					    last_dst, flags);
A
Alex Deucher 已提交
752

753 754
			last_pe_start = pe_start;
			last_pe_end = pe_start + 8 * nptes;
A
Alex Deucher 已提交
755 756
			last_dst = dst;
		} else {
757
			last_pe_end += 8 * nptes;
A
Alex Deucher 已提交
758 759 760 761 762 763
		}

		addr += nptes;
		dst += nptes * AMDGPU_GPU_PAGE_SIZE;
	}

764 765
	amdgpu_vm_frag_ptes(adev, src, pages_addr, ib, last_pe_start,
			    last_pe_end, last_dst, flags);
A
Alex Deucher 已提交
766 767 768 769 770 771
}

/**
 * amdgpu_vm_bo_update_mapping - update a mapping in the vm page table
 *
 * @adev: amdgpu_device pointer
772 773
 * @src: address where to copy page table entries from
 * @pages_addr: DMA addresses to use for mapping
A
Alex Deucher 已提交
774
 * @vm: requested vm
775 776 777
 * @start: start of mapped range
 * @last: last mapped entry
 * @flags: flags for the entries
A
Alex Deucher 已提交
778 779 780
 * @addr: addr to set the area to
 * @fence: optional resulting fence
 *
781
 * Fill in the page table entries between @start and @last.
A
Alex Deucher 已提交
782 783 784
 * Returns 0 for success, -EINVAL for failure.
 */
static int amdgpu_vm_bo_update_mapping(struct amdgpu_device *adev,
785 786
				       uint64_t src,
				       dma_addr_t *pages_addr,
A
Alex Deucher 已提交
787
				       struct amdgpu_vm *vm,
788 789 790
				       uint64_t start, uint64_t last,
				       uint32_t flags, uint64_t addr,
				       struct fence **fence)
A
Alex Deucher 已提交
791
{
792
	struct amdgpu_ring *ring;
793
	void *owner = AMDGPU_FENCE_OWNER_VM;
A
Alex Deucher 已提交
794
	unsigned nptes, ncmds, ndw;
795
	struct amdgpu_job *job;
C
Chunming Zhou 已提交
796
	struct amdgpu_ib *ib;
797
	struct fence *f = NULL;
A
Alex Deucher 已提交
798 799
	int r;

800 801
	ring = container_of(vm->entity.sched, struct amdgpu_ring, sched);

802 803 804 805
	/* sync to everything on unmapping */
	if (!(flags & AMDGPU_PTE_VALID))
		owner = AMDGPU_FENCE_OWNER_UNDEFINED;

806
	nptes = last - start + 1;
A
Alex Deucher 已提交
807 808 809 810 811 812 813 814 815 816

	/*
	 * reserve space for one command every (1 << BLOCK_SIZE)
	 *  entries or 2k dwords (whatever is smaller)
	 */
	ncmds = (nptes >> min(amdgpu_vm_block_size, 11)) + 1;

	/* padding, etc. */
	ndw = 64;

817
	if (src) {
A
Alex Deucher 已提交
818 819 820
		/* only copy commands needed */
		ndw += ncmds * 7;

821
	} else if (pages_addr) {
A
Alex Deucher 已提交
822 823 824 825 826 827 828 829 830 831 832 833 834 835
		/* header for write data commands */
		ndw += ncmds * 4;

		/* body of write data command */
		ndw += nptes * 2;

	} else {
		/* set page commands needed */
		ndw += ncmds * 10;

		/* two extra commands for begin/end of fragment */
		ndw += 2 * 10;
	}

836 837
	r = amdgpu_job_alloc_with_ib(adev, ndw * 4, &job);
	if (r)
A
Alex Deucher 已提交
838
		return r;
839 840

	ib = &job->ibs[0];
C
Chunming Zhou 已提交
841

842
	r = amdgpu_sync_resv(adev, &job->sync, vm->page_directory->tbo.resv,
843 844 845
			     owner);
	if (r)
		goto error_free;
A
Alex Deucher 已提交
846

847 848 849 850
	r = reservation_object_reserve_shared(vm->page_directory->tbo.resv);
	if (r)
		goto error_free;

851 852
	amdgpu_vm_update_ptes(adev, src, pages_addr, vm, ib, start,
			      last + 1, addr, flags);
A
Alex Deucher 已提交
853

854
	amdgpu_ring_pad_ib(ring, ib);
C
Chunming Zhou 已提交
855
	WARN_ON(ib->length_dw > ndw);
856 857
	r = amdgpu_job_submit(job, ring, &vm->entity,
			      AMDGPU_FENCE_OWNER_VM, &f);
858 859
	if (r)
		goto error_free;
A
Alex Deucher 已提交
860

861
	amdgpu_bo_fence(vm->page_directory, f, true);
862 863 864 865
	if (fence) {
		fence_put(*fence);
		*fence = fence_get(f);
	}
866
	fence_put(f);
A
Alex Deucher 已提交
867
	return 0;
C
Chunming Zhou 已提交
868 869

error_free:
870
	amdgpu_job_free(job);
871
	return r;
A
Alex Deucher 已提交
872 873
}

874 875 876 877
/**
 * amdgpu_vm_bo_split_mapping - split a mapping into smaller chunks
 *
 * @adev: amdgpu_device pointer
878 879
 * @gtt_flags: flags as they are used for GTT
 * @pages_addr: DMA addresses to use for mapping
880 881 882
 * @vm: requested vm
 * @mapping: mapped range and flags to use for the update
 * @addr: addr to set the area to
883
 * @flags: HW flags for the mapping
884 885 886 887 888 889 890 891
 * @fence: optional resulting fence
 *
 * Split the mapping into smaller chunks so that each update fits
 * into a SDMA IB.
 * Returns 0 for success, -EINVAL for failure.
 */
static int amdgpu_vm_bo_split_mapping(struct amdgpu_device *adev,
				      uint32_t gtt_flags,
892
				      dma_addr_t *pages_addr,
893 894
				      struct amdgpu_vm *vm,
				      struct amdgpu_bo_va_mapping *mapping,
895 896
				      uint32_t flags, uint64_t addr,
				      struct fence **fence)
897 898 899
{
	const uint64_t max_size = 64ULL * 1024ULL * 1024ULL / AMDGPU_GPU_PAGE_SIZE;

900
	uint64_t src = 0, start = mapping->it.start;
901 902 903 904 905 906 907 908 909 910 911 912
	int r;

	/* normally,bo_va->flags only contians READABLE and WIRTEABLE bit go here
	 * but in case of something, we filter the flags in first place
	 */
	if (!(mapping->flags & AMDGPU_PTE_READABLE))
		flags &= ~AMDGPU_PTE_READABLE;
	if (!(mapping->flags & AMDGPU_PTE_WRITEABLE))
		flags &= ~AMDGPU_PTE_WRITEABLE;

	trace_amdgpu_vm_bo_update(mapping);

913
	if (pages_addr) {
914 915 916 917
		if (flags == gtt_flags)
			src = adev->gart.table_addr + (addr >> 12) * 8;
		addr = 0;
	}
918 919
	addr += mapping->offset;

920
	if (!pages_addr || src)
921
		return amdgpu_vm_bo_update_mapping(adev, src, pages_addr, vm,
922 923 924 925 926 927
						   start, mapping->it.last,
						   flags, addr, fence);

	while (start != mapping->it.last + 1) {
		uint64_t last;

928
		last = min((uint64_t)mapping->it.last, start + max_size - 1);
929
		r = amdgpu_vm_bo_update_mapping(adev, src, pages_addr, vm,
930 931 932 933 934 935
						start, last, flags, addr,
						fence);
		if (r)
			return r;

		start = last + 1;
936
		addr += max_size * AMDGPU_GPU_PAGE_SIZE;
937 938 939 940 941
	}

	return 0;
}

A
Alex Deucher 已提交
942 943 944 945 946 947 948 949 950 951 952 953 954 955 956 957 958 959
/**
 * amdgpu_vm_bo_update - update all BO mappings in the vm page table
 *
 * @adev: amdgpu_device pointer
 * @bo_va: requested BO and VM object
 * @mem: ttm mem
 *
 * Fill in the page table entries for @bo_va.
 * Returns 0 for success, -EINVAL for failure.
 *
 * Object have to be reserved and mutex must be locked!
 */
int amdgpu_vm_bo_update(struct amdgpu_device *adev,
			struct amdgpu_bo_va *bo_va,
			struct ttm_mem_reg *mem)
{
	struct amdgpu_vm *vm = bo_va->vm;
	struct amdgpu_bo_va_mapping *mapping;
960
	dma_addr_t *pages_addr = NULL;
961
	uint32_t gtt_flags, flags;
A
Alex Deucher 已提交
962 963 964 965
	uint64_t addr;
	int r;

	if (mem) {
966 967
		struct ttm_dma_tt *ttm;

968
		addr = (u64)mem->start << PAGE_SHIFT;
969 970
		switch (mem->mem_type) {
		case TTM_PL_TT:
971 972 973
			ttm = container_of(bo_va->bo->tbo.ttm, struct
					   ttm_dma_tt, ttm);
			pages_addr = ttm->dma_address;
974 975 976
			break;

		case TTM_PL_VRAM:
A
Alex Deucher 已提交
977
			addr += adev->vm_manager.vram_base_offset;
978 979 980 981 982
			break;

		default:
			break;
		}
A
Alex Deucher 已提交
983 984 985 986 987
	} else {
		addr = 0;
	}

	flags = amdgpu_ttm_tt_pte_flags(adev, bo_va->bo->tbo.ttm, mem);
988
	gtt_flags = (adev == bo_va->bo->adev) ? flags : 0;
A
Alex Deucher 已提交
989

990 991 992 993 994 995
	spin_lock(&vm->status_lock);
	if (!list_empty(&bo_va->vm_status))
		list_splice_init(&bo_va->valids, &bo_va->invalids);
	spin_unlock(&vm->status_lock);

	list_for_each_entry(mapping, &bo_va->invalids, list) {
996 997 998
		r = amdgpu_vm_bo_split_mapping(adev, gtt_flags, pages_addr, vm,
					       mapping, flags, addr,
					       &bo_va->last_pt_update);
A
Alex Deucher 已提交
999 1000 1001 1002
		if (r)
			return r;
	}

1003 1004 1005 1006 1007 1008 1009 1010
	if (trace_amdgpu_vm_bo_mapping_enabled()) {
		list_for_each_entry(mapping, &bo_va->valids, list)
			trace_amdgpu_vm_bo_mapping(mapping);

		list_for_each_entry(mapping, &bo_va->invalids, list)
			trace_amdgpu_vm_bo_mapping(mapping);
	}

A
Alex Deucher 已提交
1011
	spin_lock(&vm->status_lock);
1012
	list_splice_init(&bo_va->invalids, &bo_va->valids);
A
Alex Deucher 已提交
1013
	list_del_init(&bo_va->vm_status);
1014 1015
	if (!mem)
		list_add(&bo_va->vm_status, &vm->cleared);
A
Alex Deucher 已提交
1016 1017 1018 1019 1020 1021 1022 1023 1024 1025 1026 1027 1028 1029 1030 1031 1032 1033 1034 1035 1036 1037 1038 1039 1040 1041
	spin_unlock(&vm->status_lock);

	return 0;
}

/**
 * amdgpu_vm_clear_freed - clear freed BOs in the PT
 *
 * @adev: amdgpu_device pointer
 * @vm: requested vm
 *
 * Make sure all freed BOs are cleared in the PT.
 * Returns 0 for success.
 *
 * PTs have to be reserved and mutex must be locked!
 */
int amdgpu_vm_clear_freed(struct amdgpu_device *adev,
			  struct amdgpu_vm *vm)
{
	struct amdgpu_bo_va_mapping *mapping;
	int r;

	while (!list_empty(&vm->freed)) {
		mapping = list_first_entry(&vm->freed,
			struct amdgpu_bo_va_mapping, list);
		list_del(&mapping->list);
1042

1043
		r = amdgpu_vm_bo_split_mapping(adev, 0, NULL, vm, mapping,
1044
					       0, 0, NULL);
A
Alex Deucher 已提交
1045 1046 1047 1048 1049 1050 1051 1052 1053 1054 1055 1056 1057 1058 1059 1060 1061 1062 1063 1064 1065
		kfree(mapping);
		if (r)
			return r;

	}
	return 0;

}

/**
 * amdgpu_vm_clear_invalids - clear invalidated BOs in the PT
 *
 * @adev: amdgpu_device pointer
 * @vm: requested vm
 *
 * Make sure all invalidated BOs are cleared in the PT.
 * Returns 0 for success.
 *
 * PTs have to be reserved and mutex must be locked!
 */
int amdgpu_vm_clear_invalids(struct amdgpu_device *adev,
1066
			     struct amdgpu_vm *vm, struct amdgpu_sync *sync)
A
Alex Deucher 已提交
1067
{
1068
	struct amdgpu_bo_va *bo_va = NULL;
1069
	int r = 0;
A
Alex Deucher 已提交
1070 1071 1072 1073 1074 1075

	spin_lock(&vm->status_lock);
	while (!list_empty(&vm->invalidated)) {
		bo_va = list_first_entry(&vm->invalidated,
			struct amdgpu_bo_va, vm_status);
		spin_unlock(&vm->status_lock);
1076

A
Alex Deucher 已提交
1077 1078 1079 1080 1081 1082 1083 1084
		r = amdgpu_vm_bo_update(adev, bo_va, NULL);
		if (r)
			return r;

		spin_lock(&vm->status_lock);
	}
	spin_unlock(&vm->status_lock);

1085
	if (bo_va)
1086
		r = amdgpu_sync_fence(adev, sync, bo_va->last_pt_update);
1087 1088

	return r;
A
Alex Deucher 已提交
1089 1090 1091 1092 1093 1094 1095 1096 1097
}

/**
 * amdgpu_vm_bo_add - add a bo to a specific vm
 *
 * @adev: amdgpu_device pointer
 * @vm: requested vm
 * @bo: amdgpu buffer object
 *
1098
 * Add @bo into the requested vm.
A
Alex Deucher 已提交
1099 1100 1101 1102 1103 1104 1105 1106 1107 1108 1109 1110 1111 1112 1113 1114 1115 1116 1117
 * Add @bo to the list of bos associated with the vm
 * Returns newly added bo_va or NULL for failure
 *
 * Object has to be reserved!
 */
struct amdgpu_bo_va *amdgpu_vm_bo_add(struct amdgpu_device *adev,
				      struct amdgpu_vm *vm,
				      struct amdgpu_bo *bo)
{
	struct amdgpu_bo_va *bo_va;

	bo_va = kzalloc(sizeof(struct amdgpu_bo_va), GFP_KERNEL);
	if (bo_va == NULL) {
		return NULL;
	}
	bo_va->vm = vm;
	bo_va->bo = bo;
	bo_va->ref_count = 1;
	INIT_LIST_HEAD(&bo_va->bo_list);
1118 1119
	INIT_LIST_HEAD(&bo_va->valids);
	INIT_LIST_HEAD(&bo_va->invalids);
A
Alex Deucher 已提交
1120
	INIT_LIST_HEAD(&bo_va->vm_status);
1121

A
Alex Deucher 已提交
1122 1123 1124 1125 1126 1127 1128 1129 1130 1131 1132 1133 1134 1135 1136 1137 1138
	list_add_tail(&bo_va->bo_list, &bo->va);

	return bo_va;
}

/**
 * amdgpu_vm_bo_map - map bo inside a vm
 *
 * @adev: amdgpu_device pointer
 * @bo_va: bo_va to store the address
 * @saddr: where to map the BO
 * @offset: requested offset in the BO
 * @flags: attributes of pages (read/write/valid/etc.)
 *
 * Add a mapping of the BO at the specefied addr into the VM.
 * Returns 0 for success, error for failure.
 *
1139
 * Object has to be reserved and unreserved outside!
A
Alex Deucher 已提交
1140 1141 1142 1143 1144 1145 1146 1147 1148 1149 1150 1151 1152
 */
int amdgpu_vm_bo_map(struct amdgpu_device *adev,
		     struct amdgpu_bo_va *bo_va,
		     uint64_t saddr, uint64_t offset,
		     uint64_t size, uint32_t flags)
{
	struct amdgpu_bo_va_mapping *mapping;
	struct amdgpu_vm *vm = bo_va->vm;
	struct interval_tree_node *it;
	unsigned last_pfn, pt_idx;
	uint64_t eaddr;
	int r;

1153 1154
	/* validate the parameters */
	if (saddr & AMDGPU_GPU_PAGE_MASK || offset & AMDGPU_GPU_PAGE_MASK ||
1155
	    size == 0 || size & AMDGPU_GPU_PAGE_MASK)
1156 1157
		return -EINVAL;

A
Alex Deucher 已提交
1158
	/* make sure object fit at this offset */
1159
	eaddr = saddr + size - 1;
1160
	if ((saddr >= eaddr) || (offset + size > amdgpu_bo_size(bo_va->bo)))
A
Alex Deucher 已提交
1161 1162 1163
		return -EINVAL;

	last_pfn = eaddr / AMDGPU_GPU_PAGE_SIZE;
1164 1165
	if (last_pfn >= adev->vm_manager.max_pfn) {
		dev_err(adev->dev, "va above limit (0x%08X >= 0x%08X)\n",
A
Alex Deucher 已提交
1166 1167 1168 1169 1170 1171 1172
			last_pfn, adev->vm_manager.max_pfn);
		return -EINVAL;
	}

	saddr /= AMDGPU_GPU_PAGE_SIZE;
	eaddr /= AMDGPU_GPU_PAGE_SIZE;

1173
	it = interval_tree_iter_first(&vm->va, saddr, eaddr);
A
Alex Deucher 已提交
1174 1175 1176 1177 1178 1179 1180 1181
	if (it) {
		struct amdgpu_bo_va_mapping *tmp;
		tmp = container_of(it, struct amdgpu_bo_va_mapping, it);
		/* bo and tmp overlap, invalid addr */
		dev_err(adev->dev, "bo %p va 0x%010Lx-0x%010Lx conflict with "
			"0x%010lx-0x%010lx\n", bo_va->bo, saddr, eaddr,
			tmp->it.start, tmp->it.last + 1);
		r = -EINVAL;
1182
		goto error;
A
Alex Deucher 已提交
1183 1184 1185 1186 1187
	}

	mapping = kmalloc(sizeof(*mapping), GFP_KERNEL);
	if (!mapping) {
		r = -ENOMEM;
1188
		goto error;
A
Alex Deucher 已提交
1189 1190 1191 1192
	}

	INIT_LIST_HEAD(&mapping->list);
	mapping->it.start = saddr;
1193
	mapping->it.last = eaddr;
A
Alex Deucher 已提交
1194 1195 1196
	mapping->offset = offset;
	mapping->flags = flags;

1197
	list_add(&mapping->list, &bo_va->invalids);
A
Alex Deucher 已提交
1198 1199 1200 1201 1202 1203 1204 1205 1206 1207 1208 1209 1210
	interval_tree_insert(&mapping->it, &vm->va);

	/* Make sure the page tables are allocated */
	saddr >>= amdgpu_vm_block_size;
	eaddr >>= amdgpu_vm_block_size;

	BUG_ON(eaddr >= amdgpu_vm_num_pdes(adev));

	if (eaddr > vm->max_pde_used)
		vm->max_pde_used = eaddr;

	/* walk over the address space and allocate the page tables */
	for (pt_idx = saddr; pt_idx <= eaddr; ++pt_idx) {
1211
		struct reservation_object *resv = vm->page_directory->tbo.resv;
1212
		struct amdgpu_bo_list_entry *entry;
A
Alex Deucher 已提交
1213 1214
		struct amdgpu_bo *pt;

1215 1216
		entry = &vm->page_tables[pt_idx].entry;
		if (entry->robj)
A
Alex Deucher 已提交
1217 1218 1219 1220
			continue;

		r = amdgpu_bo_create(adev, AMDGPU_VM_PTE_COUNT * 8,
				     AMDGPU_GPU_PAGE_SIZE, true,
1221 1222
				     AMDGPU_GEM_DOMAIN_VRAM,
				     AMDGPU_GEM_CREATE_NO_CPU_ACCESS,
1223
				     NULL, resv, &pt);
1224
		if (r)
A
Alex Deucher 已提交
1225
			goto error_free;
1226

1227 1228 1229 1230 1231
		/* Keep a reference to the page table to avoid freeing
		 * them up in the wrong order.
		 */
		pt->parent = amdgpu_bo_ref(vm->page_directory);

1232
		r = amdgpu_vm_clear_bo(adev, vm, pt);
A
Alex Deucher 已提交
1233 1234 1235 1236 1237
		if (r) {
			amdgpu_bo_unref(&pt);
			goto error_free;
		}

1238 1239 1240 1241
		entry->robj = pt;
		entry->priority = 0;
		entry->tv.bo = &entry->robj->tbo;
		entry->tv.shared = true;
1242
		entry->user_pages = NULL;
A
Alex Deucher 已提交
1243 1244 1245 1246 1247 1248 1249 1250
		vm->page_tables[pt_idx].addr = 0;
	}

	return 0;

error_free:
	list_del(&mapping->list);
	interval_tree_remove(&mapping->it, &vm->va);
1251
	trace_amdgpu_vm_bo_unmap(bo_va, mapping);
A
Alex Deucher 已提交
1252 1253
	kfree(mapping);

1254
error:
A
Alex Deucher 已提交
1255 1256 1257 1258 1259 1260 1261 1262 1263 1264 1265 1266 1267
	return r;
}

/**
 * amdgpu_vm_bo_unmap - remove bo mapping from vm
 *
 * @adev: amdgpu_device pointer
 * @bo_va: bo_va to remove the address from
 * @saddr: where to the BO is mapped
 *
 * Remove a mapping of the BO at the specefied addr from the VM.
 * Returns 0 for success, error for failure.
 *
1268
 * Object has to be reserved and unreserved outside!
A
Alex Deucher 已提交
1269 1270 1271 1272 1273 1274 1275
 */
int amdgpu_vm_bo_unmap(struct amdgpu_device *adev,
		       struct amdgpu_bo_va *bo_va,
		       uint64_t saddr)
{
	struct amdgpu_bo_va_mapping *mapping;
	struct amdgpu_vm *vm = bo_va->vm;
1276
	bool valid = true;
A
Alex Deucher 已提交
1277

1278
	saddr /= AMDGPU_GPU_PAGE_SIZE;
1279

1280
	list_for_each_entry(mapping, &bo_va->valids, list) {
A
Alex Deucher 已提交
1281 1282 1283 1284
		if (mapping->it.start == saddr)
			break;
	}

1285 1286 1287 1288 1289 1290 1291 1292
	if (&mapping->list == &bo_va->valids) {
		valid = false;

		list_for_each_entry(mapping, &bo_va->invalids, list) {
			if (mapping->it.start == saddr)
				break;
		}

1293
		if (&mapping->list == &bo_va->invalids)
1294
			return -ENOENT;
A
Alex Deucher 已提交
1295
	}
1296

A
Alex Deucher 已提交
1297 1298
	list_del(&mapping->list);
	interval_tree_remove(&mapping->it, &vm->va);
1299
	trace_amdgpu_vm_bo_unmap(bo_va, mapping);
A
Alex Deucher 已提交
1300

1301
	if (valid)
A
Alex Deucher 已提交
1302
		list_add(&mapping->list, &vm->freed);
1303
	else
A
Alex Deucher 已提交
1304 1305 1306 1307 1308 1309 1310 1311 1312 1313 1314
		kfree(mapping);

	return 0;
}

/**
 * amdgpu_vm_bo_rmv - remove a bo to a specific vm
 *
 * @adev: amdgpu_device pointer
 * @bo_va: requested bo_va
 *
1315
 * Remove @bo_va->bo from the requested vm.
A
Alex Deucher 已提交
1316 1317 1318 1319 1320 1321 1322 1323 1324 1325 1326 1327 1328 1329 1330
 *
 * Object have to be reserved!
 */
void amdgpu_vm_bo_rmv(struct amdgpu_device *adev,
		      struct amdgpu_bo_va *bo_va)
{
	struct amdgpu_bo_va_mapping *mapping, *next;
	struct amdgpu_vm *vm = bo_va->vm;

	list_del(&bo_va->bo_list);

	spin_lock(&vm->status_lock);
	list_del(&bo_va->vm_status);
	spin_unlock(&vm->status_lock);

1331
	list_for_each_entry_safe(mapping, next, &bo_va->valids, list) {
A
Alex Deucher 已提交
1332 1333
		list_del(&mapping->list);
		interval_tree_remove(&mapping->it, &vm->va);
1334
		trace_amdgpu_vm_bo_unmap(bo_va, mapping);
1335 1336 1337 1338 1339 1340
		list_add(&mapping->list, &vm->freed);
	}
	list_for_each_entry_safe(mapping, next, &bo_va->invalids, list) {
		list_del(&mapping->list);
		interval_tree_remove(&mapping->it, &vm->va);
		kfree(mapping);
A
Alex Deucher 已提交
1341
	}
1342

1343
	fence_put(bo_va->last_pt_update);
A
Alex Deucher 已提交
1344 1345 1346 1347 1348 1349 1350 1351 1352 1353
	kfree(bo_va);
}

/**
 * amdgpu_vm_bo_invalidate - mark the bo as invalid
 *
 * @adev: amdgpu_device pointer
 * @vm: requested vm
 * @bo: amdgpu buffer object
 *
1354
 * Mark @bo as invalid.
A
Alex Deucher 已提交
1355 1356 1357 1358 1359 1360 1361
 */
void amdgpu_vm_bo_invalidate(struct amdgpu_device *adev,
			     struct amdgpu_bo *bo)
{
	struct amdgpu_bo_va *bo_va;

	list_for_each_entry(bo_va, &bo->va, bo_list) {
1362 1363
		spin_lock(&bo_va->vm->status_lock);
		if (list_empty(&bo_va->vm_status))
A
Alex Deucher 已提交
1364
			list_add(&bo_va->vm_status, &bo_va->vm->invalidated);
1365
		spin_unlock(&bo_va->vm->status_lock);
A
Alex Deucher 已提交
1366 1367 1368 1369 1370 1371 1372 1373 1374
	}
}

/**
 * amdgpu_vm_init - initialize a vm instance
 *
 * @adev: amdgpu_device pointer
 * @vm: requested vm
 *
1375
 * Init @vm fields.
A
Alex Deucher 已提交
1376 1377 1378 1379 1380
 */
int amdgpu_vm_init(struct amdgpu_device *adev, struct amdgpu_vm *vm)
{
	const unsigned align = min(AMDGPU_VM_PTB_ALIGN_SIZE,
		AMDGPU_VM_PTE_COUNT * 8);
1381
	unsigned pd_size, pd_entries;
1382 1383
	unsigned ring_instance;
	struct amdgpu_ring *ring;
1384
	struct amd_sched_rq *rq;
A
Alex Deucher 已提交
1385 1386
	int i, r;

1387 1388
	for (i = 0; i < AMDGPU_MAX_RINGS; ++i)
		vm->ids[i] = NULL;
A
Alex Deucher 已提交
1389
	vm->va = RB_ROOT;
1390
	vm->client_id = atomic64_inc_return(&adev->vm_manager.client_counter);
A
Alex Deucher 已提交
1391 1392
	spin_lock_init(&vm->status_lock);
	INIT_LIST_HEAD(&vm->invalidated);
1393
	INIT_LIST_HEAD(&vm->cleared);
A
Alex Deucher 已提交
1394
	INIT_LIST_HEAD(&vm->freed);
1395

A
Alex Deucher 已提交
1396 1397 1398 1399
	pd_size = amdgpu_vm_directory_size(adev);
	pd_entries = amdgpu_vm_num_pdes(adev);

	/* allocate page table array */
1400
	vm->page_tables = drm_calloc_large(pd_entries, sizeof(struct amdgpu_vm_pt));
A
Alex Deucher 已提交
1401 1402 1403 1404 1405
	if (vm->page_tables == NULL) {
		DRM_ERROR("Cannot allocate memory for page table array\n");
		return -ENOMEM;
	}

1406
	/* create scheduler entity for page table updates */
1407 1408 1409 1410

	ring_instance = atomic_inc_return(&adev->vm_manager.vm_pte_next_ring);
	ring_instance %= adev->vm_manager.vm_pte_num_rings;
	ring = adev->vm_manager.vm_pte_rings[ring_instance];
1411 1412 1413 1414 1415 1416
	rq = &ring->sched.sched_rq[AMD_SCHED_PRIORITY_KERNEL];
	r = amd_sched_entity_init(&ring->sched, &vm->entity,
				  rq, amdgpu_sched_jobs);
	if (r)
		return r;

1417 1418
	vm->page_directory_fence = NULL;

A
Alex Deucher 已提交
1419
	r = amdgpu_bo_create(adev, pd_size, align, true,
1420 1421
			     AMDGPU_GEM_DOMAIN_VRAM,
			     AMDGPU_GEM_CREATE_NO_CPU_ACCESS,
1422
			     NULL, NULL, &vm->page_directory);
A
Alex Deucher 已提交
1423
	if (r)
1424 1425
		goto error_free_sched_entity;

1426
	r = amdgpu_bo_reserve(vm->page_directory, false);
1427 1428 1429 1430
	if (r)
		goto error_free_page_directory;

	r = amdgpu_vm_clear_bo(adev, vm, vm->page_directory);
1431
	amdgpu_bo_unreserve(vm->page_directory);
1432 1433
	if (r)
		goto error_free_page_directory;
A
Alex Deucher 已提交
1434 1435

	return 0;
1436 1437 1438 1439 1440 1441 1442 1443 1444

error_free_page_directory:
	amdgpu_bo_unref(&vm->page_directory);
	vm->page_directory = NULL;

error_free_sched_entity:
	amd_sched_entity_fini(&ring->sched, &vm->entity);

	return r;
A
Alex Deucher 已提交
1445 1446 1447 1448 1449 1450 1451 1452
}

/**
 * amdgpu_vm_fini - tear down a vm instance
 *
 * @adev: amdgpu_device pointer
 * @vm: requested vm
 *
1453
 * Tear down @vm.
A
Alex Deucher 已提交
1454 1455 1456 1457 1458
 * Unbind the VM and remove all bos from the vm bo list
 */
void amdgpu_vm_fini(struct amdgpu_device *adev, struct amdgpu_vm *vm)
{
	struct amdgpu_bo_va_mapping *mapping, *tmp;
1459
	struct amdgpu_vm_id *id, *id_tmp;
A
Alex Deucher 已提交
1460 1461
	int i;

1462
	amd_sched_entity_fini(vm->entity.sched, &vm->entity);
1463

A
Alex Deucher 已提交
1464 1465 1466 1467 1468 1469 1470 1471 1472 1473 1474 1475 1476 1477
	if (!RB_EMPTY_ROOT(&vm->va)) {
		dev_err(adev->dev, "still active bo inside vm\n");
	}
	rbtree_postorder_for_each_entry_safe(mapping, tmp, &vm->va, it.rb) {
		list_del(&mapping->list);
		interval_tree_remove(&mapping->it, &vm->va);
		kfree(mapping);
	}
	list_for_each_entry_safe(mapping, tmp, &vm->freed, list) {
		list_del(&mapping->list);
		kfree(mapping);
	}

	for (i = 0; i < amdgpu_vm_num_pdes(adev); i++)
1478
		amdgpu_bo_unref(&vm->page_tables[i].entry.robj);
1479
	drm_free_large(vm->page_tables);
A
Alex Deucher 已提交
1480 1481

	amdgpu_bo_unref(&vm->page_directory);
1482
	fence_put(vm->page_directory_fence);
1483

1484 1485 1486
	mutex_lock(&adev->vm_manager.lock);
	list_for_each_entry_safe(id, id_tmp, &adev->vm_manager.ids_lru,
				 list) {
1487 1488
		if (!id)
			continue;
1489
		if (atomic_long_read(&id->owner) == vm->client_id) {
1490 1491 1492
			atomic_long_set(&id->owner, 0);
			id->pd_gpu_addr = 0;
		}
A
Alex Deucher 已提交
1493
	}
1494
	mutex_unlock(&adev->vm_manager.lock);
A
Alex Deucher 已提交
1495
}
1496

1497 1498 1499 1500 1501 1502 1503 1504 1505 1506 1507 1508 1509 1510
/**
 * amdgpu_vm_manager_init - init the VM manager
 *
 * @adev: amdgpu_device pointer
 *
 * Initialize the VM manager structures
 */
void amdgpu_vm_manager_init(struct amdgpu_device *adev)
{
	unsigned i;

	INIT_LIST_HEAD(&adev->vm_manager.ids_lru);

	/* skip over VMID 0, since it is the system VM */
1511 1512
	for (i = 1; i < adev->vm_manager.num_ids; ++i) {
		amdgpu_vm_reset_id(adev, i);
1513
		amdgpu_sync_create(&adev->vm_manager.ids[i].active);
1514 1515
		list_add_tail(&adev->vm_manager.ids[i].list,
			      &adev->vm_manager.ids_lru);
1516
	}
1517 1518

	atomic_set(&adev->vm_manager.vm_pte_next_ring, 0);
1519
	atomic64_set(&adev->vm_manager.client_counter, AMDGPU_CLIENT_ID_RESERVED);
1520 1521
}

1522 1523 1524 1525 1526 1527 1528 1529 1530 1531 1532
/**
 * amdgpu_vm_manager_fini - cleanup VM manager
 *
 * @adev: amdgpu_device pointer
 *
 * Cleanup the VM manager and free resources.
 */
void amdgpu_vm_manager_fini(struct amdgpu_device *adev)
{
	unsigned i;

1533 1534 1535
	for (i = 0; i < AMDGPU_NUM_VM; ++i) {
		struct amdgpu_vm_id *id = &adev->vm_manager.ids[i];

1536 1537
		fence_put(adev->vm_manager.ids[i].first);
		amdgpu_sync_free(&adev->vm_manager.ids[i].active);
1538 1539
		fence_put(id->flushed_updates);
	}
1540
}