pan/midgard: Generalize texture registers across GPUs
[mesa.git] / src / panfrost / midgard / midgard_ra.c
index f8355a89720ea36b6f7771c4a707d3460f9f6fc6..7035a2a160b8beb8676a66055da262575a9873e7 100644 (file)
 
 #include "compiler.h"
 #include "midgard_ops.h"
-#include "util/register_allocate.h"
 #include "util/u_math.h"
 #include "util/u_memory.h"
-
-/* For work registers, we can subdivide in various ways. So we create
- * classes for the various sizes and conflict accordingly, keeping in
- * mind that physical registers are divided along 128-bit boundaries.
- * The important part is that 128-bit boundaries are not crossed.
- *
- * For each 128-bit register, we can subdivide to 32-bits 10 ways
- *
- * vec4: xyzw
- * vec3: xyz, yzw
- * vec2: xy, yz, zw,
- * vec1: x, y, z, w
- *
- * For each 64-bit register, we can subdivide similarly to 16-bit
- * (TODO: half-float RA, not that we support fp16 yet)
- */
-
-#define WORK_STRIDE 10
-
-/* We have overlapping register classes for special registers, handled via
- * shadows */
-
-#define SHADOW_R0  17
-#define SHADOW_R28 18
-#define SHADOW_R29 19
-
-/* Prepacked masks/swizzles for virtual register types */
-static unsigned reg_type_to_mask[WORK_STRIDE] = {
-        0xF,                                    /* xyzw */
-        0x7, 0x7 << 1,                          /* xyz */
-                 0x3, 0x3 << 1, 0x3 << 2,                /* xy */
-                 0x1, 0x1 << 1, 0x1 << 2, 0x1 << 3       /* x */
-};
-
-static unsigned reg_type_to_swizzle[WORK_STRIDE] = {
-        SWIZZLE(COMPONENT_X, COMPONENT_Y, COMPONENT_Z, COMPONENT_W),
-
-        SWIZZLE(COMPONENT_X, COMPONENT_Y, COMPONENT_Z, COMPONENT_W),
-        SWIZZLE(COMPONENT_Y, COMPONENT_Z, COMPONENT_W, COMPONENT_W),
-
-        SWIZZLE(COMPONENT_X, COMPONENT_Y, COMPONENT_Z, COMPONENT_W),
-        SWIZZLE(COMPONENT_Y, COMPONENT_Z, COMPONENT_Z, COMPONENT_W),
-        SWIZZLE(COMPONENT_Z, COMPONENT_W, COMPONENT_Z, COMPONENT_W),
-
-        SWIZZLE(COMPONENT_X, COMPONENT_Y, COMPONENT_Z, COMPONENT_W),
-        SWIZZLE(COMPONENT_Y, COMPONENT_Y, COMPONENT_Z, COMPONENT_W),
-        SWIZZLE(COMPONENT_Z, COMPONENT_Y, COMPONENT_Z, COMPONENT_W),
-        SWIZZLE(COMPONENT_W, COMPONENT_Y, COMPONENT_Z, COMPONENT_W),
-};
+#include "lcra.h"
 
 struct phys_reg {
+        /* Physical register: 0-31 */
         unsigned reg;
-        unsigned mask;
-        unsigned swizzle;
-};
 
-/* Given the mask/swizzle of both the register and the original source,
- * compose to find the actual mask/swizzle to give the hardware */
+        /* Byte offset into the physical register: 0-15 */
+        unsigned offset;
 
-static unsigned
-compose_writemask(unsigned mask, struct phys_reg reg)
-{
-        /* Note: the reg mask is guaranteed to be contiguous. So we shift
-         * into the X place, compose via a simple AND, and shift back */
+        /* Number of bytes in a component of this register */
+        unsigned size;
+};
 
-        unsigned shift = __builtin_ctz(reg.mask);
-        return ((reg.mask >> shift) & mask) << shift;
-}
+/* Shift up by reg_offset and horizontally by dst_offset. */
 
