ttm_bo.c 41.8 KB
Newer Older
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
/**************************************************************************
 *
 * Copyright (c) 2006-2009 VMware, Inc., Palo Alto, CA., USA
 * All Rights Reserved.
 *
 * 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, sub license, 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 (including the
 * next paragraph) 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 NON-INFRINGEMENT. IN NO EVENT SHALL
 * THE COPYRIGHT HOLDERS, AUTHORS AND/OR ITS SUPPLIERS 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: Thomas Hellstrom <thellstrom-at-vmware-dot-com>
 */

J
Joe Perches 已提交
31 32
#define pr_fmt(fmt) "[TTM] " fmt

33 34 35
#include <drm/ttm/ttm_module.h>
#include <drm/ttm/ttm_bo_driver.h>
#include <drm/ttm/ttm_placement.h>
36 37 38 39 40 41
#include <linux/jiffies.h>
#include <linux/slab.h>
#include <linux/sched.h>
#include <linux/mm.h>
#include <linux/file.h>
#include <linux/module.h>
A
Arun Sharma 已提交
42
#include <linux/atomic.h>
43
#include <linux/reservation.h>
44 45 46 47 48 49

#define TTM_ASSERT_LOCKED(param)
#define TTM_DEBUG(fmt, arg...)
#define TTM_BO_HASH_ORDER 13

static int ttm_bo_swapout(struct ttm_mem_shrink *shrink);
50 51 52 53 54 55 56
static void ttm_bo_global_kobj_release(struct kobject *kobj);

static struct attribute ttm_bo_count = {
	.name = "bo_count",
	.mode = S_IRUGO
};

57 58
static inline int ttm_mem_type_from_place(const struct ttm_place *place,
					  uint32_t *mem_type)
59 60 61 62
{
	int i;

	for (i = 0; i <= TTM_PL_PRIV5; i++)
63
		if (place->flags & (1 << i)) {
64 65 66 67 68 69
			*mem_type = i;
			return 0;
		}
	return -EINVAL;
}

70
static void ttm_mem_type_debug(struct ttm_bo_device *bdev, int mem_type)
71
{
72 73
	struct ttm_mem_type_manager *man = &bdev->man[mem_type];

J
Joe Perches 已提交
74 75 76
	pr_err("    has_type: %d\n", man->has_type);
	pr_err("    use_type: %d\n", man->use_type);
	pr_err("    flags: 0x%08X\n", man->flags);
77
	pr_err("    gpu_offset: 0x%08llX\n", man->gpu_offset);
J
Joe Perches 已提交
78 79 80
	pr_err("    size: %llu\n", man->size);
	pr_err("    available_caching: 0x%08X\n", man->available_caching);
	pr_err("    default_caching: 0x%08X\n", man->default_caching);
81 82
	if (mem_type != TTM_PL_SYSTEM)
		(*man->func->debug)(man, TTM_PFX);
83 84 85 86 87 88 89
}

static void ttm_bo_mem_space_debug(struct ttm_buffer_object *bo,
					struct ttm_placement *placement)
{
	int i, ret, mem_type;

J
Joe Perches 已提交
90 91 92
	pr_err("No space for %p (%lu pages, %luK, %luM)\n",
	       bo, bo->mem.num_pages, bo->mem.size >> 10,
	       bo->mem.size >> 20);
93
	for (i = 0; i < placement->num_placement; i++) {
94
		ret = ttm_mem_type_from_place(&placement->placement[i],
95 96 97
						&mem_type);
		if (ret)
			return;
J
Joe Perches 已提交
98
		pr_err("  placement[%d]=0x%08X (%d)\n",
99
		       i, placement->placement[i].flags, mem_type);
100
		ttm_mem_type_debug(bo->bdev, mem_type);
101 102 103
	}
}

104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119
static ssize_t ttm_bo_global_show(struct kobject *kobj,
				  struct attribute *attr,
				  char *buffer)
{
	struct ttm_bo_global *glob =
		container_of(kobj, struct ttm_bo_global, kobj);

	return snprintf(buffer, PAGE_SIZE, "%lu\n",
			(unsigned long) atomic_read(&glob->bo_count));
}

static struct attribute *ttm_bo_global_attrs[] = {
	&ttm_bo_count,
	NULL
};

120
static const struct sysfs_ops ttm_bo_global_ops = {
121 122 123 124 125 126 127 128 129
	.show = &ttm_bo_global_show
};

static struct kobj_type ttm_bo_glob_kobj_type  = {
	.release = &ttm_bo_global_kobj_release,
	.sysfs_ops = &ttm_bo_global_ops,
	.default_attrs = ttm_bo_global_attrs
};

130 131 132 133 134 135 136 137 138 139 140

static inline uint32_t ttm_bo_type_flags(unsigned type)
{
	return 1 << (type);
}

static void ttm_bo_release_list(struct kref *list_kref)
{
	struct ttm_buffer_object *bo =
	    container_of(list_kref, struct ttm_buffer_object, list_kref);
	struct ttm_bo_device *bdev = bo->bdev;
141
	size_t acc_size = bo->acc_size;
142 143 144 145 146 147 148 149

	BUG_ON(atomic_read(&bo->list_kref.refcount));
	BUG_ON(atomic_read(&bo->kref.refcount));
	BUG_ON(atomic_read(&bo->cpu_writers));
	BUG_ON(bo->mem.mm_node != NULL);
	BUG_ON(!list_empty(&bo->lru));
	BUG_ON(!list_empty(&bo->ddestroy));

150
	ttm_tt_destroy(bo->ttm);
151
	atomic_dec(&bo->glob->bo_count);
152
	fence_put(bo->moving);
153 154
	if (bo->resv == &bo->ttm_resv)
		reservation_object_fini(&bo->ttm_resv);
155
	mutex_destroy(&bo->wu_mutex);
156 157 158 159 160
	if (bo->destroy)
		bo->destroy(bo);
	else {
		kfree(bo);
	}
161
	ttm_mem_global_free(bdev->glob->mem_glob, acc_size);
162 163
}

164
void ttm_bo_add_to_lru(struct ttm_buffer_object *bo)
165 166 167
{
	struct ttm_bo_device *bdev = bo->bdev;

168
	lockdep_assert_held(&bo->resv->lock.base);
169 170 171 172 173

	if (!(bo->mem.placement & TTM_PL_FLAG_NO_EVICT)) {

		BUG_ON(!list_empty(&bo->lru));

174
		list_add(&bo->lru, bdev->driver->lru_tail(bo));
175 176
		kref_get(&bo->list_kref);

177
		if (bo->ttm && !(bo->ttm->page_flags & TTM_PAGE_FLAG_SG)) {
178
			list_add(&bo->swap, bdev->driver->swap_lru_tail(bo));
179 180 181 182
			kref_get(&bo->list_kref);
		}
	}
}
183
EXPORT_SYMBOL(ttm_bo_add_to_lru);
184

185
int ttm_bo_del_from_lru(struct ttm_buffer_object *bo)
186
{
187
	struct ttm_bo_device *bdev = bo->bdev;
188 189
	int put_count = 0;

190 191 192
	if (bdev->driver->lru_removal)
		bdev->driver->lru_removal(bo);

193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209
	if (!list_empty(&bo->swap)) {
		list_del_init(&bo->swap);
		++put_count;
	}
	if (!list_empty(&bo->lru)) {
		list_del_init(&bo->lru);
		++put_count;
	}

	return put_count;
}

static void ttm_bo_ref_bug(struct kref *list_kref)
{
	BUG();
}

210 211 212
void ttm_bo_list_ref_sub(struct ttm_buffer_object *bo, int count,
			 bool never_free)
{
213 214
	kref_sub(&bo->list_kref, count,
		 (never_free) ? ttm_bo_ref_bug : ttm_bo_release_list);
215 216
}

217
void ttm_bo_del_sub_from_lru(struct ttm_buffer_object *bo)
218
{
219
	int put_count;
220

221 222 223 224
	spin_lock(&bo->glob->lru_lock);
	put_count = ttm_bo_del_from_lru(bo);
	spin_unlock(&bo->glob->lru_lock);
	ttm_bo_list_ref_sub(bo, put_count, true);
225
}
226
EXPORT_SYMBOL(ttm_bo_del_sub_from_lru);
227

