drm/sched: Add a basic test for checking credit limit

Add a basic test for checking whether scheduler respects the configured
credit limit.

Signed-off-by: Tvrtko Ursulin <tvrtko.ursulin@igalia.com>
Cc: Christian König <christian.koenig@amd.com>
Cc: Danilo Krummrich <dakr@kernel.org>
Cc: Matthew Brost <matthew.brost@intel.com>
Cc: Philipp Stanner <phasta@kernel.org>
Acked-by: Christian König <christian.koenig@amd.com>
Signed-off-by: Philipp Stanner <phasta@kernel.org>
Link: https://patchwork.freedesktop.org/patch/msgid/20250324092633.49746-7-tvrtko.ursulin@igalia.com

diff --git a/drivers/gpu/drm/scheduler/tests/tests_basic.c b/drivers/gpu/drm/scheduler/tests/tests_basic.c
index 996cac00bb5235b2d2788b2ad1c219ae62f98876..7230057e0594c6246f02608f07fcb1f8d738ac75 100644 (file)
--- a/drivers/gpu/drm/scheduler/tests/tests_basic.c
+++ b/drivers/gpu/drm/scheduler/tests/tests_basic.c
@@ -412,7 +412,65 @@ static struct kunit_suite drm_sched_modify_sched = {
        .test_cases = drm_sched_modify_sched_tests,
 };
 
+static void drm_sched_test_credits(struct kunit *test)
+{
+       struct drm_mock_sched_entity *entity;
+       struct drm_mock_scheduler *sched;
+       struct drm_mock_sched_job *job[2];
+       bool done;
+       int i;
+
+       /*
+        * Check that the configured credit limit is respected.
+        */
+
+       sched = drm_mock_sched_new(test, MAX_SCHEDULE_TIMEOUT);
+       sched->base.credit_limit = 1;
+
+       entity = drm_mock_sched_entity_new(test,
+                                          DRM_SCHED_PRIORITY_NORMAL,
+                                          sched);
+
+       job[0] = drm_mock_sched_job_new(test, entity);
+       job[1] = drm_mock_sched_job_new(test, entity);
+
+       drm_mock_sched_job_submit(job[0]);
+       drm_mock_sched_job_submit(job[1]);
+
+       done = drm_mock_sched_job_wait_scheduled(job[0], HZ);
+       KUNIT_ASSERT_TRUE(test, done);
+
+       done = drm_mock_sched_job_wait_scheduled(job[1], HZ);
+       KUNIT_ASSERT_FALSE(test, done);
+
+       i = drm_mock_sched_advance(sched, 1);
+       KUNIT_ASSERT_EQ(test, i, 1);
+
+       done = drm_mock_sched_job_wait_scheduled(job[1], HZ);
+       KUNIT_ASSERT_TRUE(test, done);
+
+       i = drm_mock_sched_advance(sched, 1);
+       KUNIT_ASSERT_EQ(test, i, 1);
+
+       done = drm_mock_sched_job_wait_finished(job[1], HZ);
+       KUNIT_ASSERT_TRUE(test, done);
+
+       drm_mock_sched_entity_free(entity);
+       drm_mock_sched_fini(sched);
+}
+
+static struct kunit_case drm_sched_credits_tests[] = {
+       KUNIT_CASE(drm_sched_test_credits),
+       {}
+};
+
+static struct kunit_suite drm_sched_credits = {
+       .name = "drm_sched_basic_credits_tests",
+       .test_cases = drm_sched_credits_tests,
+};
+
 kunit_test_suites(&drm_sched_basic,
                  &drm_sched_timeout,
                  &drm_sched_priority,
-                 &drm_sched_modify_sched);
+                 &drm_sched_modify_sched,
+                 &drm_sched_credits);