-static unsigned
-compose_swizzle(unsigned swizzle, unsigned mask,
-                struct phys_reg reg, struct phys_reg dst)
+static void
+offset_swizzle(unsigned *swizzle, unsigned reg_offset, unsigned srcsize, unsigned dst_offset)
 {
-        unsigned out = pan_compose_swizzle(swizzle, reg.swizzle);
+        unsigned out[MIR_VEC_COMPONENTS];
 
-        /* Based on the register mask, we need to adjust over. E.g if we're
-         * writing to yz, a base swizzle of xy__ becomes _xy_. Save the
-         * original first component (x). But to prevent duplicate shifting
-         * (only applies to ALU -- mask param is set to xyzw out on L/S to
-         * prevent changes), we have to account for the shift inherent to the
-         * original writemask */
+        signed reg_comp = reg_offset / srcsize;
+        signed dst_comp = dst_offset / srcsize;
 
-        unsigned rep = out & 0x3;
-        unsigned shift = __builtin_ctz(dst.mask) - __builtin_ctz(mask);
-        unsigned shifted = out << (2*shift);
+        unsigned max_component = (16 / srcsize) - 1;
 
-        /* ..but we fill in the gaps so it appears to replicate */
+        assert(reg_comp * srcsize == reg_offset);
+        assert(dst_comp * srcsize == dst_offset);
 
-        for (unsigned s = 0; s < shift; ++s)
-                shifted |= rep << (2*s);
+        for (signed c = 0; c < MIR_VEC_COMPONENTS; ++c) {
+                signed comp = MAX2(c - dst_comp, 0);
+                out[c] = MIN2(swizzle[comp] + reg_comp, max_component);
+        }
 
-        return shifted;
+        memcpy(swizzle, out, sizeof(out));
 }
 
 /* Helper to return the default phys_reg for a given register */
 
 static struct phys_reg