228 229 230
void ttm_bo_move_to_lru_tail(struct ttm_buffer_object *bo)
{
	struct ttm_bo_device *bdev = bo->bdev;
231
	int put_count = 0;
232 233 234

	lockdep_assert_held(&bo->resv->lock.base);

235 236 237
	if (bdev->driver->lru_removal)
		bdev->driver->lru_removal(bo);

238 239 240
	put_count = ttm_bo_del_from_lru(bo);
	ttm_bo_list_ref_sub(bo, put_count, true);
	ttm_bo_add_to_lru(bo);
241 242 243
}
EXPORT_SYMBOL(ttm_bo_move_to_lru_tail);

244 245 246 247 248 249 250 251 252 253 254 255
struct list_head *ttm_bo_default_lru_tail(struct ttm_buffer_object *bo)
{
	return bo->bdev->man[bo->mem.mem_type].lru.prev;
}
EXPORT_SYMBOL(ttm_bo_default_lru_tail);

struct list_head *ttm_bo_default_swap_lru_tail(struct ttm_buffer_object *bo)
{
	return bo->glob->swap_lru.prev;
}
EXPORT_SYMBOL(ttm_bo_default_swap_lru_tail);

256 257 258 259 260 261
/*
 * Call bo->mutex locked.
 */
static int ttm_bo_add_ttm(struct ttm_buffer_object *bo, bool zero_alloc)
{
	struct ttm_bo_device *bdev = bo->bdev;
262
	struct ttm_bo_global *glob = bo->glob;
263 264 265 266 267 268
	int ret = 0;
	uint32_t page_flags = 0;

	TTM_ASSERT_LOCKED(&bo->mutex);
	bo->ttm = NULL;

D
Dave Airlie 已提交
269 270 271
	if (bdev->need_dma32)
		page_flags |= TTM_PAGE_FLAG_DMA32;

272 273 274 275 276
	switch (bo->type) {
	case ttm_bo_type_device:
		if (zero_alloc)
			page_flags |= TTM_PAGE_FLAG_ZERO_ALLOC;
	case ttm_bo_type_kernel:
277 278
		bo->ttm = bdev->driver->ttm_tt_create(bdev, bo->num_pages << PAGE_SHIFT,
						      page_flags, glob->dummy_read_page);
279 280 281
		if (unlikely(bo->ttm == NULL))
			ret = -ENOMEM;
		break;
282 283 284 285 286 287 288 289 290 291
	case ttm_bo_type_sg:
		bo->ttm = bdev->driver->ttm_tt_create(bdev, bo->num_pages << PAGE_SHIFT,
						      page_flags | TTM_PAGE_FLAG_SG,
						      glob->dummy_read_page);
		if (unlikely(bo->ttm == NULL)) {
			ret = -ENOMEM;
			break;
		}
		bo->ttm->sg = bo->sg;
		break;
292
	default:
J
Joe Perches 已提交
293
		pr_err("Illegal buffer object type\n");
294 295 296 297 298 299 300 301 302
		ret = -EINVAL;
		break;
	}

	return ret;
}

static int ttm_bo_handle_move_mem(struct ttm_buffer_object *bo,
				  struct ttm_mem_reg *mem,
303
				  bool evict, bool interruptible,
304
				  bool no_wait_gpu)
305 306 307 308 309 310 311 312 313
{
	struct ttm_bo_device *bdev = bo->bdev;
	bool old_is_pci = ttm_mem_reg_is_pci(bdev, &bo->mem);
	bool new_is_pci = ttm_mem_reg_is_pci(bdev, mem);
	struct ttm_mem_type_manager *old_man = &bdev->man[bo->mem.mem_type];
	struct ttm_mem_type_manager *new_man = &bdev->man[mem->mem_type];
	int ret = 0;

	if (old_is_pci || new_is_pci ||
314 315 316 317 318 319 320
	    ((mem->placement & bo->mem.placement & TTM_PL_MASK_CACHING) == 0)) {
		ret = ttm_mem_io_lock(old_man, true);
		if (unlikely(ret != 0))
			goto out_err;
		ttm_bo_unmap_virtual_locked(bo);
		ttm_mem_io_unlock(old_man);
	}
321 322 323 324 325

	/*
	 * Create and bind a ttm if required.
	 */

326 327
	if (!(new_man->flags & TTM_MEMTYPE_FLAG_FIXED)) {
		if (bo->ttm == NULL) {
328 329
			bool zero = !(old_man->flags & TTM_MEMTYPE_FLAG_FIXED);
			ret = ttm_bo_add_ttm(bo, zero);
330 331 332
			if (ret)
				goto out_err;
		}
333 334 335

		ret = ttm_tt_set_placement_caching(bo->ttm, mem->placement);
		if (ret)
336
			goto out_err;
337 338 339 340 341 342 343 344

		if (mem->mem_type != TTM_PL_SYSTEM) {
			ret = ttm_tt_bind(bo->ttm, mem);
			if (ret)
				goto out_err;
		}

		if (bo->mem.mem_type == TTM_PL_SYSTEM) {
345 346
			if (bdev->driver->move_notify)
				bdev->driver->move_notify(bo, mem);
347
			bo->mem = *mem;
348 349 350 351 352
			mem->mm_node = NULL;
			goto moved;
		}
	}

353 354 355
	if (bdev->driver->move_notify)
		bdev->driver->move_notify(bo, mem);

356 357
	if (!(old_man->flags & TTM_MEMTYPE_FLAG_FIXED) &&
	    !(new_man->flags & TTM_MEMTYPE_FLAG_FIXED))
358
		ret = ttm_bo_move_ttm(bo, evict, no_wait_gpu, mem);
359 360
	else if (bdev->driver->move)
		ret = bdev->driver->move(bo, evict, interruptible,
361
					 no_wait_gpu, mem);
362
	else
363 364
		ret = ttm_bo_move_memcpy(bo, evict, interruptible,
					 no_wait_gpu, mem);
365

366 367 368 369 370 371 372
	if (ret) {
		if (bdev->driver->move_notify) {
			struct ttm_mem_reg tmp_mem = *mem;
			*mem = bo->mem;
			bo->mem = tmp_mem;
			bdev->driver->move_notify(bo, mem);
			bo->mem = *mem;
373
			*mem = tmp_mem;
374
		}
375

376 377
		goto out_err;
	}
378

379 380
moved:
	if (bo->evicted) {
381 382 383 384 385
		if (bdev->driver->invalidate_caches) {
			ret = bdev->driver->invalidate_caches(bdev, bo->mem.placement);
			if (ret)
				pr_err("Can not flush read caches\n");
		}
386 387 388 389
		bo->evicted = false;
	}

	if (bo->mem.mm_node) {
390
		bo->offset = (bo->mem.start << PAGE_SHIFT) +
391 392
		    bdev->man[bo->mem.mem_type].gpu_offset;
		bo->cur_placement = bo->mem.placement;
393 394
	} else
		bo->offset = 0;
395 396 397 398 399

	return 0;

out_err:
	new_man = &bdev->man[bo->mem.mem_type];
400
	if (new_man->flags & TTM_MEMTYPE_FLAG_FIXED) {
401 402 403 404 405 406 407
		ttm_tt_destroy(bo->ttm);
		bo->ttm = NULL;
	}

	return ret;
}

408
/**
409
 * Call bo::reserved.
410
 * Will release GPU memory type usage on destruction.
411 412 413
 * This is the place to put in driver specific hooks to release
 * driver private resources.
 * Will release the bo::reserved lock.
414 415 416 417
 */

static void ttm_bo_cleanup_memtype_use(struct ttm_buffer_object *bo)
{
418 419 420
	if (bo->bdev->driver->move_notify)
		bo->bdev->driver->move_notify(bo, NULL);

421 422
	ttm_tt_destroy(bo->ttm);
	bo->ttm = NULL;
423
	ttm_bo_mem_put(bo, &bo->mem);
424

425
	ww_mutex_unlock (&bo->resv->lock);
426 427
}

428 429 430 431 432 433 434 435 436 437 438 439 440 441 442 443 444 445 446 447
static void ttm_bo_flush_all_fences(struct ttm_buffer_object *bo)
{
	struct reservation_object_list *fobj;
	struct fence *fence;
	int i;

	fobj = reservation_object_get_list(bo->resv);
	fence = reservation_object_get_excl(bo->resv);
	if (fence && !fence->ops->signaled)
		fence_enable_sw_signaling(fence);

	for (i = 0; fobj && i < fobj->shared_count; ++i) {
		fence = rcu_dereference_protected(fobj->shared[i],
					reservation_object_held(bo->resv));

		if (!fence->ops->signaled)
			fence_enable_sw_signaling(fence);
	}
}

