pan/midgard: Only one conditional per bundle allowed
[mesa.git] / src / panfrost / midgard / midgard_schedule.c
index 92756e15189a924a89d296c1abafbb3e03469e54..ff71e0dcd3f80c4a6258bb491a1d422ac19fa8eb 100644 (file)
@@ -168,22 +168,6 @@ mir_create_dependency_graph(midgard_instruction **instructions, unsigned count,
         }
 }
 
-/* Create a mask of accessed components from a swizzle to figure out vector
- * dependencies */
-
-static unsigned
-swizzle_to_access_mask(unsigned swizzle)
-{
-        unsigned component_mask = 0;
-
-        for (int i = 0; i < 4; ++i) {
-                unsigned c = (swizzle >> (2 * i)) & 3;
-                component_mask |= (1 << c);
-        }
-
-        return component_mask;
-}
-
 /* Does the mask cover more than a scalar? */
 
 static bool
@@ -199,159 +183,11 @@ is_single_component_mask(unsigned mask)
         return components == 1;
 }
 
-/* Checks for an SSA data hazard between two adjacent instructions, keeping in
- * mind that we are a vector architecture and we can write to different
- * components simultaneously */
-
-static bool
-can_run_concurrent_ssa(midgard_instruction *first, midgard_instruction *second)
-{
-        /* Writeout has its own rules anyway */
-        if (first->compact_branch || second->compact_branch)
-                return true;
-
-        /* Each instruction reads some registers and writes to a register. See
-         * where the first writes */
-
-        int source = first->dest;
-        int source_mask = first->mask;
-
-        /* As long as the second doesn't read from the first, we're okay */
-        for (unsigned i = 0; i < ARRAY_SIZE(second->src); ++i) {
-                if (second->src[i] != source)
-                        continue;
-
-                if (first->type != TAG_ALU_4)
-                        return false;
-
-                /* Figure out which components we just read from */
-
-                int q = (i == 0) ? second->alu.src1 : second->alu.src2;
-                midgard_vector_alu_src *m = (midgard_vector_alu_src *) &q;
-
-                /* Check if there are components in common, and fail if so */
-                if (swizzle_to_access_mask(m->swizzle) & source_mask)
-                        return false;
-        }
-
-        /* Otherwise, it's safe in that regard. Another data hazard is both
-         * writing to the same place, of course */
-
-        if (second->dest == source) {
-                /* ...but only if the components overlap */
-
-                if (second->mask & source_mask)
-                        return false;
-        }
-
-        /* ...That's it */
-        return true;
-}
-
-static bool
-midgard_has_hazard(
-        midgard_instruction **segment, unsigned segment_size,
-        midgard_instruction *ains)
-{
-        for (int s = 0; s < segment_size; ++s)
-                if (!can_run_concurrent_ssa(segment[s], ains))
-                        return true;
-
-        return false;
-
-
-}
-
-/* Fragment writeout (of r0) is allowed when:
- *
- *  - All components of r0 are written in the bundle
- *  - No components of r0 are written in VLUT
- *  - Non-pipelined dependencies of r0 are not written in the bundle
- *
- * This function checks if these requirements are satisfied given the content
- * of a scheduled bundle.
- */
-
-static bool
-can_writeout_fragment(compiler_context *ctx, midgard_instruction **bundle, unsigned count, unsigned node_count, unsigned r0)
-{
-        /* First scan for which components of r0 are written out. Initially
-         * none are written */
-
-        uint8_t r0_written_mask = 0x0;
-
-        /* Simultaneously we scan for the set of dependencies */
-
-        size_t sz = sizeof(BITSET_WORD) * BITSET_WORDS(node_count);
-        BITSET_WORD *dependencies = calloc(1, sz);
-        memset(dependencies, 0, sz);
-
-        bool success = false;
-
-        for (unsigned i = 0; i < count; ++i) {
-                midgard_instruction *ins = bundle[i];
-
-                if (ins->dest != r0)
-                        continue;
-
-                /* Record written out mask */
-                r0_written_mask |= ins->mask;
-
-                /* Record dependencies, but only if they won't become pipeline
-                 * registers. We know we can't be live after this, because
-                 * we're writeout at the very end of the shader. So check if
-                 * they were written before us. */
-
-                unsigned src0 = ins->src[0];
-                unsigned src1 = ins->src[1];
-
-                if (!mir_is_written_before(ctx, bundle[0], src0))
-                        src0 = ~0;
-
-                if (!mir_is_written_before(ctx, bundle[0], src1))
-                        src1 = ~0;
-
-                if (src0 < node_count)
-                        BITSET_SET(dependencies, src0);
-
-                if (src1 < node_count)
-                        BITSET_SET(dependencies, src1);
-
-                /* Requirement 2 */
-                if (ins->unit == UNIT_VLUT)
-                        goto done;
-        }
-
-        /* Requirement 1 */
-        if ((r0_written_mask & 0xF) != 0xF)
-                goto done;
-
-        /* Requirement 3 */
-
-        for (unsigned i = 0; i < count; ++i) {
-                unsigned dest = bundle[i]->dest;
-
-                if (dest < node_count && BITSET_TEST(dependencies, dest))
-                        goto done;
-        }
-
-        /* Otherwise, we're good to go */
-        success = true;
-
-done:
-        free(dependencies);
-        return success;
-}
-
 /* Helpers for scheudling */
 
 static bool
 mir_is_scalar(midgard_instruction *ains)
 {
-        /* Does the op support scalar units? */
-        if (!(alu_opcode_props[ains->alu.op].props & UNITS_SCALAR))
-                return false;
-
         /* Do we try to use it as a vector op? */
         if (!is_single_component_mask(ains->mask))
                 return false;
@@ -399,340 +235,6 @@ bytes_for_instruction(midgard_instruction *ains)
                 return sizeof(midgard_reg_info) + sizeof(midgard_scalar_alu);
 }
 
