amdgpu_vm.c 40.0 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
/*
 * 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
 */
28
#include <linux/fence-array.h>
A
Alex Deucher 已提交
29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53
#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.
 */

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

57 58 59 60 61 62 63 64 65 66 67 68
/* Local structure. Encapsulate some VM table update parameters to reduce
 * the number of function parameters
 */
struct amdgpu_vm_update_params {
	/* address where to copy page table entries from */
	uint64_t src;
	/* DMA addresses to use for mapping */
	dma_addr_t *pages_addr;
	/* indirect buffer to fill with commands */
	struct amdgpu_ib *ib;
};

A
Alex Deucher 已提交
69 70 71 72 73
/**
 * amdgpu_vm_num_pde - return the number of page directory entries
 *
 * @adev: amdgpu_device pointer
 *
74
 * Calculate the number of page directory entries.
A
Alex Deucher 已提交
75 76 77 78 79 80 81 82 83 84 85
 */
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
 *
86
 * Calculate the size of the page directory in bytes.
A
Alex Deucher 已提交
87 88 89 90 91 92 93
 */
static unsigned amdgpu_vm_directory_size(struct amdgpu_device *adev)
{
	return AMDGPU_GPU_PAGE_ALIGN(amdgpu_vm_num_pdes(adev) * 8);
}

/**
94
 * amdgpu_vm_get_pd_bo - add the VM PD to a validation list
A
Alex Deucher 已提交
95 96
 *
 * @vm: vm providing the BOs
97
 * @validated: head of validation list
98
 * @entry: entry to add
A
Alex Deucher 已提交
99 100
 *
 * Add the page directory to the list of BOs to
101
 * validate for command submission.
A
Alex Deucher 已提交
102
 */
103 104 105
void amdgpu_vm_get_pd_bo(struct amdgpu_vm *vm,
			 struct list_head *validated,
			 struct amdgpu_bo_list_entry *entry)
A
Alex Deucher 已提交
106
{
107 108 109 110
	entry->robj = vm->page_directory;
	entry->priority = 0;
	entry->tv.bo = &vm->page_directory->tbo;
	entry->tv.shared = true;
111
	entry->user_pages = NULL;
112 113
	list_add(&entry->tv.head, validated);
}
A
Alex Deucher 已提交
114

115
/**
116
 * amdgpu_vm_get_bos - add the vm BOs to a duplicates list
117 118
 *
 * @vm: vm providing the BOs
119
 * @duplicates: head of duplicates list
A
Alex Deucher 已提交
120
 *
121 122
 * Add the page directory to the BO duplicates list
 * for command submission.
A
Alex Deucher 已提交
123
 */
124
void amdgpu_vm_get_pt_bos(struct amdgpu_vm *vm, struct list_head *duplicates)
A
Alex Deucher 已提交
125
{
126
	unsigned i;
A
Alex Deucher 已提交
127 128

	/* add the vm page table to the list */
129 130 131 132
	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 已提交
133 134
			continue;

135
		list_add(&entry->tv.head, duplicates);
A
Alex Deucher 已提交
136
	}
137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163

}

/**
 * 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 已提交
164 165 166 167 168 169
}

/**
 * amdgpu_vm_grab_id - allocate the next free VMID
 *
 * @vm: vm to allocate id for
170 171
 * @ring: ring we want to submit job to
 * @sync: sync object where we add dependencies
172
 * @fence: fence protecting ID from reuse
A
Alex Deucher 已提交
173
 *
174
 * Allocate an id for the vm, adding fences to the sync obj as necessary.
A
Alex Deucher 已提交
175
 */
176
int amdgpu_vm_grab_id(struct amdgpu_vm *vm, struct amdgpu_ring *ring,
177 178
		      struct amdgpu_sync *sync, struct fence *fence,
		      unsigned *vm_id, uint64_t *vm_pd_addr)
A
Alex Deucher 已提交
179
{
180
	uint64_t pd_addr = amdgpu_bo_gpu_offset(vm->page_directory);
A
Alex Deucher 已提交
181
	struct amdgpu_device *adev = ring->adev;
182
	struct fence *updates = sync->last_vm_update;
183
	struct amdgpu_vm_id *id, *idle;
184 185 186 187 188 189 190 191
	struct fence **fences;
	unsigned i;
	int r = 0;

	fences = kmalloc_array(sizeof(void *), adev->vm_manager.num_ids,
			       GFP_KERNEL);
	if (!fences)
		return -ENOMEM;
A
Alex Deucher 已提交
192

193 194
	mutex_lock(&adev->vm_manager.lock);

195
	/* Check if we have an idle VMID */
196
	i = 0;
197
	list_for_each_entry(idle, &adev->vm_manager.ids_lru, list) {
198 199
		fences[i] = amdgpu_sync_peek_fence(&idle->active, ring);
		if (!fences[i])
200
			break;
201
		++i;
202 203
	}

204
	/* If we can't find a idle VMID to use, wait till one becomes available */
205
	if (&idle->list == &adev->vm_manager.ids_lru) {
206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239
		u64 fence_context = adev->vm_manager.fence_context + ring->idx;
		unsigned seqno = ++adev->vm_manager.seqno[ring->idx];
		struct fence_array *array;
		unsigned j;

		for (j = 0; j < i; ++j)
			fence_get(fences[j]);

		array = fence_array_create(i, fences, fence_context,
					   seqno, true);
		if (!array) {
			for (j = 0; j < i; ++j)
				fence_put(fences[j]);
			kfree(fences);
			r = -ENOMEM;
			goto error;
		}


		r = amdgpu_sync_fence(ring->adev, sync, &array->base);
		fence_put(&array->base);
		if (r)
			goto error;

		mutex_unlock(&adev->vm_manager.lock);
		return 0;

	}
	kfree(fences);

	/* Check if we can use a VMID already assigned to this VM */
	i = ring->idx;
	do {
		struct fence *flushed;
240
		bool same_ring = ring->idx == i;
241 242 243 244

		id = vm->ids[i++];
		if (i == AMDGPU_MAX_RINGS)
			i = 0;
245

246 247 248 249 250 251 252 253 254 255
		/* Check all the prerequisites to using this VMID */
		if (!id)
			continue;

		if (atomic64_read(&id->owner) != vm->client_id)
			continue;

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

256
		if (!same_ring &&
257 258 259 260 261 262 263 264
		    (!id->last_flush || !fence_is_signaled(id->last_flush)))
			continue;

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

265 266 267
		/* Good we can use this VMID. Remember this submission as
		 * user of the VMID.
		 */
268 269 270
		r = amdgpu_sync_fence(ring->adev, &id->active, fence);
		if (r)
			goto error;
271

272 273
		list_move_tail(&id->list, &adev->vm_manager.ids_lru);
		vm->ids[ring->idx] = id;
274

275 276 277
		*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);
278

279 280
		mutex_unlock(&adev->vm_manager.lock);
		return 0;
281

282
	} while (i != ring->idx);
