ttm_device_test.c

  1// SPDX-License-Identifier: GPL-2.0 AND MIT
  2/*
  3 * Copyright © 2023 Intel Corporation
  4 */
  5#include <drm/ttm/ttm_resource.h>
  6#include <drm/ttm/ttm_device.h>
  7#include <drm/ttm/ttm_placement.h>
  8
  9#include "ttm_kunit_helpers.h"
 10#include "../ttm_pool_internal.h"
 11
 12struct ttm_device_test_case {
 13	const char *description;
 14	unsigned int alloc_flags;
 15	bool pools_init_expected;
 16};
 17
 18static void ttm_device_init_basic(struct kunit *test)
 19{
 20	struct ttm_test_devices *priv = test->priv;
 21	struct ttm_device *ttm_dev;
 22	struct ttm_resource_manager *ttm_sys_man;
 23	int err;
 24
 25	ttm_dev = kunit_kzalloc(test, sizeof(*ttm_dev), GFP_KERNEL);
 26	KUNIT_ASSERT_NOT_NULL(test, ttm_dev);
 27
 28	err = ttm_device_kunit_init(priv, ttm_dev, 0);
 29	KUNIT_ASSERT_EQ(test, err, 0);
 30
 31	KUNIT_EXPECT_PTR_EQ(test, ttm_dev->funcs, &ttm_dev_funcs);
 32	KUNIT_ASSERT_NOT_NULL(test, ttm_dev->wq);
 33	KUNIT_ASSERT_NOT_NULL(test, ttm_dev->man_drv[TTM_PL_SYSTEM]);
 34
 35	ttm_sys_man = &ttm_dev->sysman;
 36	KUNIT_ASSERT_NOT_NULL(test, ttm_sys_man);
 37	KUNIT_EXPECT_TRUE(test, ttm_sys_man->use_tt);
 38	KUNIT_EXPECT_TRUE(test, ttm_sys_man->use_type);
 39	KUNIT_ASSERT_NOT_NULL(test, ttm_sys_man->func);
 40
 41	KUNIT_EXPECT_PTR_EQ(test, ttm_dev->dev_mapping,
 42			    priv->drm->anon_inode->i_mapping);
 43
 44	ttm_device_fini(ttm_dev);
 45}
 46
 47static void ttm_device_init_multiple(struct kunit *test)
 48{
 49	struct ttm_test_devices *priv = test->priv;
 50	struct ttm_device *ttm_devs;
 51	unsigned int i, num_dev = 3;
 52	int err;
 53
 54	ttm_devs = kunit_kcalloc(test, num_dev, sizeof(*ttm_devs), GFP_KERNEL);
 55	KUNIT_ASSERT_NOT_NULL(test, ttm_devs);
 56
 57	for (i = 0; i < num_dev; i++) {
 58		err = ttm_device_kunit_init(priv, &ttm_devs[i], 0);
 59		KUNIT_ASSERT_EQ(test, err, 0);
 60
 61		KUNIT_EXPECT_PTR_EQ(test, ttm_devs[i].dev_mapping,
 62				    priv->drm->anon_inode->i_mapping);
 63		KUNIT_ASSERT_NOT_NULL(test, ttm_devs[i].wq);
 64		KUNIT_EXPECT_PTR_EQ(test, ttm_devs[i].funcs, &ttm_dev_funcs);
 65		KUNIT_ASSERT_NOT_NULL(test, ttm_devs[i].man_drv[TTM_PL_SYSTEM]);
 66	}
 67
 68	KUNIT_ASSERT_EQ(test, list_count_nodes(&ttm_devs[0].device_list), num_dev);
 69
 70	for (i = 0; i < num_dev; i++)
 71		ttm_device_fini(&ttm_devs[i]);
 72}
 73
 74static void ttm_device_fini_basic(struct kunit *test)
 75{
 76	struct ttm_test_devices *priv = test->priv;
 77	struct ttm_device *ttm_dev;
 78	struct ttm_resource_manager *man;
 79	int err;
 80
 81	ttm_dev = kunit_kzalloc(test, sizeof(*ttm_dev), GFP_KERNEL);
 82	KUNIT_ASSERT_NOT_NULL(test, ttm_dev);
 83
 84	err = ttm_device_kunit_init(priv, ttm_dev, 0);
 85	KUNIT_ASSERT_EQ(test, err, 0);
 86
 87	man = ttm_manager_type(ttm_dev, TTM_PL_SYSTEM);
 88	KUNIT_ASSERT_NOT_NULL(test, man);
 89
 90	ttm_device_fini(ttm_dev);
 91
 92	KUNIT_ASSERT_FALSE(test, man->use_type);
 93	KUNIT_ASSERT_TRUE(test, list_empty(&man->lru[0]));
 94	KUNIT_ASSERT_NULL(test, ttm_dev->man_drv[TTM_PL_SYSTEM]);
 95}
 96
 97static void ttm_device_init_no_vma_man(struct kunit *test)
 98{
 99	struct ttm_test_devices *priv = test->priv;
100	struct drm_device *drm = priv->drm;
101	struct ttm_device *ttm_dev;
102	struct drm_vma_offset_manager *vma_man;
103	int err;
104
105	ttm_dev = kunit_kzalloc(test, sizeof(*ttm_dev), GFP_KERNEL);
106	KUNIT_ASSERT_NOT_NULL(test, ttm_dev);
107
108	/* Let's pretend there's no VMA manager allocated */
109	vma_man = drm->vma_offset_manager;
110	drm->vma_offset_manager = NULL;
111
112	err = ttm_device_kunit_init(priv, ttm_dev, 0);
113	KUNIT_EXPECT_EQ(test, err, -EINVAL);
114
115	/* Bring the manager back for a graceful cleanup */
116	drm->vma_offset_manager = vma_man;
117}
118
119static const struct ttm_device_test_case ttm_device_cases[] = {
120	{
121		.description = "No DMA allocations, no DMA32 required",
122		.pools_init_expected = false,
123	},
124	{
125		.description = "DMA allocations, DMA32 required",
126		.alloc_flags = TTM_ALLOCATION_POOL_USE_DMA_ALLOC |
127			       TTM_ALLOCATION_POOL_USE_DMA32,
128		.pools_init_expected = true,
129	},
130	{
131		.description = "No DMA allocations, DMA32 required",
132		.alloc_flags = TTM_ALLOCATION_POOL_USE_DMA32,
133		.pools_init_expected = false,
134	},
135	{
136		.description = "DMA allocations, no DMA32 required",
137		.alloc_flags = TTM_ALLOCATION_POOL_USE_DMA_ALLOC,
138		.pools_init_expected = true,
139	},
140};
141
142static void ttm_device_case_desc(const struct ttm_device_test_case *t, char *desc)
143{
144	strscpy(desc, t->description, KUNIT_PARAM_DESC_SIZE);
145}
146
147KUNIT_ARRAY_PARAM(ttm_device, ttm_device_cases, ttm_device_case_desc);
148
149static void ttm_device_init_pools(struct kunit *test)
150{
151	struct ttm_test_devices *priv = test->priv;
152	const struct ttm_device_test_case *params = test->param_value;
153	struct ttm_device *ttm_dev;
154	struct ttm_pool *pool;
155	struct ttm_pool_type pt;
156	int err;
157
158	ttm_dev = kunit_kzalloc(test, sizeof(*ttm_dev), GFP_KERNEL);
159	KUNIT_ASSERT_NOT_NULL(test, ttm_dev);
160
161	err = ttm_device_kunit_init(priv, ttm_dev, params->alloc_flags);
162	KUNIT_ASSERT_EQ(test, err, 0);
163
164	pool = &ttm_dev->pool;
165	KUNIT_ASSERT_NOT_NULL(test, pool);
166	KUNIT_EXPECT_PTR_EQ(test, pool->dev, priv->dev);
167	KUNIT_EXPECT_EQ(test, pool->alloc_flags, params->alloc_flags);
168
169	if (params->pools_init_expected) {
170		for (int i = 0; i < TTM_NUM_CACHING_TYPES; ++i) {
171			for (int j = 0; j < NR_PAGE_ORDERS; ++j) {
172				pt = pool->caching[i].orders[j];
173				KUNIT_EXPECT_PTR_EQ(test, pt.pool, pool);
174				KUNIT_EXPECT_EQ(test, pt.caching, i);
175				KUNIT_EXPECT_EQ(test, pt.order, j);
176
177				if (ttm_pool_uses_dma_alloc(pool))
178					KUNIT_ASSERT_FALSE(test,
179							   list_empty(&pt.pages));
180			}
181		}
182	}
183
184	ttm_device_fini(ttm_dev);
185}
186
187static struct kunit_case ttm_device_test_cases[] = {
188	KUNIT_CASE(ttm_device_init_basic),
189	KUNIT_CASE(ttm_device_init_multiple),
190	KUNIT_CASE(ttm_device_fini_basic),
191	KUNIT_CASE(ttm_device_init_no_vma_man),
192	KUNIT_CASE_PARAM(ttm_device_init_pools, ttm_device_gen_params),
193	{}
194};
195
196static struct kunit_suite ttm_device_test_suite = {
197	.name = "ttm_device",
198	.init = ttm_test_devices_init,
199	.exit = ttm_test_devices_fini,
200	.test_cases = ttm_device_test_cases,
201};
202
203kunit_test_suites(&ttm_device_test_suite);
204
205MODULE_DESCRIPTION("KUnit tests for ttm_device APIs");
206MODULE_LICENSE("GPL and additional rights");