drm/xe/kunit: Use xe kunit helper in WA test

Use xe helper code to allocate fake xe device.

Cc: Lucas De Marchi <lucas.demarchi@intel.com>
Reviewed-by: Lucas De Marchi <lucas.demarchi@intel.com>
Link: https://lore.kernel.org/r/20231218190629.502-9-michal.wajdeczko@intel.com
Signed-off-by: Michal Wajdeczko <michal.wajdeczko@intel.com>

diff --git a/drivers/gpu/drm/xe/tests/xe_wa_test.c b/drivers/gpu/drm/xe/tests/xe_wa_test.c
index a53c22a1958247cbd703264aeb49195620b6754a..3d7f3ef62ae9379995462c1881d58c976f775bae 100644 (file)
--- a/drivers/gpu/drm/xe/tests/xe_wa_test.c
+++ b/drivers/gpu/drm/xe/tests/xe_wa_test.c
@@ -9,6 +9,7 @@
 #include <kunit/test.h>
 
 #include "xe_device.h"
+#include "xe_kunit_helpers.h"
 #include "xe_pci_test.h"
 #include "xe_reg_sr.h"
 #include "xe_tuning.h"
@@ -108,9 +109,7 @@ static int xe_wa_test_init(struct kunit *test)
        dev = drm_kunit_helper_alloc_device(test);
        KUNIT_ASSERT_NOT_ERR_OR_NULL(test, dev);
 
-       xe = drm_kunit_helper_alloc_drm_device(test, dev,
-                                              struct xe_device,
-                                              drm, DRIVER_GEM);
+       xe = xe_kunit_helper_alloc_xe_device(test, dev);
        KUNIT_ASSERT_NOT_ERR_OR_NULL(test, xe);
 
        test->priv = &data;