-default_phys_reg(int reg)
+default_phys_reg(int reg, midgard_reg_mode size)
 {
         struct phys_reg r = {
                 .reg = reg,
-                .mask = 0xF, /* xyzw */
-                .swizzle = 0xE4 /* xyzw */
+                .offset = 0,
+                .size = mir_bytes_for_mode(size)
         };
 
         return r;
@@ -139,206 +80,37 @@ default_phys_reg(int reg)
  * register corresponds to */
 
 static struct phys_reg
-index_to_reg(compiler_context *ctx, struct ra_graph *g, unsigned reg)
+index_to_reg(compiler_context *ctx, struct lcra_state *l, unsigned reg, midgard_reg_mode size)
 {
         /* Check for special cases */
         if (reg == ~0)
-                return default_phys_reg(REGISTER_UNUSED);
+                return default_phys_reg(REGISTER_UNUSED, size);
         else if (reg >= SSA_FIXED_MINIMUM)
-                return default_phys_reg(SSA_REG_FROM_FIXED(reg));
-        else if (!g)
-                return default_phys_reg(REGISTER_UNUSED);
-
-        /* Special cases aside, we pick the underlying register */
-        int virt = ra_get_node_reg(g, reg);
-
-        /* Divide out the register and classification */
-        int phys = virt / WORK_STRIDE;
-        int type = virt % WORK_STRIDE;
-
-        /* Apply shadow registers */
-
-        if (phys >= SHADOW_R28 && phys <= SHADOW_R29)
-                phys += 28 - SHADOW_R28;
-        else if (phys == SHADOW_R0)
-                phys = 0;
+                return default_phys_reg(SSA_REG_FROM_FIXED(reg), size);
+        else if (!l)
+                return default_phys_reg(REGISTER_UNUSED, size);
 
         struct phys_reg r = {
-                .reg = phys,
-                .mask = reg_type_to_mask[type],
-                .swizzle = reg_type_to_swizzle[type]
+                .reg = l->solutions[reg] / 16,
+                .offset = l->solutions[reg] & 0xF,
+                .size = mir_bytes_for_mode(size)
         };
 
         /* Report that we actually use this register, and return it */
 
-        if (phys < 16)
-                ctx->work_registers = MAX2(ctx->work_registers, phys);
+        if (r.reg < 16)
+                ctx->work_registers = MAX2(ctx->work_registers, r.reg);
 
         return r;
 }
 
-/* This routine creates a register set. Should be called infrequently since
- * it's slow and can be cached. For legibility, variables are named in terms of
- * work registers, although it is also used to create the register set for
- * special register allocation */
-
-static void
-add_shadow_conflicts (struct ra_regs *regs, unsigned base, unsigned shadow, unsigned shadow_count)
-{
-        for (unsigned a = 0; a < WORK_STRIDE; ++a) {
-                unsigned reg_a = (WORK_STRIDE * base) + a;
-
-                for (unsigned b = 0; b < shadow_count; ++b) {
-                        unsigned reg_b = (WORK_STRIDE * shadow) + b;
-
-                        ra_add_reg_conflict(regs, reg_a, reg_b);
-                        ra_add_reg_conflict(regs, reg_b, reg_a);
-                }
-        }
-}
-
-static struct ra_regs *
-create_register_set(unsigned work_count, unsigned *classes)
-{
-        int virtual_count = 32 * WORK_STRIDE;
-
-        /* First, initialize the RA */
-        struct ra_regs *regs = ra_alloc_reg_set(NULL, virtual_count, true);
-
-        for (unsigned c = 0; c < (NR_REG_CLASSES - 1); ++c) {
-                int work_vec4 = ra_alloc_reg_class(regs);
-                int work_vec3 = ra_alloc_reg_class(regs);
-                int work_vec2 = ra_alloc_reg_class(regs);
-                int work_vec1 = ra_alloc_reg_class(regs);
-
-                classes[4*c + 0] = work_vec1;
-                classes[4*c + 1] = work_vec2;
-                classes[4*c + 2] = work_vec3;
-                classes[4*c + 3] = work_vec4;
-
-                /* Special register classes have other register counts */
-                unsigned count =
-                        (c == REG_CLASS_WORK)   ? work_count : 2;
-
-                unsigned first_reg =
-                        (c == REG_CLASS_LDST)   ? 26 :
-                        (c == REG_CLASS_TEXR)   ? 28 :
-                        (c == REG_CLASS_TEXW)   ? SHADOW_R28 :
-                        0;
-
-                /* Add the full set of work registers */
-                for (unsigned i = first_reg; i < (first_reg + count); ++i) {
-                        int base = WORK_STRIDE * i;
-
-                        /* Build a full set of subdivisions */
-                        ra_class_add_reg(regs, work_vec4, base);
-                        ra_class_add_reg(regs, work_vec3, base + 1);
-                        ra_class_add_reg(regs, work_vec3, base + 2);
-                        ra_class_add_reg(regs, work_vec2, base + 3);
-                        ra_class_add_reg(regs, work_vec2, base + 4);
-                        ra_class_add_reg(regs, work_vec2, base + 5);
-                        ra_class_add_reg(regs, work_vec1, base + 6);
-                        ra_class_add_reg(regs, work_vec1, base + 7);
-                        ra_class_add_reg(regs, work_vec1, base + 8);
-                        ra_class_add_reg(regs, work_vec1, base + 9);
-
-                        for (unsigned a = 0; a < 10; ++a) {
-                                unsigned mask1 = reg_type_to_mask[a];
-
-                                for (unsigned b = 0; b < 10; ++b) {
-                                        unsigned mask2 = reg_type_to_mask[b];
-
-                                        if (mask1 & mask2)
-                                                ra_add_reg_conflict(regs,
-                                                                    base + a, base + b);
-                                }
-                        }
-                }
-        }
-
-        int fragc = ra_alloc_reg_class(regs);
-
-        classes[4*REG_CLASS_FRAGC + 0] = fragc;
-        classes[4*REG_CLASS_FRAGC + 1] = fragc;
-        classes[4*REG_CLASS_FRAGC + 2] = fragc;
-        classes[4*REG_CLASS_FRAGC + 3] = fragc;
-        ra_class_add_reg(regs, fragc, WORK_STRIDE * SHADOW_R0);
-
-        /* We have duplicate classes */
-        add_shadow_conflicts(regs,  0, SHADOW_R0,  1);
-        add_shadow_conflicts(regs, 28, SHADOW_R28, WORK_STRIDE);
-        add_shadow_conflicts(regs, 29, SHADOW_R29, WORK_STRIDE);
-
-        /* We're done setting up */
-        ra_set_finalize(regs, NULL);
-
-        return regs;
-}
-
-/* This routine gets a precomputed register set off the screen if it's able, or
- * otherwise it computes one on the fly */
-
-static struct ra_regs *
-get_register_set(struct midgard_screen *screen, unsigned work_count, unsigned **classes)
-{
-        /* Bounds check */
-        assert(work_count >= 8);
-        assert(work_count <= 16);
-
-        /* Compute index */
-        unsigned index = work_count - 8;
-
-        /* Find the reg set */
-        struct ra_regs *cached = screen->regs[index];
-
-        if (cached) {
-                assert(screen->reg_classes[index]);
-                *classes = screen->reg_classes[index];
-                return cached;
-        }
-
-        /* Otherwise, create one */
-        struct ra_regs *created = create_register_set(work_count, screen->reg_classes[index]);
-
-        /* Cache it and use it */
-        screen->regs[index] = created;
-
-        *classes = screen->reg_classes[index];
-        return created;
-}
-
-/* Assign a (special) class, ensuring that it is compatible with whatever class
- * was already set */
-
 static void
 set_class(unsigned *classes, unsigned node, unsigned class)
 {
-        /* Check that we're even a node */
-        if (node >= SSA_FIXED_MINIMUM)
-                return;
-
-        /* First 4 are work, next 4 are load/store.. */
-        unsigned current_class = classes[node] >> 2;
-
-        /* Nothing to do */
-        if (class == current_class)
-                return;
-
-        /* If we're changing, we haven't assigned a special class */
-        assert(current_class == REG_CLASS_WORK);
-
-        classes[node] &= 0x3;
-        classes[node] |= (class << 2);
-}
-
-static void
-force_vec4(unsigned *classes, unsigned node)
-{
-        if (node >= SSA_FIXED_MINIMUM)
-                return;
-
-        /* Force vec4 = 3 */
-        classes[node] |= 0x3;
+        if (node < SSA_FIXED_MINIMUM && class != classes[node]) {
+                assert(classes[node] == REG_CLASS_WORK);
+                classes[node] = class;
+        }
 }
 
 /* Special register classes impose special constraints on who can read their
@@ -351,9 +123,7 @@ check_read_class(unsigned *classes, unsigned tag, unsigned node)
         if (node >= SSA_FIXED_MINIMUM)
                 return true;
 
-        unsigned current_class = classes[node] >> 2;
-
-        switch (current_class) {
+        switch (classes[node]) {
         case REG_CLASS_LDST:
                 return (tag == TAG_LOAD_STORE_4);
         case REG_CLASS_TEXR:
@@ -374,9 +144,7 @@ check_write_class(unsigned *classes, unsigned tag, unsigned node)
         if (node >= SSA_FIXED_MINIMUM)
                 return true;
 
-        unsigned current_class = classes[node] >> 2;
-
-        switch (current_class) {
+        switch (classes[node]) {
         case REG_CLASS_TEXR:
                 return true;
         case REG_CLASS_TEXW:
@@ -499,8 +267,7 @@ mir_lower_special_reads(compiler_context *ctx)
                         unsigned idx = spill_idx++;
 
                         midgard_instruction m = hazard_write ?
-                                v_mov(idx, blank_alu_src, i) :
-                                v_mov(i, blank_alu_src, idx);
+                                v_mov(idx, i) : v_mov(i, idx);
 
                         /* Insert move before each read/write, depending on the
                          * hazard we're trying to account for */
@@ -524,7 +291,7 @@ mir_lower_special_reads(compiler_context *ctx)
                                         mir_rewrite_index_dst_single(pre_use, i, idx);
                                 } else {
                                         idx = spill_idx++;
-                                        m = v_mov(i, blank_alu_src, idx);
+                                        m = v_mov(i, idx);
                                         m.mask = mir_from_bytemask(mir_bytemask_of_read_components(pre_use, i), midgard_reg_mode_32);
                                         mir_insert_instruction_before(ctx, pre_use, m);
                                         mir_rewrite_index_src_single(pre_use, i, idx);
@@ -541,10 +308,73 @@ mir_lower_special_reads(compiler_context *ctx)
         free(texw);
 }
 