283

284 285
	/* Still no ID to use? Then use the idle one found earlier */
	id = idle;
286

287 288
	/* Remember this submission as user of the VMID */
	r = amdgpu_sync_fence(ring->adev, &id->active, fence);
289 290
	if (r)
		goto error;
291

292 293
	fence_put(id->first);
	id->first = fence_get(fence);
294

295 296 297
	fence_put(id->last_flush);
	id->last_flush = NULL;

298 299
	fence_put(id->flushed_updates);
	id->flushed_updates = fence_get(updates);
300

301
	id->pd_gpu_addr = pd_addr;
302

303
	list_move_tail(&id->list, &adev->vm_manager.ids_lru);
304
	atomic64_set(&id->owner, vm->client_id);
305
	vm->ids[ring->idx] = id;
A
Alex Deucher 已提交
306

307 308 309 310 311
	*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:
312
	mutex_unlock(&adev->vm_manager.lock);
313
	return r;
A
Alex Deucher 已提交
314 315 316 317 318 319
}

/**
 * amdgpu_vm_flush - hardware flush the vm
 *
 * @ring: ring to use for flush
320
 * @vm_id: vmid number to use
321
 * @pd_addr: address of the page directory
A
Alex Deucher 已提交
322
 *
323
 * Emit a VM flush when it is necessary.
A
Alex Deucher 已提交
324
 */
325 326 327 328 329
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 已提交
330
{
331
	struct amdgpu_device *adev = ring->adev;
332
	struct amdgpu_vm_id *id = &adev->vm_manager.ids[vm_id];
333
	bool gds_switch_needed = ring->funcs->emit_gds_switch && (
334 335 336 337 338 339
		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);
340
	int r;
341 342

	if (ring->funcs->emit_pipeline_sync && (
343 344
	    pd_addr != AMDGPU_VM_NO_FLUSH || gds_switch_needed ||
		    ring->type == AMDGPU_RING_TYPE_COMPUTE))
345
		amdgpu_ring_emit_pipeline_sync(ring);
346

347 348
	if (ring->funcs->emit_vm_flush &&
	    pd_addr != AMDGPU_VM_NO_FLUSH) {
349 350
		struct fence *fence;

351 352
		trace_amdgpu_vm_flush(pd_addr, ring->idx, vm_id);
		amdgpu_ring_emit_vm_flush(ring, vm_id, pd_addr);
353

354 355 356 357
		r = amdgpu_fence_emit(ring, &fence);
		if (r)
			return r;

358
		mutex_lock(&adev->vm_manager.lock);
359 360
		fence_put(id->last_flush);
		id->last_flush = fence;
361
		mutex_unlock(&adev->vm_manager.lock);
A
Alex Deucher 已提交
362
	}
363

364
	if (gds_switch_needed) {
365 366 367 368 369 370
		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;
371 372 373 374
		amdgpu_ring_emit_gds_switch(ring, vm_id,
					    gds_base, gds_size,
					    gws_base, gws_size,
					    oa_base, oa_size);
375
	}
376 377

	return 0;
378 379 380 381 382 383 384 385 386 387 388 389
}

/**
 * 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)
{
390 391 392 393 394 395 396 397
	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 已提交
398 399 400 401 402 403 404 405
}

/**
 * amdgpu_vm_bo_find - find the bo_va for a specific vm & bo
 *
 * @vm: requested vm
 * @bo: requested buffer object
 *
406
 * Find @bo inside the requested vm.
A
Alex Deucher 已提交
407 408 409 410 411 412 413 414 415 416 417 418 419 420 421 422 423 424 425 426 427 428
 * 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
429
 * @vm_update_params: see amdgpu_vm_update_params definition
A
Alex Deucher 已提交
430 431 432 433 434 435 436 437 438 439
 * @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,
440 441
				   struct amdgpu_vm_update_params
					*vm_update_params,
A
Alex Deucher 已提交
442 443
				   uint64_t pe, uint64_t addr,
				   unsigned count, uint32_t incr,
444
				   uint32_t flags)
A
Alex Deucher 已提交
445 446 447
{
	trace_amdgpu_vm_set_page(pe, addr, count, incr, flags);

448 449 450
	if (vm_update_params->src) {
		amdgpu_vm_copy_pte(adev, vm_update_params->ib,
			pe, (vm_update_params->src + (addr >> 12) * 8), count);
A
Alex Deucher 已提交
451

452 453 454 455
	} else if (vm_update_params->pages_addr) {
		amdgpu_vm_write_pte(adev, vm_update_params->ib,
			vm_update_params->pages_addr,
			pe, addr, count, incr, flags);
456 457

	} else if (count < 3) {
458
		amdgpu_vm_write_pte(adev, vm_update_params->ib, NULL, pe, addr,
459
				    count, incr, flags);
A
Alex Deucher 已提交
460 461

	} else {
462
		amdgpu_vm_set_pte_pde(adev, vm_update_params->ib, pe, addr,
A
Alex Deucher 已提交
463 464 465 466 467 468 469 470 471
				      count, incr, flags);
	}
}

/**
 * amdgpu_vm_clear_bo - initially clear the page dir/table
 *
 * @adev: amdgpu_device pointer
 * @bo: bo to clear
472 473
 *
 * need to reserve bo first before calling it.
A
Alex Deucher 已提交
474 475
 */
