nv50: support for geometry programs in nv50_program
[mesa.git] / src / gallium / drivers / nv50 / nv50_program.c
index 3c1e9eca8740959456a0e9ca7680c9243239f723..6647f759b5abb428f7da2d417004077d6cc568e4 100644 (file)
@@ -92,6 +92,9 @@ struct nv50_reg {
 
        int rhw; /* result hw for FP outputs, or interpolant index */
        int acc; /* instruction where this reg is last read (first insn == 1) */
+
+       int vtx; /* vertex index, for GP inputs (TGSI Dimension.Index) */
+       int indirect[2]; /* index into pc->addr, or -1 */
 };
 
 #define NV50_MOD_NEG 1
@@ -135,7 +138,6 @@ struct nv50_pc {
        int immd_nr;
        struct nv50_reg **addr;
        int addr_nr;
-       uint8_t addr_alloc; /* set bit indicates used for TGSI_FILE_ADDRESS */
 
        struct nv50_reg *temp_temp[16];
        struct nv50_program_exec *temp_temp_exec[16];
@@ -171,6 +173,8 @@ struct nv50_pc {
        uint8_t edgeflag_out;
 };
 
+static struct nv50_reg *get_address_reg(struct nv50_pc *, struct nv50_reg *);
+
 static INLINE void
 ctor_reg(struct nv50_reg *reg, unsigned type, int index, int hw)
 {
@@ -179,7 +183,9 @@ ctor_reg(struct nv50_reg *reg, unsigned type, int index, int hw)
        reg->hw = hw;
        reg->mod = 0;
        reg->rhw = -1;
+       reg->vtx = -1;
        reg->acc = 0;
+       reg->indirect[0] = reg->indirect[1] = -1;
 }
 
 static INLINE unsigned
@@ -197,7 +203,8 @@ terminate_mbb(struct nv50_pc *pc)
 
        /* remove records of temporary address register values */
        for (i = 0; i < NV50_SU_MAX_ADDR; ++i)
-               pc->r_addr[i].rhw = -1;
+               if (pc->r_addr[i].index < 0)
+                       pc->r_addr[i].acc = 0;
 }
 
 static void
@@ -246,7 +253,8 @@ alloc_reg(struct nv50_pc *pc, struct nv50_reg *reg)
                }
        }
 
-       assert(0);
+       NOUVEAU_ERR("out of registers\n");
+       abort();
 }
 
 static INLINE struct nv50_reg *
@@ -259,6 +267,7 @@ reg_instance(struct nv50_pc *pc, struct nv50_reg *reg)
        if (reg) {
                alloc_reg(pc, reg);
                *ri = *reg;
+               reg->indirect[0] = reg->indirect[1] = -1;
                reg->mod = 0;
        }
        return ri;
@@ -286,7 +295,8 @@ alloc_temp(struct nv50_pc *pc, struct nv50_reg *dst)
                }
        }
 
-       assert(0);
+       NOUVEAU_ERR("out of registers\n");
+       abort();
        return NULL;
 }
 
@@ -523,11 +533,33 @@ set_immd(struct nv50_pc *pc, struct nv50_reg *imm, struct nv50_program_exec *e)
 static INLINE void
 set_addr(struct nv50_program_exec *e, struct nv50_reg *a)
 {
+       assert(a->type == P_ADDR);
+
        assert(!(e->inst[0] & 0x0c000000));
        assert(!(e->inst[1] & 0x00000004));
 
        e->inst[0] |= (a->hw & 3) << 26;
-       e->inst[1] |= (a->hw >> 2) << 2;
+       e->inst[1] |= a->hw & 4;
+}
+
+static void
+emit_arl(struct nv50_pc *, struct nv50_reg *, struct nv50_reg *, uint8_t);
+
+static void
+emit_shl_imm(struct nv50_pc *, struct nv50_reg *, struct nv50_reg *, int);
+
+static void
+emit_mov_from_addr(struct nv50_pc *pc, struct nv50_reg *dst,
+                  struct nv50_reg *src)
+{
+       struct nv50_program_exec *e = exec(pc);
+
+       e->inst[1] = 0x40000000;
+       set_long(pc, e);
+       set_dst(pc, dst, e);
+       set_addr(e, src);
+
+       emit(pc, e);
 }
 
 static void
@@ -546,72 +578,6 @@ emit_add_addr_imm(struct nv50_pc *pc, struct nv50_reg *dst,
        emit(pc, e);
 }
 
-static struct nv50_reg *
-alloc_addr(struct nv50_pc *pc, struct nv50_reg *ref)
-{
-       struct nv50_reg *a_tgsi = NULL, *a = NULL;
-       int i;
-       uint8_t avail = ~pc->addr_alloc;
-
-       if (!ref) {
-               /* allocate for TGSI_FILE_ADDRESS */
-               while (avail) {
-                       i = ffs(avail) - 1;
-
-                       if (pc->r_addr[i].rhw < 0 ||
-                           pc->r_addr[i].acc != pc->insn_cur) {
-                               pc->addr_alloc |= (1 << i);
-
-                               pc->r_addr[i].rhw = -1;
-                               pc->r_addr[i].index = i;
-                               return &pc->r_addr[i];
-                       }
-                       avail &= ~(1 << i);
-               }
-               assert(0);
-               return NULL;
-       }
-
-       /* Allocate and set an address reg so we can access 'ref'.
-        *
-        * If and r_addr->index will be -1 or the hw index the value
-        * value in rhw is relative to. If rhw < 0, the reg has not
-        * been initialized or is in use for TGSI_FILE_ADDRESS.
-        */
-       while (avail) { /* only consider regs that are not TGSI */
-               i = ffs(avail) - 1;
-               avail &= ~(1 << i);
-
-               if ((!a || a->rhw >= 0) && pc->r_addr[i].rhw < 0) {
-                       /* prefer an usused reg with low hw index */
-                       a = &pc->r_addr[i];
-                       continue;
-               }
-               if (!a && pc->r_addr[i].acc != pc->insn_cur)
-                       a = &pc->r_addr[i];
-
-               if (ref->hw - pc->r_addr[i].rhw >= 128)
-                       continue;
-
-               if ((ref->acc >= 0 && pc->r_addr[i].index < 0) ||
-                   (ref->acc < 0 && pc->r_addr[i].index == ref->index)) {
-                       pc->r_addr[i].acc = pc->insn_cur;
-                       return &pc->r_addr[i];
-               }
-       }
-       assert(a);
-
-       if (ref->acc < 0)
-               a_tgsi = pc->addr[ref->index];
-
-       emit_add_addr_imm(pc, a, a_tgsi, (ref->hw & ~0x7f) * 4);
-
-       a->rhw = ref->hw & ~0x7f;
-       a->acc = pc->insn_cur;
-       a->index = a_tgsi ? ref->index : -1;
-       return a;
-}
-
 #define INTERP_LINEAR          0
 #define INTERP_FLAT            1
 #define INTERP_PERSPECTIVE     2