+/* We register allocate after scheduling, so we need to ensure instructions
+ * executing in parallel within a segment of a bundle don't clobber each
+ * other's registers. This is mostly a non-issue thanks to scheduling, but
+ * there are edge cases. In particular, after a register is written in a
+ * segment, it interferes with anything reading. */
+
+static void
+mir_compute_segment_interference(
+                compiler_context *ctx,
+                struct lcra_state *l,
+                midgard_bundle *bun,
+                unsigned pivot,
+                unsigned i)
+{
+        for (unsigned j = pivot; j < i; ++j) {
+                mir_foreach_src(bun->instructions[j], s) {
+                        if (bun->instructions[j]->src[s] >= ctx->temp_count)
+                                continue;
+
+                        for (unsigned q = pivot; q < i; ++q) {
+                                if (bun->instructions[q]->dest >= ctx->temp_count)
+                                        continue;
+
+                                /* See dEQP-GLES2.functional.shaders.return.output_write_in_func_dynamic_fragment */
+
+                                if (q >= j) {
+                                        if (!(bun->instructions[j]->unit == UNIT_SMUL && bun->instructions[q]->unit == UNIT_VLUT))
+                                                continue;
+                                }
+
+                                unsigned mask = mir_bytemask(bun->instructions[q]);
+                                unsigned rmask = mir_bytemask_of_read_components(bun->instructions[j], bun->instructions[j]->src[s]);
+                                lcra_add_node_interference(l, bun->instructions[q]->dest, mask, bun->instructions[j]->src[s], rmask);
+                        }
+                }
+        }
+}
+
+static void
+mir_compute_bundle_interference(
+                compiler_context *ctx,
+                struct lcra_state *l,
+                midgard_bundle *bun)
+{
+        if (!IS_ALU(bun->tag))
+                return;
+
+        bool old = bun->instructions[0]->unit >= UNIT_VADD;
+        unsigned pivot = 0;
+
+        for (unsigned i = 1; i < bun->instruction_count; ++i) {
+                bool new = bun->instructions[i]->unit >= UNIT_VADD;
+
+                if (old != new) {
+                        mir_compute_segment_interference(ctx, l, bun, 0, i);
+                        pivot = i;
+                        break;
+                }
+        }
+
+        mir_compute_segment_interference(ctx, l, bun, pivot, bun->instruction_count);
+}
+
 static void
 mir_compute_interference(
                 compiler_context *ctx,
-                struct ra_graph *g)
+                struct lcra_state *l)
 {
         /* First, we need liveness information to be computed per block */
         mir_compute_liveness(ctx);
@@ -564,14 +394,19 @@ mir_compute_interference(
 
                         if (dest < ctx->temp_count) {
                                 for (unsigned i = 0; i < ctx->temp_count; ++i)
-                                        if (live[i])
-                                                ra_add_node_interference(g, dest, i);
+                                        if (live[i]) {
+                                                unsigned mask = mir_bytemask(ins);
+                                                lcra_add_node_interference(l, dest, mask, i, live[i]);
+                                        }
                         }
 
                         /* Update live_in */
                         mir_liveness_ins_update(live, ins, ctx->temp_count);
                 }
 
+                mir_foreach_bundle_in_block(blk, bun)
+                        mir_compute_bundle_interference(ctx, l, bun);
+
                 free(live);
         }
 }
