1 files changed, 144 insertions, 0 deletions

diff --git a/tests/amdgpu/amdgpu_test.h b/tests/amdgpu/amdgpu_test.h
index e30e2312..62875736 100644
--- a/tests/amdgpu/amdgpu_test.h
+++ b/tests/amdgpu/amdgpu_test.h
@@ -85,6 +85,11 @@ int suite_cs_tests_init();
 int suite_cs_tests_clean();
 
 /**
+ * Decide if the suite is enabled by default or not.
+ */
+CU_BOOL suite_cs_tests_enable(void);
+
+/**
  * Tests in cs test suite
  */
 extern CU_TestInfo cs_tests[];
@@ -100,11 +105,96 @@ int suite_vce_tests_init();
 int suite_vce_tests_clean();
 
 /**
+ * Decide if the suite is enabled by default or not.
+ */
+CU_BOOL suite_vce_tests_enable(void);
+
+/**
  * Tests in vce test suite
  */
 extern CU_TestInfo vce_tests[];
 
 /**
++ * Initialize vcn test suite
++ */
+int suite_vcn_tests_init();
+
+/**
++ * Deinitialize vcn test suite
++ */
+int suite_vcn_tests_clean();
+
+/**
+ * Decide if the suite is enabled by default or not.
+ */
+CU_BOOL suite_vcn_tests_enable(void);
+
+/**
++ * Tests in vcn test suite
++ */
+extern CU_TestInfo vcn_tests[];
+
+/**
+ * Initialize uvd enc test suite
+ */
+int suite_uvd_enc_tests_init();
+
+/**
+ * Deinitialize uvd enc test suite
+ */
+int suite_uvd_enc_tests_clean();
+
+/**
+ * Decide if the suite is enabled by default or not.
+ */
+CU_BOOL suite_uvd_enc_tests_enable(void);
+
+/**
+ * Tests in uvd enc test suite
+ */
+extern CU_TestInfo uvd_enc_tests[];
+
+/**
+ * Initialize deadlock test suite
+ */
+int suite_deadlock_tests_init();
+
+/**
+ * Deinitialize deadlock test suite
+ */
+int suite_deadlock_tests_clean();
+
+/**
+ * Decide if the suite is enabled by default or not.
+ */
+CU_BOOL suite_deadlock_tests_enable(void);
+
+/**
+ * Tests in uvd enc test suite
+ */
+extern CU_TestInfo deadlock_tests[];
+
+/**
+ * Initialize vm test suite
+ */
+int suite_vm_tests_init();
+
+/**
+ * Deinitialize deadlock test suite
+ */
+int suite_vm_tests_clean();
+
+/**
+ * Decide if the suite is enabled by default or not.
+ */
+CU_BOOL suite_vm_tests_enable(void);
+
+/**
+ * Tests in vm test suite
+ */
+extern CU_TestInfo vm_tests[];
+
+/**
  * Helper functions
  */
 static inline amdgpu_bo_handle gpu_mem_alloc(
@@ -162,6 +252,29 @@ static inline int gpu_mem_free(amdgpu_bo_handle bo,
 }
 
 static inline int
+amdgpu_bo_alloc_wrap(amdgpu_device_handle dev, unsigned size,
+                     unsigned alignment, unsigned heap, uint64_t flags,
+                     amdgpu_bo_handle *bo)
+{
+        struct amdgpu_bo_alloc_request request = {};
+        amdgpu_bo_handle buf_handle;
+        int r;
+
+        request.alloc_size = size;
+        request.phys_alignment = alignment;
+        request.preferred_heap = heap;
+        request.flags = flags;
+
+        r = amdgpu_bo_alloc(dev, &request, &buf_handle);
+        if (r)
+                return r;
+
+        *bo = buf_handle;
+
+        return 0;
+}
+
+static inline int
 amdgpu_bo_alloc_and_map(amdgpu_device_handle dev, unsigned size,
                         unsigned alignment, unsigned heap, uint64_t flags,
                         amdgpu_bo_handle *bo, void **cpu, uint64_t *mc_address,
@@ -236,4 +349,35 @@ amdgpu_get_bo_list(amdgpu_device_handle dev, amdgpu_bo_handle bo1,
         return amdgpu_bo_list_create(dev, bo2 ? 2 : 1, resources, NULL, list);
 }
 
+
+static inline CU_ErrorCode amdgpu_set_suite_active(const char *suite_name,
+                                                          CU_BOOL active)
+{
+        CU_ErrorCode r = CU_set_suite_active(CU_get_suite(suite_name), active);
+
+        if (r != CUE_SUCCESS)
+                fprintf(stderr, "Failed to obtain suite %s\n", suite_name);
+
+        return r;
+}
+
+static inline CU_ErrorCode amdgpu_set_test_active(const char *suite_name,
+                                  const char *test_name, CU_BOOL active)
+{
+        CU_ErrorCode r;
+        CU_pSuite pSuite = CU_get_suite(suite_name);
+
+        if (!pSuite) {
+                fprintf(stderr, "Failed to obtain suite %s\n",
+                                suite_name);
+                return CUE_NOSUITE;
+        }
+
+        r = CU_set_test_active(CU_get_test(pSuite, test_name), active);
+        if (r != CUE_SUCCESS)
+                fprintf(stderr, "Failed to obtain test %s\n", test_name);
+
+        return r;
+}
+
 #endif  /* #ifdef _AMDGPU_TEST_H_ */