ir3: Plumb through bindless support
[mesa.git] / src / freedreno / ir3 / ir3_a6xx.c
index e84c4808de1e2cc3d102a4681dabbe255335a8a1..23c1970bb360c2a52e0b63f72322b82bd0410d83 100644 (file)
  * encoding compared to a4xx/a5xx.
  */
 
+static void
+handle_bindless_cat6(struct ir3_instruction *instr, nir_src rsrc)
+{
+       nir_intrinsic_instr *intrin = ir3_bindless_resource(rsrc);
+       if (!intrin)
+               return;
+
+       instr->flags |= IR3_INSTR_B;
+       instr->cat6.base = nir_intrinsic_desc_set(intrin);
+}
+
+static struct ir3_instruction *
+ssbo_idx(struct ir3_context *ctx, nir_src src)
+{
+       if (ir3_bindless_resource(src)) {
+               ctx->so->bindless_ibo = true;
+               return ir3_get_src(ctx, &src)[0];
+       } else {
+               /* can this be non-const buffer_index?  how do we handle that? */
+               int ibo_idx = ir3_ssbo_to_ibo(ctx->so->shader, nir_src_as_uint(src));
+               return create_immed(ctx->block, ibo_idx);
+       }
+}
+
+static struct ir3_instruction *
+image_idx(struct ir3_context *ctx, nir_src src)
+{
+       if (ir3_bindless_resource(src)) {
+               ctx->so->bindless_ibo = true;
+               return ir3_get_src(ctx, &src)[0];
+       } else {
+               /* can this be non-const buffer_index?  how do we handle that? */
+               int ibo_idx = ir3_image_to_ibo(ctx->so->shader, nir_src_as_uint(src));
+               return create_immed(ctx->block, ibo_idx);
+       }
+}
 
 /* src[] = { buffer_index, offset }. No const_index */
 static void
@@ -47,18 +83,16 @@ emit_intrinsic_load_ssbo(struct ir3_context *ctx, nir_intrinsic_instr *intr,
        struct ir3_instruction *offset;
        struct ir3_instruction *ldib;
 
-       /* can this be non-const buffer_index?  how do we handle that? */
-       int ibo_idx = ir3_ssbo_to_ibo(ctx->so->shader, nir_src_as_uint(intr->src[0]));
-
        offset = ir3_get_src(ctx, &intr->src[2])[0];
 
-       ldib = ir3_LDIB(b, create_immed(b, ibo_idx), 0, offset, 0);
+       ldib = ir3_LDIB(b, ssbo_idx(ctx, intr->src[0]), 0, offset, 0);
        ldib->regs[0]->wrmask = MASK(intr->num_components);
        ldib->cat6.iim_val = intr->num_components;
        ldib->cat6.d = 1;
        ldib->cat6.type = TYPE_U32;
        ldib->barrier_class = IR3_BARRIER_BUFFER_R;
        ldib->barrier_conflict = IR3_BARRIER_BUFFER_W;
+       handle_bindless_cat6(ldib, intr->src[0]);
 
        ir3_split_dest(b, dst, ldib, 0, intr->num_components);
 }
@@ -76,20 +110,18 @@ emit_intrinsic_store_ssbo(struct ir3_context *ctx, nir_intrinsic_instr *intr)
        unsigned wrmask = intr->const_index[0];
        unsigned ncomp = ffs(~wrmask) - 1;
 
-       /* can this be non-const buffer_index?  how do we handle that? */
-       int ibo_idx = ir3_ssbo_to_ibo(ctx->so->shader, nir_src_as_uint(intr->src[1]));
-
        /* src0 is offset, src1 is value:
         */
        val = ir3_create_collect(ctx, ir3_get_src(ctx, &intr->src[0]), ncomp);
        offset = ir3_get_src(ctx, &intr->src[3])[0];
 
-       stib = ir3_STIB(b, create_immed(b, ibo_idx), 0, offset, 0, val, 0);
+       stib = ir3_STIB(b, ssbo_idx(ctx, intr->src[1]), 0, offset, 0, val, 0);
        stib->cat6.iim_val = ncomp;
        stib->cat6.d = 1;
        stib->cat6.type = TYPE_U32;
        stib->barrier_class = IR3_BARRIER_BUFFER_W;
        stib->barrier_conflict = IR3_BARRIER_BUFFER_R | IR3_BARRIER_BUFFER_W;
+       handle_bindless_cat6(stib, intr->src[1]);
 
        array_insert(b, b->keeps, stib);
 }
@@ -118,10 +150,7 @@ emit_intrinsic_atomic_ssbo(struct ir3_context *ctx, nir_intrinsic_instr *intr)
        struct ir3_instruction *atomic, *ibo, *src0, *src1, *data, *dummy;
        type_t type = TYPE_U32;
 