448
static void ttm_bo_cleanup_refs_or_queue(struct ttm_buffer_object *bo)
449 450
{
	struct ttm_bo_device *bdev = bo->bdev;
451
	struct ttm_bo_global *glob = bo->glob;
452
	int put_count;
453 454
	int ret;

455
	spin_lock(&glob->lru_lock);
456
	ret = __ttm_bo_reserve(bo, false, true, NULL);
457

M
Maarten Lankhorst 已提交
458
	if (!ret) {
459
		if (!ttm_bo_wait(bo, false, true)) {
M
Maarten Lankhorst 已提交
460
			put_count = ttm_bo_del_from_lru(bo);
461

M
Maarten Lankhorst 已提交
462 463
			spin_unlock(&glob->lru_lock);
			ttm_bo_cleanup_memtype_use(bo);
464

M
Maarten Lankhorst 已提交
465
			ttm_bo_list_ref_sub(bo, put_count, true);
466

M
Maarten Lankhorst 已提交
467
			return;
468 469
		} else
			ttm_bo_flush_all_fences(bo);
470 471 472 473 474 475 476 477 478 479 480

		/*
		 * Make NO_EVICT bos immediately available to
		 * shrinkers, now that they are queued for
		 * destruction.
		 */
		if (bo->mem.placement & TTM_PL_FLAG_NO_EVICT) {
			bo->mem.placement &= ~TTM_PL_FLAG_NO_EVICT;
			ttm_bo_add_to_lru(bo);
		}

481
		__ttm_bo_unreserve(bo);
482
	}
483 484 485 486 487 488 489 490 491 492

	kref_get(&bo->list_kref);
	list_add_tail(&bo->ddestroy, &bdev->ddestroy);
	spin_unlock(&glob->lru_lock);

	schedule_delayed_work(&bdev->wq,
			      ((HZ / 100) < 1) ? 1 : HZ / 100);
}

/**
493
 * function ttm_bo_cleanup_refs_and_unlock
494 495 496
 * If bo idle, remove from delayed- and lru lists, and unref.
 * If not idle, do nothing.
 *
497 498 499
 * Must be called with lru_lock and reservation held, this function
 * will drop both before returning.
 *
500 501 502 503
 * @interruptible         Any sleeps should occur interruptibly.
 * @no_wait_gpu           Never wait for gpu. Return -EBUSY instead.
 */

504 505 506
static int ttm_bo_cleanup_refs_and_unlock(struct ttm_buffer_object *bo,
					  bool interruptible,
					  bool no_wait_gpu)
507 508 509
{
	struct ttm_bo_global *glob = bo->glob;
	int put_count;
510
	int ret;
511

512
	ret = ttm_bo_wait(bo, false, true);
513

514
	if (ret && !no_wait_gpu) {
M
Maarten Lankhorst 已提交
515 516 517 518 519 520 521 522 523 524 525 526 527
		long lret;
		ww_mutex_unlock(&bo->resv->lock);
		spin_unlock(&glob->lru_lock);

		lret = reservation_object_wait_timeout_rcu(bo->resv,
							   true,
							   interruptible,
							   30 * HZ);

		if (lret < 0)
			return lret;
		else if (lret == 0)
			return -EBUSY;
528

529
		spin_lock(&glob->lru_lock);
530
		ret = __ttm_bo_reserve(bo, false, true, NULL);
531

532 533 534 535 536 537 538 539 540 541 542 543
		/*
		 * We raced, and lost, someone else holds the reservation now,
		 * and is probably busy in ttm_bo_cleanup_memtype_use.
		 *
		 * Even if it's not the case, because we finished waiting any
		 * delayed destruction would succeed, so just return success
		 * here.
		 */
		if (ret) {
			spin_unlock(&glob->lru_lock);
			return 0;
		}
544 545 546 547 548

		/*
		 * remove sync_obj with ttm_bo_wait, the wait should be
		 * finished, and no new wait object should have been added.
		 */
549
		ret = ttm_bo_wait(bo, false, true);
550 551
		WARN_ON(ret);
	}
552

553
	if (ret || unlikely(list_empty(&bo->ddestroy))) {
554
		__ttm_bo_unreserve(bo);
555
		spin_unlock(&glob->lru_lock);
556
		return ret;
557 558
	}

559 560 561 562 563 564 565
	put_count = ttm_bo_del_from_lru(bo);
	list_del_init(&bo->ddestroy);
	++put_count;

	spin_unlock(&glob->lru_lock);
	ttm_bo_cleanup_memtype_use(bo);

566
	ttm_bo_list_ref_sub(bo, put_count, true);
567 568

	return 0;
569 570 571 572 573 574 575 576 577
}

/**
 * Traverse the delayed list, and call ttm_bo_cleanup_refs on all
 * encountered buffers.
 */

static int ttm_bo_delayed_delete(struct ttm_bo_device *bdev, bool remove_all)
{
578
	struct ttm_bo_global *glob = bdev->glob;
579 580
	struct ttm_buffer_object *entry = NULL;
	int ret = 0;
581

582
	spin_lock(&glob->lru_lock);
583 584 585 586 587 588 589 590 591 592 593 594 595
	if (list_empty(&bdev->ddestroy))
		goto out_unlock;

	entry = list_first_entry(&bdev->ddestroy,
		struct ttm_buffer_object, ddestroy);
	kref_get(&entry->list_kref);

	for (;;) {
		struct ttm_buffer_object *nentry = NULL;

		if (entry->ddestroy.next != &bdev->ddestroy) {
			nentry = list_first_entry(&entry->ddestroy,
				struct ttm_buffer_object, ddestroy);
596 597 598
			kref_get(&nentry->list_kref);
		}

599
		ret = __ttm_bo_reserve(entry, false, true, NULL);
600 601
		if (remove_all && ret) {
			spin_unlock(&glob->lru_lock);
602
			ret = __ttm_bo_reserve(entry, false, false, NULL);
603 604 605
			spin_lock(&glob->lru_lock);
		}

606 607 608 609 610 611
		if (!ret)
			ret = ttm_bo_cleanup_refs_and_unlock(entry, false,
							     !remove_all);
		else
			spin_unlock(&glob->lru_lock);

612
		kref_put(&entry->list_kref, ttm_bo_release_list);
613 614 615 616
		entry = nentry;

		if (ret || !entry)
			goto out;
617

618
		spin_lock(&glob->lru_lock);
619
		if (list_empty(&entry->ddestroy))
620 621 622
			break;
	}

623 624 625 626 627
out_unlock:
	spin_unlock(&glob->lru_lock);
out:
	if (entry)
		kref_put(&entry->list_kref, ttm_bo_release_list);
628 629 630 631 632 633 634 635 636 637 638 639 640 641 642 643 644 645 646
	return ret;
}

static void ttm_bo_delayed_workqueue(struct work_struct *work)
{
	struct ttm_bo_device *bdev =
	    container_of(work, struct ttm_bo_device, wq.work);

	if (ttm_bo_delayed_delete(bdev, false)) {
		schedule_delayed_work(&bdev->wq,
				      ((HZ / 100) < 1) ? 1 : HZ / 100);
	}
}

static void ttm_bo_release(struct kref *kref)
{
	struct ttm_buffer_object *bo =
	    container_of(kref, struct ttm_buffer_object, kref);
	struct ttm_bo_device *bdev = bo->bdev;
647
	struct ttm_mem_type_manager *man = &bdev->man[bo->mem.mem_type];
648

649
	drm_vma_offset_remove(&bdev->vma_manager, &bo->vma_node);
650 651 652
	ttm_mem_io_lock(man, false);
	ttm_mem_io_free_vm(bo);
	ttm_mem_io_unlock(man);
653
	ttm_bo_cleanup_refs_or_queue(bo);
654 655 656 657 658 659 660 661 662 663 664 665
	kref_put(&bo->list_kref, ttm_bo_release_list);
}

void ttm_bo_unref(struct ttm_buffer_object **p_bo)
{
	struct ttm_buffer_object *bo = *p_bo;

	*p_bo = NULL;
	kref_put(&bo->kref, ttm_bo_release);
}
EXPORT_SYMBOL(ttm_bo_unref);

666 667 668 669 670 671 672 673 674 675 676 677 678 679
int ttm_bo_lock_delayed_workqueue(struct ttm_bo_device *bdev)
{
	return cancel_delayed_work_sync(&bdev->wq);
}
EXPORT_SYMBOL(ttm_bo_lock_delayed_workqueue);

