| // SPDX-License-Identifier: GPL-2.0 AND MIT |
| /* |
| * Copyright © 2023 Intel Corporation |
| */ |
| |
| #include <kunit/test.h> |
| |
| #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); |