-       /* can this be non-const buffer_index?  how do we handle that? */
-       int ibo_idx = ir3_ssbo_to_ibo(ctx->so->shader,
-                       nir_src_as_uint(intr->src[0]));
-       ibo = create_immed(b, ibo_idx);
+       ibo = ssbo_idx(ctx, intr->src[0]);
 
        data   = ir3_get_src(ctx, &intr->src[2])[0];
 
@@ -196,6 +225,7 @@ emit_intrinsic_atomic_ssbo(struct ir3_context *ctx, nir_intrinsic_instr *intr)
        atomic->cat6.type = type;
        atomic->barrier_class = IR3_BARRIER_BUFFER_W;
        atomic->barrier_conflict = IR3_BARRIER_BUFFER_R | IR3_BARRIER_BUFFER_W;
+       handle_bindless_cat6(atomic, intr->src[0]);
 
        /* even if nothing consume the result, we can't DCE the instruction: */
        array_insert(b, b->keeps, atomic);
@@ -203,6 +233,30 @@ emit_intrinsic_atomic_ssbo(struct ir3_context *ctx, nir_intrinsic_instr *intr)
        return atomic;
 }
 
+/* src[] = { deref, coord, sample_index }. const_index[] = {} */
+static void
+emit_intrinsic_load_image(struct ir3_context *ctx, nir_intrinsic_instr *intr,
+               struct ir3_instruction **dst)
+{
+       struct ir3_block *b = ctx->block;
+       struct ir3_instruction *ldib;
+       struct ir3_instruction * const *coords = ir3_get_src(ctx, &intr->src[1]);
+       unsigned ncoords = ir3_get_image_coords(intr, NULL);
+
+       ldib = ir3_LDIB(b, image_idx(ctx, intr->src[0]), 0,
+                                       ir3_create_collect(ctx, coords, ncoords), 0);
+       ldib->regs[0]->wrmask = MASK(intr->num_components);
+       ldib->cat6.iim_val = intr->num_components;
+       ldib->cat6.d = ncoords;
+       ldib->cat6.type = ir3_get_type_for_image_intrinsic(intr);
+       ldib->cat6.typed = true;
+       ldib->barrier_class = IR3_BARRIER_IMAGE_R;
+       ldib->barrier_conflict = IR3_BARRIER_IMAGE_W;
+       handle_bindless_cat6(ldib, intr->src[0]);
+
+       ir3_split_dest(b, dst, ldib, 0, intr->num_components);
+}
+
 /* src[] = { deref, coord, sample_index, value }. const_index[] = {} */
 static void
 emit_intrinsic_store_image(struct ir3_context *ctx, nir_intrinsic_instr *intr)
@@ -212,14 +266,12 @@ emit_intrinsic_store_image(struct ir3_context *ctx, nir_intrinsic_instr *intr)
        struct ir3_instruction * const *value = ir3_get_src(ctx, &intr->src[3]);
        struct ir3_instruction * const *coords = ir3_get_src(ctx, &intr->src[1]);
        unsigned ncoords = ir3_get_image_coords(intr, NULL);
-       unsigned slot = nir_src_as_uint(intr->src[0]);
-       unsigned ibo_idx = ir3_image_to_ibo(ctx->so->shader, slot);
        enum pipe_format format = nir_intrinsic_format(intr);
        unsigned ncomp = ir3_get_num_components_for_image_format(format);
 
        /* src0 is offset, src1 is value:
         */
-       stib = ir3_STIB(b, create_immed(b, ibo_idx), 0,
+       stib = ir3_STIB(b, image_idx(ctx, intr->src[0]), 0,
                        ir3_create_collect(ctx, coords, ncoords), 0,
                        ir3_create_collect(ctx, value, ncomp), 0);
        stib->cat6.iim_val = ncomp;
@@ -228,6 +280,7 @@ emit_intrinsic_store_image(struct ir3_context *ctx, nir_intrinsic_instr *intr)
        stib->cat6.typed = true;
        stib->barrier_class = IR3_BARRIER_IMAGE_W;
        stib->barrier_conflict = IR3_BARRIER_IMAGE_R | IR3_BARRIER_IMAGE_W;
+       handle_bindless_cat6(stib, intr->src[0]);
 
        array_insert(b, b->keeps, stib);
 }
@@ -241,10 +294,8 @@ emit_intrinsic_atomic_image(struct ir3_context *ctx, nir_intrinsic_instr *intr)
        struct ir3_instruction * const *coords = ir3_get_src(ctx, &intr->src[1]);
        struct ir3_instruction *value = ir3_get_src(ctx, &intr->src[3])[0];
        unsigned ncoords = ir3_get_image_coords(intr, NULL);
-       unsigned slot = nir_src_as_uint(intr->src[0]);
-       unsigned ibo_idx = ir3_image_to_ibo(ctx->so->shader, slot);
 