void ttm_bo_unlock_delayed_workqueue(struct ttm_bo_device *bdev, int resched)
{
	if (resched)
		schedule_delayed_work(&bdev->wq,
				      ((HZ / 100) < 1) ? 1 : HZ / 100);
}
EXPORT_SYMBOL(ttm_bo_unlock_delayed_workqueue);

680
static int ttm_bo_evict(struct ttm_buffer_object *bo, bool interruptible,
681
			bool no_wait_gpu)
682 683 684
{
	struct ttm_bo_device *bdev = bo->bdev;
	struct ttm_mem_reg evict_mem;
685 686
	struct ttm_placement placement;
	int ret = 0;
687

688
	lockdep_assert_held(&bo->resv->lock.base);
689 690 691

	evict_mem = bo->mem;
	evict_mem.mm_node = NULL;
692 693
	evict_mem.bus.io_reserved_vm = false;
	evict_mem.bus.io_reserved_count = 0;
694

695 696
	placement.num_placement = 0;
	placement.num_busy_placement = 0;
697 698
	bdev->driver->evict_flags(bo, &placement);
	ret = ttm_bo_mem_space(bo, &placement, &evict_mem, interruptible,
699
				no_wait_gpu);
700
	if (ret) {
701
		if (ret != -ERESTARTSYS) {
J
Joe Perches 已提交
702 703
			pr_err("Failed to find memory space for buffer 0x%p eviction\n",
			       bo);
704 705
			ttm_bo_mem_space_debug(bo, &placement);
		}
706 707 708 709
		goto out;
	}

	ret = ttm_bo_handle_move_mem(bo, &evict_mem, true, interruptible,
710
				     no_wait_gpu);
711
	if (unlikely(ret)) {
712
		if (ret != -ERESTARTSYS)
J
Joe Perches 已提交
713
			pr_err("Buffer eviction failed\n");
714
		ttm_bo_mem_put(bo, &evict_mem);
715 716
		goto out;
	}
717 718 719 720 721 722 723
	bo->evicted = true;
out:
	return ret;
}

static int ttm_mem_evict_first(struct ttm_bo_device *bdev,
				uint32_t mem_type,
724
				const struct ttm_place *place,
725
				bool interruptible,
726
				bool no_wait_gpu)
727 728 729 730
{
	struct ttm_bo_global *glob = bdev->glob;
	struct ttm_mem_type_manager *man = &bdev->man[mem_type];
	struct ttm_buffer_object *bo;
731
	int ret = -EBUSY, put_count;
732

733
	spin_lock(&glob->lru_lock);
734
	list_for_each_entry(bo, &man->lru, lru) {
735
		ret = __ttm_bo_reserve(bo, false, true, NULL);
736 737 738 739 740 741 742 743 744 745 746 747 748
		if (!ret) {
			if (place && (place->fpfn || place->lpfn)) {
				/* Don't evict this BO if it's outside of the
				 * requested placement range
				 */
				if (place->fpfn >= (bo->mem.start + bo->mem.size) ||
				    (place->lpfn && place->lpfn <= bo->mem.start)) {
					__ttm_bo_unreserve(bo);
					ret = -EBUSY;
					continue;
				}
			}

749
			break;
750
		}
751 752 753
	}

	if (ret) {
754
		spin_unlock(&glob->lru_lock);
755
		return ret;
756 757
	}

758
	kref_get(&bo->list_kref);
759

760
	if (!list_empty(&bo->ddestroy)) {
761 762
		ret = ttm_bo_cleanup_refs_and_unlock(bo, interruptible,
						     no_wait_gpu);
763
		kref_put(&bo->list_kref, ttm_bo_release_list);
764
		return ret;
765 766
	}

767
	put_count = ttm_bo_del_from_lru(bo);
768
	spin_unlock(&glob->lru_lock);
769 770 771

	BUG_ON(ret != 0);

772
	ttm_bo_list_ref_sub(bo, put_count, true);
773

774
	ret = ttm_bo_evict(bo, interruptible, no_wait_gpu);
775
	ttm_bo_unreserve(bo);
776

777
	kref_put(&bo->list_kref, ttm_bo_release_list);
778 779 780
	return ret;
}

781 782
void ttm_bo_mem_put(struct ttm_buffer_object *bo, struct ttm_mem_reg *mem)
{
783
	struct ttm_mem_type_manager *man = &bo->bdev->man[mem->mem_type];
784

785 786
	if (mem->mm_node)
		(*man->func->put_node)(man, mem);
787 788 789
}
EXPORT_SYMBOL(ttm_bo_mem_put);

790 791 792 793
/**
 * Repeatedly evict memory from the LRU for @mem_type until we create enough
 * space, or we've evicted everything and there isn't enough space.
 */
794 795
static int ttm_bo_mem_force_space(struct ttm_buffer_object *bo,
					uint32_t mem_type,
796
					const struct ttm_place *place,
797
					struct ttm_mem_reg *mem,
798 799
					bool interruptible,
					bool no_wait_gpu)
800
{
801
	struct ttm_bo_device *bdev = bo->bdev;
802 803 804 805
	struct ttm_mem_type_manager *man = &bdev->man[mem_type];
	int ret;

	do {
806
		ret = (*man->func->get_node)(man, bo, place, mem);
807 808
		if (unlikely(ret != 0))
			return ret;
809
		if (mem->mm_node)
810
			break;
811
		ret = ttm_mem_evict_first(bdev, mem_type, place,
812
					  interruptible, no_wait_gpu);
813 814 815
		if (unlikely(ret != 0))
			return ret;
	} while (1);
816
	if (mem->mm_node == NULL)
817 818 819 820 821
		return -ENOMEM;
	mem->mem_type = mem_type;
	return 0;
}

822 823 824 825 826 827 828 829 830 831 832 833 834 835 836 837 838 839 840 841 842 843 844 845 846
static uint32_t ttm_bo_select_caching(struct ttm_mem_type_manager *man,
				      uint32_t cur_placement,
				      uint32_t proposed_placement)
{
	uint32_t caching = proposed_placement & TTM_PL_MASK_CACHING;
	uint32_t result = proposed_placement & ~TTM_PL_MASK_CACHING;

	/**
	 * Keep current caching if possible.
	 */

	if ((cur_placement & caching) != 0)
		result |= (cur_placement & caching);
	else if ((man->default_caching & caching) != 0)
		result |= man->default_caching;
	else if ((TTM_PL_FLAG_CACHED & caching) != 0)
		result |= TTM_PL_FLAG_CACHED;
	else if ((TTM_PL_FLAG_WC & caching) != 0)
		result |= TTM_PL_FLAG_WC;
	else if ((TTM_PL_FLAG_UNCACHED & caching) != 0)
		result |= TTM_PL_FLAG_UNCACHED;

	return result;
}

847 848
static bool ttm_bo_mt_compatible(struct ttm_mem_type_manager *man,
				 uint32_t mem_type,
849
				 const struct ttm_place *place,
850
				 uint32_t *masked_placement)
851 852 853
{
	uint32_t cur_flags = ttm_bo_type_flags(mem_type);

854
	if ((cur_flags & place->flags & TTM_PL_MASK_MEM) == 0)
855 856
		return false;

857
	if ((place->flags & man->available_caching) == 0)
858 859
		return false;

860
	cur_flags |= (place->flags & man->available_caching);
861 862

	*masked_placement = cur_flags;
863 864 865 866 867 868 869 870 871 872 873 874
	return true;
}

/**
 * Creates space for memory region @mem according to its type.
 *
 * This function first searches for free space in compatible memory types in
 * the priority order defined by the driver.  If free space isn't found, then
 * ttm_bo_mem_force_space is attempted in priority order to evict and find
 * space.
 */
int ttm_bo_mem_space(struct ttm_buffer_object *bo,
875 876
			struct ttm_placement *placement,
			struct ttm_mem_reg *mem,
877
			bool interruptible,
878
			bool no_wait_gpu)
