Mesa (master): nir/builder: Handle 16-bit floats in nir_imm_floatN_t

[GitLab Mirror]

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

Author: Jason Ekstrand <jason.ekstrand at intel.com>
Date:   Mon Oct 22 14:08:44 2018 -0500

nir/builder: Handle 16-bit floats in nir_imm_floatN_t

Reviewed-by: Ian Romanick <ian.d.romanick at intel.com>
Reviewed-by: Eric Anholt <eric at anholt.net>

---

 src/compiler/nir/nir_builder.h | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/src/compiler/nir/nir_builder.h b/src/compiler/nir/nir_builder.h
index 5ea0a5a263..3271a48052 100644
--- a/src/compiler/nir/nir_builder.h
+++ b/src/compiler/nir/nir_builder.h
@@ -25,6 +25,7 @@
 #define NIR_BUILDER_H
 
 #include "nir_control_flow.h"
+#include "util/half_float.h"
 
 struct exec_list;
 
@@ -229,6 +230,17 @@ nir_imm_false(nir_builder *build)
 }
 
 static inline nir_ssa_def *
+nir_imm_float16(nir_builder *build, float x)
+{
+   nir_const_value v;
+
+   memset(&v, 0, sizeof(v));
+   v.u16[0] = _mesa_float_to_half(x);
+
+   return nir_build_imm(build, 1, 16, v);
+}
+
+static inline nir_ssa_def *
 nir_imm_float(nir_builder *build, float x)
 {
    nir_const_value v;
@@ -254,6 +266,8 @@ static inline nir_ssa_def *
 nir_imm_floatN_t(nir_builder *build, double x, unsigned bit_size)
 {
    switch (bit_size) {
+   case 16:
+      return nir_imm_float16(build, x);
    case 32:
       return nir_imm_float(build, x);
    case 64: