Mesa (master): zink: add spirv builder for unops with a const operand

[GitLab Mirror]

Module: Mesa
Branch: master
Commit: 0fba3bc2365fe0131f1ca890dd213cbd5664a45c
URL:    http://cgit.freedesktop.org/mesa/mesa/commit/?id=0fba3bc2365fe0131f1ca890dd213cbd5664a45c

Author: Mike Blumenkrantz <michael.blumenkrantz at gmail.com>
Date:   Mon Apr  5 09:54:05 2021 -0400

zink: add spirv builder for unops with a const operand

Reviewed-by: Dave Airlie <airlied at redhat.com>
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/10510>

---

 src/gallium/drivers/zink/nir_to_spirv/spirv_builder.c | 13 +++++++++++++
 src/gallium/drivers/zink/nir_to_spirv/spirv_builder.h |  3 +++
 2 files changed, 16 insertions(+)

diff --git a/src/gallium/drivers/zink/nir_to_spirv/spirv_builder.c b/src/gallium/drivers/zink/nir_to_spirv/spirv_builder.c
index ed8e0c707a9..f76dda8be67 100644
--- a/src/gallium/drivers/zink/nir_to_spirv/spirv_builder.c
+++ b/src/gallium/drivers/zink/nir_to_spirv/spirv_builder.c
@@ -467,6 +467,19 @@ spirv_builder_emit_interlock(struct spirv_builder *b, bool end)
    spirv_buffer_emit_word(&b->instructions, (end ? SpvOpEndInvocationInterlockEXT : SpvOpBeginInvocationInterlockEXT) | (1 << 16));
 }
 
+
+SpvId
+spirv_builder_emit_unop_const(struct spirv_builder *b, SpvOp op, SpvId result_type, uint64_t operand)
+{
+   SpvId result = spirv_builder_new_id(b);
+   spirv_buffer_prepare(&b->instructions, b->mem_ctx, 4);
+   spirv_buffer_emit_word(&b->instructions, op | (4 << 16));
+   spirv_buffer_emit_word(&b->instructions, result_type);
+   spirv_buffer_emit_word(&b->instructions, result);
+   spirv_buffer_emit_word(&b->instructions, spirv_builder_const_uint(b, 32, operand));
+   return result;
+}
+
 SpvId
 spirv_builder_emit_unop(struct spirv_builder *b, SpvOp op, SpvId result_type,
                         SpvId operand)
diff --git a/src/gallium/drivers/zink/nir_to_spirv/spirv_builder.h b/src/gallium/drivers/zink/nir_to_spirv/spirv_builder.h
index 60bad0388d7..683607badbb 100644
--- a/src/gallium/drivers/zink/nir_to_spirv/spirv_builder.h
+++ b/src/gallium/drivers/zink/nir_to_spirv/spirv_builder.h
@@ -190,6 +190,9 @@ spirv_builder_emit_access_chain(struct spirv_builder *b, SpvId result_type,
 void
 spirv_builder_emit_interlock(struct spirv_builder *b, bool end);
 
+SpvId
+spirv_builder_emit_unop_const(struct spirv_builder *b, SpvOp op, SpvId result_type, uint64_t operand);
+
 SpvId
 spirv_builder_emit_unop(struct spirv_builder *b, SpvOp op, SpvId result_type,
                         SpvId operand);