static int amdgpu_vm_clear_bo(struct amdgpu_device *adev,
476
			      struct amdgpu_vm *vm,
A
Alex Deucher 已提交
477 478
			      struct amdgpu_bo *bo)
{
479
	struct amdgpu_ring *ring;
480
	struct fence *fence = NULL;
481
	struct amdgpu_job *job;
482
	struct amdgpu_vm_update_params vm_update_params;
A
Alex Deucher 已提交
483 484 485 486
	unsigned entries;
	uint64_t addr;
	int r;

487
	memset(&vm_update_params, 0, sizeof(vm_update_params));
488 489
	ring = container_of(vm->entity.sched, struct amdgpu_ring, sched);

M
monk.liu 已提交
490 491 492 493
	r = reservation_object_reserve_shared(bo->tbo.resv);
	if (r)
		return r;

A
Alex Deucher 已提交
494 495
	r = ttm_bo_validate(&bo->tbo, &bo->placement, true, false);
	if (r)
496
		goto error;
A
Alex Deucher 已提交
497 498 499 500

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

501 502
	r = amdgpu_job_alloc_with_ib(adev, 64, &job);
	if (r)
503
		goto error;
A
Alex Deucher 已提交
504

505 506
	vm_update_params.ib = &job->ibs[0];
	amdgpu_vm_update_pages(adev, &vm_update_params, addr, 0, entries,
507 508 509 510
			       0, 0);
	amdgpu_ring_pad_ib(ring, &job->ibs[0]);

	WARN_ON(job->ibs[0].length_dw > 64);
511 512
	r = amdgpu_job_submit(job, ring, &vm->entity,
			      AMDGPU_FENCE_OWNER_VM, &fence);
A
Alex Deucher 已提交
513 514 515
	if (r)
		goto error_free;

516
	amdgpu_bo_fence(bo, fence, true);
517
	fence_put(fence);
518
	return 0;
519

A
Alex Deucher 已提交
520
error_free:
521
	amdgpu_job_free(job);
A
Alex Deucher 已提交
522

523
error:
A
Alex Deucher 已提交
524 525 526 527
	return r;
}

/**
528
 * amdgpu_vm_map_gart - Resolve gart mapping of addr
A
Alex Deucher 已提交
529
 *
530
 * @pages_addr: optional DMA address to use for lookup
A
Alex Deucher 已提交
531 532 533
 * @addr: the unmapped addr
 *
 * Look up the physical address of the page that the pte resolves
534
 * to and return the pointer for the page table entry.
A
Alex Deucher 已提交
535
 */
