[PATCH i-g-t, v2] lib/intel_compute: Use LR mode for compute when using Xe
Zbigniew Kempczyński
zbigniew.kempczynski at intel.com
Mon Jan 27 05:13:29 UTC 2025
On Fri, Jan 24, 2025 at 12:31:40PM +0100, Francois Dugast wrote:
> When Xe is used, create the VM in LR mode as this is what the
> compute UMD does to run compute kernels. This makes those tests
> more representative of real world scenarios. A side effect is
> that user fences must be used.
>
> v2: Minimize changes, stick to xe_vm_bind_userptr_async()
>
> Signed-off-by: Francois Dugast <francois.dugast at intel.com>
> ---
> lib/intel_compute.c | 98 +++++++++++++++++++++++++++++++++++++--------
> 1 file changed, 81 insertions(+), 17 deletions(-)
>
> diff --git a/lib/intel_compute.c b/lib/intel_compute.c
> index f1520aad4..a7d5d3e0d 100644
> --- a/lib/intel_compute.c
> +++ b/lib/intel_compute.c
> @@ -27,6 +27,7 @@
> #define SIZE_BATCH 0x1000
> #define SIZE_BUFFER_INPUT MAX(sizeof(float) * SIZE_DATA, 0x1000)
> #define SIZE_BUFFER_OUTPUT MAX(sizeof(float) * SIZE_DATA, 0x1000)
> +#define ADDR_SYNC 0x010000ULL
> #define ADDR_BATCH 0x100000ULL
> #define ADDR_INPUT 0x200000ULL
> #define ADDR_OUTPUT 0x300000ULL
> @@ -43,6 +44,8 @@
> #define XE2_ADDR_STATE_CONTEXT_DATA_BASE 0x900000ULL
> #define OFFSET_STATE_SIP 0xFFFF0000
>
> +#define USER_FENCE_VALUE 0xdeadbeefdeadbeefull
> +
> /*
> * TGP - ThreadGroup Preemption
> * WMTP - Walker Mid Thread Preemption
> @@ -58,6 +61,10 @@ struct bo_dict_entry {
> uint32_t handle;
> };
>
> +struct bo_sync {
> + uint64_t sync;
> +};
> +
> struct bo_execenv {
> int fd;
> enum intel_driver driver;
> @@ -81,7 +88,7 @@ static void bo_execenv_create(int fd, struct bo_execenv *execenv,
> execenv->driver = get_intel_driver(fd);
>
> if (execenv->driver == INTEL_DRIVER_XE) {
> - execenv->vm = xe_vm_create(fd, 0, 0);
> + execenv->vm = xe_vm_create(fd, DRM_XE_VM_CREATE_FLAG_LR_MODE, 0);
>
> if (eci) {
> execenv->exec_queue = xe_exec_queue_create(fd, execenv->vm,
> @@ -107,8 +114,8 @@ static void bo_execenv_destroy(struct bo_execenv *execenv)
> igt_assert(execenv);
>
> if (execenv->driver == INTEL_DRIVER_XE) {
> - xe_vm_destroy(execenv->fd, execenv->vm);
> xe_exec_queue_destroy(execenv->fd, execenv->exec_queue);
> + xe_vm_destroy(execenv->fd, execenv->vm);
> }
What's for this reorder?
--
Zbigniew
> }
>
> @@ -119,18 +126,30 @@ static void bo_execenv_bind(struct bo_execenv *execenv,
>
> if (execenv->driver == INTEL_DRIVER_XE) {
> uint32_t vm = execenv->vm;
> + uint32_t exec_queue = execenv->exec_queue;
> uint64_t alignment = xe_get_default_alignment(fd);
> - struct drm_xe_sync sync = { 0 };
> -
> - sync.type = DRM_XE_SYNC_TYPE_SYNCOBJ;
> - sync.flags = DRM_XE_SYNC_FLAG_SIGNAL;
> - sync.handle = syncobj_create(fd, 0);
> + struct bo_sync *bo_sync;
> + size_t bo_size = sizeof(*bo_sync);
> + uint32_t bo = 0;
> + struct drm_xe_sync sync = {
> + .type = DRM_XE_SYNC_TYPE_USER_FENCE,
> + .flags = DRM_XE_SYNC_FLAG_SIGNAL,
> + .timeline_value = USER_FENCE_VALUE,
> + };
> +
> + bo_size = xe_bb_size(fd, bo_size);
> + bo = xe_bo_create(fd, execenv->vm, bo_size, vram_if_possible(fd, 0),
> + DRM_XE_GEM_CREATE_FLAG_NEEDS_VISIBLE_VRAM);
> + bo_sync = xe_bo_map(fd, bo, bo_size);
> + sync.addr = to_user_pointer(&bo_sync->sync);
>
> for (int i = 0; i < entries; i++) {
> + bo_sync->sync = 0;
> bo_dict[i].data = aligned_alloc(alignment, bo_dict[i].size);
> xe_vm_bind_userptr_async(fd, vm, 0, to_user_pointer(bo_dict[i].data),
> bo_dict[i].addr, bo_dict[i].size, &sync, 1);
> - syncobj_wait(fd, &sync.handle, 1, INT64_MAX, 0, NULL);
> + xe_wait_ufence(fd, &bo_sync->sync, USER_FENCE_VALUE, exec_queue,
> + NSEC_PER_SEC);
> memset(bo_dict[i].data, 0, bo_dict[i].size);
>
> igt_debug("[i: %2d name: %20s] data: %p, addr: %16llx, size: %llx\n",
> @@ -139,7 +158,8 @@ static void bo_execenv_bind(struct bo_execenv *execenv,
> (long long)bo_dict[i].size);
> }
>
> - syncobj_destroy(fd, sync.handle);
> + munmap(bo_sync, bo_size);
> + gem_close(fd, bo);
> } else {
> struct drm_i915_gem_execbuffer2 *execbuf = &execenv->execbuf;
> struct drm_i915_gem_exec_object2 *obj;
> @@ -177,19 +197,32 @@ static void bo_execenv_unbind(struct bo_execenv *execenv,
>
> if (execenv->driver == INTEL_DRIVER_XE) {
> uint32_t vm = execenv->vm;
> - struct drm_xe_sync sync = { 0 };
> -
> - sync.type = DRM_XE_SYNC_TYPE_SYNCOBJ;
> - sync.flags = DRM_XE_SYNC_FLAG_SIGNAL;
> - sync.handle = syncobj_create(fd, 0);
> + uint32_t exec_queue = execenv->exec_queue;
> + struct bo_sync *bo_sync;
> + size_t bo_size = sizeof(*bo_sync);
> + uint32_t bo = 0;
> + struct drm_xe_sync sync = {
> + .type = DRM_XE_SYNC_TYPE_USER_FENCE,
> + .flags = DRM_XE_SYNC_FLAG_SIGNAL,
> + .timeline_value = USER_FENCE_VALUE,
> + };
> +
> + bo_size = xe_bb_size(fd, bo_size);
> + bo = xe_bo_create(fd, execenv->vm, bo_size, vram_if_possible(fd, 0),
> + DRM_XE_GEM_CREATE_FLAG_NEEDS_VISIBLE_VRAM);
> + bo_sync = xe_bo_map(fd, bo, bo_size);
> + sync.addr = to_user_pointer(&bo_sync->sync);
>
> for (int i = 0; i < entries; i++) {
> + bo_sync->sync = 0;
> xe_vm_unbind_async(fd, vm, 0, 0, bo_dict[i].addr, bo_dict[i].size, &sync, 1);
> - syncobj_wait(fd, &sync.handle, 1, INT64_MAX, 0, NULL);
> + xe_wait_ufence(fd, &bo_sync->sync, USER_FENCE_VALUE, exec_queue,
> + NSEC_PER_SEC);
> free(bo_dict[i].data);
> }
>
> - syncobj_destroy(fd, sync.handle);
> + munmap(bo_sync, bo_size);
> + gem_close(fd, bo);
> } else {
> for (int i = 0; i < entries; i++) {
> gem_close(fd, bo_dict[i].handle);
> @@ -204,7 +237,38 @@ static void bo_execenv_exec(struct bo_execenv *execenv, uint64_t start_addr)
> int fd = execenv->fd;
>
> if (execenv->driver == INTEL_DRIVER_XE) {
> - xe_exec_wait(fd, execenv->exec_queue, start_addr);
> + uint32_t exec_queue = execenv->exec_queue;
> + struct bo_sync *bo_sync;
> + size_t bo_size = sizeof(*bo_sync);
> + uint32_t bo = 0;
> + struct drm_xe_sync sync = {
> + .type = DRM_XE_SYNC_TYPE_USER_FENCE,
> + .flags = DRM_XE_SYNC_FLAG_SIGNAL,
> + .timeline_value = USER_FENCE_VALUE,
> + };
> + struct drm_xe_exec exec = {
> + .num_batch_buffer = 1,
> + .num_syncs = 1,
> + .syncs = to_user_pointer(&sync),
> + .exec_queue_id = exec_queue,
> + .address = start_addr,
> + };
> +
> + bo_size = xe_bb_size(fd, bo_size);
> + bo = xe_bo_create(fd, execenv->vm, bo_size, vram_if_possible(fd, 0),
> + DRM_XE_GEM_CREATE_FLAG_NEEDS_VISIBLE_VRAM);
> + bo_sync = xe_bo_map(fd, bo, bo_size);
> + sync.addr = to_user_pointer(&bo_sync->sync);
> + xe_vm_bind_async(fd, execenv->vm, 0, bo, 0, ADDR_SYNC, bo_size, &sync, 1);
> + xe_wait_ufence(fd, &bo_sync->sync, USER_FENCE_VALUE, exec_queue, NSEC_PER_SEC);
> +
> + sync.addr = ADDR_SYNC;
> + bo_sync->sync = 0;
> + xe_exec(fd, &exec);
> + xe_wait_ufence(fd, &bo_sync->sync, USER_FENCE_VALUE, exec_queue, NSEC_PER_SEC);
> +
> + munmap(bo_sync, bo_size);
> + gem_close(fd, bo);
> } else {
> struct drm_i915_gem_execbuffer2 *execbuf = &execenv->execbuf;
> struct drm_i915_gem_exec_object2 *obj = execenv->obj;
> --
> 2.43.0
>
More information about the igt-dev
mailing list