879 880 881 882 883 884 885
{
	struct ttm_bo_device *bdev = bo->bdev;
	struct ttm_mem_type_manager *man;
	uint32_t mem_type = TTM_PL_SYSTEM;
	uint32_t cur_flags = 0;
	bool type_found = false;
	bool type_ok = false;
886
	bool has_erestartsys = false;
887
	int i, ret;
888 889

	mem->mm_node = NULL;
890
	for (i = 0; i < placement->num_placement; ++i) {
891 892 893
		const struct ttm_place *place = &placement->placement[i];

		ret = ttm_mem_type_from_place(place, &mem_type);
894 895
		if (ret)
			return ret;
896
		man = &bdev->man[mem_type];
897 898
		if (!man->has_type || !man->use_type)
			continue;
899

900
		type_ok = ttm_bo_mt_compatible(man, mem_type, place,
901
						&cur_flags);
902 903 904 905

		if (!type_ok)
			continue;

906
		type_found = true;
907 908
		cur_flags = ttm_bo_select_caching(man, bo->mem.placement,
						  cur_flags);
909 910 911 912
		/*
		 * Use the access and other non-mapping-related flag bits from
		 * the memory placement flags to the current flags
		 */
913
		ttm_flag_masked(&cur_flags, place->flags,
914
				~TTM_PL_MASK_MEMTYPE);
915

916 917 918
		if (mem_type == TTM_PL_SYSTEM)
			break;

919 920 921 922
		ret = (*man->func->get_node)(man, bo, place, mem);
		if (unlikely(ret))
			return ret;
		
923
		if (mem->mm_node)
924 925 926
			break;
	}

927
	if ((type_ok && (mem_type == TTM_PL_SYSTEM)) || mem->mm_node) {
928 929 930 931 932
		mem->mem_type = mem_type;
		mem->placement = cur_flags;
		return 0;
	}

933
	for (i = 0; i < placement->num_busy_placement; ++i) {
934 935 936
		const struct ttm_place *place = &placement->busy_placement[i];

		ret = ttm_mem_type_from_place(place, &mem_type);
937 938
		if (ret)
			return ret;
939
		man = &bdev->man[mem_type];
940
		if (!man->has_type || !man->use_type)
941
			continue;
942
		if (!ttm_bo_mt_compatible(man, mem_type, place, &cur_flags))
943 944
			continue;

945
		type_found = true;
946 947
		cur_flags = ttm_bo_select_caching(man, bo->mem.placement,
						  cur_flags);
948 949 950 951
		/*
		 * Use the access and other non-mapping-related flag bits from
		 * the memory placement flags to the current flags
		 */
952
		ttm_flag_masked(&cur_flags, place->flags,
953
				~TTM_PL_MASK_MEMTYPE);
954

955 956 957 958 959 960 961
		if (mem_type == TTM_PL_SYSTEM) {
			mem->mem_type = mem_type;
			mem->placement = cur_flags;
			mem->mm_node = NULL;
			return 0;
		}

962
		ret = ttm_bo_mem_force_space(bo, mem_type, place, mem,
963
						interruptible, no_wait_gpu);
964 965 966 967
		if (ret == 0 && mem->mm_node) {
			mem->placement = cur_flags;
			return 0;
		}
968 969
		if (ret == -ERESTARTSYS)
			has_erestartsys = true;
970
	}
971 972 973 974 975 976 977

	if (!type_found) {
		printk(KERN_ERR TTM_PFX "No compatible memory type found.\n");
		return -EINVAL;
	}

	return (has_erestartsys) ? -ERESTARTSYS : -ENOMEM;
978 979 980
}
EXPORT_SYMBOL(ttm_bo_mem_space);

981
static int ttm_bo_move_buffer(struct ttm_buffer_object *bo,
982
			struct ttm_placement *placement,
983
			bool interruptible,
984
			bool no_wait_gpu)
985 986 987 988
{
	int ret = 0;
	struct ttm_mem_reg mem;

989
	lockdep_assert_held(&bo->resv->lock.base);
990 991 992 993

	mem.num_pages = bo->num_pages;
	mem.size = mem.num_pages << PAGE_SHIFT;
	mem.page_alignment = bo->mem.page_alignment;
994 995
	mem.bus.io_reserved_vm = false;
	mem.bus.io_reserved_count = 0;
996 997 998
	/*
	 * Determine where to move the buffer.
	 */
999 1000
	ret = ttm_bo_mem_space(bo, placement, &mem,
			       interruptible, no_wait_gpu);
1001 1002
	if (ret)
		goto out_unlock;
1003 1004
	ret = ttm_bo_handle_move_mem(bo, &mem, false,
				     interruptible, no_wait_gpu);
1005
out_unlock:
1006 1007
	if (ret && mem.mm_node)
		ttm_bo_mem_put(bo, &mem);
1008 1009 1010
	return ret;
}

1011 1012 1013
static bool ttm_bo_mem_compat(struct ttm_placement *placement,
			      struct ttm_mem_reg *mem,
			      uint32_t *new_flags)
1014
{
1015
	int i;
1016

1017
	for (i = 0; i < placement->num_placement; i++) {
1018
		const struct ttm_place *heap = &placement->placement[i];
1019
		if (mem->mm_node &&
1020
		    (mem->start < heap->fpfn ||
1021
		     (heap->lpfn != 0 && (mem->start + mem->num_pages) > heap->lpfn)))
1022 1023 1024
			continue;

		*new_flags = heap->flags;
1025 1026 1027 1028 1029 1030
		if ((*new_flags & mem->placement & TTM_PL_MASK_CACHING) &&
		    (*new_flags & mem->placement & TTM_PL_MASK_MEM))
			return true;
	}

	for (i = 0; i < placement->num_busy_placement; i++) {
1031
		const struct ttm_place *heap = &placement->busy_placement[i];
1032
		if (mem->mm_node &&
1033
		    (mem->start < heap->fpfn ||
1034
		     (heap->lpfn != 0 && (mem->start + mem->num_pages) > heap->lpfn)))
1035 1036 1037
			continue;

		*new_flags = heap->flags;
1038 1039 1040
		if ((*new_flags & mem->placement & TTM_PL_MASK_CACHING) &&
		    (*new_flags & mem->placement & TTM_PL_MASK_MEM))
			return true;
1041
	}
1042 1043

	return false;
1044 1045
}

1046 1047
int ttm_bo_validate(struct ttm_buffer_object *bo,
			struct ttm_placement *placement,
1048
			bool interruptible,
1049
			bool no_wait_gpu)
1050 1051
{
	int ret;
1052
	uint32_t new_flags;
1053

1054
	lockdep_assert_held(&bo->resv->lock.base);
1055 1056 1057
	/*
	 * Check whether we need to move buffer.
	 */
1058
	if (!ttm_bo_mem_compat(placement, &bo->mem, &new_flags)) {
1059 1060
		ret = ttm_bo_move_buffer(bo, placement, interruptible,
					 no_wait_gpu);
1061
		if (ret)
1062
			return ret;
1063 1064 1065 1066 1067
	} else {
		/*
		 * Use the access and other non-mapping-related flag bits from
		 * the compatible memory placement flags to the active flags
		 */
1068
		ttm_flag_masked(&bo->mem.placement, new_flags,
1069
				~TTM_PL_MASK_MEMTYPE);
1070 1071 1072 1073 1074 1075 1076 1077 1078 1079 1080
	}
	/*
	 * We might need to add a TTM.
	 */
	if (bo->mem.mem_type == TTM_PL_SYSTEM && bo->ttm == NULL) {
		ret = ttm_bo_add_ttm(bo, true);
		if (ret)
			return ret;
	}
	return 0;
}
1081
EXPORT_SYMBOL(ttm_bo_validate);
1082

1083 1084 1085 1086 1087 1088 1089
int ttm_bo_init(struct ttm_bo_device *bdev,
		struct ttm_buffer_object *bo,
		unsigned long size,
		enum ttm_bo_type type,
		struct ttm_placement *placement,
		uint32_t page_alignment,
		bool interruptible,
J
Jan Engelhardt 已提交
1090
		struct file *persistent_swap_storage,
1091
		size_t acc_size,
1092
		struct sg_table *sg,
1093
		struct reservation_object *resv,
1094
		void (*destroy) (struct ttm_buffer_object *))