536
uint64_t amdgpu_vm_map_gart(const dma_addr_t *pages_addr, uint64_t addr)
A
Alex Deucher 已提交
537 538 539
{
	uint64_t result;

540 541 542 543 544 545 546 547 548 549 550
	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 已提交
551

552
	result &= 0xFFFFFFFFFFFFF000ULL;
A
Alex Deucher 已提交
553 554 555 556 557 558 559 560 561 562 563 564 565

	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
566
 * and updates the page directory.
A
Alex Deucher 已提交
567 568 569 570 571
 * Returns 0 for success, error for failure.
 */
int amdgpu_vm_update_page_directory(struct amdgpu_device *adev,
				    struct amdgpu_vm *vm)
{
572
	struct amdgpu_ring *ring;
A
Alex Deucher 已提交
573 574 575 576 577
	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;
578
	struct amdgpu_job *job;
579
	struct amdgpu_vm_update_params vm_update_params;
580
	struct fence *fence = NULL;
C
Chunming Zhou 已提交
581

A
Alex Deucher 已提交
582 583
	int r;

584
	memset(&vm_update_params, 0, sizeof(vm_update_params));
585 586
	ring = container_of(vm->entity.sched, struct amdgpu_ring, sched);

A
Alex Deucher 已提交
587 588 589 590 591 592
	/* padding, etc. */
	ndw = 64;

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

593 594
	r = amdgpu_job_alloc_with_ib(adev, ndw * 4, &job);
	if (r)
A
Alex Deucher 已提交
595
		return r;
596

597
	vm_update_params.ib = &job->ibs[0];
A
Alex Deucher 已提交
598 599 600

	/* walk over the address space and update the page directory */
	for (pt_idx = 0; pt_idx <= vm->max_pde_used; ++pt_idx) {
601
		struct amdgpu_bo *bo = vm->page_tables[pt_idx].entry.robj;
A
Alex Deucher 已提交
602 603 604 605 606 607 608 609 610 611 612 613 614 615 616
		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) {
617
				amdgpu_vm_update_pages(adev, &vm_update_params,
618 619 620
						       last_pde, last_pt,
						       count, incr,
						       AMDGPU_PTE_VALID);
A
Alex Deucher 已提交
621 622 623 624 625 626 627 628 629 630 631
			}

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

	if (count)
632 633 634
		amdgpu_vm_update_pages(adev, &vm_update_params,
					last_pde, last_pt,
					count, incr, AMDGPU_PTE_VALID);
A
Alex Deucher 已提交
635

636 637
	if (vm_update_params.ib->length_dw != 0) {
		amdgpu_ring_pad_ib(ring, vm_update_params.ib);
638 639
		amdgpu_sync_resv(adev, &job->sync, pd->tbo.resv,
				 AMDGPU_FENCE_OWNER_VM);
640
		WARN_ON(vm_update_params.ib->length_dw > ndw);
641 642
		r = amdgpu_job_submit(job, ring, &vm->entity,
				      AMDGPU_FENCE_OWNER_VM, &fence);
643 644
		if (r)
			goto error_free;
645

646
		amdgpu_bo_fence(pd, fence, true);
647 648
		fence_put(vm->page_directory_fence);
		vm->page_directory_fence = fence_get(fence);
649
		fence_put(fence);
C
Chunming Zhou 已提交
650

651 652
	} else {
		amdgpu_job_free(job);
C
Chunming Zhou 已提交
653
	}
A
Alex Deucher 已提交
654 655

	return 0;
C
Chunming Zhou 已提交
656 657

error_free:
658
	amdgpu_job_free(job);
659
	return r;
A
Alex Deucher 已提交
660 661 662 663 664 665
}

/**
 * amdgpu_vm_frag_ptes - add fragment information to PTEs
 *
 * @adev: amdgpu_device pointer
666
 * @vm_update_params: see amdgpu_vm_update_params definition
A
Alex Deucher 已提交
667 668 669 670 671 672
 * @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,
673 674
				struct amdgpu_vm_update_params
					*vm_update_params,
A
Alex Deucher 已提交
675
				uint64_t pe_start, uint64_t pe_end,
676
				uint64_t addr, uint32_t flags)
A
Alex Deucher 已提交
677 678 679 680 681 682 683 684 685 686 687 688 689 690 691 692 693 694 695 696 697 698 699 700 701 702 703 704 705
{
	/**
	 * 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;

706 707 708 709
	/* Abort early if there isn't anything to do */
	if (pe_start == pe_end)
		return;

A
Alex Deucher 已提交
710
	/* system pages are non continuously */
711 712
	if (vm_update_params->src || vm_update_params->pages_addr ||
		!(flags & AMDGPU_PTE_VALID) || (frag_start >= frag_end)) {
A
Alex Deucher 已提交
713 714

		count = (pe_end - pe_start) / 8;
715
		amdgpu_vm_update_pages(adev, vm_update_params, pe_start,
716 717
				       addr, count, AMDGPU_GPU_PAGE_SIZE,
				       flags);
A
Alex Deucher 已提交
718 719 720 721 722 723
		return;
	}

	/* handle the 4K area at the beginning */
	if (pe_start != frag_start) {
		count = (frag_start - pe_start) / 8;
724
		amdgpu_vm_update_pages(adev, vm_update_params, pe_start, addr,
725
				       count, AMDGPU_GPU_PAGE_SIZE, flags);
A
Alex Deucher 已提交
726 727 728 729 730
		addr += AMDGPU_GPU_PAGE_SIZE * count;
	}

	/* handle the area in the middle */
	count = (frag_end - frag_start) / 8;
731
	amdgpu_vm_update_pages(adev, vm_update_params, frag_start, addr, count,
732
			       AMDGPU_GPU_PAGE_SIZE, flags | frag_flags);
A
Alex Deucher 已提交
733 734 735 736 737

	/* handle the 4K area at the end */
	if (frag_end != pe_end) {
		addr += AMDGPU_GPU_PAGE_SIZE * count;
		count = (pe_end - frag_end) / 8;
738
		amdgpu_vm_update_pages(adev, vm_update_params, frag_end, addr,
739
				       count, AMDGPU_GPU_PAGE_SIZE, flags);
A
Alex Deucher 已提交
740 741 742 743 744 745 746
	}
}

/**
 * amdgpu_vm_update_ptes - make sure that page tables are valid
 *
 * @adev: amdgpu_device pointer
747
 * @vm_update_params: see amdgpu_vm_update_params definition
A
Alex Deucher 已提交
748 749 750
 * @vm: requested vm
 * @start: start of GPU address range
 * @end: end of GPU address range
751
 * @dst: destination address to map to, the next dst inside the function
A
Alex Deucher 已提交
752 753
 * @flags: mapping flags
 *
754
 * Update the page tables in the range @start - @end.
A
Alex Deucher 已提交
755
 */
756
static void amdgpu_vm_update_ptes(struct amdgpu_device *adev,
757 758
				  struct amdgpu_vm_update_params
					*vm_update_params,
759 760 761
				  struct amdgpu_vm *vm,
				  uint64_t start, uint64_t end,
				  uint64_t dst, uint32_t flags)