-/* Schedules, but does not emit, a single basic block. After scheduling, the
- * final tag and size of the block are known, which are necessary for branching
- * */
-
-static midgard_bundle
-schedule_bundle(compiler_context *ctx, midgard_block *block, midgard_instruction *ins, int *skip)
-{
-        int instructions_emitted = 0, packed_idx = 0;
-        midgard_bundle bundle = { 0 };
-
-        midgard_instruction *scheduled[5] = { NULL };
-
-        uint8_t tag = ins->type;
-
-        /* Default to the instruction's tag */
-        bundle.tag = tag;
-
-        switch (ins->type) {
-        case TAG_ALU_4: {
-                uint32_t control = 0;
-                size_t bytes_emitted = sizeof(control);
-
-                /* TODO: Constant combining */
-                int index = 0, last_unit = 0;
-
-                /* Previous instructions, for the purpose of parallelism */
-                midgard_instruction *segment[4] = {0};
-                int segment_size = 0;
-
-                instructions_emitted = -1;
-                midgard_instruction *pins = ins;
-
-                unsigned constant_count = 0;
-
-                for (;;) {
-                        midgard_instruction *ains = pins;
-
-                        /* Advance instruction pointer */
-                        if (index) {
-                                ains = mir_next_op(pins);
-                                pins = ains;
-                        }
-
-                        /* Out-of-work condition */
-                        if ((struct list_head *) ains == &block->instructions)
-                                break;
-
-                        /* Ensure that the chain can continue */
-                        if (ains->type != TAG_ALU_4) break;
-
-                        /* If there's already something in the bundle and we
-                         * have weird scheduler constraints, break now */
-                        if (ains->precede_break && index) break;
-
-                        /* According to the presentation "The ARM
-                         * Mali-T880 Mobile GPU" from HotChips 27,
-                         * there are two pipeline stages. Branching
-                         * position determined experimentally. Lines
-                         * are executed in parallel:
-                         *
-                         * [ VMUL ] [ SADD ]
-                         * [ VADD ] [ SMUL ] [ LUT ] [ BRANCH ]
-                         *
-                         * Verify that there are no ordering dependencies here.
-                         *
-                         * TODO: Allow for parallelism!!!
-                         */
-
-                        /* Pick a unit for it if it doesn't force a particular unit */
-
-                        int unit = ains->unit;
-
-                        if (!unit) {
-                                int op = ains->alu.op;
-                                int units = alu_opcode_props[op].props;
-                                bool scalar = mir_is_scalar(ains);
-
-                                if (!scalar) {
-                                        if (last_unit >= UNIT_VADD) {
-                                                if (units & UNIT_VLUT)
-                                                        unit = UNIT_VLUT;
-                                                else
-                                                        break;
-                                        } else {
-                                                if ((units & UNIT_VMUL) && last_unit < UNIT_VMUL)
-                                                        unit = UNIT_VMUL;
-                                                else if ((units & UNIT_VADD) && !(control & UNIT_VADD))
-                                                        unit = UNIT_VADD;
-                                                else if (units & UNIT_VLUT)
-                                                        unit = UNIT_VLUT;
-                                                else
-                                                        break;
-                                        }
-                                } else {
-                                        if (last_unit >= UNIT_VADD) {
-                                                if ((units & UNIT_SMUL) && !(control & UNIT_SMUL))
-                                                        unit = UNIT_SMUL;
-                                                else if (units & UNIT_VLUT)
-                                                        unit = UNIT_VLUT;
-                                                else
-                                                        break;
-                                        } else {
-                                                if ((units & UNIT_VMUL) && (last_unit < UNIT_VMUL))
-                                                        unit = UNIT_VMUL;
-                                                else if ((units & UNIT_SADD) && !(control & UNIT_SADD) && !midgard_has_hazard(segment, segment_size, ains))
-                                                        unit = UNIT_SADD;
-                                                else if (units & UNIT_VADD)
-                                                        unit = UNIT_VADD;
-                                                else if (units & UNIT_SMUL)
-                                                        unit = UNIT_SMUL;
-                                                else if (units & UNIT_VLUT)
-                                                        unit = UNIT_VLUT;
-                                                else
-                                                        break;
-                                        }
-                                }
-
-                                assert(unit & units);
-                        }
-
-                        /* Late unit check, this time for encoding (not parallelism) */
-                        if (unit <= last_unit) break;
-
-                        /* Clear the segment */
-                        if (last_unit < UNIT_VADD && unit >= UNIT_VADD)
-                                segment_size = 0;
-
-                        if (midgard_has_hazard(segment, segment_size, ains))
-                                break;
-
-                        /* We're good to go -- emit the instruction */
-                        ains->unit = unit;
-
-                        segment[segment_size++] = ains;
-
-                        /* We try to reuse constants if possible, by adjusting
-                         * the swizzle */
-
-                        if (ains->has_blend_constant) {
-                                /* Everything conflicts with the blend constant */
-                                if (bundle.has_embedded_constants)
-                                        break;
-
-                                bundle.has_blend_constant = 1;
-                                bundle.has_embedded_constants = 1;
-                        } else if (ains->has_constants && ains->alu.reg_mode == midgard_reg_mode_16) {
-                                /* TODO: DRY with the analysis pass */
-
-                                if (bundle.has_blend_constant)
-                                        break;
-
-                                if (constant_count)
-                                        break;
-
-                                /* TODO: Fix packing XXX */
-                                uint16_t *bundles = (uint16_t *) bundle.constants;
-                                uint32_t *constants = (uint32_t *) ains->constants;
-
-                                /* Copy them wholesale */
-                                for (unsigned i = 0; i < 4; ++i)
-                                        bundles[i] = constants[i];
-
-                                bundle.has_embedded_constants = true;
-                                constant_count = 4;
-                        } else if (ains->has_constants) {
-                                /* By definition, blend constants conflict with
-                                 * everything, so if there are already
-                                 * constants we break the bundle *now* */
-
-                                if (bundle.has_blend_constant)
-                                        break;
-
-                                /* For anything but blend constants, we can do
-                                 * proper analysis, however */
-
-                                /* TODO: Mask by which are used */
-                                uint32_t *constants = (uint32_t *) ains->constants;
-                                uint32_t *bundles = (uint32_t *) bundle.constants;
-
-                                uint32_t indices[4] = { 0 };
-                                bool break_bundle = false;
-
-                                for (unsigned i = 0; i < 4; ++i) {
-                                        uint32_t cons = constants[i];
-                                        bool constant_found = false;
-
-                                        /* Search for the constant */
-                                        for (unsigned j = 0; j < constant_count; ++j) {
-                                                if (bundles[j] != cons)
-                                                        continue;
-
-                                                /* We found it, reuse */
-                                                indices[i] = j;
-                                                constant_found = true;
-                                                break;
-                                        }
-
-                                        if (constant_found)
-                                                continue;
-
-                                        /* We didn't find it, so allocate it */
-                                        unsigned idx = constant_count++;
-
-                                        if (idx >= 4) {
-                                                /* Uh-oh, out of space */
-                                                break_bundle = true;
-                                                break;
-                                        }
-
-                                        /* We have space, copy it in! */
-                                        bundles[idx] = cons;
-                                        indices[i] = idx;
-                                }
-
-                                if (break_bundle)
-                                        break;
-
-                                /* Cool, we have it in. So use indices as a
-                                 * swizzle */
-
-                                unsigned swizzle = SWIZZLE_FROM_ARRAY(indices);
-                                unsigned r_constant = SSA_FIXED_REGISTER(REGISTER_CONSTANT);
-
-                                if (ains->src[0] == r_constant)
-                                        ains->alu.src1 = vector_alu_apply_swizzle(ains->alu.src1, swizzle);
-
-                                if (ains->src[1] == r_constant)
-                                        ains->alu.src2 = vector_alu_apply_swizzle(ains->alu.src2, swizzle);
-
-                                bundle.has_embedded_constants = true;
-                        }
-
-                        if (ains->compact_branch) {
-                                /* All of r0 has to be written out along with
-                                 * the branch writeout */
-
-                                if (ains->writeout && !can_writeout_fragment(ctx, scheduled, index, ctx->temp_count, ains->src[0])) {
-                                        /* We only work on full moves
-                                         * at the beginning. We could
-                                         * probably do better */
-                                        if (index != 0)
-                                                break;
-
-                                        /* Inject a move */
-                                        midgard_instruction ins = v_mov(0, blank_alu_src, SSA_FIXED_REGISTER(0));
-                                        ins.unit = UNIT_VMUL;
-                                        control |= ins.unit;
-
-                                        /* TODO don't leak */
-                                        midgard_instruction *move =
-                                                mem_dup(&ins, sizeof(midgard_instruction));
-                                        bytes_emitted += bytes_for_instruction(move);
-                                        bundle.instructions[packed_idx++] = move;
-                                }
-                        }
-
-                        bytes_emitted += bytes_for_instruction(ains);
-
-                        /* Defer marking until after writing to allow for break */
-                        scheduled[index] = ains;
-                        control |= ains->unit;
-                        last_unit = ains->unit;
-                        ++instructions_emitted;
-                        ++index;
-                }
-
-                int padding = 0;
-
-                /* Pad ALU op to nearest word */
-
-                if (bytes_emitted & 15) {
-                        padding = 16 - (bytes_emitted & 15);
-                        bytes_emitted += padding;
-                }
-
-                /* Constants must always be quadwords */
-                if (bundle.has_embedded_constants)
-                        bytes_emitted += 16;
-
-                /* Size ALU instruction for tag */
-                bundle.tag = (TAG_ALU_4) + (bytes_emitted / 16) - 1;
-                bundle.padding = padding;
-                bundle.control = bundle.tag | control;
-
-                break;
-        }
-
-        case TAG_LOAD_STORE_4: {
-                /* Load store instructions have two words at once. If
-                 * we only have one queued up, we need to NOP pad.
-                 * Otherwise, we store both in succession to save space
-                 * and cycles -- letting them go in parallel -- skip
-                 * the next. The usefulness of this optimisation is
-                 * greatly dependent on the quality of the instruction
-                 * scheduler.
-                 */
-
-                midgard_instruction *next_op = mir_next_op(ins);
-
-                if ((struct list_head *) next_op != &block->instructions && next_op->type == TAG_LOAD_STORE_4) {
-                        /* TODO: Concurrency check */
-                        instructions_emitted++;
-                }
-
-                break;
-        }
-
-        case TAG_TEXTURE_4: {
-                /* Which tag we use depends on the shader stage */
-                bool in_frag = ctx->stage == MESA_SHADER_FRAGMENT;
-                bundle.tag = in_frag ? TAG_TEXTURE_4 : TAG_TEXTURE_4_VTX;
-                break;
-        }
-
-        default:
-                unreachable("Unknown tag");
-                break;
-        }
-
-        /* Copy the instructions into the bundle */
-        bundle.instruction_count = instructions_emitted + 1 + packed_idx;
-
-        midgard_instruction *uins = ins;
-        for (; packed_idx < bundle.instruction_count; ++packed_idx) {
-                assert(&uins->link != &block->instructions);
-                bundle.instructions[packed_idx] = uins;
-                uins = mir_next_op(uins);
-        }
-
-        *skip = instructions_emitted;
-
-        return bundle;
-}
-
 /* We would like to flatten the linked list of midgard_instructions in a bundle
  * to an array of pointers on the heap for easy indexing */
 
