feat(infra): record the shader revision in DB

rayanht · May 2, 2022 · b0df8c5 · b0df8c5
1 parent 5ceedfc
commit b0df8c5
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 0 deletions.
diff --git a/amber_server.py b/amber_server.py
@@ -70,6 +70,7 @@ def insert_BQ_entry(
     INSERT INTO `spirvsmith.spirv.shader_data`
     VALUES (
         {shader_id},
+        {original_entry.shader_rev},
         {original_entry.generator_id},
         {original_entry.generator_version},
         {original_entry.n_buffers},

diff --git a/infra/main.tf b/infra/main.tf
@@ -95,6 +95,12 @@ resource "google_bigquery_table" "spirv_shader_data_table" {
     "mode": "REQUIRED",
     "description": "Shader identifier. Not a primary key."
   },
+  {
+    "name": "shader_rev",
+    "type": "INT64",
+    "mode": "REQUIRED",
+    "description": "Shader revision. A shader can have different revisions due to the reduce & recondition process."
+  }
   {
     "name": "generator_id",
     "type": "STRING",

diff --git a/src/fuzzing_server.py b/src/fuzzing_server.py
@@ -193,6 +193,7 @@ def broadcast_shader_data(generator_id: ULID, shader: SPIRVShader, monitor: Moni
         INSERT INTO `spirvsmith.spirv.shader_data`
         VALUES (
             {shader.id},
+            1,
             {generator_id},
             {shader.context.config.misc.version},
             {len(shader.context.get_storage_buffers())},