A
Alex Deucher 已提交
762
{
763 764
	const uint64_t mask = AMDGPU_VM_PTE_COUNT - 1;

765 766
	uint64_t cur_pe_start = ~0, cur_pe_end = ~0, cur_dst = ~0;
	uint64_t addr; /* next GPU address to be updated */
A
Alex Deucher 已提交
767 768 769 770

	/* walk over the address space and update the page tables */
	for (addr = start; addr < end; ) {
		uint64_t pt_idx = addr >> amdgpu_vm_block_size;
771
		struct amdgpu_bo *pt = vm->page_tables[pt_idx].entry.robj;
772 773
		unsigned nptes; /* next number of ptes to be updated */
		uint64_t next_pe_start;
A
Alex Deucher 已提交
774 775 776 777 778 779

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

780 781
		next_pe_start = amdgpu_bo_gpu_offset(pt);
		next_pe_start += (addr & mask) * 8;
A
Alex Deucher 已提交
782

783 784 785 786 787 788 789
		if (cur_pe_end == next_pe_start) {
			/* The next ptb is consecutive to current ptb.
			 * Don't call amdgpu_vm_frag_ptes now.
			 * Will update two ptbs together in future.
			*/
			cur_pe_end += 8 * nptes;
		} else {
790
			amdgpu_vm_frag_ptes(adev, vm_update_params,
791 792
					    cur_pe_start, cur_pe_end,
					    cur_dst, flags);
A
Alex Deucher 已提交
793

794 795 796
			cur_pe_start = next_pe_start;
			cur_pe_end = next_pe_start + 8 * nptes;
			cur_dst = dst;
A
Alex Deucher 已提交
797 798 799 800 801 802
		}

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

803 804
	amdgpu_vm_frag_ptes(adev, vm_update_params, cur_pe_start,
			    cur_pe_end, cur_dst, flags);
A
Alex Deucher 已提交
805 806 807 808 809 810
}

/**
 * amdgpu_vm_bo_update_mapping - update a mapping in the vm page table
 *
 * @adev: amdgpu_device pointer
811 812
 * @src: address where to copy page table entries from
 * @pages_addr: DMA addresses to use for mapping
A
Alex Deucher 已提交
813
 * @vm: requested vm
814 815 816
 * @start: start of mapped range
 * @last: last mapped entry
 * @flags: flags for the entries
A
Alex Deucher 已提交
817 818 819
 * @addr: addr to set the area to
 * @fence: optional resulting fence
 *
820
 * Fill in the page table entries between @start and @last.
A
Alex Deucher 已提交
821 822 823
 * Returns 0 for success, -EINVAL for failure.
 */
static int amdgpu_vm_bo_update_mapping(struct amdgpu_device *adev,
824 825
				       uint64_t src,
				       dma_addr_t *pages_addr,
A
Alex Deucher 已提交
826
				       struct amdgpu_vm *vm,
827 828 829
				       uint64_t start, uint64_t last,
				       uint32_t flags, uint64_t addr,
				       struct fence **fence)
A
Alex Deucher 已提交
830
{
831
	struct amdgpu_ring *ring;
832
	void *owner = AMDGPU_FENCE_OWNER_VM;
A
Alex Deucher 已提交
833
	unsigned nptes, ncmds, ndw;
834
	struct amdgpu_job *job;
835
	struct amdgpu_vm_update_params vm_update_params;
836
	struct fence *f = NULL;
A
Alex Deucher 已提交
837 838
	int r;

839
	ring = container_of(vm->entity.sched, struct amdgpu_ring, sched);
840 841 842
	memset(&vm_update_params, 0, sizeof(vm_update_params));
	vm_update_params.src = src;
	vm_update_params.pages_addr = pages_addr;
843

844 845 846 847
	/* sync to everything on unmapping */
	if (!(flags & AMDGPU_PTE_VALID))
		owner = AMDGPU_FENCE_OWNER_UNDEFINED;

848
	nptes = last - start + 1;
A
Alex Deucher 已提交
849 850 851 852 853 854 855 856 857 858

	/*
	 * 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;

859
	if (vm_update_params.src) {
A
Alex Deucher 已提交
860 861 862
		/* only copy commands needed */
		ndw += ncmds * 7;

863
	} else if (vm_update_params.pages_addr) {
A
Alex Deucher 已提交
864 865 866 867 868 869 870 871 872 873 874 875 876 877
		/* 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;
	}

878 879
	r = amdgpu_job_alloc_with_ib(adev, ndw * 4, &job);
	if (r)
A
Alex Deucher 已提交
880
		return r;
881

882
	vm_update_params.ib = &job->ibs[0];
C
Chunming Zhou 已提交
883

884
	r = amdgpu_sync_resv(adev, &job->sync, vm->page_directory->tbo.resv,
885 886 887
			     owner);
	if (r)
		goto error_free;
A
Alex Deucher 已提交
888

889 890 891 892
	r = reservation_object_reserve_shared(vm->page_directory->tbo.resv);
	if (r)
		goto error_free;

893
	amdgpu_vm_update_ptes(adev, &vm_update_params, vm, start,
894
			      last + 1, addr, flags);
A
Alex Deucher 已提交
895

896 897
	amdgpu_ring_pad_ib(ring, vm_update_params.ib);
	WARN_ON(vm_update_params.ib->length_dw > ndw);
898 899
	r = amdgpu_job_submit(job, ring, &vm->entity,
			      AMDGPU_FENCE_OWNER_VM, &f);
900 901
	if (r)
		goto error_free;
A
Alex Deucher 已提交
902

903
	amdgpu_bo_fence(vm->page_directory, f, true);
904 905 906 907
	if (fence) {
		fence_put(*fence);
		*fence = fence_get(f);
	}
908
	fence_put(f);
A
Alex Deucher 已提交
909
	return 0;
C
Chunming Zhou 已提交
910 911

error_free:
912
	amdgpu_job_free(job);
913
	return r;
A
Alex Deucher 已提交
914 915
}

916 917 918 919
/**
 * amdgpu_vm_bo_split_mapping - split a mapping into smaller chunks
 *
 * @adev: amdgpu_device pointer
920 921
 * @gtt_flags: flags as they are used for GTT
 * @pages_addr: DMA addresses to use for mapping
922 923 924
 * @vm: requested vm
 * @mapping: mapped range and flags to use for the update
 * @addr: addr to set the area to
925
 * @flags: HW flags for the mapping
926 927 928 929 930 931 932 933
 * @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,
934
				      dma_addr_t *pages_addr,
935 936
				      struct amdgpu_vm *vm,
				      struct amdgpu_bo_va_mapping *mapping,
937 938
				      uint32_t flags, uint64_t addr,
				      struct fence **fence)
939 940 941
{
	const uint64_t max_size = 64ULL * 1024ULL * 1024ULL / AMDGPU_GPU_PAGE_SIZE;

942
	uint64_t src = 0, start = mapping->it.start;
943 944 945 946 947 948 949 950 951 952 953 954
	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);

955
	if (pages_addr) {
956 957 958 959
		if (flags == gtt_flags)
			src = adev->gart.table_addr + (addr >> 12) * 8;
		addr = 0;
	}
960 961
	addr += mapping->offset;

962
	if (!pages_addr || src)
963
		return amdgpu_vm_bo_update_mapping(adev, src, pages_addr, vm,
964 965 966 967 968 969
						   start, mapping->it.last,
						   flags, addr, fence);

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

970
		last = min((uint64_t)mapping->it.last, start + max_size - 1);
971
		r = amdgpu_vm_bo_update_mapping(adev, src, pages_addr, vm,
972 973 974 975 976 977
						start, last, flags, addr,
						fence);
		if (r)
			return r;

		start = last + 1;
978
		addr += max_size * AMDGPU_GPU_PAGE_SIZE;
979 980 981 982 983
	}

	return 0;
}

A
Alex Deucher 已提交
984 985 986 987 988 989 990 991 992 993 994 995 996 997 998 999 1000 1001
/**
 * 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;
1002
	dma_addr_t *pages_addr = NULL;
1003
	uint32_t gtt_flags, flags;
A
Alex Deucher 已提交
1004 1005 1006 1007
	uint64_t addr;
	int r;

	if (mem) {
1008 1009
		struct ttm_dma_tt *ttm;

1010
		addr = (u64)mem->start << PAGE_SHIFT;
1011 1012
		switch (mem->mem_type) {
		case TTM_PL_TT:
1013 1014 1015
			ttm = container_of(bo_va->bo->tbo.ttm, struct
					   ttm_dma_tt, ttm);
			pages_addr = ttm->dma_address;
1016 1017 1018
			break;

		case TTM_PL_VRAM:
A
Alex Deucher 已提交
1019
			addr += adev->vm_manager.vram_base_offset;
1020 1021 1022 1023 1024
			break;

		default:
			break;
		}
A
Alex Deucher 已提交
1025 1026 1027 1028 1029
	} else {
		addr = 0;
	}

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

1032 1033 1034 1035 1036 1037
	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) {
1038 1039 1040
		r = amdgpu_vm_bo_split_mapping(adev, gtt_flags, pages_addr, vm,
					       mapping, flags, addr,
					       &bo_va->last_pt_update);
A
Alex Deucher 已提交
1041 1042 1043 1044
		if (r)
			return r;
	}

1045 1046 1047 1048 1049 1050 1051 1052
	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 已提交
1053
	spin_lock(&vm->status_lock);
1054
	list_splice_init(&bo_va->invalids, &bo_va->valids);
A
Alex Deucher 已提交
1055
	list_del_init(&bo_va->vm_status);
1056 1057
	if (!mem)
		list_add(&bo_va->vm_status, &vm->cleared);
A
Alex Deucher 已提交
1058 1059 1060 1061 1062 1063 1064 1065 1066 1067 1068 1069 1070 1071 1072 1073 1074 1075 1076 1077 1078 1079 1080 1081 1082 1083
	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);
1084

1085
		r = amdgpu_vm_bo_split_mapping(adev, 0, NULL, vm, mapping,
1086
					       0, 0, NULL);
A
Alex Deucher 已提交
1087 1088 1089 1090 1091 1092 1093 1094 1095 1096 1097 1098 1099 1100 1101 1102 1103 1104 1105 1106 1107
		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,
1108
			     struct amdgpu_vm *vm, struct amdgpu_sync *sync)
A
Alex Deucher 已提交
1109
{
1110
	struct amdgpu_bo_va *bo_va = NULL;
1111
	int r = 0;
A
Alex Deucher 已提交
1112 1113 1114 1115 1116 1117

	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);
1118

A
Alex Deucher 已提交
1119 1120 1121 1122 1123 1124 1125 1126
		r = amdgpu_vm_bo_update(adev, bo_va, NULL);
		if (r)
			return r;

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

1127
	if (bo_va)
1128
		r = amdgpu_sync_fence(adev, sync, bo_va->last_pt_update);
1129 1130

	return r;
A
Alex Deucher 已提交
1131 1132 1133 1134 1135 1136 1137 1138 1139
}

/**
 * amdgpu_vm_bo_add - add a bo to a specific vm
 *
 * @adev: amdgpu_device pointer
 * @vm: requested vm
 * @bo: amdgpu buffer object
 *
1140
 * Add @bo into the requested vm.
A
Alex Deucher 已提交
1141 1142 1143 1144 1145 1146 1147 1148 1149 1150 1151 1152 1153 1154 1155 1156 1157 1158 1159
 * 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);
1160 1161
	INIT_LIST_HEAD(&bo_va->valids);
	INIT_LIST_HEAD(&bo_va->invalids);
A
Alex Deucher 已提交
1162
	INIT_LIST_HEAD(&bo_va->vm_status);
1163

A
Alex Deucher 已提交
1164 1165 1166 1167 1168 1169 1170 1171 1172 1173 1174 1175 1176 1177 1178 1179 1180
	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.
 *
1181
 * Object has to be reserved and unreserved outside!
A
Alex Deucher 已提交
1182 1183 1184 1185 1186 1187 1188 1189 1190 1191 1192 1193 1194
 */
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;

1195 1196
	/* validate the parameters */
	if (saddr & AMDGPU_GPU_PAGE_MASK || offset & AMDGPU_GPU_PAGE_MASK ||
1197
	    size == 0 || size & AMDGPU_GPU_PAGE_MASK)
1198 1199
		return -EINVAL;

A
Alex Deucher 已提交
1200
	/* make sure object fit at this offset */
1201
	eaddr = saddr + size - 1;
1202
	if ((saddr >= eaddr) || (offset + size > amdgpu_bo_size(bo_va->bo)))
A
Alex Deucher 已提交
1203 1204 1205
		return -EINVAL;

	last_pfn = eaddr / AMDGPU_GPU_PAGE_SIZE;
1206 1207
	if (last_pfn >= adev->vm_manager.max_pfn) {
		dev_err(adev->dev, "va above limit (0x%08X >= 0x%08X)\n",
A
Alex Deucher 已提交
1208 1209 1210 1211 1212 1213 1214
			last_pfn, adev->vm_manager.max_pfn);
		return -EINVAL;
	}

	saddr /= AMDGPU_GPU_PAGE_SIZE;
	eaddr /= AMDGPU_GPU_PAGE_SIZE;

1215
	it = interval_tree_iter_first(&vm->va, saddr, eaddr);
A
Alex Deucher 已提交
1216 1217 1218 1219 1220 1221 1222 1223
	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;
1224
		goto error;
A
Alex Deucher 已提交
1225 1226 1227 1228 1229
	}

	mapping = kmalloc(sizeof(*mapping), GFP_KERNEL);
	if (!mapping) {
		r = -ENOMEM;
1230
		goto error;
A
Alex Deucher 已提交
1231 1232 1233 1234
	}

	INIT_LIST_HEAD(&mapping->list);
	mapping->it.start = saddr;
1235
	mapping->it.last = eaddr;
A
Alex Deucher 已提交
1236 1237 1238
	mapping->offset = offset;
	mapping->flags = flags;

1239
	list_add(&mapping->list, &bo_va->invalids);
A
Alex Deucher 已提交
1240 1241 1242 1243 1244 1245 1246 1247 1248 1249 1250 1251 1252
	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) {
1253
		struct reservation_object *resv = vm->page_directory->tbo.resv;
1254
		struct amdgpu_bo_list_entry *entry;
A
Alex Deucher 已提交
1255 1256
		struct amdgpu_bo *pt;

1257 1258
		entry = &vm->page_tables[pt_idx].entry;
		if (entry->robj)
A
Alex Deucher 已提交
1259 1260 1261 1262
			continue;

		r = amdgpu_bo_create(adev, AMDGPU_VM_PTE_COUNT * 8,
				     AMDGPU_GPU_PAGE_SIZE, true,
1263 1264
				     AMDGPU_GEM_DOMAIN_VRAM,
				     AMDGPU_GEM_CREATE_NO_CPU_ACCESS,
1265
				     NULL, resv, &pt);
1266
		if (r)
A
Alex Deucher 已提交
1267
			goto error_free;
1268

1269 1270 1271 1272 1273
		/* Keep a reference to the page table to avoid freeing
		 * them up in the wrong order.
		 */
		pt->parent = amdgpu_bo_ref(vm->page_directory);

1274
		r = amdgpu_vm_clear_bo(adev, vm, pt);
A
Alex Deucher 已提交
1275 1276 1277 1278 1279
		if (r) {
			amdgpu_bo_unref(&pt);
			goto error_free;
		}

1280 1281 1282 1283
		entry->robj = pt;
		entry->priority = 0;
		entry->tv.bo = &entry->robj->tbo;
		entry->tv.shared = true;
1284
		entry->user_pages = NULL;
A
Alex Deucher 已提交
1285 1286 1287 1288 1289 1290 1291 1292
		vm->page_tables[pt_idx].addr = 0;
	}

	return 0;

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