1095
{
1096
	int ret = 0;
1097
	unsigned long num_pages;
1098
	struct ttm_mem_global *mem_glob = bdev->glob->mem_glob;
1099
	bool locked;
1100 1101 1102

	ret = ttm_mem_global_alloc(mem_glob, acc_size, false, false);
	if (ret) {
J
Joe Perches 已提交
1103
		pr_err("Out of kernel memory\n");
1104 1105 1106 1107 1108 1109
		if (destroy)
			(*destroy)(bo);
		else
			kfree(bo);
		return -ENOMEM;
	}
1110 1111 1112

	num_pages = (size + PAGE_SIZE - 1) >> PAGE_SHIFT;
	if (num_pages == 0) {
J
Joe Perches 已提交
1113
		pr_err("Illegal buffer object size\n");
1114 1115 1116 1117
		if (destroy)
			(*destroy)(bo);
		else
			kfree(bo);
1118
		ttm_mem_global_free(mem_glob, acc_size);
1119 1120 1121 1122 1123 1124 1125 1126 1127 1128
		return -EINVAL;
	}
	bo->destroy = destroy;

	kref_init(&bo->kref);
	kref_init(&bo->list_kref);
	atomic_set(&bo->cpu_writers, 0);
	INIT_LIST_HEAD(&bo->lru);
	INIT_LIST_HEAD(&bo->ddestroy);
	INIT_LIST_HEAD(&bo->swap);
1129
	INIT_LIST_HEAD(&bo->io_reserve_lru);
1130
	mutex_init(&bo->wu_mutex);
1131
	bo->bdev = bdev;
1132
	bo->glob = bdev->glob;
1133 1134
	bo->type = type;
	bo->num_pages = num_pages;
1135
	bo->mem.size = num_pages << PAGE_SHIFT;
1136 1137 1138 1139
	bo->mem.mem_type = TTM_PL_SYSTEM;
	bo->mem.num_pages = bo->num_pages;
	bo->mem.mm_node = NULL;
	bo->mem.page_alignment = page_alignment;
1140 1141
	bo->mem.bus.io_reserved_vm = false;
	bo->mem.bus.io_reserved_count = 0;
1142
	bo->moving = NULL;
1143
	bo->mem.placement = (TTM_PL_FLAG_SYSTEM | TTM_PL_FLAG_CACHED);
J
Jan Engelhardt 已提交
1144
	bo->persistent_swap_storage = persistent_swap_storage;
1145
	bo->acc_size = acc_size;
1146
	bo->sg = sg;
1147 1148 1149 1150 1151 1152 1153
	if (resv) {
		bo->resv = resv;
		lockdep_assert_held(&bo->resv->lock.base);
	} else {
		bo->resv = &bo->ttm_resv;
		reservation_object_init(&bo->ttm_resv);
	}
1154
	atomic_inc(&bo->glob->bo_count);
1155
	drm_vma_node_reset(&bo->vma_node);
1156 1157 1158 1159 1160

	/*
	 * For ttm_bo_type_device buffers, allocate
	 * address space from the device.
	 */
1161 1162
	if (bo->type == ttm_bo_type_device ||
	    bo->type == ttm_bo_type_sg)
1163 1164
		ret = drm_vma_offset_add(&bdev->vma_manager, &bo->vma_node,
					 bo->mem.num_pages);
1165

1166 1167 1168 1169 1170 1171 1172
	/* passed reservation objects should already be locked,
	 * since otherwise lockdep will be angered in radeon.
	 */
	if (!resv) {
		locked = ww_mutex_trylock(&bo->resv->lock);
		WARN_ON(!locked);
	}
1173

1174 1175
	if (likely(!ret))
		ret = ttm_bo_validate(bo, placement, interruptible, false);
1176

1177
	if (!resv) {
1178
		ttm_bo_unreserve(bo);
1179

1180 1181 1182 1183 1184 1185
	} else if (!(bo->mem.placement & TTM_PL_FLAG_NO_EVICT)) {
		spin_lock(&bo->glob->lru_lock);
		ttm_bo_add_to_lru(bo);
		spin_unlock(&bo->glob->lru_lock);
	}

1186 1187
	if (unlikely(ret))
		ttm_bo_unref(&bo);
1188 1189 1190

	return ret;
}
1191
EXPORT_SYMBOL(ttm_bo_init);
1192

1193 1194 1195
size_t ttm_bo_acc_size(struct ttm_bo_device *bdev,
		       unsigned long bo_size,
		       unsigned struct_size)
1196
{
1197 1198
	unsigned npages = (PAGE_ALIGN(bo_size)) >> PAGE_SHIFT;
	size_t size = 0;
1199

1200
	size += ttm_round_pot(struct_size);
F
Felix Kuehling 已提交
1201
	size += ttm_round_pot(npages * sizeof(void *));
1202 1203
	size += ttm_round_pot(sizeof(struct ttm_tt));
	return size;
1204
}
1205 1206 1207 1208 1209 1210 1211 1212 1213 1214
EXPORT_SYMBOL(ttm_bo_acc_size);

size_t ttm_bo_dma_acc_size(struct ttm_bo_device *bdev,
			   unsigned long bo_size,
			   unsigned struct_size)
{
	unsigned npages = (PAGE_ALIGN(bo_size)) >> PAGE_SHIFT;
	size_t size = 0;

	size += ttm_round_pot(struct_size);
F
Felix Kuehling 已提交
1215
	size += ttm_round_pot(npages * (2*sizeof(void *) + sizeof(dma_addr_t)));
1216 1217 1218 1219
	size += ttm_round_pot(sizeof(struct ttm_dma_tt));
	return size;
}
EXPORT_SYMBOL(ttm_bo_dma_acc_size);
1220

1221 1222 1223 1224 1225 1226
int ttm_bo_create(struct ttm_bo_device *bdev,
			unsigned long size,
			enum ttm_bo_type type,
			struct ttm_placement *placement,
			uint32_t page_alignment,
			bool interruptible,
J
Jan Engelhardt 已提交
1227
			struct file *persistent_swap_storage,
1228
			struct ttm_buffer_object **p_bo)
1229 1230
{
	struct ttm_buffer_object *bo;
1231
	size_t acc_size;
1232
	int ret;
1233 1234

	bo = kzalloc(sizeof(*bo), GFP_KERNEL);
1235
	if (unlikely(bo == NULL))
1236 1237
		return -ENOMEM;

1238
	acc_size = ttm_bo_acc_size(bdev, size, sizeof(struct ttm_buffer_object));
1239
	ret = ttm_bo_init(bdev, bo, size, type, placement, page_alignment,
1240
			  interruptible, persistent_swap_storage, acc_size,
1241
			  NULL, NULL, NULL);
1242 1243 1244 1245 1246
	if (likely(ret == 0))
		*p_bo = bo;

	return ret;
}
T
Thomas Hellstrom 已提交
1247
EXPORT_SYMBOL(ttm_bo_create);
1248 1249

static int ttm_bo_force_list_clean(struct ttm_bo_device *bdev,
1250
					unsigned mem_type, bool allow_errors)
1251
{
1252
	struct ttm_mem_type_manager *man = &bdev->man[mem_type];
1253
	struct ttm_bo_global *glob = bdev->glob;
1254 1255 1256 1257 1258 1259
	int ret;

	/*
	 * Can't use standard list traversal since we're unlocking.
	 */

1260
	spin_lock(&glob->lru_lock);
1261
	while (!list_empty(&man->lru)) {
1262
		spin_unlock(&glob->lru_lock);
1263
		ret = ttm_mem_evict_first(bdev, mem_type, NULL, false, false);
1264 1265 1266 1267
		if (ret) {
			if (allow_errors) {
				return ret;
			} else {
J
Joe Perches 已提交
1268
				pr_err("Cleanup eviction failed\n");
1269 1270
			}
		}
1271
		spin_lock(&glob->lru_lock);
1272
	}
1273
	spin_unlock(&glob->lru_lock);
1274 1275 1276 1277 1278
	return 0;
}

int ttm_bo_clean_mm(struct ttm_bo_device *bdev, unsigned mem_type)
{
R
Roel Kluin 已提交
1279
	struct ttm_mem_type_manager *man;
1280 1281 1282
	int ret = -EINVAL;

	if (mem_type >= TTM_NUM_MEM_TYPES) {
J
Joe Perches 已提交
1283
		pr_err("Illegal memory type %d\n", mem_type);
1284 1285
		return ret;
	}
R
Roel Kluin 已提交
1286
	man = &bdev->man[mem_type];
1287 1288

	if (!man->has_type) {
J
Joe Perches 已提交
1289 1290
		pr_err("Trying to take down uninitialized memory manager type %u\n",
		       mem_type);
1291 1292 1293 1294 1295 1296 1297 1298
		return ret;
	}

	man->use_type = false;
	man->has_type = false;

	ret = 0;
	if (mem_type > 0) {
1299
		ttm_bo_force_list_clean(bdev, mem_type, false);
1300

1301
		ret = (*man->func->takedown)(man);
1302 1303 1304 1305 1306 1307 1308 1309 1310 1311 1312
	}

	return ret;
}
EXPORT_SYMBOL(ttm_bo_clean_mm);

