[igt-dev] [PATCH v4 i-g-t 2/2] tests/intel/xe_create: create-big-vram subtest

[Marcin Bernatowicz]

Validates the creation of significant Buffer Object (BO) within VRAM,
considering the entire available CPU-visible VRAM size.
The size of the created BO can be adjusted using command line
parameters, with '-S' representing BO size in MB,
and '-p' representing BO size as a percentage of the VRAM size.

v2: rebased, updated to uAPI changes (DRM_XE_VM_CREATE_FLAG_ASYNC_DEFAULT),
    after review corrections: 1024UL -> 1024ULL,
    int -> unsigned int (Kamil)

v3: provided a flag to allocate the memory within the CPU-visible
      portion of VRAM (Matt)
    __create_bo replaced with xe_bo_create_flags (Lukasz)
    removed the percent command line parameter (Lukasz)
    renamed size_MB to size_mb (Lukasz)
    added helper function to query available CPU-visible VRAM size,
    renamed 'xe_vram_available' to 'xe_available_vram_size' for
      consistency with other function names

v4: split lib and test changes into separate patches (Lukasz, Kamil)
    added prefixes to titles (Kamil)
    restored percent command line parameter (Kamil)
    whitespace correction (Kamil)

Cc: Kamil Konieczny <kamil.konieczny at linux.intel.com>
Cc: Laguna, Lukasz <lukasz.laguna at intel.com>
Cc: Matthew Auld <matthew.auld at intel.com>
Signed-off-by: Marcin Bernatowicz <marcin.bernatowicz at intel.com>
---
 tests/intel/xe_create.c | 84 ++++++++++++++++++++++++++++++++++++++++-
 1 file changed, 83 insertions(+), 1 deletion(-)

diff --git a/tests/intel/xe_create.c b/tests/intel/xe_create.c
index f4633cfb3..194c677ee 100644
--- a/tests/intel/xe_create.c
+++ b/tests/intel/xe_create.c
@@ -18,6 +18,14 @@
 
 #define PAGE_SIZE 0x1000
 
+static struct param {
+	unsigned int size_mb;
+	unsigned int vram_percent;
+} params = {
+	.size_mb = 0,
+	.vram_percent = 0,
+};
+
 static int __create_bo(int fd, uint32_t vm, uint64_t size, uint32_t flags,
 		       uint32_t *handlep)
 {
@@ -214,7 +222,78 @@ static void create_massive_size(int fd)
 	}
 }
 
-igt_main
+/**
+ * SUBTEST: create-big-vram
+ * Functionality: BO creation
+ * Test category: functionality test
+ * Description: Verifies the creation of substantial BO within VRAM,
+ *		constituting all available CPU-visible VRAM.
+ */
+static void create_big_vram(int fd)
+{
+	uint64_t bo_size, size, visible_avail_size, alignment;
+	uint32_t bo_handle;
+	char *bo_ptr = NULL;
+	uint64_t vm = 0;
+	int gt;
+
+	igt_require(xe_has_vram(fd));
+	alignment = xe_get_default_alignment(fd);
+	vm = xe_vm_create(fd, DRM_XE_VM_CREATE_FLAG_ASYNC_DEFAULT, 0);
+
+	xe_for_each_gt(fd, gt) {
+		visible_avail_size = xe_visible_available_vram_size(fd, gt);
+		bo_size = params.size_mb
+			  ? params.size_mb * 1024ULL * 1024ULL
+			  : params.vram_percent
+				? ALIGN(visible_avail_size * params.vram_percent / 100, alignment)
+				: ALIGN_DOWN(visible_avail_size, alignment);
+		igt_info("gt%u bo_size=%lu visible_available_vram_size=%lu\n",
+			gt, bo_size, visible_avail_size);
+
+		bo_handle = xe_bo_create_flags(fd, vm, bo_size, visible_vram_memory(fd, gt));
+		bo_ptr = xe_bo_map(fd, bo_handle, bo_size);
+
+		size = bo_size - 1;
+		while (size > SZ_64K) {
+			igt_assert_eq(0, READ_ONCE(bo_ptr[size]));
+			WRITE_ONCE(bo_ptr[size], 'A');
+			igt_assert_eq('A', READ_ONCE(bo_ptr[size]));
+			size >>= 1;
+		}
+		igt_assert_eq(0, bo_ptr[0]);
+
+		munmap(bo_ptr, bo_size);
+		gem_close(fd, bo_handle);
+	}
+
+	xe_vm_destroy(fd, vm);
+}
+
+static int opt_handler(int opt, int opt_index, void *data)
+{
+	switch (opt) {
+	case 'S':
+		params.size_mb = atoi(optarg);
+		igt_debug("Size MB: %d\n", params.size_mb);
+		break;
+	case 'p':
+		params.vram_percent = atoi(optarg);
+		igt_debug("Percent of VRAM: %d\n", params.vram_percent);
+		break;
+	default:
+		return IGT_OPT_HANDLER_ERROR;
+	}
+
+	return IGT_OPT_HANDLER_SUCCESS;
+}
+
+const char *help_str =
+	"  -S\tBO size in MB\n"
+	"  -p\tPercent of VRAM for BO\n"
+	;
+
+igt_main_args("S:p:", NULL, help_str, opt_handler, NULL)
 {
 	int xe;
 
@@ -253,6 +332,9 @@ igt_main
 		igt_waitchildren();
 	}
 
+	igt_subtest("create-big-vram") {
+		create_big_vram(xe);
+	}
 
 	igt_fixture
 		drm_close_driver(xe);
-- 
2.31.1