@@ -655,12 +621,12 @@ set_data(struct nv50_pc *pc, struct nv50_reg *src, unsigned m, unsigned s,
        e->param.shift = s;
        e->param.mask = m << (s % 32);
 
-       if (src->hw > 127)
-               set_addr(e, alloc_addr(pc, src));
+       if (src->hw < 0 || src->hw > 127) /* need (additional) address reg */
+               set_addr(e, get_address_reg(pc, src));
        else
        if (src->acc < 0) {
                assert(src->type == P_CONST);
-               set_addr(e, pc->addr[src->index]);
+               set_addr(e, pc->addr[src->indirect[0]]);
        }
 
        e->inst[1] |= (((src->type == P_IMMD) ? 0 : 1) << 22);
@@ -692,6 +658,12 @@ emit_mov(struct nv50_pc *pc, struct nv50_reg *dst, struct nv50_reg *src)
                if (src->type == P_ATTR) {
                        set_long(pc, e);
                        e->inst[1] |= 0x00200000;
+
+                       if (src->vtx >= 0) {
+                               /* indirect (vertex base + c) load from p[] */
+                               e->inst[0] |= 0x01800000;
+                               set_addr(e, get_address_reg(pc, src));
+                       }
                }
 
                alloc_reg(pc, src);
@@ -806,6 +778,11 @@ set_src_0(struct nv50_pc *pc, struct nv50_reg *src, struct nv50_program_exec *e)
        if (src->type == P_ATTR) {
                set_long(pc, e);
                e->inst[1] |= 0x00200000;
+
+               if (src->vtx >= 0) {
+                       e->inst[0] |= 0x01800000; /* src from p[] */
+                       set_addr(e, get_address_reg(pc, src));
+               }
        } else
        if (src->type == P_CONST || src->type == P_IMMD) {
                struct nv50_reg *temp = temp_temp(pc, e);
@@ -830,13 +807,13 @@ set_src_1(struct nv50_pc *pc, struct nv50_reg *src, struct nv50_program_exec *e)
                src = temp;
        } else
        if (src->type == P_CONST || src->type == P_IMMD) {
-               assert(!(e->inst[0] & 0x00800000));
-               if (e->inst[0] & 0x01000000) {
+               if (e->inst[0] & 0x01800000) {
                        struct nv50_reg *temp = temp_temp(pc, e);
 
                        emit_mov(pc, temp, src);
                        src = temp;
                } else {
+                       assert(!(e->inst[0] & 0x00800000));
                        set_data(pc, src, 0x7f, 16, e);
                        e->inst[0] |= 0x00800000;
                }
@@ -860,13 +837,13 @@ set_src_2(struct nv50_pc *pc, struct nv50_reg *src, struct nv50_program_exec *e)
                src = temp;
        } else
        if (src->type == P_CONST || src->type == P_IMMD) {
-               assert(!(e->inst[0] & 0x01000000));
-               if (e->inst[0] & 0x00800000) {
+               if (e->inst[0] & 0x01800000) {
                        struct nv50_reg *temp = temp_temp(pc, e);
 
                        emit_mov(pc, temp, src);
                        src = temp;
                } else {
+                       assert(!(e->inst[0] & 0x01000000));
                        set_data(pc, src, 0x7f, 32+14, e);
                        e->inst[0] |= 0x01000000;
                }
@@ -876,6 +853,26 @@ set_src_2(struct nv50_pc *pc, struct nv50_reg *src, struct nv50_program_exec *e)
        e->inst[1] |= ((src->hw & 127) << 14);
 }
 
+static void
+set_half_src(struct nv50_pc *pc, struct nv50_reg *src, int lh,
+            struct nv50_program_exec *e, int pos)
+{
+       struct nv50_reg *r = src;
+
+       alloc_reg(pc, r);
+       if (r->type != P_TEMP) {
+               r = temp_temp(pc, e);
+               emit_mov(pc, r, src);
+       }
+
+       if (r->hw > (NV50_SU_MAX_TEMP / 2)) {
+               NOUVEAU_ERR("out of low GPRs\n");
+               abort();
+       }
+
+       e->inst[pos / 32] |= ((src->hw * 2) + lh) << (pos % 32);
+}
+
 static void
 emit_mov_from_pred(struct nv50_pc *pc, struct nv50_reg *dst, int pred)
 {
@@ -975,11 +972,125 @@ emit_arl(struct nv50_pc *pc, struct nv50_reg *dst, struct nv50_reg *src,
 
        e->inst[0] |= dst->hw << 2;
        e->inst[0] |= s << 16; /* shift left */
-       set_src_0_restricted(pc, src, e);
+       set_src_0(pc, src, e);
 
        emit(pc, e);
 }
 
+static boolean
+address_reg_suitable(struct nv50_reg *a, struct nv50_reg *r)
+{
+       if (!r)
+               return FALSE;
+
+       if (r->vtx != a->vtx)
+               return FALSE;
+       if (r->vtx >= 0)
+               return (r->indirect[1] == a->indirect[1]);
+
+       if (r->hw < a->rhw || (r->hw - a->rhw) >= 128)
+               return FALSE;
+
+       if (a->index >= 0)
+               return (a->index == r->indirect[0]);
+       return (a->indirect[0] == r->indirect[0]);
+}
+
+static void
+load_vertex_base(struct nv50_pc *pc, struct nv50_reg *dst,
+                struct nv50_reg *a, int shift)
+{
+       struct nv50_reg mem, *temp;
+
+       ctor_reg(&mem, P_ATTR, -1, dst->vtx);
+
+       assert(dst->type == P_ADDR);
+       if (!a) {
+               emit_arl(pc, dst, &mem, 0);
+               return;
+       }
+       temp = alloc_temp(pc, NULL);
+
+       if (shift) {
+               emit_mov_from_addr(pc, temp, a);
+               if (shift < 0)
+                       emit_shl_imm(pc, temp, temp, shift);
+               emit_arl(pc, dst, temp, MAX2(shift, 0));
+       }
+       emit_mov(pc, temp, &mem);
+       set_addr(pc->p->exec_tail, dst);
+
+       emit_arl(pc, dst, temp, 0);
+       free_temp(pc, temp);
+}
+
+/* case (ref == NULL): allocate address register for TGSI_FILE_ADDRESS
+ * case (vtx >= 0, acc >= 0): load vertex base from a[vtx * 4] to $aX
+ * case (vtx >= 0, acc < 0): load vertex base from s[$aY + vtx * 4] to $aX
+ * case (vtx < 0, acc >= 0): memory address too high to encode
+ * case (vtx < 0, acc < 0): get source register for TGSI_FILE_ADDRESS
+ */
+static struct nv50_reg *
+get_address_reg(struct nv50_pc *pc, struct nv50_reg *ref)
+{
+       int i;
+       struct nv50_reg *a_ref, *a = NULL;
+
+       for (i = 0; i < NV50_SU_MAX_ADDR; ++i) {
+               if (pc->r_addr[i].acc == 0)
+                       a = &pc->r_addr[i]; /* an unused address reg */
+               else
+               if (address_reg_suitable(&pc->r_addr[i], ref)) {
+                       pc->r_addr[i].acc = pc->insn_cur;
+                       return &pc->r_addr[i];
+               } else
+               if (!a && pc->r_addr[i].index < 0 &&
+                   pc->r_addr[i].acc < pc->insn_cur)
+                       a = &pc->r_addr[i];
+       }
+       if (!a) {
+               /* We'll be able to spill address regs when this
+                * mess is replaced with a proper compiler ...
+                */
+               NOUVEAU_ERR("out of address regs\n");
+               abort();
+               return NULL;
+       }
+
+       /* initialize and reserve for this TGSI instruction */
+       a->rhw = 0;
+       a->index = a->indirect[0] = a->indirect[1] = -1;
+       a->acc = pc->insn_cur;
+
+       if (!ref) {
+               a->vtx = -1;
+               return a;
+       }
+       a->vtx = ref->vtx;
+
+       /* now put in the correct value ... */
+
+       if (ref->vtx >= 0) {
+               a->indirect[1] = ref->indirect[1];
+
+               /* For an indirect vertex index, we need to shift address right
+                * by 2, the address register will contain vtx * 16, we need to
+                * load from a[vtx * 4].
+                */
+               load_vertex_base(pc, a, (ref->acc < 0) ?
+                                pc->addr[ref->indirect[1]] : NULL, -2);
+       } else {
+               assert(ref->acc < 0 || ref->indirect[0] < 0);
+
+               a->rhw = ref->hw & ~0x7f;
+               a->indirect[0] = ref->indirect[0];
+               a_ref = (ref->acc < 0) ? pc->addr[ref->indirect[0]] : NULL;
+
+               emit_add_addr_imm(pc, a, a_ref, a->rhw * 4);
+       }
+       return a;
+}
+
 #define NV50_MAX_F32 0x880
 #define NV50_MAX_S32 0x08c
 #define NV50_MAX_U32 0x084
@@ -1058,6 +1169,20 @@ emit_bitop2(struct nv50_pc *pc, struct nv50_reg *dst, struct nv50_reg *src0,
        emit(pc, e);
 }
 
+static void
+emit_not(struct nv50_pc *pc, struct nv50_reg *dst, struct nv50_reg *src)
+{
+       struct nv50_program_exec *e = exec(pc);
+
+       e->inst[0] = 0xd0000000;
+       e->inst[1] = 0x0402c000;
+       set_long(pc, e);
+       set_dst(pc, dst, e);
+       set_src_1(pc, src, e);
+
+       emit(pc, e);
+}
+
 static void
 emit_shift(struct nv50_pc *pc, struct nv50_reg *dst,
           struct nv50_reg *src0, struct nv50_reg *src1, unsigned dir)
@@ -1086,6 +1211,27 @@ emit_shift(struct nv50_pc *pc, struct nv50_reg *dst,
        emit(pc, e);
 }
 
+static void
+emit_shl_imm(struct nv50_pc *pc, struct nv50_reg *dst,
+            struct nv50_reg *src, int s)
+{
+       struct nv50_program_exec *e = exec(pc);
+
+       e->inst[0] = 0x30000000;
+       e->inst[1] = 0xc4100000;
+       if (s < 0) {
+               e->inst[1] |= 1 << 29;
+               s = -s;
+       }
+       e->inst[1] |= ((s & 0x7f) << 16);
+
+       set_long(pc, e);
+       set_dst(pc, dst, e);
+       set_src_0(pc, src, e);
+
+       emit(pc, e);
+}
+
 static void
 emit_mad(struct nv50_pc *pc, struct nv50_reg *dst, struct nv50_reg *src0,
         struct nv50_reg *src1, struct nv50_reg *src2)
@@ -1361,6 +1507,43 @@ emit_add_b32(struct nv50_pc *pc, struct nv50_reg *dst,
        emit(pc, e);
 }
 
+static void
+emit_mad_u16(struct nv50_pc *pc, struct nv50_reg *dst,
+            struct nv50_reg *src0, int lh_0, struct nv50_reg *src1, int lh_1,
+            struct nv50_reg *src2)
+{
+       struct nv50_program_exec *e = exec(pc);
+
+       e->inst[0] = 0x60000000;
+       if (!pc->allow32)
+               set_long(pc, e);
+       set_dst(pc, dst, e);
+
+       set_half_src(pc, src0, lh_0, e, 9);
+       set_half_src(pc, src1, lh_1, e, 16);
+       alloc_reg(pc, src2);
+       if (is_long(e) || (src2->type != P_TEMP) || (src2->hw != dst->hw))
+               set_src_2(pc, src2, e);
+
+       emit(pc, e);
+}
+
+static void
+emit_mul_u16(struct nv50_pc *pc, struct nv50_reg *dst,
+            struct nv50_reg *src0, int lh_0, struct nv50_reg *src1, int lh_1)
+{
+       struct nv50_program_exec *e = exec(pc);
+
+       e->inst[0] = 0x40000000;
+       set_long(pc, e);
+       set_dst(pc, dst, e);
+
+       set_half_src(pc, src0, lh_0, e, 9);
+       set_half_src(pc, src1, lh_1, e, 16);
+
+       emit(pc, e);
+}
+
 static void
 emit_sad(struct nv50_pc *pc, struct nv50_reg *dst,
         struct nv50_reg *src0, struct nv50_reg *src1, struct nv50_reg *src2)
@@ -1368,6 +1551,9 @@ emit_sad(struct nv50_pc *pc, struct nv50_reg *dst,
        struct nv50_program_exec *e = exec(pc);
 
        e->inst[0] = 0x50000000;
+       if (!pc->allow32)
+               set_long(pc, e);
+       check_swap_src_0_1(pc, &src0, &src1);
        set_dst(pc, dst, e);
        set_src_0(pc, src0, e);
        set_src_1(pc, src1, e);
@@ -1379,6 +1565,8 @@ emit_sad(struct nv50_pc *pc, struct nv50_reg *dst,
                e->inst[1] |= 0x0c << 24;
        else
                e->inst[0] |= 0x81 << 8;
+
+       emit(pc, e);
 }
 
 static INLINE void
@@ -1530,6 +1718,18 @@ emit_ret(struct nv50_pc *pc, int pred, unsigned cc)
        emit_control_flow(pc, 0x3, pred, cc);
 }
 
+static void
+emit_prim_cmd(struct nv50_pc *pc, unsigned cmd)
+{
+       struct nv50_program_exec *e = exec(pc);
+
+       e->inst[0] = 0xf0000000 | (cmd << 9);
+       e->inst[1] = 0xc0000000;
+       set_long(pc, e);
+
+       emit(pc, e);
+}
+
 #define QOP_ADD 0
 #define QOP_SUBR 1
 #define QOP_SUB 2
@@ -1890,7 +2090,11 @@ convert_to_long(struct nv50_pc *pc, struct nv50_program_exec *e)
        case 0x5:
                /* SAD */
                m = ~(0x81 << 8);
-               q = 0x0c << 24;
+               q = (0x0c << 24) | ((e->inst[0] & (0x7f << 2)) << 12);
+               break;
+       case 0x6:
+               /* MAD u16 */
+               q = (e->inst[0] & (0x7f << 2)) << 12;
                break;
        case 0x8:
                /* INTERP (move centroid, perspective and flat bits) */
@@ -1970,8 +2174,11 @@ get_supported_mods(const struct tgsi_full_instruction *insn, int i)
        case TGSI_OPCODE_IMAX:
        case TGSI_OPCODE_IMIN:
        case TGSI_OPCODE_ISHR:
+       case TGSI_OPCODE_NOT:
+       case TGSI_OPCODE_UMAD:
        case TGSI_OPCODE_UMAX:
        case TGSI_OPCODE_UMIN:
+       case TGSI_OPCODE_UMUL:
        case TGSI_OPCODE_USHR:
                return NV50_MOD_I32;
        default:
@@ -2065,8 +2272,9 @@ tgsi_dst(struct nv50_pc *pc, int c, const struct tgsi_full_dst_register *dst)
        {
                struct nv50_reg *r = pc->addr[dst->Register.Index * 4 + c];
                if (!r) {
-                       r = alloc_addr(pc, NULL);
-                       pc->addr[dst->Register.Index * 4 + c] = r;
+                       r = get_address_reg(pc, NULL);
+                       r->index = dst->Register.Index * 4 + c;
+                       pc->addr[r->index] = r;
                }
                assert(r);
                return r;
@@ -2102,6 +2310,18 @@ tgsi_src(struct nv50_pc *pc, int chan, const struct tgsi_full_src_register *src,
                switch (src->Register.File) {
                case TGSI_FILE_INPUT:
                        r = &pc->attr[src->Register.Index * 4 + c];
+
+                       if (!src->Dimension.Dimension)
+                               break;
+                       r = reg_instance(pc, r);
+                       r->vtx = src->Dimension.Index;
+
+                       if (!src->Dimension.Indirect)
+                               break;
+                       swz = tgsi_util_get_src_register_swizzle(
+                               &src->DimIndirect, 0);
+                       r->acc = -1;
+                       r->indirect[1] = src->DimIndirect.Index * 4 + swz;
                        break;
                case TGSI_FILE_TEMPORARY:
                        r = &pc->temp[src->Register.Index * 4 + c];
@@ -2115,12 +2335,12 @@ tgsi_src(struct nv50_pc *pc, int chan, const struct tgsi_full_src_register *src,
                         * use the index field to select the address reg.
                         */
                        r = reg_instance(pc, NULL);
+                       ctor_reg(r, P_CONST, -1, src->Register.Index * 4 + c);
+
                        swz = tgsi_util_get_src_register_swizzle(
-                                                &src->Indirect, 0);
-                       ctor_reg(r, P_CONST,
-                                src->Indirect.Index * 4 + swz,
-                                src->Register.Index * 4 + c);
+                               &src->Indirect, 0);
                        r->acc = -1;
+                       r->indirect[0] = src->Indirect.Index * 4 + swz;
                        break;
                case TGSI_FILE_IMMEDIATE:
                        r = &pc->immd[src->Register.Index * 4 + c];
@@ -2167,7 +2387,7 @@ tgsi_src(struct nv50_pc *pc, int chan, const struct tgsi_full_src_register *src,
                r->mod |= mod & NV50_MOD_I32;
 
        assert(r);
-       if (r->acc >= 0 && r != temp)
+       if (r->acc >= 0 && r->vtx < 0 && r != temp)
                return reg_instance(pc, r); /* will clear r->mod */
        return r;
 }
@@ -2389,10 +2609,14 @@ nv50_program_tx_insn(struct nv50_pc *pc,
                }
                break;
        case TGSI_OPCODE_ARL:
-               assert(src[0][0]);
                temp = temp_temp(pc, NULL);
-               emit_cvt(pc, temp, src[0][0], -1, CVT_FLOOR | CVT_S32_F32);
-               emit_arl(pc, dst[0], temp, 4);
+               for (c = 0; c < 4; c++) {
+                       if (!(mask & (1 << c)))
+                               continue;
+                       emit_cvt(pc, temp, src[0][c], -1,
+                                CVT_FLOOR | CVT_S32_F32);
+                       emit_arl(pc, dst[c], temp, 4);
+               }
                break;
        case TGSI_OPCODE_BGNLOOP:
                pc->loop_brka[pc->loop_lvl] = emit_breakaddr(pc);
@@ -2499,6 +2723,9 @@ nv50_program_tx_insn(struct nv50_pc *pc,
                pc->if_insn[pc->if_lvl++] = pc->p->exec_tail;
                terminate_mbb(pc);
                break;
+       case TGSI_OPCODE_EMIT:
+               emit_prim_cmd(pc, 1);
+               break;
        case TGSI_OPCODE_ENDIF:
                pc->if_insn[--pc->if_lvl]->param.index = pc->p->exec_size;
 
@@ -2522,8 +2749,12 @@ nv50_program_tx_insn(struct nv50_pc *pc,
                pc->loop_brka[pc->loop_lvl]->param.index = pc->p->exec_size;
                terminate_mbb(pc);
                break;
+       case TGSI_OPCODE_ENDPRIM:
+               emit_prim_cmd(pc, 2);
+               break;
        case TGSI_OPCODE_ENDSUB:
                assert(pc->in_subroutine);
+               terminate_mbb(pc);
                pc->in_subroutine = FALSE;
                break;
        case TGSI_OPCODE_EX2:
@@ -2713,6 +2944,13 @@ nv50_program_tx_insn(struct nv50_pc *pc,
                        emit_mul(pc, dst[c], src[0][c], src[1][c]);
                }
                break;
+       case TGSI_OPCODE_NOT:
+               for (c = 0; c < 4; c++) {
+                       if (!(mask & (1 << c)))
+                               continue;
+                       emit_not(pc, dst[c], src[0][c]);
+               }
+               break;
        case TGSI_OPCODE_POW:
                emit_pow(pc, brdc, src[0][0], src[1][0]);
                break;
@@ -2857,6 +3095,39 @@ nv50_program_tx_insn(struct nv50_pc *pc,
                        emit_minmax(pc, 0x0a4, dst[c], src[0][c], src[1][c]);
                }
                break;
+       case TGSI_OPCODE_UMAD:
+       {
+               assert(!temp);
+               temp = temp_temp(pc, NULL);
+               for (c = 0; c < 4; c++) {
+                       if (!(mask & (1 << c)))
+                               continue;
+                       emit_mul_u16(pc, temp, src[0][c], 0, src[1][c], 1);
+                       emit_mad_u16(pc, temp, src[0][c], 1, src[1][c], 0,
+                                    temp);
+                       emit_shl_imm(pc, temp, temp, 16);
+                       emit_mad_u16(pc, temp, src[0][c], 0, src[1][c], 0,
+                                    temp);
+                       emit_add_b32(pc, dst[c], temp, src[2][c]);
+               }
+       }
+               break;
+       case TGSI_OPCODE_UMUL:
+       {
+               assert(!temp);
+               temp = temp_temp(pc, NULL);
+               for (c = 0; c < 4; c++) {
+                       if (!(mask & (1 << c)))
+                               continue;
+                       emit_mul_u16(pc, temp, src[0][c], 0, src[1][c], 1);
+                       emit_mad_u16(pc, temp, src[0][c], 1, src[1][c], 0,
+                                    temp);
+                       emit_shl_imm(pc, temp, temp, 16);
+                       emit_mad_u16(pc, dst[c], src[0][c], 0, src[1][c], 0,
+                                    temp);
+               }
+       }
+               break;
        case TGSI_OPCODE_XPD:
                temp = temp_temp(pc, NULL);
                if (mask & (1 << 0)) {
@@ -2886,6 +3157,8 @@ nv50_program_tx_insn(struct nv50_pc *pc,
                        emit_nop(pc);
 
                pc->p->exec_tail->inst[1] |= 1; /* set exit bit */
+
+               terminate_mbb(pc);
                break;
        default:
                NOUVEAU_ERR("invalid opcode %d\n", inst->Instruction.Opcode);
@@ -3194,6 +3467,24 @@ load_frontfacing(struct nv50_pc *pc, struct nv50_reg *a)
        FREE(one);
 }
 
+static void
+copy_semantic_info(struct nv50_program *p)
+{
+       unsigned i, id;
+
+       for (i = 0; i < p->cfg.in_nr; ++i) {
+               id = p->cfg.in[i].id;
+               p->cfg.in[i].sn = p->info.input_semantic_name[id];
+               p->cfg.in[i].si = p->info.input_semantic_index[id];
+       }
+
+       for (i = 0; i < p->cfg.out_nr; ++i) {
+               id = p->cfg.out[i].id;
+               p->cfg.out[i].sn = p->info.output_semantic_name[id];
+               p->cfg.out[i].si = p->info.output_semantic_index[id];
+       }
+}
+
 static boolean
 nv50_program_tx_prep(struct nv50_pc *pc)
 {
@@ -3240,13 +3531,13 @@ nv50_program_tx_prep(struct nv50_pc *pc)
                                switch (d->Semantic.Name) {
                                case TGSI_SEMANTIC_BCOLOR:
                                        p->cfg.two_side[si].hw = first;
-                                       if (p->cfg.io_nr > first)
-                                               p->cfg.io_nr = first;
+                                       if (p->cfg.out_nr > first)
+                                               p->cfg.out_nr = first;
                                        break;
                                case TGSI_SEMANTIC_PSIZE:
                                        p->cfg.psiz = first;
-                                       if (p->cfg.io_nr > first)
-                                               p->cfg.io_nr = first;
+                                       if (p->cfg.out_nr > first)
+                                               p->cfg.out_nr = first;
                                        break;
                                case TGSI_SEMANTIC_EDGEFLAG:
                                        pc->edgeflag_out = first;
@@ -3306,68 +3597,86 @@ nv50_program_tx_prep(struct nv50_pc *pc)
                }
        }
 
-       if (p->type == PIPE_SHADER_VERTEX) {
+       if (p->type == PIPE_SHADER_VERTEX || p->type == PIPE_SHADER_GEOMETRY) {
                int rid = 0;
 
-               for (i = 0; i < pc->attr_nr * 4; ++i) {
-                       if (pc->attr[i].acc) {
-                               pc->attr[i].hw = rid++;
-                               p->cfg.attr[i / 32] |= 1 << (i % 32);
+               if (p->type == PIPE_SHADER_GEOMETRY) {
+                       for (i = 0; i < pc->attr_nr; ++i) {
+                               p->cfg.in[i].hw = rid;
+                               p->cfg.in[i].id = i;
+
+                               for (c = 0; c < 4; ++c) {
+                                       int n = i * 4 + c;
+                                       if (!pc->attr[n].acc)
+                                               continue;
+                                       pc->attr[n].hw = rid++;
+                                       p->cfg.in[i].mask |= 1 << c;
+                               }
+                       }
+               } else {
+                       for (i = 0; i < pc->attr_nr * 4; ++i) {
+                               if (pc->attr[i].acc) {
+                                       pc->attr[i].hw = rid++;
+                                       p->cfg.attr[i / 32] |= 1 << (i % 32);
+                               }
                        }
                }
 
                for (i = 0, rid = 0; i < pc->result_nr; ++i) {
-                       p->cfg.io[i].hw = rid;
-                       p->cfg.io[i].id = i;
+                       p->cfg.out[i].hw = rid;
+                       p->cfg.out[i].id = i;
 
                        for (c = 0; c < 4; ++c) {
                                int n = i * 4 + c;
                                if (!pc->result[n].acc)
                                        continue;
                                pc->result[n].hw = rid++;
-                               p->cfg.io[i].mask |= 1 << c;
+                               p->cfg.out[i].mask |= 1 << c;
                        }
                }
 
                for (c = 0; c < 2; ++c)
                        if (p->cfg.two_side[c].hw < 0x40)
-                               p->cfg.two_side[c] = p->cfg.io[
+                               p->cfg.two_side[c] = p->cfg.out[
                                        p->cfg.two_side[c].hw];
 
                if (p->cfg.psiz < 0x40)
-                       p->cfg.psiz = p->cfg.io[p->cfg.psiz].hw;
+                       p->cfg.psiz = p->cfg.out[p->cfg.psiz].hw;
+
+               copy_semantic_info(p);
        } else
        if (p->type == PIPE_SHADER_FRAGMENT) {
-               int rid, aid;
+               int rid, aid, base;
                unsigned n = 0, m = pc->attr_nr - flat_nr;
 
                pc->allow32 = TRUE;
 
-               int base = (TGSI_SEMANTIC_POSITION ==
-                           p->info.input_semantic_name[0]) ? 0 : 1;
+               base = (TGSI_SEMANTIC_POSITION ==
+                       p->info.input_semantic_name[0]) ? 0 : 1;
 
                /* non-flat interpolants have to be mapped to
                 * the lower hardware IDs, so sort them:
                 */
                for (i = 0; i < pc->attr_nr; i++) {
                        if (pc->interp_mode[i] == INTERP_FLAT)
-                               p->cfg.io[m++].id = i;
+                               p->cfg.in[m++].id = i;
                        else {
                                if (!(pc->interp_mode[i] & INTERP_PERSPECTIVE))
-                                       p->cfg.io[n].linear = TRUE;
-                               p->cfg.io[n++].id = i;
+                                       p->cfg.in[n].linear = TRUE;
+                               p->cfg.in[n++].id = i;
                        }
                }
+               copy_semantic_info(p);
 
                if (!base) /* set w-coordinate mask from perspective interp */
-                       p->cfg.io[0].mask |= p->cfg.regs[1] >> 24;
+                       p->cfg.in[0].mask |= p->cfg.regs[1] >> 24;
 
                aid = popcnt4( /* if fcrd isn't contained in cfg.io */
-                       base ? (p->cfg.regs[1] >> 24) : p->cfg.io[0].mask);
+                       base ? (p->cfg.regs[1] >> 24) : p->cfg.in[0].mask);
 
                for (n = 0; n < pc->attr_nr; ++n) {
-                       p->cfg.io[n].hw = rid = aid;
-                       i = p->cfg.io[n].id;
+                       p->cfg.in[n].hw = rid = aid;
+                       i = p->cfg.in[n].id;
 
                        if (p->info.input_semantic_name[n] ==
                            TGSI_SEMANTIC_FACE) {
@@ -3379,15 +3688,15 @@ nv50_program_tx_prep(struct nv50_pc *pc)
                                if (!pc->attr[i * 4 + c].acc)
                                        continue;
                                pc->attr[i * 4 + c].rhw = rid++;
-                               p->cfg.io[n].mask |= 1 << c;
+                               p->cfg.in[n].mask |= 1 << c;
 
                                load_interpolant(pc, &pc->attr[i * 4 + c]);
                        }
-                       aid += popcnt4(p->cfg.io[n].mask);
+                       aid += popcnt4(p->cfg.in[n].mask);
                }
 
                if (!base)
-                       p->cfg.regs[1] |= p->cfg.io[0].mask << 24;
+                       p->cfg.regs[1] |= p->cfg.in[0].mask << 24;
 
                m = popcnt4(p->cfg.regs[1] >> 24);
 
@@ -3397,30 +3706,23 @@ nv50_program_tx_prep(struct nv50_pc *pc)
                p->cfg.regs[1] |= aid - m;
 
                if (flat_nr) {
-                       i = p->cfg.io[pc->attr_nr - flat_nr].hw;
+                       i = p->cfg.in[pc->attr_nr - flat_nr].hw;
                        p->cfg.regs[1] |= (i - m) << 16;
                } else
                        p->cfg.regs[1] |= p->cfg.regs[1] << 16;
 
                /* mark color semantic for light-twoside */
-               n = 0x40;
-               for (i = 0; i < pc->attr_nr; i++) {
-                       ubyte si, sn;
-
-                       sn = p->info.input_semantic_name[p->cfg.io[i].id];
-                       si = p->info.input_semantic_index[p->cfg.io[i].id];
-
-                       if (sn == TGSI_SEMANTIC_COLOR) {
-                               p->cfg.two_side[si] = p->cfg.io[i];
-
-                               /* increase colour count */
-                               p->cfg.regs[0] += popcnt4(
-                                       p->cfg.two_side[si].mask) << 16;
-
-                               n = MIN2(n, p->cfg.io[i].hw - m);
+               n = 0x80;
+               for (i = 0; i < p->cfg.in_nr; i++) {
+                       if (p->cfg.in[i].sn == TGSI_SEMANTIC_COLOR) {
+                               n = MIN2(n, p->cfg.in[i].hw - m);
+                               p->cfg.two_side[p->cfg.in[i].si] = p->cfg.in[i];
+
+                               p->cfg.regs[0] += /* increase colour count */
+                                       popcnt4(p->cfg.in[i].mask) << 16;
                        }
                }
-               if (n < 0x40)
+               if (n < 0x80)
                        p->cfg.regs[0] += n;
 
                /* Initialize FP results:
@@ -3476,10 +3778,29 @@ free_nv50_pc(struct nv50_pc *pc)
                FREE(pc->attr);
        if (pc->temp)
                FREE(pc->temp);
+       if (pc->insn_pos)
+               FREE(pc->insn_pos);
 
        FREE(pc);
 }
 
+static INLINE uint32_t
+nv50_map_gs_output_prim(unsigned pprim)
+{
+       switch (pprim) {
+       case PIPE_PRIM_POINTS:
+               return NV50TCL_GP_OUTPUT_PRIMITIVE_TYPE_POINTS;
+       case PIPE_PRIM_LINE_STRIP:
+               return NV50TCL_GP_OUTPUT_PRIMITIVE_TYPE_LINE_STRIP;
+       case PIPE_PRIM_TRIANGLE_STRIP:
+               return NV50TCL_GP_OUTPUT_PRIMITIVE_TYPE_TRIANGLE_STRIP;
+       default:
+               NOUVEAU_ERR("invalid GS_OUTPUT_PRIMITIVE: %u\n", pprim);
+               abort();
+               return 0;
+       }
+}
+
 static boolean
 ctor_nv50_pc(struct nv50_pc *pc, struct nv50_program *p)
 {
@@ -3501,17 +3822,44 @@ ctor_nv50_pc(struct nv50_pc *pc, struct nv50_program *p)
 
        p->cfg.edgeflag_in = pc->edgeflag_out = 0xff;
 
+       for (i = 0; i < p->info.num_properties; ++i) {
+               unsigned *data = &p->info.properties[i].data[0];
+
+               switch (p->info.properties[i].name) {
+               case TGSI_PROPERTY_GS_OUTPUT_PRIM:
+                       p->cfg.prim_type = nv50_map_gs_output_prim(data[0]);
+                       break;
+               case TGSI_PROPERTY_GS_MAX_VERTICES:
+                       p->cfg.vert_count = data[0];
+                       break;
+               default:
+                       break;
+               }
+       }
+
        switch (p->type) {
        case PIPE_SHADER_VERTEX:
                p->cfg.psiz = 0x40;
                p->cfg.clpd = 0x40;
-               p->cfg.io_nr = pc->result_nr;
+               p->cfg.out_nr = pc->result_nr;
+               break;
+       case PIPE_SHADER_GEOMETRY:
+               assert(p->cfg.prim_type);
+               assert(p->cfg.vert_count);
+
+               p->cfg.psiz = 0x80;
+               p->cfg.clpd = 0x80;
+               p->cfg.out_nr = pc->result_nr;
+               p->cfg.in_nr = pc->attr_nr;
+
+               p->cfg.two_side[0].hw = 0x80;
+               p->cfg.two_side[1].hw = 0x80;
                break;
        case PIPE_SHADER_FRAGMENT:
                rtype[0] = rtype[1] = P_TEMP;
 
                p->cfg.regs[0] = 0x01000004;
-               p->cfg.io_nr = pc->attr_nr;
+               p->cfg.in_nr = pc->attr_nr;
 
                if (p->info.writes_z) {
                        p->cfg.regs[2] |= 0x00000100;
@@ -3569,7 +3917,7 @@ ctor_nv50_pc(struct nv50_pc *pc, struct nv50_program *p)
                        return FALSE;
        }
        for (i = 0; i < NV50_SU_MAX_ADDR; ++i)
-               ctor_reg(&pc->r_addr[i], P_ADDR, -256, i + 1);
+               ctor_reg(&pc->r_addr[i], P_ADDR, -1, i + 1);
 
        return TRUE;
 }
@@ -3731,13 +4079,17 @@ nv50_program_validate_data(struct nv50_context *nv50, struct nv50_program *p)
 
        if (p->param_nr) {
                unsigned cb;
-               uint32_t *map = pipe_buffer_map(pscreen, nv50->constbuf[p->type],
+               uint32_t *map = pipe_buffer_map(pscreen,
+                                               nv50->constbuf[p->type],
                                                PIPE_BUFFER_USAGE_CPU_READ);
-
-               if (p->type == PIPE_SHADER_VERTEX)
+               switch (p->type) {
+               case PIPE_SHADER_GEOMETRY: cb = NV50_CB_PGP; break;
+               case PIPE_SHADER_FRAGMENT: cb = NV50_CB_PFP; break;
+               default:
                        cb = NV50_CB_PVP;
-               else
-                       cb = NV50_CB_PFP;
+                       assert(p->type == PIPE_SHADER_VERTEX);
+                       break;
+               }
 
                nv50_program_upload_data(nv50, map, 0, p->param_nr, cb);
                pipe_buffer_unmap(pscreen, nv50->constbuf[p->type]);
@@ -3831,19 +4183,18 @@ nv50_vertprog_validate(struct nv50_context *nv50)
        nv50_program_validate_data(nv50, p);
        nv50_program_validate_code(nv50, p);
 
-       so = so_new(5, 8, 2);
+       so = so_new(5, 7, 2);
        so_method(so, tesla, NV50TCL_VP_ADDRESS_HIGH, 2);
        so_reloc (so, p->bo, 0, NOUVEAU_BO_VRAM | NOUVEAU_BO_RD |
-                     NOUVEAU_BO_HIGH, 0, 0);
+                 NOUVEAU_BO_HIGH, 0, 0);
        so_reloc (so, p->bo, 0, NOUVEAU_BO_VRAM | NOUVEAU_BO_RD |
-                     NOUVEAU_BO_LOW, 0, 0);
+                 NOUVEAU_BO_LOW, 0, 0);
        so_method(so, tesla, NV50TCL_VP_ATTR_EN_0, 2);
        so_data  (so, p->cfg.attr[0]);
        so_data  (so, p->cfg.attr[1]);
        so_method(so, tesla, NV50TCL_VP_REG_ALLOC_RESULT, 1);
        so_data  (so, p->cfg.high_result);
-       so_method(so, tesla, NV50TCL_VP_RESULT_MAP_SIZE, 2);
-       so_data  (so, p->cfg.high_result); //8);
+       so_method(so, tesla, NV50TCL_VP_REG_ALLOC_TEMP, 1);
        so_data  (so, p->cfg.high_temp);
        so_method(so, tesla, NV50TCL_VP_START_ID, 1);
        so_data  (so, 0); /* program start offset */
@@ -3887,42 +4238,74 @@ nv50_fragprog_validate(struct nv50_context *nv50)
        so_ref(NULL, &so);
 }
 
+void
+nv50_geomprog_validate(struct nv50_context *nv50)
+{
+       struct nouveau_grobj *tesla = nv50->screen->tesla;
+       struct nv50_program *p = nv50->geomprog;
+       struct nouveau_stateobj *so;
+
+       if (!p->translated) {
+               nv50_program_validate(nv50, p);
+               if (!p->translated)
+                       assert(0);
+       }
+
+       nv50_program_validate_data(nv50, p);
+       nv50_program_validate_code(nv50, p);
+
+       so = so_new(6, 7, 2);
+       so_method(so, tesla, NV50TCL_GP_ADDRESS_HIGH, 2);
+       so_reloc (so, p->bo, 0, NOUVEAU_BO_VRAM | NOUVEAU_BO_RD |
+                 NOUVEAU_BO_HIGH, 0, 0);
+       so_reloc (so, p->bo, 0, NOUVEAU_BO_VRAM | NOUVEAU_BO_RD |
+                 NOUVEAU_BO_LOW, 0, 0);
+       so_method(so, tesla, NV50TCL_GP_REG_ALLOC_TEMP, 1);
+       so_data  (so, p->cfg.high_temp);
+       so_method(so, tesla, NV50TCL_GP_REG_ALLOC_RESULT, 1);
+       so_data  (so, p->cfg.high_result);
+       so_method(so, tesla, NV50TCL_GP_OUTPUT_PRIMITIVE_TYPE, 1);
+       so_data  (so, p->cfg.prim_type);
+       so_method(so, tesla, NV50TCL_GP_VERTEX_OUTPUT_COUNT, 1);
+       so_data  (so, p->cfg.vert_count);
+       so_method(so, tesla, NV50TCL_GP_START_ID, 1);
+       so_data  (so, 0);
+       so_ref(so, &nv50->state.geomprog);
+       so_ref(NULL, &so);
+}
+
 static uint32_t
 nv50_pntc_replace(struct nv50_context *nv50, uint32_t pntc[8], unsigned base)
 {
+       struct nv50_program *vp;
        struct nv50_program *fp = nv50->fragprog;
-       struct nv50_program *vp = nv50->vertprog;
        unsigned i, c, m = base;
        uint32_t origin = 0x00000010;
 
+       vp = nv50->geomprog ? nv50->geomprog : nv50->vertprog;
+
        /* XXX: this might not work correctly in all cases yet - we'll
         * just assume that an FP generic input that is not written in
         * the VP is PointCoord.
         */
        memset(pntc, 0, 8 * sizeof(uint32_t));
 
-       for (i = 0; i < fp->cfg.io_nr; i++) {
-               uint8_t sn, si;
-               uint8_t j, k = fp->cfg.io[i].id;
-               unsigned n = popcnt4(fp->cfg.io[i].mask);
+       for (i = 0; i < fp->cfg.in_nr; i++) {
+               unsigned j, n = popcnt4(fp->cfg.in[i].mask);
 
-               if (fp->info.input_semantic_name[k] != TGSI_SEMANTIC_GENERIC) {
+               if (fp->cfg.in[i].sn != TGSI_SEMANTIC_GENERIC) {
                        m += n;
                        continue;
                }
 
-               for (j = 0; j < vp->info.num_outputs; ++j) {
-                       sn = vp->info.output_semantic_name[j];
-                       si = vp->info.output_semantic_index[j];
-
-                       if (sn == fp->info.input_semantic_name[k] &&
-                           si == fp->info.input_semantic_index[k])
+               for (j = 0; j < vp->cfg.out_nr; ++j)
+                       if (vp->cfg.out[j].sn ==  fp->cfg.in[i].sn &&
+                           vp->cfg.out[j].si == fp->cfg.in[i].si)
                                break;
-               }
 
-               if (j < vp->info.num_outputs) {
-                       ubyte mode =
-                               nv50->rasterizer->pipe.sprite_coord_mode[si];
+               if (j < vp->cfg.out_nr) {
+                       ubyte mode = nv50->rasterizer->pipe.sprite_coord_mode[
+                               vp->cfg.out[j].si];
 
                        if (mode == PIPE_SPRITE_COORD_NONE) {
                                m += n;
@@ -3934,7 +4317,7 @@ nv50_pntc_replace(struct nv50_context *nv50, uint32_t pntc[8], unsigned base)
 
                /* this is either PointCoord or replaced by sprite coords */
                for (c = 0; c < 4; c++) {
-                       if (!(fp->cfg.io[i].mask & (1 << c)))
+                       if (!(fp->cfg.in[i].mask & (1 << c)))
                                continue;
                        pntc[m / 8] |= (c + 1) << ((m % 8) * 4);
                        ++m;
@@ -3944,18 +4327,22 @@ nv50_pntc_replace(struct nv50_context *nv50, uint32_t pntc[8], unsigned base)
 }
 
 static int
-nv50_sreg4_map(uint32_t *p_map, int mid, uint32_t lin[4],
-              struct nv50_sreg4 *fpi, struct nv50_sreg4 *vpo)
+nv50_vec4_map(uint32_t *map32, int mid, uint8_t zval, uint32_t lin[4],
+             struct nv50_sreg4 *fpi, struct nv50_sreg4 *vpo)
 {
        int c;
        uint8_t mv = vpo->mask, mf = fpi->mask, oid = vpo->hw;
-       uint8_t *map = (uint8_t *)p_map;
+       uint8_t *map = (uint8_t *)map32;
 
        for (c = 0; c < 4; ++c) {
                if (mf & 1) {
                        if (fpi->linear == TRUE)
                                lin[mid / 32] |= 1 << (mid % 32);
-                       map[mid++] = (mv & 1) ? oid : ((c == 3) ? 0x41 : 0x40);
+                       if (mv & 1)
+                               map[mid] = oid;
+                       else
+                               map[mid] = (c == 3) ? (zval + 1) : zval;
+                       ++mid;
                }
 
                oid += mv & 1;
@@ -3967,16 +4354,21 @@ nv50_sreg4_map(uint32_t *p_map, int mid, uint32_t lin[4],
 }
 
 void
-nv50_linkage_validate(struct nv50_context *nv50)
+nv50_fp_linkage_validate(struct nv50_context *nv50)
 {
        struct nouveau_grobj *tesla = nv50->screen->tesla;
        struct nv50_program *vp = nv50->vertprog;
        struct nv50_program *fp = nv50->fragprog;
        struct nouveau_stateobj *so;
-       struct nv50_sreg4 dummy, *vpo;
+       struct nv50_sreg4 dummy;
        int i, n, c, m = 0;
        uint32_t map[16], lin[4], reg[5], pcrd[8];
+       uint8_t zval = 0x40;
 
+       if (nv50->geomprog) {
+               vp = nv50->geomprog;
+               zval = 0x80;
+       }
        memset(map, 0, sizeof(map));
        memset(lin, 0, sizeof(lin));
 
@@ -3988,13 +4380,15 @@ nv50_linkage_validate(struct nv50_context *nv50)
 
        dummy.linear = FALSE;
        dummy.mask = 0xf; /* map all components of HPOS */
-       m = nv50_sreg4_map(map, m, lin, &dummy, &vp->cfg.io[0]);
+       m = nv50_vec4_map(map, m, zval, lin, &dummy, &vp->cfg.out[0]);
 
        dummy.mask = 0x0;
 
        if (vp->cfg.clpd < 0x40) {
-               for (c = 0; c < vp->cfg.clpd_nr; ++c)
-                       map[m++] = vp->cfg.clpd + c;
+               for (c = 0; c < vp->cfg.clpd_nr; ++c) {
+                       map[m / 4] |= (vp->cfg.clpd + c) << ((m % 4) * 8);
+                       ++m;
+               }
                reg[1] = (m << 8);
        }
 
@@ -4002,35 +4396,30 @@ nv50_linkage_validate(struct nv50_context *nv50)
 
        /* if light_twoside is active, it seems FFC0_ID == BFC0_ID is bad */
        if (nv50->rasterizer->pipe.light_twoside) {
-               vpo = &vp->cfg.two_side[0];
+               struct nv50_sreg4 *vpo = &vp->cfg.two_side[0];
+               struct nv50_sreg4 *fpi = &fp->cfg.two_side[0];
 
-               m = nv50_sreg4_map(map, m, lin, &fp->cfg.two_side[0], &vpo[0]);
-               m = nv50_sreg4_map(map, m, lin, &fp->cfg.two_side[1], &vpo[1]);
+               m = nv50_vec4_map(map, m, zval, lin, &fpi[0], &vpo[0]);
+               m = nv50_vec4_map(map, m, zval, lin, &fpi[1], &vpo[1]);
        }
 
        reg[0] += m - 4; /* adjust FFC0 id */
        reg[4] |= m << 8; /* set mid where 'normal' FP inputs start */
 
-       for (i = 0; i < fp->cfg.io_nr; i++) {
-               ubyte sn = fp->info.input_semantic_name[fp->cfg.io[i].id];
-               ubyte si = fp->info.input_semantic_index[fp->cfg.io[i].id];
-
-               /* position must be mapped first */
-               assert(i == 0 || sn != TGSI_SEMANTIC_POSITION);
-
+       for (i = 0; i < fp->cfg.in_nr; i++) {
                /* maybe even remove these from cfg.io */
-               if (sn == TGSI_SEMANTIC_POSITION || sn == TGSI_SEMANTIC_FACE)
+               if (fp->cfg.in[i].sn == TGSI_SEMANTIC_POSITION ||
+                   fp->cfg.in[i].sn == TGSI_SEMANTIC_FACE)
                        continue;
 
-               /* VP outputs and vp->cfg.io are in the same order */
-               for (n = 0; n < vp->info.num_outputs; ++n) {
-                       if (vp->info.output_semantic_name[n] == sn &&
-                           vp->info.output_semantic_index[n] == si)
+               for (n = 0; n < vp->cfg.out_nr; ++n)
+                       if (vp->cfg.out[n].sn == fp->cfg.in[i].sn &&
+                           vp->cfg.out[n].si == fp->cfg.in[i].si)
                                break;
-               }
-               vpo = (n < vp->info.num_outputs) ? &vp->cfg.io[n] : &dummy;
 
-               m = nv50_sreg4_map(map, m, lin, &fp->cfg.io[i], vpo);
+               m = nv50_vec4_map(map, m, zval, lin, &fp->cfg.in[i],
+                                 (n < vp->cfg.out_nr) ?
+                                 &vp->cfg.out[n] : &dummy);
        }
 
        if (nv50->rasterizer->pipe.point_size_per_vertex) {
@@ -4038,14 +4427,22 @@ nv50_linkage_validate(struct nv50_context *nv50)
                reg[3] = (m++ << 4) | 1;
        }
 
-       /* now fill the stateobj */
-       so = so_new(7, 57, 0);
+       /* now fill the stateobj (at most 28 so_data)  */
+       so = so_new(8, 56, 0);
 
        n = (m + 3) / 4;
-       so_method(so, tesla, NV50TCL_VP_RESULT_MAP_SIZE, 1);
-       so_data  (so, m);
-       so_method(so, tesla, NV50TCL_VP_RESULT_MAP(0), n);
-       so_datap (so, map, n);
+       assert(m <= 32);
+       if (vp->type == PIPE_SHADER_GEOMETRY) {
+               so_method(so, tesla, NV50TCL_GP_RESULT_MAP_SIZE, 1);
+               so_data  (so, m);
+               so_method(so, tesla, NV50TCL_GP_RESULT_MAP(0), n);
+               so_datap (so, map, n);
+       } else {
+               so_method(so, tesla, NV50TCL_VP_RESULT_MAP_SIZE, 1);
+               so_data  (so, m);
+               so_method(so, tesla, NV50TCL_VP_RESULT_MAP(0), n);
+               so_datap (so, map, n);
+       }
 
        so_method(so, tesla, NV50TCL_MAP_SEMANTIC_0, 4);
        so_datap (so, reg, 4);
@@ -4065,8 +4462,74 @@ nv50_linkage_validate(struct nv50_context *nv50)
                so_datap (so, pcrd, 8);
        }
 
-        so_ref(so, &nv50->state.programs);
-        so_ref(NULL, &so);
+       so_method(so, tesla, NV50TCL_GP_ENABLE, 1);
+       so_data  (so, (vp->type == PIPE_SHADER_GEOMETRY) ? 1 : 0);
+
+       so_ref(so, &nv50->state.fp_linkage);
+       so_ref(NULL, &so);
+}
+
+static int
+construct_vp_gp_mapping(uint32_t *map32, int m,
+                       struct nv50_program *vp, struct nv50_program *gp)
+{
+       uint8_t *map = (uint8_t *)map32;
+       int i, j, c;
+
+        for (i = 0; i < gp->cfg.in_nr; ++i) {
+                uint8_t oid, mv = 0, mg = gp->cfg.in[i].mask;
+
+                for (j = 0; j < vp->cfg.out_nr; ++j) {
+                        if (vp->cfg.out[j].sn == gp->cfg.in[i].sn &&
+                            vp->cfg.out[j].si == gp->cfg.in[i].si) {
+                               mv = vp->cfg.out[j].mask;
+                               oid = vp->cfg.out[j].hw;
+                                break;
+                       }
+               }
+
+                for (c = 0; c < 4; ++c, mv >>= 1, mg >>= 1) {
+                       if (mg & mv & 1)
+                               map[m++] = oid;
+                       else
+                       if (mg & 1)
+                               map[m++] = (c == 3) ? 0x41 : 0x40;
+                        oid += mv & 1;
+                }
+        }
+       return m;
+}
+
+void
+nv50_gp_linkage_validate(struct nv50_context *nv50)
+{
+       struct nouveau_grobj *tesla = nv50->screen->tesla;
+       struct nouveau_stateobj *so;
+       struct nv50_program *vp = nv50->vertprog;
+       struct nv50_program *gp = nv50->geomprog;
+       uint32_t map[16];
+       int m = 0;
+
+       if (!gp) {
+               so_ref(NULL, &nv50->state.gp_linkage);
+               return;
+       }
+       memset(map, 0, sizeof(map));
+
+       m = construct_vp_gp_mapping(map, m, vp, gp);
+
+       so = so_new(2, 14, 0);
+
+       assert(m <= 32);
+       so_method(so, tesla, NV50TCL_VP_RESULT_MAP_SIZE, 1);
+       so_data  (so, m);
+
+       m = (m + 3) / 4;
+       so_method(so, tesla, NV50TCL_VP_RESULT_MAP(0), m);
+       so_datap (so, map, m);
+
+       so_ref(so, &nv50->state.gp_linkage);
+       so_ref(NULL, &so);
 }
 
 void