@@ -579,31 +414,32 @@ mir_compute_interference(
 /* This routine performs the actual register allocation. It should be succeeded
  * by install_registers */
 
-struct ra_graph *
+struct lcra_state *
 allocate_registers(compiler_context *ctx, bool *spilled)
 {
         /* The number of vec4 work registers available depends on when the
          * uniforms start, so compute that first */
         int work_count = 16 - MAX2((ctx->uniform_cutoff - 8), 0);
-        unsigned *classes = NULL;
-        struct ra_regs *regs = get_register_set(ctx->screen, work_count, &classes);
-
-        assert(regs != NULL);
-        assert(classes != NULL);
 
        /* No register allocation to do with no SSA */
 
         if (!ctx->temp_count)
                 return NULL;
 
-        /* Let's actually do register allocation */
-        int nodes = ctx->temp_count;
-        struct ra_graph *g = ra_alloc_interference_graph(regs, nodes);
-        
-        /* Register class (as known to the Mesa register allocator) is actually
-         * the product of both semantic class (work, load/store, texture..) and
-         * size (vec2/vec3..). First, we'll go through and determine the
-         * minimum size needed to hold values */
+        struct lcra_state *l = lcra_alloc_equations(ctx->temp_count, 1, 8, 16, 5);
+
+        /* Starts of classes, in bytes */
+        l->class_start[REG_CLASS_WORK]  = 16 * 0;
+        l->class_start[REG_CLASS_LDST]  = 16 * 26;
+        l->class_start[REG_CLASS_TEXR]  = 16 * 28;
+        l->class_start[REG_CLASS_TEXW]  = 16 * 28;
+
+        l->class_size[REG_CLASS_WORK] = 16 * work_count;
+        l->class_size[REG_CLASS_LDST]  = 16 * 2;
+        l->class_size[REG_CLASS_TEXR]  = 16 * 2;
+        l->class_size[REG_CLASS_TEXW]  = 16 * 2;
+
+        lcra_set_disjoint_class(l, REG_CLASS_TEXR, REG_CLASS_TEXW);
 
         unsigned *found_class = calloc(sizeof(unsigned), ctx->temp_count);
 
@@ -618,8 +454,19 @@ allocate_registers(compiler_context *ctx, bool *spilled)
 
                 int dest = ins->dest;
                 found_class[dest] = MAX2(found_class[dest], class);
+
+                lcra_set_alignment(l, dest, 2); /* (1 << 2) = 4 */
+
+                /* XXX: Ensure swizzles align the right way with more LCRA constraints? */
+                if (ins->type == TAG_ALU_4 && ins->alu.reg_mode != midgard_reg_mode_32)
+                        lcra_set_alignment(l, dest, 3); /* (1 << 3) = 8 */
         }
 
+        for (unsigned i = 0; i < ctx->temp_count; ++i)
+                lcra_restrict_range(l, i, (found_class[i] + 1) * 4);
+        
+        free(found_class);
+
         /* Next, we'll determine semantic class. We default to zero (work).
          * But, if we're used with a special operation, that will force us to a
          * particular class. Each node must be assigned to exactly one class; a
@@ -631,57 +478,38 @@ allocate_registers(compiler_context *ctx, bool *spilled)
                 /* Check if this operation imposes any classes */
 
                 if (ins->type == TAG_LOAD_STORE_4) {
-                        bool force_vec4_only = OP_IS_VEC4_ONLY(ins->load_store.op);
+                        set_class(l->class, ins->src[0], REG_CLASS_LDST);
+                        set_class(l->class, ins->src[1], REG_CLASS_LDST);
+                        set_class(l->class, ins->src[2], REG_CLASS_LDST);
 
-                        set_class(found_class, ins->src[0], REG_CLASS_LDST);
-                        set_class(found_class, ins->src[1], REG_CLASS_LDST);
-                        set_class(found_class, ins->src[2], REG_CLASS_LDST);
-
-                        if (force_vec4_only) {
-                                force_vec4(found_class, ins->dest);
-                                force_vec4(found_class, ins->src[0]);
-                                force_vec4(found_class, ins->src[1]);
-                                force_vec4(found_class, ins->src[2]);
-                        }
+                        if (OP_IS_VEC4_ONLY(ins->load_store.op))
+                                lcra_restrict_range(l, ins->dest, 16);
                 } else if (ins->type == TAG_TEXTURE_4) {
-                        set_class(found_class, ins->dest, REG_CLASS_TEXW);
-                        set_class(found_class, ins->src[0], REG_CLASS_TEXR);
-                        set_class(found_class, ins->src[1], REG_CLASS_TEXR);
-                        set_class(found_class, ins->src[2], REG_CLASS_TEXR);
+                        set_class(l->class, ins->dest, REG_CLASS_TEXW);
+                        set_class(l->class, ins->src[0], REG_CLASS_TEXR);
+                        set_class(l->class, ins->src[1], REG_CLASS_TEXR);
+                        set_class(l->class, ins->src[2], REG_CLASS_TEXR);
                 }
         }
 
         /* Check that the semantics of the class are respected */
         mir_foreach_instr_global(ctx, ins) {
-                assert(check_write_class(found_class, ins->type, ins->dest));
-                assert(check_read_class(found_class, ins->type, ins->src[0]));
-                assert(check_read_class(found_class, ins->type, ins->src[1]));
-                assert(check_read_class(found_class, ins->type, ins->src[2]));
+                assert(check_write_class(l->class, ins->type, ins->dest));
+                assert(check_read_class(l->class, ins->type, ins->src[0]));
+                assert(check_read_class(l->class, ins->type, ins->src[1]));
+                assert(check_read_class(l->class, ins->type, ins->src[2]));
         }
 
         /* Mark writeout to r0 */
         mir_foreach_instr_global(ctx, ins) {
-                if (ins->compact_branch && ins->writeout)
-                        set_class(found_class, ins->src[0], REG_CLASS_FRAGC);
+                if (ins->compact_branch && ins->writeout && ins->src[0] < ctx->temp_count)
+                        l->solutions[ins->src[0]] = 0;
         }
