iris: import program cache code

5 files changed, 399 insertions, 7 deletions

diff --git a/src/gallium/drivers/iris/iris_context.c b/src/gallium/drivers/iris/iris_context.c
index df273b29f0b..ed8172ad441 100644
--- a/src/gallium/drivers/iris/iris_context.c
+++ b/src/gallium/drivers/iris/iris_context.c
@@ -24,6 +24,7 @@
 #include <time.h>
 #include "pipe/p_defines.h"
 #include "pipe/p_state.h"
+#include "util/ralloc.h"
 #include "util/u_inlines.h"
 #include "util/u_format.h"
 #include "util/u_upload_mgr.h"
@@ -68,17 +69,19 @@ iris_set_debug_callback(struct pipe_context *ctx,
 static void
 iris_destroy_context(struct pipe_context *ctx)
 {
+   struct iris_context *ice = (struct iris_context *)ctx;
+
    if (ctx->stream_uploader)
       u_upload_destroy(ctx->stream_uploader);
 
-   free(ctx);
+   ralloc_free(ice);
 }
 
 struct pipe_context *
 iris_create_context(struct pipe_screen *pscreen, void *priv, unsigned flags)
 {
    struct iris_screen *screen = (struct iris_screen*)pscreen;
-   struct iris_context *ice = calloc(1, sizeof(struct iris_context));
+   struct iris_context *ice = rzalloc(NULL, struct iris_context);
 
    if (!ice)
       return NULL;
@@ -106,6 +109,8 @@ iris_create_context(struct pipe_screen *pscreen, void *priv, unsigned flags)
    iris_init_state_functions(ctx);
    iris_init_query_functions(ctx);
 
+   iris_init_program_cache(ice);
+
    iris_batch_init(&ice->render_batch, screen, &ice->dbg);
 
    return ctx;
diff --git a/src/gallium/drivers/iris/iris_context.h b/src/gallium/drivers/iris/iris_context.h
index d90bfc5d398..e5c7791dcd7 100644
--- a/src/gallium/drivers/iris/iris_context.h
+++ b/src/gallium/drivers/iris/iris_context.h
@@ -65,10 +65,23 @@ enum iris_dirty {
    IRIS_DIRTY_UNCOMPILED_GS            = (1ull << 25),
    IRIS_DIRTY_UNCOMPILED_FS            = (1ull << 26),
    IRIS_DIRTY_UNCOMPILED_CS            = (1ull << 27),
+   IRIS_DIRTY_VS                       = (1ull << 28),
+   IRIS_DIRTY_TCS                      = (1ull << 29),
+   IRIS_DIRTY_TES                      = (1ull << 30),
+   IRIS_DIRTY_GS                       = (1ull << 31),
+   IRIS_DIRTY_FS                       = (1ull << 32),
+   IRIS_DIRTY_CS                       = (1ull << 33),
 };
 
 struct iris_depth_stencil_alpha_state;
 
+struct iris_program_cache {
+   struct hash_table *table;
+   struct iris_bo *bo;
+   void *map;
+   uint32_t next_offset;
+};
+
 struct iris_context {
    struct pipe_context ctx;
 
@@ -77,7 +90,10 @@ struct iris_context {
    struct {
       struct iris_uncompiled_shader *progs[MESA_SHADER_STAGES];
       struct brw_stage_prog_data *prog_data[MESA_SHADER_STAGES];
+      uint32_t *prog_offset[MESA_SHADER_STAGES];
       struct brw_vue_map *last_vue_map;
+
+      struct iris_program_cache cache;
    } shaders;
 
    /** The main batch for rendering */
@@ -133,4 +149,37 @@ void iris_update_compiled_shaders(struct iris_context *ice);
 
 void iris_draw_vbo(struct pipe_context *ctx, const struct pipe_draw_info *info);
 
+enum iris_program_cache_id {
+   IRIS_CACHE_VS  = MESA_SHADER_VERTEX,
+   IRIS_CACHE_TCS = MESA_SHADER_TESS_CTRL,
+   IRIS_CACHE_TES = MESA_SHADER_TESS_EVAL,
+   IRIS_CACHE_GS  = MESA_SHADER_GEOMETRY,
+   IRIS_CACHE_FS  = MESA_SHADER_FRAGMENT,
+   IRIS_CACHE_CS  = MESA_SHADER_COMPUTE,
+   IRIS_CACHE_BLORP,
+};
+
+void iris_init_program_cache(struct iris_context *ice);
+void iris_destroy_program_cache(struct iris_context *ice);
+void iris_print_program_cache(struct iris_context *ice);
+bool iris_search_cache(struct iris_context *ice,
+                       enum iris_program_cache_id cache_id,
+                       const void *key,
+                       unsigned key_size,
+                       uint64_t dirty_flag,
+                       uint32_t *inout_assembly_offset,
+                       void *inout_prog_data);
+void iris_upload_cache(struct iris_context *ice,
+                       enum iris_program_cache_id cache_id,
+                       const void *key,
+                       unsigned key_size,
+                       const void *assembly,
+                       unsigned assembly_size,
+                       const void *prog_data,
+                       unsigned prog_data_size,
+                       uint32_t *out_assembly_offset,
+                       void *out_prog_data);
+const void *iris_find_previous_compile(struct iris_program_cache *cache,
+                                       enum iris_program_cache_id cache_id,
+                                       unsigned program_string_id);
 #endif
diff --git a/src/gallium/drivers/iris/iris_program.c b/src/gallium/drivers/iris/iris_program.c
index 846729ecefb..eed24334486 100644
--- a/src/gallium/drivers/iris/iris_program.c
+++ b/src/gallium/drivers/iris/iris_program.c
@@ -227,11 +227,10 @@ iris_compile_vs(struct iris_context *ice,
    /* The param and pull_param arrays will be freed by the shader cache. */
    ralloc_steal(NULL, prog_data->param);
    ralloc_steal(NULL, prog_data->pull_param);
-   //brw_upload_cache(&brw->cache, BRW_CACHE_VS_PROG,
-                    //key, sizeof(struct brw_vs_prog_key),
-                    //program, prog_data.base.base.program_size,
-                    //&prog_data, sizeof(prog_data),
-                    //&brw->vs.base.prog_offset, &brw->vs.base.prog_data);
+   iris_upload_cache(ice, IRIS_CACHE_VS, key, sizeof(*key), program,
+                     prog_data->program_size, prog_data, sizeof(vs_prog_data),
+                     &ice->shaders.prog_offset[MESA_SHADER_VERTEX],
+                     &ice->shaders.prog_data[MESA_SHADER_VERTEX]);
    ralloc_free(mem_ctx);
 
    return true;
@@ -249,6 +248,11 @@ iris_update_compiled_vs(struct iris_context *ice)
    struct brw_vs_prog_key key;
    iris_populate_vs_key(ice, &key);
 
+   if (iris_search_cache(ice, IRIS_CACHE_VS, &key, sizeof(key), IRIS_DIRTY_VS,
+                         &ice->shaders.prog_offset[MESA_SHADER_VERTEX],
+                         &ice->shaders.prog_data[MESA_SHADER_VERTEX]))
+      return;
+
    UNUSED bool success =
       iris_compile_vs(ice, ice->shaders.progs[MESA_SHADER_VERTEX], &key);
 }
diff --git a/src/gallium/drivers/iris/iris_program_cache.c b/src/gallium/drivers/iris/iris_program_cache.c
new file mode 100644
index 00000000000..2cbd635684d
--- /dev/null
+++ b/src/gallium/drivers/iris/iris_program_cache.c
@@ -0,0 +1,333 @@
+/*
+ * Copyright © 2017 Intel Corporation
+ *
+ * Permission is hereby granted, free of charge, to any person obtaining a
+ * copy of this software and associated documentation files (the "Software"),
+ * to deal in the Software without restriction, including without limitation
+ * on the rights to use, copy, modify, merge, publish, distribute, sub
+ * license, and/or sell copies of the Software, and to permit persons to whom
+ * the Software is furnished to do so, subject to the following conditions:
+ *
+ * The above copyright notice and this permission notice (including the next
+ * paragraph) shall be included in all copies or substantial portions of the
+ * Software.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ * FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT. IN NO EVENT SHALL
+ * THE AUTHOR(S) AND/OR THEIR SUPPLIERS BE LIABLE FOR ANY CLAIM,
+ * DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR
+ * OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE
+ * USE OR OTHER DEALINGS IN THE SOFTWARE.
+ */
+#include <stdio.h>
+#include <errno.h>
+#include <i915_drm.h>
+#include "pipe/p_defines.h"
+#include "pipe/p_state.h"
+#include "pipe/p_context.h"
+#include "pipe/p_screen.h"
+#include "util/u_atomic.h"
+#include "compiler/nir/nir.h"
+#include "compiler/nir/nir_builder.h"
+#include "intel/compiler/brw_compiler.h"
+#include "intel/compiler/brw_eu.h"
+#include "intel/compiler/brw_nir.h"
+#include "iris_context.h"
+
+struct keybox {
+   uint8_t size;
+   enum iris_program_cache_id cache_id;
+   uint8_t data[0];
+};
+
+// XXX: put packets here, expose this somehow and simplify the interfaces
+struct iris_program_cache_item {
+   uint32_t assembly_offset;
+   uint32_t assembly_size;
+   void *prog_data;
+};
+
+static struct keybox *
+make_keybox(struct iris_program_cache *cache,
+            enum iris_program_cache_id cache_id,
+            const void *key, unsigned key_size)
+{
+   struct keybox *keybox =
+      ralloc_size(cache->table, sizeof(struct keybox) + key_size);
+
+   keybox->cache_id = cache_id;
+   keybox->size = key_size;
+   memcpy(keybox->data, key, key_size);
+
+   return keybox;
+}
+
+static uint32_t
+keybox_hash(const void *void_key)
+{
+   const struct keybox *key = void_key;
+   return _mesa_hash_data(&key->cache_id, key->size + sizeof(key->cache_id));
+}
+
+static bool
+keybox_equals(const void *void_a, const void *void_b)
+{
+   const struct keybox *a = void_a, *b = void_b;
+   if (a->size != b->size)
+      return false;
+
+   return memcmp(a->data, b->data, a->size) == 0;
+}
+
+static unsigned
+get_program_string_id(enum iris_program_cache_id cache_id, const void *key)
+{
+   switch (cache_id) {
+   case IRIS_CACHE_VS:
+      return ((struct brw_vs_prog_key *) key)->program_string_id;
+   case IRIS_CACHE_TCS:
+      return ((struct brw_tcs_prog_key *) key)->program_string_id;
+   case IRIS_CACHE_TES:
+      return ((struct brw_tes_prog_key *) key)->program_string_id;
+   case IRIS_CACHE_GS:
+      return ((struct brw_gs_prog_key *) key)->program_string_id;
+   case IRIS_CACHE_CS:
+      return ((struct brw_cs_prog_key *) key)->program_string_id;
+   case IRIS_CACHE_FS:
+      return ((struct brw_wm_prog_key *) key)->program_string_id;
+   default:
+      unreachable("no program string id for this kind of program");
+   }
+}
+
+/**
+ * Returns the buffer object matching cache_id and key, or NULL.
+ */
+// XXX: rename to iris_bind_cached_shader?
+bool
+iris_search_cache(struct iris_context *ice,
+                  enum iris_program_cache_id cache_id,
+                  const void *key,
+                  unsigned key_size,
+                  uint64_t dirty_flag,
+                  uint32_t *inout_assembly_offset,
+                  void *inout_prog_data)
+{
+   struct iris_program_cache *cache = &ice->shaders.cache;
+
+   struct keybox *keybox = make_keybox(cache, cache_id, key, key_size);
+
+   struct hash_entry *entry = _mesa_hash_table_search(cache->table, keybox);
+
+   if (entry == NULL)
+      return false;
+
+   struct iris_program_cache_item *item = entry->data;
+
+   if (item->assembly_offset != *inout_assembly_offset ||
+       item->prog_data != *((void **) inout_prog_data)) {
+      *inout_assembly_offset = item->assembly_offset;
+      *((void **) inout_prog_data) = item->prog_data;
+      ice->state.dirty |= dirty_flag;
+   }
+
+   return true;
+}
+
+static void
+recreate_cache_bo(struct iris_context *ice, uint32_t size)
+{
+   struct iris_program_cache *cache = &ice->shaders.cache;
+   struct iris_screen *screen = (struct iris_screen *)ice->ctx.screen;
+   struct iris_bo *old_bo = cache->bo;
+   void *old_map = cache->map;
+
+   cache->bo = iris_bo_alloc(screen->bufmgr, "program cache", size, 64);
+   cache->bo->kflags = EXEC_OBJECT_CAPTURE;
+   cache->map = iris_bo_map(&ice->dbg, cache->bo,
+                            MAP_READ | MAP_WRITE | MAP_ASYNC | MAP_PERSISTENT);
+
+   /* Copy any existing data that needs to be saved. */
+   if (old_bo) {
+      perf_debug(&ice->dbg,
+                 "Copying to larger program cache: %u kB -> %u kB\n",
+                 (unsigned) old_bo->size / 1024,
+                 (unsigned) cache->bo->size / 1024);
+
+      memcpy(cache->map, old_map, cache->next_offset);
+
+      iris_bo_unreference(old_bo);
+      iris_bo_unmap(old_bo);
+   }
+}
+
+const void *
+iris_find_previous_compile(struct iris_program_cache *cache,
+                           enum iris_program_cache_id cache_id,
+                           unsigned program_string_id)
+{
+   hash_table_foreach(cache->table, entry) {
+      const struct keybox *keybox = entry->key;
+      if (keybox->cache_id == cache_id &&
+          get_program_string_id(cache_id, keybox->data) == program_string_id) {
+         return keybox->data;
+      }
+   }
+
+   return NULL;
+}
+
+/**
+ * Look for an existing entry in the cache that has identical assembly code.
+ *
+ * This is useful for programs generating shaders at runtime, where multiple
+ * distinct shaders (from an API perspective) may compile to the same assembly
+ * in our backend.
+ */
+static const struct iris_program_cache_item *
+find_existing_assembly(const struct iris_program_cache *cache,
+                       const void *assembly,
+                       unsigned assembly_size)
+{
+   hash_table_foreach(cache->table, entry) {
+      const struct iris_program_cache_item *item = entry->data;
+      if (item->assembly_size == assembly_size &&
+          memcmp(cache->map + item->assembly_offset,
+                 assembly, assembly_size) == 0)
+         return item;
+   }
+   return NULL;
+}
+
+static uint32_t
+upload_new_assembly(struct iris_context *ice,
+                    const void *assembly,
+                    unsigned size)
+{
+   struct iris_program_cache *cache = &ice->shaders.cache;
+
+   /* Allocate space in the cache BO for our new program. */
+   if (cache->next_offset + size > cache->bo->size) {
+      uint32_t new_size = cache->bo->size * 2;
+
+      while (cache->next_offset + size > new_size)
+         new_size *= 2;
+
+      recreate_cache_bo(ice, new_size);
+   }
+
+   uint32_t offset = cache->next_offset;
+
+   /* Programs are always 64-byte aligned, so set up the next one now */
+   cache->next_offset = ALIGN(offset + size, 64);
+
+   /* Copy data to the buffer */
+   memcpy(cache->map + offset, assembly, size);
+
+   return offset;
+}
+
+void
+iris_upload_cache(struct iris_context *ice,
+                  enum iris_program_cache_id cache_id,
+                  const void *key,
+                  unsigned key_size,
+                  const void *assembly,
+                  unsigned assembly_size,
+                  const void *prog_data,
+                  unsigned prog_data_size,
+                  uint32_t *out_assembly_offset,
+                  void *out_prog_data)
+{
+   struct iris_program_cache *cache = &ice->shaders.cache;
+   struct iris_program_cache_item *item =
+      ralloc(cache->table, struct iris_program_cache_item);
+   const struct iris_program_cache_item *matching_data =
+      find_existing_assembly(cache, assembly, assembly_size);
+
+   /* If we can find a matching prog in the cache already, then reuse the
+    * existing stuff without creating new copy into the underlying buffer
+    * object.  This is notably useful for programs generating shaders at
+    * runtime, where multiple shaders may compile to the same thing in our
+    * backend.
+    */
+   if (matching_data) {
+      item->assembly_offset = matching_data->assembly_offset;
+   } else {
+      item->assembly_offset =
+         upload_new_assembly(ice, assembly, assembly_size);
+   }
+
+   item->assembly_size = assembly_size;
+   item->prog_data = ralloc_size(item, prog_data_size);
+   memcpy(item->prog_data, prog_data, prog_data_size);
+
+   if (cache_id != IRIS_CACHE_BLORP) {
+      struct brw_stage_prog_data *stage_prog_data = prog_data;
+      ralloc_steal(item->prog_data, stage_prog_data->param);
+      ralloc_steal(item->prog_data, stage_prog_data->pull_param);
+   }
+
+   struct keybox *keybox = make_keybox(cache, cache_id, key, key_size);
+   _mesa_hash_table_insert(cache->table, keybox, item);
+
+   *out_assembly_offset = item->assembly_offset;
+   *(void **)out_prog_data = item->prog_data;
+}
+
+void
+iris_init_program_cache(struct iris_context *ice)
+{
+   struct iris_program_cache *cache = &ice->shaders.cache;
+
+   cache->table = _mesa_hash_table_create(ice, keybox_hash, keybox_equals);
+
+   recreate_cache_bo(ice, 16384);
+}
+
+void
+iris_destroy_program_cache(struct iris_context *ice)
+{
+   struct iris_program_cache *cache = &ice->shaders.cache;
+
+   /* This can be NULL if context creation failed early on. */
+   if (cache->bo) {
+      iris_bo_unmap(cache->bo);
+      iris_bo_unreference(cache->bo);
+      cache->bo = NULL;
+      cache->map = NULL;
+   }
+
+   cache->next_offset = 0;
+
+   /* Also, NULL out any stale program pointers. */
+   for (int i = 0; i < MESA_SHADER_STAGES; i++) {
+      ice->shaders.prog_data[i] = NULL;
+   }
+}
+
+static const char *
+cache_name(enum iris_program_cache_id cache_id)
+{
+   if (cache_id == IRIS_CACHE_BLORP)
+      return "BLORP";
+
+   return _mesa_shader_stage_to_string(cache_id);
+}
+
+void
+iris_print_program_cache(struct iris_context *ice)
+{
+   struct iris_program_cache *cache = &ice->shaders.cache;
+   struct iris_screen *screen = (struct iris_screen *)ice->ctx.screen;
+   const struct gen_device_info *devinfo = &screen->devinfo;
+
+   hash_table_foreach(cache->table, entry) {
+      const struct keybox *keybox = entry->key;
+      struct iris_program_cache_item *item = entry->data;
+      fprintf(stderr, "%s:\n", cache_name(keybox->cache_id));
+      brw_disassemble(devinfo, cache->map,
+                      item->assembly_offset, item->assembly_size, stderr);
+   }
+}
diff --git a/src/gallium/drivers/iris/meson.build b/src/gallium/drivers/iris/meson.build
index fc33e9c59c1..af8bbcfe758 100644
--- a/src/gallium/drivers/iris/meson.build
+++ b/src/gallium/drivers/iris/meson.build
@@ -31,6 +31,7 @@ files_libiris = files(
   'iris_formats.c',
   'iris_pipe.h',
   'iris_program.c',
+  'iris_program_cache.c',
   'iris_query.c',
   'iris_resource.c',
   'iris_resource.h',