1296
error:
A
Alex Deucher 已提交
1297 1298 1299 1300 1301 1302 1303 1304 1305 1306 1307 1308 1309
	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.
 *
1310
 * Object has to be reserved and unreserved outside!
A
Alex Deucher 已提交
1311 1312 1313 1314 1315 1316 1317
 */
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;
1318
	bool valid = true;
A
Alex Deucher 已提交
1319

1320
	saddr /= AMDGPU_GPU_PAGE_SIZE;
1321

1322
	list_for_each_entry(mapping, &bo_va->valids, list) {
A
Alex Deucher 已提交
1323 1324 1325 1326
		if (mapping->it.start == saddr)
			break;
	}

1327 1328 1329 1330 1331 1332 1333 1334
	if (&mapping->list == &bo_va->valids) {
		valid = false;

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

1335
		if (&mapping->list == &bo_va->invalids)
1336
			return -ENOENT;
A
Alex Deucher 已提交
1337
	}
1338

A
Alex Deucher 已提交
1339 1340
	list_del(&mapping->list);
	interval_tree_remove(&mapping->it, &vm->va);
1341
	trace_amdgpu_vm_bo_unmap(bo_va, mapping);
A
Alex Deucher 已提交
1342

1343
	if (valid)
A
Alex Deucher 已提交
1344
		list_add(&mapping->list, &vm->freed);
