nir: Move nir_lower_mediump_outputs from ir3

[mesa.git] / src / compiler / nir / nir_opt_vectorize.c

diff --git a/src/compiler/nir/nir_opt_vectorize.c b/src/compiler/nir/nir_opt_vectorize.c
index 275202b0d8597ca31ffe75e9b8068ff9303c2912..20e03db0d38f1d87166b6736040a6b90c373674d 100644 (file)
--- a/src/compiler/nir/nir_opt_vectorize.c
+++ b/src/compiler/nir/nir_opt_vectorize.c
@@ -33,8 +33,9 @@ static uint32_t
 hash_src(uint32_t hash, const nir_src *src)
 {
    assert(src->is_ssa);
+   void *hash_data = nir_src_is_const(*src) ? NULL : src->ssa;
 
-   return HASH(hash, src->ssa);
+   return HASH(hash, hash_data);
 }
 
 static uint32_t
@@ -79,7 +80,8 @@ srcs_equal(const nir_src *src1, const nir_src *src2)
    assert(src1->is_ssa);
    assert(src2->is_ssa);
 
-   return src1->ssa == src2->ssa;
+   return src1->ssa == src2->ssa ||
+      nir_src_is_const(*src1) == nir_src_is_const(*src2);
 }
 
 static bool
@@ -185,6 +187,27 @@ instr_try_combine(nir_instr *instr1, nir_instr *instr2)
    new_alu->dest.write_mask = (1 << total_components) - 1;
 
    for (unsigned i = 0; i < nir_op_infos[alu1->op].num_inputs; i++) {
+      /* handle constant merging case */
+      if (alu1->src[i].src.ssa != alu2->src[i].src.ssa) {
+         nir_const_value *c1 = nir_src_as_const_value(alu1->src[i].src);
+         nir_const_value *c2 = nir_src_as_const_value(alu2->src[i].src);
+         assert(c1 && c2);
+         nir_const_value value[4];
+         unsigned bit_size = alu1->src[i].src.ssa->bit_size;
+
+         for (unsigned j = 0; j < total_components; j++) {
+            value[j].u64 = j < alu1_components ?
+                              c1[alu1->src[i].swizzle[j]].u64 :
+                              c2[alu2->src[i].swizzle[j - alu1_components]].u64;
+         }
+         nir_ssa_def *def = nir_build_imm(&b, total_components, bit_size, value);
+
+         new_alu->src[i].src = nir_src_for_ssa(def);
+         for (unsigned j = 0; j < total_components; j++)
+            new_alu->src[i].swizzle[j] = j;
+         continue;
+      }
+
       new_alu->src[i].src = alu1->src[i].src;
 
       for (unsigned j = 0; j < alu1_components; j++)
@@ -225,8 +248,8 @@ instr_try_combine(nir_instr *instr1, nir_instr *instr2)
       nir_if_rewrite_condition(src->parent_if, nir_src_for_ssa(new_alu1));
    }
 
-   assert(list_empty(&alu1->dest.dest.ssa.uses));
-   assert(list_empty(&alu1->dest.dest.ssa.if_uses));
+   assert(list_is_empty(&alu1->dest.dest.ssa.uses));
+   assert(list_is_empty(&alu1->dest.dest.ssa.if_uses));
 
    nir_foreach_use_safe(src, &alu2->dest.dest.ssa) {
       if (src->parent_instr->type == nir_instr_type_alu) {
@@ -262,8 +285,8 @@ instr_try_combine(nir_instr *instr1, nir_instr *instr2)
       nir_if_rewrite_condition(src->parent_if, nir_src_for_ssa(new_alu2));
    }
 
-   assert(list_empty(&alu2->dest.dest.ssa.uses));
-   assert(list_empty(&alu2->dest.dest.ssa.if_uses));
+   assert(list_is_empty(&alu2->dest.dest.ssa.uses));
+   assert(list_is_empty(&alu2->dest.dest.ssa.if_uses));
 
    nir_instr_remove(instr1);
    nir_instr_remove(instr2);
@@ -312,7 +335,7 @@ vec_instr_stack_push(struct util_dynarray *stack, nir_instr *instr)
 static void
 vec_instr_stack_pop(struct util_dynarray *stack, nir_instr *instr)
 {
-   nir_instr *last = util_dynarray_pop(stack, nir_instr *);
+   ASSERTED nir_instr *last = util_dynarray_pop(stack, nir_instr *);
    assert(last == instr);
 }