int ttm_bo_evict_mm(struct ttm_bo_device *bdev, unsigned mem_type)
{
	struct ttm_mem_type_manager *man = &bdev->man[mem_type];

	if (mem_type == 0 || mem_type >= TTM_NUM_MEM_TYPES) {
J
Joe Perches 已提交
1313
		pr_err("Illegal memory manager memory type %u\n", mem_type);
1314 1315 1316 1317
		return -EINVAL;
	}

	if (!man->has_type) {
J
Joe Perches 已提交
1318
		pr_err("Memory type %u has not been initialized\n", mem_type);
1319 1320 1321
		return 0;
	}

1322
	return ttm_bo_force_list_clean(bdev, mem_type, true);
1323 1324 1325 1326
}
EXPORT_SYMBOL(ttm_bo_evict_mm);

int ttm_bo_init_mm(struct ttm_bo_device *bdev, unsigned type,
1327
			unsigned long p_size)
1328 1329 1330 1331
{
	int ret = -EINVAL;
	struct ttm_mem_type_manager *man;

1332
	BUG_ON(type >= TTM_NUM_MEM_TYPES);
1333
	man = &bdev->man[type];
1334
	BUG_ON(man->has_type);
1335 1336 1337 1338
	man->io_reserve_fastpath = true;
	man->use_io_reserve_lru = false;
	mutex_init(&man->io_reserve_mutex);
	INIT_LIST_HEAD(&man->io_reserve_lru);
1339 1340 1341 1342

	ret = bdev->driver->init_mem_type(bdev, type, man);
	if (ret)
		return ret;
1343
	man->bdev = bdev;
1344 1345 1346

	ret = 0;
	if (type != TTM_PL_SYSTEM) {
1347
		ret = (*man->func->init)(man, p_size);
1348 1349 1350 1351 1352 1353 1354 1355 1356 1357 1358 1359 1360
		if (ret)
			return ret;
	}
	man->has_type = true;
	man->use_type = true;
	man->size = p_size;

	INIT_LIST_HEAD(&man->lru);

	return 0;
}
EXPORT_SYMBOL(ttm_bo_init_mm);

1361 1362 1363 1364 1365 1366 1367 1368 1369 1370
static void ttm_bo_global_kobj_release(struct kobject *kobj)
{
	struct ttm_bo_global *glob =
		container_of(kobj, struct ttm_bo_global, kobj);

	ttm_mem_unregister_shrink(glob->mem_glob, &glob->shrink);
	__free_page(glob->dummy_read_page);
	kfree(glob);
}

1371
void ttm_bo_global_release(struct drm_global_reference *ref)
1372 1373 1374 1375 1376 1377 1378 1379
{
	struct ttm_bo_global *glob = ref->object;

	kobject_del(&glob->kobj);
	kobject_put(&glob->kobj);
}
EXPORT_SYMBOL(ttm_bo_global_release);

1380
int ttm_bo_global_init(struct drm_global_reference *ref)
1381 1382 1383 1384 1385 1386 1387 1388 1389 1390 1391 1392 1393 1394 1395 1396 1397 1398 1399 1400 1401 1402
{
	struct ttm_bo_global_ref *bo_ref =
		container_of(ref, struct ttm_bo_global_ref, ref);
	struct ttm_bo_global *glob = ref->object;
	int ret;

	mutex_init(&glob->device_list_mutex);
	spin_lock_init(&glob->lru_lock);
	glob->mem_glob = bo_ref->mem_glob;
	glob->dummy_read_page = alloc_page(__GFP_ZERO | GFP_DMA32);

	if (unlikely(glob->dummy_read_page == NULL)) {
		ret = -ENOMEM;
		goto out_no_drp;
	}

	INIT_LIST_HEAD(&glob->swap_lru);
	INIT_LIST_HEAD(&glob->device_list);

	ttm_mem_init_shrink(&glob->shrink, ttm_bo_swapout);
	ret = ttm_mem_register_shrink(glob->mem_glob, &glob->shrink);
	if (unlikely(ret != 0)) {
J
Joe Perches 已提交
1403
		pr_err("Could not register buffer object swapout\n");
1404 1405 1406 1407 1408
		goto out_no_shrink;
	}

	atomic_set(&glob->bo_count, 0);

1409 1410
	ret = kobject_init_and_add(
		&glob->kobj, &ttm_bo_glob_kobj_type, ttm_get_kobj(), "buffer_objects");
1411 1412 1413 1414 1415 1416 1417 1418 1419 1420 1421 1422
	if (unlikely(ret != 0))
		kobject_put(&glob->kobj);
	return ret;
out_no_shrink:
	__free_page(glob->dummy_read_page);
out_no_drp:
	kfree(glob);
	return ret;
}
EXPORT_SYMBOL(ttm_bo_global_init);


1423 1424 1425 1426 1427
int ttm_bo_device_release(struct ttm_bo_device *bdev)
{
	int ret = 0;
	unsigned i = TTM_NUM_MEM_TYPES;
	struct ttm_mem_type_manager *man;
1428
	struct ttm_bo_global *glob = bdev->glob;
1429 1430 1431 1432 1433 1434 1435

	while (i--) {
		man = &bdev->man[i];
		if (man->has_type) {
			man->use_type = false;
			if ((i != TTM_PL_SYSTEM) && ttm_bo_clean_mm(bdev, i)) {
				ret = -EBUSY;
J
Joe Perches 已提交
1436 1437
				pr_err("DRM memory manager type %d is not clean\n",
				       i);
1438 1439 1440 1441 1442
			}
			man->has_type = false;
		}
	}

1443 1444 1445 1446
	mutex_lock(&glob->device_list_mutex);
	list_del(&bdev->device_list);
	mutex_unlock(&glob->device_list_mutex);

1447
	cancel_delayed_work_sync(&bdev->wq);
1448 1449 1450 1451

	while (ttm_bo_delayed_delete(bdev, true))
		;

1452
	spin_lock(&glob->lru_lock);
1453 1454 1455 1456 1457
	if (list_empty(&bdev->ddestroy))
		TTM_DEBUG("Delayed destroy list was clean\n");

	if (list_empty(&bdev->man[0].lru))
		TTM_DEBUG("Swap list was clean\n");
1458
	spin_unlock(&glob->lru_lock);
1459

1460
	drm_vma_offset_manager_destroy(&bdev->vma_manager);
1461 1462 1463 1464 1465 1466

	return ret;
}
EXPORT_SYMBOL(ttm_bo_device_release);

int ttm_bo_device_init(struct ttm_bo_device *bdev,
1467 1468
		       struct ttm_bo_global *glob,
		       struct ttm_bo_driver *driver,
1469
		       struct address_space *mapping,
D
Dave Airlie 已提交
1470
		       uint64_t file_page_offset,
D
Dave Airlie 已提交
1471
		       bool need_dma32)
1472 1473 1474 1475 1476 1477 1478 1479 1480 1481 1482
{
	int ret = -EINVAL;

	bdev->driver = driver;

	memset(bdev->man, 0, sizeof(bdev->man));

	/*
	 * Initialize the system memory buffer type.
	 * Other types need to be driver / IOCTL initialized.
	 */
1483
	ret = ttm_bo_init_mm(bdev, TTM_PL_SYSTEM, 0);
1484
	if (unlikely(ret != 0))
1485
		goto out_no_sys;
1486

1487 1488
	drm_vma_offset_manager_init(&bdev->vma_manager, file_page_offset,
				    0x10000000);
1489 1490
	INIT_DELAYED_WORK(&bdev->wq, ttm_bo_delayed_workqueue);
	INIT_LIST_HEAD(&bdev->ddestroy);
1491
	bdev->dev_mapping = mapping;
1492
	bdev->glob = glob;
D
Dave Airlie 已提交
1493
	bdev->need_dma32 = need_dma32;
1494 1495 1496
	mutex_lock(&glob->device_list_mutex);
	list_add_tail(&bdev->device_list, &glob->device_list);
	mutex_unlock(&glob->device_list_mutex);
1497 1498

	return 0;
1499
out_no_sys:
1500 1501 1502 1503 1504 1505 1506 1507 1508 1509 1510 1511 1512 1513 1514 1515 1516 1517 1518 1519 1520 1521 1522 1523 1524
	return ret;
}
EXPORT_SYMBOL(ttm_bo_device_init);

/*
 * buffer object vm functions.
 */

bool ttm_mem_reg_is_pci(struct ttm_bo_device *bdev, struct ttm_mem_reg *mem)
{
	struct ttm_mem_type_manager *man = &bdev->man[mem->mem_type];

	if (!(man->flags & TTM_MEMTYPE_FLAG_FIXED)) {
		if (mem->mem_type == TTM_PL_SYSTEM)
			return false;

		if (man->flags & TTM_MEMTYPE_FLAG_CMA)
			return false;

		if (mem->placement & TTM_PL_FLAG_CACHED)
			return false;
	}
	return true;
}