+        
+        mir_compute_interference(ctx, l);
 
-        for (unsigned i = 0; i < ctx->temp_count; ++i) {
-                unsigned class = found_class[i];
-                ra_set_node_class(g, i, classes[class]);
-        }
-
-        mir_compute_interference(ctx, g);
-
-        if (!ra_allocate(g)) {
-                *spilled = true;
-        } else {
-                *spilled = false;
-        }
-
-        /* Whether we were successful or not, report the graph so we can
-         * compute spill nodes */
-
-        return g;
+        *spilled = !lcra_solve(l);
+        return l;
 }
 
 /* Once registers have been decided via register allocation
@@ -691,7 +519,7 @@ allocate_registers(compiler_context *ctx, bool *spilled)
 static void
 install_registers_instr(
         compiler_context *ctx,
-        struct ra_graph *g,
+        struct lcra_state *l,
         midgard_instruction *ins)
 {
         switch (ins->type) {
@@ -702,21 +530,17 @@ install_registers_instr(
                  if (ins->compact_branch)
                          return;
 
-                struct phys_reg src1 = index_to_reg(ctx, g, ins->src[0]);
-                struct phys_reg src2 = index_to_reg(ctx, g, ins->src[1]);
-                struct phys_reg dest = index_to_reg(ctx, g, ins->dest);
+                struct phys_reg src1 = index_to_reg(ctx, l, ins->src[0], mir_srcsize(ins, 0));
+                struct phys_reg src2 = index_to_reg(ctx, l, ins->src[1], mir_srcsize(ins, 1));
+                struct phys_reg dest = index_to_reg(ctx, l, ins->dest, mir_typesize(ins));
 
-                unsigned uncomposed_mask = ins->mask;
-                ins->mask = compose_writemask(uncomposed_mask, dest);
+                mir_set_bytemask(ins, mir_bytemask(ins) << dest.offset);
 
-                /* Adjust the dest mask if necessary. Mostly this is a no-op
-                 * but it matters for dot products */
-                dest.mask = effective_writemask(&ins->alu, ins->mask);
+                unsigned dest_offset =
+                        GET_CHANNEL_COUNT(alu_opcode_props[ins->alu.op].props) ? 0 :
+                        dest.offset;
 
