// SPDX-License-Identifier: GPL-2.0 AND MIT /* * Copyright © 2023 Intel Corporation */ #include #include "xe_device.h" #include "xe_kunit_helpers.h" static int guc_dbm_test_init(struct kunit *test) { struct xe_guc_db_mgr *dbm; xe_kunit_helper_xe_device_test_init(test); dbm = &xe_device_get_gt(test->priv, 0)->uc.guc.dbm; mutex_init(dbm_mutex(dbm)); test->priv = dbm; return 0; } static void test_empty(struct kunit *test) { struct xe_guc_db_mgr *dbm = test->priv; KUNIT_ASSERT_EQ(test, xe_guc_db_mgr_init(dbm, 0), 0); KUNIT_ASSERT_EQ(test, dbm->count, 0); mutex_lock(dbm_mutex(dbm)); KUNIT_EXPECT_LT(test, xe_guc_db_mgr_reserve_id_locked(dbm), 0); mutex_unlock(dbm_mutex(dbm)); KUNIT_EXPECT_LT(test, xe_guc_db_mgr_reserve_range(dbm, 1, 0), 0); } static void test_default(struct kunit *test) { struct xe_guc_db_mgr *dbm = test->priv; KUNIT_ASSERT_EQ(test, xe_guc_db_mgr_init(dbm, ~0), 0); KUNIT_ASSERT_EQ(test, dbm->count, GUC_NUM_DOORBELLS); } static const unsigned int guc_dbm_params[] = { GUC_NUM_DOORBELLS / 64, GUC_NUM_DOORBELLS / 32, GUC_NUM_DOORBELLS / 8, GUC_NUM_DOORBELLS, }; static void uint_param_get_desc(const unsigned int *p, char *desc) { snprintf(desc, KUNIT_PARAM_DESC_SIZE, "%u", *p); } KUNIT_ARRAY_PARAM(guc_dbm, guc_dbm_params, uint_param_get_desc); static void test_size(struct kunit *test) { const unsigned int *p = test->param_value; struct xe_guc_db_mgr *dbm = test->priv; unsigned int n; int id; KUNIT_ASSERT_EQ(test, xe_guc_db_mgr_init(dbm, *p), 0); KUNIT_ASSERT_EQ(test, dbm->count, *p); mutex_lock(dbm_mutex(dbm)); for (n = 0; n < *p; n++) { KUNIT_EXPECT_GE(test, id = xe_guc_db_mgr_reserve_id_locked(dbm), 0); KUNIT_EXPECT_LT(test, id, dbm->count); } KUNIT_EXPECT_LT(test, xe_guc_db_mgr_reserve_id_locked(dbm), 0); mutex_unlock(dbm_mutex(dbm)); mutex_lock(dbm_mutex(dbm)); for (n = 0; n < *p; n++) xe_guc_db_mgr_release_id_locked(dbm, n); mutex_unlock(dbm_mutex(dbm)); } static void test_reuse(struct kunit *test) { const unsigned int *p = test->param_value; struct xe_guc_db_mgr *dbm = test->priv; unsigned int n; KUNIT_ASSERT_EQ(test, xe_guc_db_mgr_init(dbm, *p), 0); mutex_lock(dbm_mutex(dbm)); for (n = 0; n < *p; n++) KUNIT_EXPECT_GE(test, xe_guc_db_mgr_reserve_id_locked(dbm), 0); KUNIT_EXPECT_LT(test, xe_guc_db_mgr_reserve_id_locked(dbm), 0); mutex_unlock(dbm_mutex(dbm)); mutex_lock(dbm_mutex(dbm)); for (n = 0; n < *p; n++) { xe_guc_db_mgr_release_id_locked(dbm, n); KUNIT_EXPECT_EQ(test, xe_guc_db_mgr_reserve_id_locked(dbm), n); } KUNIT_EXPECT_LT(test, xe_guc_db_mgr_reserve_id_locked(dbm), 0); mutex_unlock(dbm_mutex(dbm)); mutex_lock(dbm_mutex(dbm)); for (n = 0; n < *p; n++) xe_guc_db_mgr_release_id_locked(dbm, n); mutex_unlock(dbm_mutex(dbm)); } static void test_range_overlap(struct kunit *test) { const unsigned int *p = test->param_value; struct xe_guc_db_mgr *dbm = test->priv; int id1, id2, id3; unsigned int n; KUNIT_ASSERT_EQ(test, xe_guc_db_mgr_init(dbm, ~0), 0); KUNIT_ASSERT_LE(test, *p, dbm->count); KUNIT_ASSERT_GE(test, id1 = xe_guc_db_mgr_reserve_range(dbm, *p, 0), 0); for (n = 0; n < dbm->count - *p; n++) { KUNIT_ASSERT_GE(test, id2 = xe_guc_db_mgr_reserve_range(dbm, 1, 0), 0); KUNIT_ASSERT_NE(test, id2, id1); KUNIT_ASSERT_NE_MSG(test, id2 < id1, id2 > id1 + *p - 1, "id1=%d id2=%d", id1, id2); } KUNIT_ASSERT_LT(test, xe_guc_db_mgr_reserve_range(dbm, 1, 0), 0); xe_guc_db_mgr_release_range(dbm, 0, dbm->count); if (*p >= 1) { KUNIT_ASSERT_GE(test, id1 = xe_guc_db_mgr_reserve_range(dbm, 1, 0), 0); KUNIT_ASSERT_GE(test, id2 = xe_guc_db_mgr_reserve_range(dbm, *p - 1, 0), 0); KUNIT_ASSERT_NE(test, id2, id1); KUNIT_ASSERT_NE_MSG(test, id1 < id2, id1 > id2 + *p - 2, "id1=%d id2=%d", id1, id2); for (n = 0; n < dbm->count - *p; n++) { KUNIT_ASSERT_GE(test, id3 = xe_guc_db_mgr_reserve_range(dbm, 1, 0), 0); KUNIT_ASSERT_NE(test, id3, id1); KUNIT_ASSERT_NE(test, id3, id2); KUNIT_ASSERT_NE_MSG(test, id3 < id2, id3 > id2 + *p - 2, "id3=%d id2=%d", id3, id2); } KUNIT_ASSERT_LT(test, xe_guc_db_mgr_reserve_range(dbm, 1, 0), 0); xe_guc_db_mgr_release_range(dbm, 0, dbm->count); } } static void test_range_compact(struct kunit *test) { const unsigned int *p = test->param_value; struct xe_guc_db_mgr *dbm = test->priv; unsigned int n; KUNIT_ASSERT_EQ(test, xe_guc_db_mgr_init(dbm, ~0), 0); KUNIT_ASSERT_NE(test, *p, 0); KUNIT_ASSERT_LE(test, *p, dbm->count); if (dbm->count % *p) kunit_skip(test, "must be divisible"); KUNIT_ASSERT_GE(test, xe_guc_db_mgr_reserve_range(dbm, *p, 0), 0); for (n = 1; n < dbm->count / *p; n++) KUNIT_ASSERT_GE(test, xe_guc_db_mgr_reserve_range(dbm, *p, 0), 0); KUNIT_ASSERT_LT(test, xe_guc_db_mgr_reserve_range(dbm, 1, 0), 0); xe_guc_db_mgr_release_range(dbm, 0, dbm->count); } static void test_range_spare(struct kunit *test) { const unsigned int *p = test->param_value; struct xe_guc_db_mgr *dbm = test->priv; int id; KUNIT_ASSERT_EQ(test, xe_guc_db_mgr_init(dbm, ~0), 0); KUNIT_ASSERT_LE(test, *p, dbm->count); KUNIT_ASSERT_LT(test, xe_guc_db_mgr_reserve_range(dbm, *p, dbm->count), 0); KUNIT_ASSERT_LT(test, xe_guc_db_mgr_reserve_range(dbm, *p, dbm->count - *p + 1), 0); KUNIT_ASSERT_EQ(test, id = xe_guc_db_mgr_reserve_range(dbm, *p, dbm->count - *p), 0); KUNIT_ASSERT_LT(test, xe_guc_db_mgr_reserve_range(dbm, 1, dbm->count - *p), 0); xe_guc_db_mgr_release_range(dbm, id, *p); } static struct kunit_case guc_dbm_test_cases[] = { KUNIT_CASE(test_empty), KUNIT_CASE(test_default), KUNIT_CASE_PARAM(test_size, guc_dbm_gen_params), KUNIT_CASE_PARAM(test_reuse, guc_dbm_gen_params), KUNIT_CASE_PARAM(test_range_overlap, guc_dbm_gen_params), KUNIT_CASE_PARAM(test_range_compact, guc_dbm_gen_params), KUNIT_CASE_PARAM(test_range_spare, guc_dbm_gen_params), {} }; static struct kunit_suite guc_dbm_suite = { .name = "guc_dbm", .test_cases = guc_dbm_test_cases, .init = guc_dbm_test_init, }; kunit_test_suites(&guc_dbm_suite);