1525
void ttm_bo_unmap_virtual_locked(struct ttm_buffer_object *bo)
1526 1527 1528
{
	struct ttm_bo_device *bdev = bo->bdev;

1529
	drm_vma_node_unmap(&bo->vma_node, bdev->dev_mapping);
1530
	ttm_mem_io_free_vm(bo);
1531
}
1532 1533 1534 1535 1536 1537 1538 1539 1540

void ttm_bo_unmap_virtual(struct ttm_buffer_object *bo)
{
	struct ttm_bo_device *bdev = bo->bdev;
	struct ttm_mem_type_manager *man = &bdev->man[bo->mem.mem_type];

	ttm_mem_io_lock(man, false);
	ttm_bo_unmap_virtual_locked(bo);
	ttm_mem_io_unlock(man);
1541
}
1542 1543


1544
EXPORT_SYMBOL(ttm_bo_unmap_virtual);
1545 1546

int ttm_bo_wait(struct ttm_buffer_object *bo,
1547
		bool interruptible, bool no_wait)
1548
{
1549 1550 1551 1552 1553
	struct reservation_object_list *fobj;
	struct reservation_object *resv;
	struct fence *excl;
	long timeout = 15 * HZ;
	int i;
1554

1555 1556 1557 1558 1559 1560 1561
	resv = bo->resv;
	fobj = reservation_object_get_list(resv);
	excl = reservation_object_get_excl(resv);
	if (excl) {
		if (!fence_is_signaled(excl)) {
			if (no_wait)
				return -EBUSY;
1562

1563 1564
			timeout = fence_wait_timeout(excl,
						     interruptible, timeout);
1565
		}
1566
	}
1567

1568 1569 1570 1571
	for (i = 0; fobj && timeout > 0 && i < fobj->shared_count; ++i) {
		struct fence *fence;
		fence = rcu_dereference_protected(fobj->shared[i],
						reservation_object_held(resv));
1572

1573 1574 1575
		if (!fence_is_signaled(fence)) {
			if (no_wait)
				return -EBUSY;
M
Maarten Lankhorst 已提交
1576

1577 1578
			timeout = fence_wait_timeout(fence,
						     interruptible, timeout);
1579 1580
		}
	}
1581 1582 1583 1584 1585 1586 1587 1588 1589

	if (timeout < 0)
		return timeout;

	if (timeout == 0)
		return -EBUSY;

	reservation_object_add_excl_fence(resv, NULL);
	return 0;
1590 1591 1592 1593 1594 1595 1596 1597
}
EXPORT_SYMBOL(ttm_bo_wait);

int ttm_bo_synccpu_write_grab(struct ttm_buffer_object *bo, bool no_wait)
{
	int ret = 0;

	/*
1598
	 * Using ttm_bo_reserve makes sure the lru lists are updated.
1599 1600
	 */

1601
	ret = ttm_bo_reserve(bo, true, no_wait, NULL);
1602 1603
	if (unlikely(ret != 0))
		return ret;
1604
	ret = ttm_bo_wait(bo, true, no_wait);
1605 1606 1607 1608 1609
	if (likely(ret == 0))
		atomic_inc(&bo->cpu_writers);
	ttm_bo_unreserve(bo);
	return ret;
}
1610
EXPORT_SYMBOL(ttm_bo_synccpu_write_grab);
1611 1612 1613

void ttm_bo_synccpu_write_release(struct ttm_buffer_object *bo)
{
1614
	atomic_dec(&bo->cpu_writers);
1615
}
1616
EXPORT_SYMBOL(ttm_bo_synccpu_write_release);
1617 1618 1619 1620 1621 1622 1623 1624

/**
 * A buffer object shrink method that tries to swap out the first
 * buffer object on the bo_global::swap_lru list.
 */

static int ttm_bo_swapout(struct ttm_mem_shrink *shrink)
{
1625 1626
	struct ttm_bo_global *glob =
	    container_of(shrink, struct ttm_bo_global, shrink);
1627 1628 1629 1630 1631
	struct ttm_buffer_object *bo;
	int ret = -EBUSY;
	int put_count;
	uint32_t swap_placement = (TTM_PL_FLAG_CACHED | TTM_PL_FLAG_SYSTEM);

1632
	spin_lock(&glob->lru_lock);
1633
	list_for_each_entry(bo, &glob->swap_lru, swap) {
1634
		ret = __ttm_bo_reserve(bo, false, true, NULL);
1635 1636 1637
		if (!ret)
			break;
	}
1638

1639 1640 1641 1642
	if (ret) {
		spin_unlock(&glob->lru_lock);
		return ret;
	}
1643

1644
	kref_get(&bo->list_kref);
1645

1646 1647 1648 1649
	if (!list_empty(&bo->ddestroy)) {
		ret = ttm_bo_cleanup_refs_and_unlock(bo, false, false);
		kref_put(&bo->list_kref, ttm_bo_release_list);
		return ret;
1650 1651 1652
	}

	put_count = ttm_bo_del_from_lru(bo);
1653
	spin_unlock(&glob->lru_lock);
1654

1655
	ttm_bo_list_ref_sub(bo, put_count, true);
1656 1657

	/**
1658
	 * Move to system cached
1659 1660 1661 1662 1663 1664 1665 1666 1667 1668 1669
	 */

	if ((bo->mem.placement & swap_placement) != swap_placement) {
		struct ttm_mem_reg evict_mem;

		evict_mem = bo->mem;
		evict_mem.mm_node = NULL;
		evict_mem.placement = TTM_PL_FLAG_SYSTEM | TTM_PL_FLAG_CACHED;
		evict_mem.mem_type = TTM_PL_SYSTEM;

		ret = ttm_bo_handle_move_mem(bo, &evict_mem, true,
1670
					     false, false);
1671 1672 1673 1674
		if (unlikely(ret != 0))
			goto out;
	}

1675 1676 1677 1678 1679 1680 1681 1682
	/**
	 * Make sure BO is idle.
	 */

	ret = ttm_bo_wait(bo, false, false);
	if (unlikely(ret != 0))
		goto out;

1683 1684 1685 1686 1687 1688 1689
	ttm_bo_unmap_virtual(bo);

	/**
	 * Swap out. Buffer will be swapped in again as soon as
	 * anyone tries to access a ttm page.
	 */

1690 1691 1692
	if (bo->bdev->driver->swap_notify)
		bo->bdev->driver->swap_notify(bo);

J
Jan Engelhardt 已提交
1693
	ret = ttm_tt_swapout(bo->ttm, bo->persistent_swap_storage);
1694 1695 1696 1697 1698 1699 1700 1701
out:

	/**
	 *
	 * Unreserve without putting on LRU to avoid swapping out an
	 * already swapped buffer.
	 */

1702
	__ttm_bo_unreserve(bo);
1703 1704 1705 1706 1707 1708
	kref_put(&bo->list_kref, ttm_bo_release_list);
	return ret;
}

void ttm_bo_swapout_all(struct ttm_bo_device *bdev)
{
1709
	while (ttm_bo_swapout(&bdev->glob->shrink) == 0)
1710 1711
		;
}
1712
EXPORT_SYMBOL(ttm_bo_swapout_all);
1713 1714 1715 1716 1717 1718 1719 1720 1721 1722 1723 1724 1725 1726 1727 1728 1729 1730 1731 1732 1733 1734 1735

/**
 * ttm_bo_wait_unreserved - interruptible wait for a buffer object to become
 * unreserved
 *
 * @bo: Pointer to buffer
 */
int ttm_bo_wait_unreserved(struct ttm_buffer_object *bo)
{
	int ret;

	/*
	 * In the absense of a wait_unlocked API,
	 * Use the bo::wu_mutex to avoid triggering livelocks due to
	 * concurrent use of this function. Note that this use of
	 * bo::wu_mutex can go away if we change locking order to
	 * mmap_sem -> bo::reserve.
	 */
	ret = mutex_lock_interruptible(&bo->wu_mutex);
	if (unlikely(ret != 0))
		return -ERESTARTSYS;
	if (!ww_mutex_is_locked(&bo->resv->lock))
		goto out_unlock;
1736
	ret = __ttm_bo_reserve(bo, true, false, NULL);
1737 1738
	if (unlikely(ret != 0))
		goto out_unlock;
1739
	__ttm_bo_unreserve(bo);
1740 1741 1742 1743 1744

out_unlock:
	mutex_unlock(&bo->wu_mutex);
	return ret;
}