@@ -818,6 +320,9 @@ struct midgard_predicate {
         /* True if we want to pop off the chosen instruction */
         bool destructive;
 
+        /* For ALU, choose only this unit */
+        unsigned unit;
+
         /* State for bundle constants. constants is the actual constants
          * for the bundle. constant_count is the number of bytes (up to
          * 16) currently in use for constants. When picking in destructive
@@ -830,6 +335,10 @@ struct midgard_predicate {
 
         /* Exclude this destination (if not ~0) */
         unsigned exclude;
+
+        /* Don't schedule instructions consuming conditionals (since we already
+         * scheduled one). Excludes conditional branches and csel */
+        bool no_cond;
 };
 
 /* For an instruction that can fit, adjust it to fit and update the constants
@@ -888,14 +397,31 @@ mir_choose_instruction(
         bool alu = tag == TAG_ALU_4;
         unsigned unit = predicate->unit;
         bool branch = alu && (unit == ALU_ENAB_BR_COMPACT);
+        bool scalar = (unit != ~0) && (unit & UNITS_SCALAR);
+        bool no_cond = predicate->no_cond;
 
         /* Iterate to find the best instruction satisfying the predicate */
         unsigned i;
         BITSET_WORD tmp;
 
         signed best_index = -1;
+        bool best_conditional = false;
+
+        /* Enforce a simple metric limiting distance to keep down register
+         * pressure. TOOD: replace with liveness tracking for much better
+         * results */
+
+        unsigned max_active = 0;
+        unsigned max_distance = 6;
 
         BITSET_FOREACH_SET(i, tmp, worklist, count) {
+                max_active = MAX2(max_active, i);
+        }
+
+        BITSET_FOREACH_SET(i, tmp, worklist, count) {
+                if ((max_active - i) >= max_distance)
+                        continue;
+
                 if (tag != ~0 && instructions[i]->type != tag)
                         continue;
 
@@ -908,11 +434,24 @@ mir_choose_instruction(
                 if (branch && !instructions[i]->compact_branch)
                         continue;
 
+                if (alu && scalar && !mir_is_scalar(instructions[i]))
+                        continue;
+
+                if (alu && !mir_adjust_constants(instructions[i], predicate, false))
+                        continue;
+
+                bool conditional = alu && !branch && OP_IS_CSEL(instructions[i]->alu.op);
+                conditional |= (branch && !instructions[i]->prepacked_branch && instructions[i]->branch.conditional);
+
+                if (conditional && no_cond)
+                        continue;
+
                 /* Simulate in-order scheduling */
                 if ((signed) i < best_index)
                         continue;
 
                 best_index = i;
+                best_conditional = conditional;
         }
 
 
@@ -926,6 +465,12 @@ mir_choose_instruction(
 
         if (predicate->destructive) {
                 BITSET_CLEAR(worklist, best_index);
+
+                if (alu)
+                        mir_adjust_constants(instructions[best_index], predicate, true);
+
+                /* Once we schedule a conditional, we can't again */
+                predicate->no_cond |= best_conditional;
         }
 
         return instructions[best_index];
@@ -957,6 +502,27 @@ mir_choose_bundle(
                 return ~0;
 }
 
+/* We want to choose an ALU instruction filling a given unit */
+static void
+mir_choose_alu(midgard_instruction **slot,
+                midgard_instruction **instructions,
+                BITSET_WORD *worklist, unsigned len,
+                struct midgard_predicate *predicate,
+                unsigned unit)
+{
+        /* Did we already schedule to this slot? */
+        if ((*slot) != NULL)
+                return;
+
+        /* Try to schedule something, if not */
+        predicate->unit = unit;
+        *slot = mir_choose_instruction(instructions, worklist, len, predicate);
+
+        /* Store unit upon scheduling */
+        if (*slot && !((*slot)->compact_branch))
+                (*slot)->unit = unit;
+}
+
 /* When we are scheduling a branch/csel, we need the consumed condition in the
  * same block as a pipeline register. There are two options to enable this:
  *
@@ -1141,17 +707,26 @@ mir_schedule_ldst(
                 .exclude = ~0
         };
 
+        /* Try to pick two load/store ops. Second not gauranteed to exist */
+
         midgard_instruction *ins =
                 mir_choose_instruction(instructions, worklist, len, &predicate);
 
-        mir_update_worklist(worklist, len, instructions, ins);
+        midgard_instruction *pair =
+                mir_choose_instruction(instructions, worklist, len, &predicate);
 
         struct midgard_bundle out = {
                 .tag = TAG_LOAD_STORE_4,
-                .instruction_count = 1,
-                .instructions = { ins }
+                .instruction_count = pair ? 2 : 1,
+                .instructions = { ins, pair }
         };
 
+        /* We have to update the worklist atomically, since the two
+         * instructions run concurrently (TODO: verify it's not pipelined) */
+
+        mir_update_worklist(worklist, len, instructions, ins);
+        mir_update_worklist(worklist, len, instructions, pair);
+
         return out;
 }
 
@@ -1168,12 +743,10 @@ mir_schedule_alu(
         struct midgard_predicate predicate = {
                 .tag = TAG_ALU_4,
                 .destructive = true,
-                .exclude = ~0
+                .exclude = ~0,
+                .constants = (uint8_t *) bundle.constants
         };
 
-        midgard_instruction *ins =
-                mir_choose_instruction(instructions, worklist, len, &predicate);
-
         midgard_instruction *vmul = NULL;
         midgard_instruction *vadd = NULL;
         midgard_instruction *vlut = NULL;
@@ -1181,88 +754,97 @@ mir_schedule_alu(
         midgard_instruction *sadd = NULL;
         midgard_instruction *branch = NULL;
 
-        mir_update_worklist(worklist, len, instructions, ins);
+        mir_choose_alu(&branch, instructions, worklist, len, &predicate, ALU_ENAB_BR_COMPACT);
+        mir_update_worklist(worklist, len, instructions, branch);
+        bool writeout = branch && branch->writeout;
 
-        if (ins->compact_branch) {
-                branch = ins;
-        } else if (!ins->unit) {
-                unsigned units = alu_opcode_props[ins->alu.op].props;
-
-                if (units & UNIT_VMUL) {
-                        ins->unit = UNIT_VMUL;
-                        vmul = ins;
-                } else if (units & UNIT_VADD) {
-                        ins->unit = UNIT_VADD;
-                        vadd = ins;
-                } else if (units & UNIT_VLUT) {
-                        ins->unit = UNIT_VLUT;
-                        vlut = ins;
-                } else
-                        assert(0);
+        if (branch && !branch->prepacked_branch && branch->branch.conditional) {
+                midgard_instruction *cond = mir_schedule_condition(ctx, &predicate, worklist, len, instructions, branch);
+
+                if (cond->unit == UNIT_VADD)
+                        vadd = cond;
+                else if (cond->unit == UNIT_SMUL)
+                        smul = cond;
+                else
+                        unreachable("Bad condition");
         }
 
-        bundle.has_embedded_constants = ins->has_constants;
-        bundle.has_blend_constant = ins->has_blend_constant;
+        mir_choose_alu(&smul, instructions, worklist, len, &predicate, UNIT_SMUL);
 
-        if (ins->alu.reg_mode == midgard_reg_mode_16) {
-              /* TODO: Fix packing XXX */
-                uint16_t *bundles = (uint16_t *) bundle.constants;
-                uint32_t *constants = (uint32_t *) ins->constants;
+        if (!writeout)
+                mir_choose_alu(&vlut, instructions, worklist, len, &predicate, UNIT_VLUT);
 
-                /* Copy them wholesale */
-                for (unsigned i = 0; i < 4; ++i)
-                        bundles[i] = constants[i];
-        } else {
-                memcpy(bundle.constants, ins->constants, sizeof(bundle.constants));
+        mir_choose_alu(&vadd, instructions, worklist, len, &predicate, UNIT_VADD);
+        
+        mir_update_worklist(worklist, len, instructions, vlut);
+        mir_update_worklist(worklist, len, instructions, vadd);
+        mir_update_worklist(worklist, len, instructions, smul);
+
+        bool vadd_csel = vadd && OP_IS_CSEL(vadd->alu.op);
+        bool smul_csel = smul && OP_IS_CSEL(smul->alu.op);
+
+        if (vadd_csel || smul_csel) {
+                midgard_instruction *ins = vadd_csel ? vadd : smul;
+                midgard_instruction *cond = mir_schedule_condition(ctx, &predicate, worklist, len, instructions, ins);
+
+                if (cond->unit == UNIT_VMUL)
+                        vmul = cond;
+                else if (cond->unit == UNIT_SADD)
+                        sadd = cond;
+                else
+                        unreachable("Bad condition");
         }
 
-        if (ins->writeout) {
-                unsigned src = (branch->src[0] == ~0) ? SSA_FIXED_REGISTER(0) : branch->src[0];
-                unsigned temp = (branch->src[0] == ~0) ? SSA_FIXED_REGISTER(0) : make_compiler_temp(ctx);
-                midgard_instruction mov = v_mov(src, blank_alu_src, temp);
-                vmul = mem_dup(&mov, sizeof(midgard_instruction));
-                vmul->unit = UNIT_VMUL;
-                vmul->mask = 0xF;
-                /* TODO: Don't leak */
-
-                /* Rewrite to use our temp */
+        /* Stage 2, let's schedule sadd before vmul for writeout */
+        mir_choose_alu(&sadd, instructions, worklist, len, &predicate, UNIT_SADD);
+
+        /* Check if writeout reads its own register */
+        bool bad_writeout = false;
+
+        if (branch && branch->writeout) {
                 midgard_instruction *stages[] = { sadd, vadd, smul };
+                unsigned src = (branch->src[0] == ~0) ? SSA_FIXED_REGISTER(0) : branch->src[0];
+                unsigned writeout_mask = 0x0;
 
                 for (unsigned i = 0; i < ARRAY_SIZE(stages); ++i) {
-                        if (stages[i])
-                                mir_rewrite_index_dst_single(stages[i], src, temp);
+                        if (!stages[i])
+                                continue;
+
+                        if (stages[i]->dest != src)
+                                continue;
+
+                        writeout_mask |= stages[i]->mask;
+                        bad_writeout |= mir_has_arg(stages[i], branch->src[0]);
                 }
 
-                mir_rewrite_index_src_single(branch, src, temp);
-        }
+                /* Add a move if necessary */
+                if (bad_writeout || writeout_mask != 0xF) {
+                        unsigned temp = (branch->src[0] == ~0) ? SSA_FIXED_REGISTER(0) : make_compiler_temp(ctx);
+                        midgard_instruction mov = v_mov(src, blank_alu_src, temp);
+                        vmul = mem_dup(&mov, sizeof(midgard_instruction));
+                        vmul->unit = UNIT_VMUL;
+                        vmul->mask = 0xF ^ writeout_mask;
+                        /* TODO: Don't leak */
 
-        if ((vadd && OP_IS_CSEL(vadd->alu.op)) || (smul && OP_IS_CSEL(smul->alu.op)) || (ins->compact_branch && !ins->prepacked_branch && ins->branch.conditional)) {
-                midgard_instruction *cond = mir_choose_instruction(instructions, worklist, len, &predicate);
-                mir_update_worklist(worklist, len, instructions, cond);
+                        /* Rewrite to use our temp */
 
-                if (!cond->unit) {
-                        unsigned units = alu_opcode_props[cond->alu.op].props;
+                        for (unsigned i = 0; i < ARRAY_SIZE(stages); ++i) {
+                                if (stages[i])
+                                        mir_rewrite_index_dst_single(stages[i], src, temp);
+                        }
 
-                        if (units & UNIT_VMUL) {
-                                cond->unit = UNIT_VMUL;
-                        } else if (units & UNIT_VADD) {
-                                cond->unit = UNIT_VADD;
-                        } else
-                                assert(0);
+                        mir_rewrite_index_src_single(branch, src, temp);
                 }
-
-                if (cond->unit & UNIT_VMUL)
-                        vmul = cond;
-                else if (cond->unit & UNIT_SADD)
-                        sadd = cond;
-                else if (cond->unit & UNIT_VADD)
-                        vadd = cond;
-                else if (cond->unit & UNIT_SMUL)
-                        smul = cond;
-                else
-                        unreachable("Bad condition");
         }
 
+        mir_choose_alu(&vmul, instructions, worklist, len, &predicate, UNIT_VMUL);
+
+        mir_update_worklist(worklist, len, instructions, vmul);
+        mir_update_worklist(worklist, len, instructions, sadd);
+
+        bundle.has_blend_constant = predicate.blend_constant;
+        bundle.has_embedded_constants = predicate.constant_count > 0;
+
         unsigned padding = 0;
 
         /* Now that we have finished scheduling, build up the bundle */
@@ -1306,6 +888,9 @@ schedule_block(compiler_context *ctx, midgard_block *block)
         unsigned len = 0;
         midgard_instruction **instructions = flatten_mir(block, &len);
 
+        if (!len)
+                return;
+
         /* Calculate dependencies and initial worklist */
         unsigned node_count = ctx->temp_count + 1;
         mir_create_dependency_graph(instructions, len, node_count);
@@ -1315,77 +900,59 @@ schedule_block(compiler_context *ctx, midgard_block *block)
         BITSET_WORD *worklist = calloc(sz, 1);
         mir_initialize_worklist(worklist, instructions, len);
 
-        util_dynarray_init(&block->bundles, NULL);
+        struct util_dynarray bundles;
+        util_dynarray_init(&bundles, NULL);
 
         block->quadword_count = 0;
+        unsigned blend_offset = 0;
+
+        for (;;) {
+                unsigned tag = mir_choose_bundle(instructions, worklist, len);
+                midgard_bundle bundle;
+
+                if (tag == TAG_TEXTURE_4)
+                        bundle = mir_schedule_texture(instructions, worklist, len);
+                else if (tag == TAG_LOAD_STORE_4)
+                        bundle = mir_schedule_ldst(instructions, worklist, len);
+                else if (tag == TAG_ALU_4)
+                        bundle = mir_schedule_alu(ctx, instructions, worklist, len);
+                else
+                        break;
 
-        int skip = 0;
-        mir_foreach_instr_in_block(block, ins) {
-                if (skip) {
-                        skip--;
-                        continue;
-                }
-
-                midgard_bundle bundle = schedule_bundle(ctx, block, ins, &skip);
-                util_dynarray_append(&block->bundles, midgard_bundle, bundle);
+                util_dynarray_append(&bundles, midgard_bundle, bundle);
 
-                if (bundle.has_blend_constant) {
-                        unsigned offset = ctx->quadword_count + block->quadword_count + quadword_size(bundle.tag) - 1;
-                        ctx->blend_constant_offset = offset * 0x10;
-                }
+                if (bundle.has_blend_constant)
+                        blend_offset = block->quadword_count;
 
                 block->quadword_count += quadword_size(bundle.tag);
         }
 
-        block->is_scheduled = true;
-        ctx->quadword_count += block->quadword_count;
-}
-
-/* The following passes reorder MIR instructions to enable better scheduling */
-
-static void
-midgard_pair_load_store(compiler_context *ctx, midgard_block *block)
-{
-        mir_foreach_instr_in_block_safe(block, ins) {
-                if (ins->type != TAG_LOAD_STORE_4) continue;
-
-                /* We've found a load/store op. Check if next is also load/store. */
-                midgard_instruction *next_op = mir_next_op(ins);
-                if (&next_op->link != &block->instructions) {
-                        if (next_op->type == TAG_LOAD_STORE_4) {
-                                /* If so, we're done since we're a pair */
-                                ins = mir_next_op(ins);
-                                continue;
-                        }
-
-                        /* Maximum search distance to pair, to avoid register pressure disasters */
-                        int search_distance = 8;
-
-                        /* Otherwise, we have an orphaned load/store -- search for another load */
-                        mir_foreach_instr_in_block_from(block, c, mir_next_op(ins)) {
-                                /* Terminate search if necessary */
-                                if (!(search_distance--)) break;
+        /* We emitted bundles backwards; copy into the block in reverse-order */
 
-                                if (c->type != TAG_LOAD_STORE_4) continue;
-
-                                /* We can only reorder if there are no sources */
+        util_dynarray_init(&block->bundles, NULL);
+        util_dynarray_foreach_reverse(&bundles, midgard_bundle, bundle) {
+                util_dynarray_append(&block->bundles, midgard_bundle, *bundle);
+        }
 
-                                bool deps = false;
+        /* Blend constant was backwards as well. blend_offset if set is
+         * strictly positive, as an offset of zero would imply constants before
+         * any instructions which is invalid in Midgard */
 
-                                for (unsigned s = 0; s < ARRAY_SIZE(ins->src); ++s)
-                                        deps |= (c->src[s] != ~0);
+        if (blend_offset)
+                ctx->blend_constant_offset = ((ctx->quadword_count + block->quadword_count) - blend_offset - 1) * 0x10;
 
-                                if (deps)
-                                        continue;
+        block->is_scheduled = true;
+        ctx->quadword_count += block->quadword_count;
 
-                                /* We found one! Move it up to pair and remove it from the old location */
+        /* Reorder instructions to match bundled. First remove existing
+         * instructions and then recreate the list */
 
-                                mir_insert_instruction_before(ctx, ins, *c);
-                                mir_remove_instruction(c);
+        mir_foreach_instr_in_block_safe(block, ins) {
+                list_del(&ins->link);
+        }
 
-                                break;
-                        }
-                }
+        mir_foreach_instr_in_block_scheduled_rev(block, ins) {
+                list_add(&ins->link, &block->instructions);
         }
 }
 
@@ -1665,10 +1232,6 @@ schedule_program(compiler_context *ctx)
 
         midgard_promote_uniforms(ctx, 16);
 
-        mir_foreach_block(ctx, block) {
-                midgard_pair_load_store(ctx, block);
-        }
-
         /* Must be lowered right before RA */
         mir_squeeze_index(ctx);
         mir_lower_special_reads(ctx);