-       ibo = create_immed(b, ibo_idx);
+       ibo = image_idx(ctx, intr->src[0]);
 
        /* So this gets a bit creative:
         *
@@ -261,7 +312,8 @@ emit_intrinsic_atomic_image(struct ir3_context *ctx, nir_intrinsic_instr *intr)
        dummy = create_immed(b, 0);
        src0 = ir3_create_collect(ctx, coords, ncoords);
 
-       if (intr->intrinsic == nir_intrinsic_image_atomic_comp_swap) {
+       if (intr->intrinsic == nir_intrinsic_image_atomic_comp_swap ||
+               intr->intrinsic == nir_intrinsic_bindless_image_atomic_comp_swap) {
                struct ir3_instruction *compare = ir3_get_src(ctx, &intr->src[4])[0];
                src1 = ir3_create_collect(ctx, (struct ir3_instruction*[]){
                        dummy, compare, value
@@ -274,29 +326,39 @@ emit_intrinsic_atomic_image(struct ir3_context *ctx, nir_intrinsic_instr *intr)
 
        switch (intr->intrinsic) {
        case nir_intrinsic_image_atomic_add:
+       case nir_intrinsic_bindless_image_atomic_add:
                atomic = ir3_ATOMIC_ADD_G(b, ibo, 0, src0, 0, src1, 0);
                break;
        case nir_intrinsic_image_atomic_imin:
        case nir_intrinsic_image_atomic_umin:
+       case nir_intrinsic_bindless_image_atomic_imin:
+       case nir_intrinsic_bindless_image_atomic_umin:
                atomic = ir3_ATOMIC_MIN_G(b, ibo, 0, src0, 0, src1, 0);
                break;
        case nir_intrinsic_image_atomic_imax:
        case nir_intrinsic_image_atomic_umax:
+       case nir_intrinsic_bindless_image_atomic_imax:
+       case nir_intrinsic_bindless_image_atomic_umax:
                atomic = ir3_ATOMIC_MAX_G(b, ibo, 0, src0, 0, src1, 0);
                break;
        case nir_intrinsic_image_atomic_and:
+       case nir_intrinsic_bindless_image_atomic_and:
                atomic = ir3_ATOMIC_AND_G(b, ibo, 0, src0, 0, src1, 0);
                break;
        case nir_intrinsic_image_atomic_or:
+       case nir_intrinsic_bindless_image_atomic_or:
                atomic = ir3_ATOMIC_OR_G(b, ibo, 0, src0, 0, src1, 0);
                break;
        case nir_intrinsic_image_atomic_xor:
+       case nir_intrinsic_bindless_image_atomic_xor:
                atomic = ir3_ATOMIC_XOR_G(b, ibo, 0, src0, 0, src1, 0);
                break;
        case nir_intrinsic_image_atomic_exchange:
+       case nir_intrinsic_bindless_image_atomic_exchange:
                atomic = ir3_ATOMIC_XCHG_G(b, ibo, 0, src0, 0, src1, 0);
                break;
        case nir_intrinsic_image_atomic_comp_swap:
+       case nir_intrinsic_bindless_image_atomic_comp_swap:
                atomic = ir3_ATOMIC_CMPXCHG_G(b, ibo, 0, src0, 0, src1, 0);
                break;
        default:
@@ -309,6 +371,7 @@ emit_intrinsic_atomic_image(struct ir3_context *ctx, nir_intrinsic_instr *intr)
        atomic->cat6.typed = true;
        atomic->barrier_class = IR3_BARRIER_IMAGE_W;
        atomic->barrier_conflict = IR3_BARRIER_IMAGE_R | IR3_BARRIER_IMAGE_W;
+       handle_bindless_cat6(atomic, intr->src[0]);
 
        /* even if nothing consume the result, we can't DCE the instruction: */
        array_insert(b, b->keeps, atomic);
@@ -320,6 +383,7 @@ const struct ir3_context_funcs ir3_a6xx_funcs = {
                .emit_intrinsic_load_ssbo = emit_intrinsic_load_ssbo,
                .emit_intrinsic_store_ssbo = emit_intrinsic_store_ssbo,
                .emit_intrinsic_atomic_ssbo = emit_intrinsic_atomic_ssbo,
+               .emit_intrinsic_load_image = emit_intrinsic_load_image,
                .emit_intrinsic_store_image = emit_intrinsic_store_image,
                .emit_intrinsic_atomic_image = emit_intrinsic_atomic_image,
 };
@@ -385,7 +449,7 @@ ir3_a6xx_fixup_atomic_dests(struct ir3 *ir, struct ir3_shader_variant *so)
                foreach_instr_safe (instr, &block->instr_list) {
                        struct ir3_register *reg;
 
-                       foreach_src(reg, instr) {
+                       foreach_src (reg, instr) {
                                struct ir3_instruction *src = reg->instr;
 
                                if (!src)