1345
	else
A
Alex Deucher 已提交
1346 1347 1348 1349 1350 1351 1352 1353 1354 1355 1356
		kfree(mapping);

	return 0;
}

/**
 * amdgpu_vm_bo_rmv - remove a bo to a specific vm
 *
 * @adev: amdgpu_device pointer
 * @bo_va: requested bo_va
 *
1357
 * Remove @bo_va->bo from the requested vm.
A
Alex Deucher 已提交
1358 1359 1360 1361 1362 1363 1364 1365 1366 1367 1368 1369 1370 1371 1372
 *
 * 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);

1373
	list_for_each_entry_safe(mapping, next, &bo_va->valids, list) {
A
Alex Deucher 已提交
1374 1375
		list_del(&mapping->list);
		interval_tree_remove(&mapping->it, &vm->va);
1376
		trace_amdgpu_vm_bo_unmap(bo_va, mapping);
1377 1378 1379 1380 1381 1382
		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 已提交
1383
	}
1384

1385
	fence_put(bo_va->last_pt_update);
A
Alex Deucher 已提交
1386 1387 1388 1389 1390 1391 1392 1393 1394 1395
	kfree(bo_va);
}

/**
 * amdgpu_vm_bo_invalidate - mark the bo as invalid
 *
 * @adev: amdgpu_device pointer
 * @vm: requested vm
 * @bo: amdgpu buffer object
 *
1396
 * Mark @bo as invalid.
A
Alex Deucher 已提交
1397 1398 1399 1400 1401 1402 1403
 */
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) {
1404 1405
		spin_lock(&bo_va->vm->status_lock);
		if (list_empty(&bo_va->vm_status))
A
Alex Deucher 已提交
1406
			list_add(&bo_va->vm_status, &bo_va->vm->invalidated);
1407
		spin_unlock(&bo_va->vm->status_lock);
A
Alex Deucher 已提交
1408 1409 1410 1411 1412 1413 1414 1415 1416
	}
}