-                midgard_vector_alu_src mod1 =
-                        vector_alu_from_unsigned(ins->alu.src1);
-                mod1.swizzle = compose_swizzle(mod1.swizzle, uncomposed_mask, src1, dest);
-                ins->alu.src1 = vector_alu_srco_unsigned(mod1);
+                offset_swizzle(ins->swizzle[0], src1.offset, src1.size, dest_offset);
 
                 ins->registers.src1_reg = src1.reg;
 
@@ -736,8 +560,7 @@ install_registers_instr(
                 } else {
                         midgard_vector_alu_src mod2 =
                                 vector_alu_from_unsigned(ins->alu.src2);
-                        mod2.swizzle = compose_swizzle(
-                                               mod2.swizzle, uncomposed_mask, src2, dest);
+                        offset_swizzle(ins->swizzle[1], src2.offset, src2.size, dest_offset);
                         ins->alu.src2 = vector_alu_srco_unsigned(mod2);
 
                         ins->registers.src2_reg = src2.reg;
@@ -755,54 +578,35 @@ install_registers_instr(
                 bool encodes_src = OP_IS_STORE(ins->load_store.op);
 
                 if (encodes_src) {
-                        struct phys_reg src = index_to_reg(ctx, g, ins->src[0]);
+                        struct phys_reg src = index_to_reg(ctx, l, ins->src[0], mir_srcsize(ins, 0));
                         assert(src.reg == 26 || src.reg == 27);
 
                         ins->load_store.reg = src.reg - 26;
-
-                        unsigned shift = __builtin_ctz(src.mask);
-                        unsigned adjusted_mask = src.mask >> shift;
-                        assert(((adjusted_mask + 1) & adjusted_mask) == 0);
-
-                        unsigned new_swizzle = 0;
-                        for (unsigned q = 0; q < 4; ++q) {
-                                unsigned c = (ins->load_store.swizzle >> (2*q)) & 3;
-                                new_swizzle |= (c + shift) << (2*q);
-                        }
-
-                        ins->load_store.swizzle = compose_swizzle(
-                                                          new_swizzle, src.mask,
-                                                          default_phys_reg(0), src);
+                        offset_swizzle(ins->swizzle[0], src.offset, src.size, 0);
                } else {
-                        struct phys_reg src = index_to_reg(ctx, g, ins->dest);
+                        struct phys_reg dst = index_to_reg(ctx, l, ins->dest, mir_typesize(ins));
 
-                        ins->load_store.reg = src.reg;
-
-                        ins->load_store.swizzle = compose_swizzle(
-                                                          ins->load_store.swizzle, 0xF,
-                                                          default_phys_reg(0), src);
-
-                        ins->mask = compose_writemask(
-                                            ins->mask, src);
+                        ins->load_store.reg = dst.reg;
+                        offset_swizzle(ins->swizzle[0], 0, 4, dst.offset);
+                        mir_set_bytemask(ins, mir_bytemask(ins) << dst.offset);
                 }
 
                 /* We also follow up by actual arguments */
 
-                int src2 =
-                        encodes_src ? ins->src[1] : ins->src[0];
+                unsigned src2 = ins->src[1];
+                unsigned src3 = ins->src[2];
 
-                int src3 =
-                        encodes_src ? ins->src[2] : ins->src[1];
-
-                if (src2 >= 0) {
-                        struct phys_reg src = index_to_reg(ctx, g, src2);
-                        unsigned component = __builtin_ctz(src.mask);
+                if (src2 != ~0) {
+                        struct phys_reg src = index_to_reg(ctx, l, src2, mir_srcsize(ins, 1));
+                        unsigned component = src.offset / src.size;
+                        assert(component * src.size == src.offset);
                         ins->load_store.arg_1 |= midgard_ldst_reg(src.reg, component);
                 }
 
-                if (src3 >= 0) {
-                        struct phys_reg src = index_to_reg(ctx, g, src3);
-                        unsigned component = __builtin_ctz(src.mask);
+                if (src3 != ~0) {
+                        struct phys_reg src = index_to_reg(ctx, l, src3, mir_srcsize(ins, 2));
+                        unsigned component = src.offset / src.size;
+                        assert(component * src.size == src.offset);
                         ins->load_store.arg_2 |= midgard_ldst_reg(src.reg, component);
                 }
  
@@ -811,35 +615,30 @@ install_registers_instr(
 
         case TAG_TEXTURE_4: {
                 /* Grab RA results */
-                struct phys_reg dest = index_to_reg(ctx, g, ins->dest);
-                struct phys_reg coord = index_to_reg(ctx, g, ins->src[0]);
-                struct phys_reg lod = index_to_reg(ctx, g, ins->src[1]);
-
-                assert(dest.reg == 28 || dest.reg == 29);
-                assert(coord.reg == 28 || coord.reg == 29);
+                struct phys_reg dest = index_to_reg(ctx, l, ins->dest, mir_typesize(ins));
+                struct phys_reg coord = index_to_reg(ctx, l, ins->src[1], mir_srcsize(ins, 1));
+                struct phys_reg lod = index_to_reg(ctx, l, ins->src[2], mir_srcsize(ins, 2));
 
                 /* First, install the texture coordinate */
                 ins->texture.in_reg_full = 1;
                 ins->texture.in_reg_upper = 0;
-                ins->texture.in_reg_select = coord.reg - 28;
-                ins->texture.in_reg_swizzle =
-                        pan_compose_swizzle(ins->texture.in_reg_swizzle, coord.swizzle);
+                ins->texture.in_reg_select = coord.reg & 1;
+                offset_swizzle(ins->swizzle[1], coord.offset, coord.size, 0);
 
                 /* Next, install the destination */
                 ins->texture.out_full = 1;
                 ins->texture.out_upper = 0;
-                ins->texture.out_reg_select = dest.reg - 28;
-                ins->texture.swizzle =
-                        compose_swizzle(ins->texture.swizzle, dest.mask, dest, dest);
-                ins->mask =
-                        compose_writemask(ins->mask, dest);
+                ins->texture.out_reg_select = dest.reg & 1;
+                offset_swizzle(ins->swizzle[0], 0, 4, dest.offset);
+                mir_set_bytemask(ins, mir_bytemask(ins) << dest.offset);
 
                 /* If there is a register LOD/bias, use it */
-                if (ins->src[1] != ~0) {
+                if (ins->src[2] != ~0) {
+                        assert(!(lod.offset & 3));
                         midgard_tex_register_select sel = {
                                 .select = lod.reg,
                                 .full = 1,
-                                .component = lod.swizzle & 3,
+                                .component = lod.offset / 4
                         };
 
                         uint8_t packed;
@@ -856,8 +655,8 @@ install_registers_instr(
 }
 
 void
-install_registers(compiler_context *ctx, struct ra_graph *g)
+install_registers(compiler_context *ctx, struct lcra_state *l)
 {
         mir_foreach_instr_global(ctx, ins)
-                install_registers_instr(ctx, g, ins);
+                install_registers_instr(ctx, l, ins);
 }