From: Tvrtko Ursulin <tvrtko.ursu...@intel.com>

Instead of hardcoding the VCS balancing engines, discover, both with the
new engines query, or with the legacy get_param in the fallback case, so
class based addressing always works.

v2:
 * Simplify has_engine_query check. (Andi)
 * Fix assert on uninitialized variable. (Andi)

Signed-off-by: Tvrtko Ursulin <tvrtko.ursu...@intel.com>
Reviewed-by: Andi Shyti <andi.sh...@intel.com>
---
 benchmarks/gem_wsim.c | 173 ++++++++++++++++++++++++++++++++++++++++--
 1 file changed, 166 insertions(+), 7 deletions(-)

diff --git a/benchmarks/gem_wsim.c b/benchmarks/gem_wsim.c
index 02c3e9d655d8..64d19ed469b0 100644
--- a/benchmarks/gem_wsim.c
+++ b/benchmarks/gem_wsim.c
@@ -365,34 +365,191 @@ static int str_to_engine(const char *str)
        return -1;
 }
 
+static bool __engines_queried;
+static unsigned int __num_engines;
+static struct i915_engine_class_instance *__engines;
+
+static int
+__i915_query(int i915, struct drm_i915_query *q)
+{
+       if (igt_ioctl(i915, DRM_IOCTL_I915_QUERY, q))
+               return -errno;
+       return 0;
+}
+
+static int
+__i915_query_items(int i915, struct drm_i915_query_item *items, uint32_t 
n_items)
+{
+       struct drm_i915_query q = {
+               .num_items = n_items,
+               .items_ptr = to_user_pointer(items),
+       };
+       return __i915_query(i915, &q);
+}
+
+static void
+i915_query_items(int i915, struct drm_i915_query_item *items, uint32_t n_items)
+{
+       igt_assert_eq(__i915_query_items(i915, items, n_items), 0);
+}
+
+static bool has_engine_query(int i915)
+{
+       struct drm_i915_query_item item = {
+               .query_id = DRM_I915_QUERY_ENGINE_INFO,
+       };
+
+       return __i915_query_items(i915, &item, 1) == 0 && item.length > 0;
+}
+
+static void query_engines(void)
+{
+       struct i915_engine_class_instance *engines;
+       unsigned int num;
+
+       if (__engines_queried)
+               return;
+
+       __engines_queried = true;
+
+       if (!has_engine_query(fd)) {
+               unsigned int num_bsd = gem_has_bsd(fd) + gem_has_bsd2(fd);
+               unsigned int i = 0;
+
+               igt_assert(num_bsd);
+
+               num = 1 + num_bsd;
+
+               if (gem_has_blt(fd))
+                       num++;
+
+               if (gem_has_vebox(fd))
+                       num++;
+
+               engines = calloc(num,
+                                sizeof(struct i915_engine_class_instance));
+               igt_assert(engines);
+
+               engines[i].engine_class = I915_ENGINE_CLASS_RENDER;
+               engines[i].engine_instance = 0;
+               i++;
+
+               if (gem_has_blt(fd)) {
+                       engines[i].engine_class = I915_ENGINE_CLASS_COPY;
+                       engines[i].engine_instance = 0;
+                       i++;
+               }
+
+               if (gem_has_bsd(fd)) {
+                       engines[i].engine_class = I915_ENGINE_CLASS_VIDEO;
+                       engines[i].engine_instance = 0;
+                       i++;
+               }
+
+               if (gem_has_bsd2(fd)) {
+                       engines[i].engine_class = I915_ENGINE_CLASS_VIDEO;
+                       engines[i].engine_instance = 1;
+                       i++;
+               }
+
+               if (gem_has_vebox(fd)) {
+                       engines[i].engine_class =
+                               I915_ENGINE_CLASS_VIDEO_ENHANCE;
+                       engines[i].engine_instance = 0;
+                       i++;
+               }
+       } else {
+               struct drm_i915_query_engine_info *engine_info;
+               struct drm_i915_query_item item = {
+                       .query_id = DRM_I915_QUERY_ENGINE_INFO,
+               };
+               const unsigned int sz = 4096;
+               unsigned int i;
+
+               engine_info = malloc(sz);
+               igt_assert(engine_info);
+               memset(engine_info, 0, sz);
+
+               item.data_ptr = to_user_pointer(engine_info);
+               item.length = sz;
+
+               i915_query_items(fd, &item, 1);
+               igt_assert(item.length > 0);
+               igt_assert(item.length <= sz);
+
+               num = engine_info->num_engines;
+
+               engines = calloc(num,
+                                sizeof(struct i915_engine_class_instance));
+               igt_assert(engines);
+
+               for (i = 0; i < num; i++) {
+                       struct drm_i915_engine_info *engine =
+                               (struct drm_i915_engine_info 
*)&engine_info->engines[i];
+
+                       engines[i] = engine->engine;
+               }
+       }
+
+       __engines = engines;
+       __num_engines = num;
+}
+
 static unsigned int num_engines_in_class(enum intel_engine_id class)
 {
+       unsigned int i, count = 0;
+
        igt_assert(class == VCS);
 
-       return 2;
+       query_engines();
+
+       for (i = 0; i < __num_engines; i++) {
+               if (__engines[i].engine_class == I915_ENGINE_CLASS_VIDEO)
+                       count++;
+       }
+
+       igt_assert(count);
+       return count;
 }
 
 static void
 fill_engines_class(struct i915_engine_class_instance *ci,
                   enum intel_engine_id class)
 {
+       unsigned int i, j = 0;
+
        igt_assert(class == VCS);
 
-       ci[0].engine_class = I915_ENGINE_CLASS_VIDEO;
-       ci[0].engine_instance = 0;
+       query_engines();
+
+       for (i = 0; i < __num_engines; i++) {
+               if (__engines[i].engine_class != I915_ENGINE_CLASS_VIDEO)
+                       continue;
 
-       ci[1].engine_class = I915_ENGINE_CLASS_VIDEO;
-       ci[1].engine_instance = 1;
+               ci[j].engine_class = __engines[i].engine_class;
+               ci[j].engine_instance = __engines[i].engine_instance;
+               j++;
+       }
 }
 
 static void
 fill_engines_id_class(enum intel_engine_id *list,
                      enum intel_engine_id class)
 {
+       enum intel_engine_id engine = VCS1;
+       unsigned int i, j = 0;
+
        igt_assert(class == VCS);
+       igt_assert(num_engines_in_class(VCS) <= 2);
+
+       query_engines();
 
-       list[0] = VCS1;
-       list[1] = VCS2;
+       for (i = 0; i < __num_engines; i++) {
+               if (__engines[i].engine_class != I915_ENGINE_CLASS_VIDEO)
+                       continue;
+
+               list[j++] = engine++;
+       }
 }
 
 static struct i915_engine_class_instance
@@ -400,6 +557,8 @@ get_engine(enum intel_engine_id engine)
 {
        struct i915_engine_class_instance ci;
 
+       query_engines();
+
        switch (engine) {
        case RCS:
                ci.engine_class = I915_ENGINE_CLASS_RENDER;
-- 
2.20.1

_______________________________________________
Intel-gfx mailing list
Intel-gfx@lists.freedesktop.org
https://lists.freedesktop.org/mailman/listinfo/intel-gfx

Reply via email to