/**
 * amdgpu_vm_init - initialize a vm instance
 *
 * @adev: amdgpu_device pointer
 * @vm: requested vm
 *
1417
 * Init @vm fields.
A
Alex Deucher 已提交
1418 1419 1420 1421 1422
 */
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);
1423
	unsigned pd_size, pd_entries;
1424 1425
	unsigned ring_instance;
	struct amdgpu_ring *ring;
1426
	struct amd_sched_rq *rq;
A
Alex Deucher 已提交
1427 1428
	int i, r;

1429 1430
	for (i = 0; i < AMDGPU_MAX_RINGS; ++i)
		vm->ids[i] = NULL;
A
Alex Deucher 已提交
1431
	vm->va = RB_ROOT;
1432
	vm->client_id = atomic64_inc_return(&adev->vm_manager.client_counter);
A
Alex Deucher 已提交
1433 1434
	spin_lock_init(&vm->status_lock);
	INIT_LIST_HEAD(&vm->invalidated);
1435
	INIT_LIST_HEAD(&vm->cleared);
A
Alex Deucher 已提交
1436
	INIT_LIST_HEAD(&vm->freed);
1437

A
Alex Deucher 已提交
1438 1439 1440 1441
	pd_size = amdgpu_vm_directory_size(adev);
	pd_entries = amdgpu_vm_num_pdes(adev);

	/* allocate page table array */
1442
	vm->page_tables = drm_calloc_large(pd_entries, sizeof(struct amdgpu_vm_pt));
A
Alex Deucher 已提交
1443 1444 1445 1446 1447
	if (vm->page_tables == NULL) {
		DRM_ERROR("Cannot allocate memory for page table array\n");
		return -ENOMEM;
	}

1448
	/* create scheduler entity for page table updates */
1449 1450 1451 1452

	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];
1453 1454 1455 1456 1457 1458
	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;

1459 1460
	vm->page_directory_fence = NULL;

A
Alex Deucher 已提交
1461
	r = amdgpu_bo_create(adev, pd_size, align, true,
1462 1463
			     AMDGPU_GEM_DOMAIN_VRAM,
			     AMDGPU_GEM_CREATE_NO_CPU_ACCESS,
1464
			     NULL, NULL, &vm->page_directory);
A
Alex Deucher 已提交
1465
	if (r)
1466 1467
		goto error_free_sched_entity;

1468
	r = amdgpu_bo_reserve(vm->page_directory, false);
1469 1470 1471 1472
	if (r)
		goto error_free_page_directory;

	r = amdgpu_vm_clear_bo(adev, vm, vm->page_directory);
1473
	amdgpu_bo_unreserve(vm->page_directory);
1474 1475
	if (r)
		goto error_free_page_directory;
A
Alex Deucher 已提交
1476 1477

	return 0;
1478 1479 1480 1481 1482 1483 1484 1485 1486

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 已提交
1487 1488 1489 1490 1491 1492 1493 1494
}

/**
 * amdgpu_vm_fini - tear down a vm instance
 *
 * @adev: amdgpu_device pointer
 * @vm: requested vm
 *
1495
 * Tear down @vm.
A
Alex Deucher 已提交
1496 1497 1498 1499 1500 1501 1502
 * 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;
	int i;

1503
	amd_sched_entity_fini(vm->entity.sched, &vm->entity);
1504

A
Alex Deucher 已提交
1505 1506 1507 1508 1509 1510 1511 1512 1513 1514 1515 1516 1517 1518
	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++)
1519
		amdgpu_bo_unref(&vm->page_tables[i].entry.robj);
1520
	drm_free_large(vm->page_tables);
A
Alex Deucher 已提交
1521 1522

	amdgpu_bo_unref(&vm->page_directory);
1523
	fence_put(vm->page_directory_fence);
A
Alex Deucher 已提交
1524
}
1525

1526 1527 1528 1529 1530 1531 1532 1533 1534 1535 1536 1537 1538 1539
/**
 * 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 */
1540 1541
	for (i = 1; i < adev->vm_manager.num_ids; ++i) {
		amdgpu_vm_reset_id(adev, i);
1542
		amdgpu_sync_create(&adev->vm_manager.ids[i].active);
1543 1544
		list_add_tail(&adev->vm_manager.ids[i].list,
			      &adev->vm_manager.ids_lru);
1545
	}
1546

1547 1548 1549 1550
	adev->vm_manager.fence_context = fence_context_alloc(AMDGPU_MAX_RINGS);
	for (i = 0; i < AMDGPU_MAX_RINGS; ++i)
		adev->vm_manager.seqno[i] = 0;

1551
	atomic_set(&adev->vm_manager.vm_pte_next_ring, 0);
1552
	atomic64_set(&adev->vm_manager.client_counter, 0);
1553 1554
}

1555 1556 1557 1558 1559 1560 1561 1562 1563 1564 1565
/**
 * 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;

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

1569 1570
		fence_put(adev->vm_manager.ids[i].first);
		amdgpu_sync_free(&adev->vm_manager.ids[i].active);
1571 1572
		fence_put(id->flushed_updates);
	}
1573
}