Mesa (master): pan/mdg: Replace zext with a type enum
GitLab Mirror
gitlab-mirror at kemper.freedesktop.org
Fri Jan 1 03:14:24 UTC 2021
Module: Mesa
Branch: master
Commit: e5e8a628c20901822ccdd6d317baf8605c0c0204
URL: http://cgit.freedesktop.org/mesa/mesa/commit/?id=e5e8a628c20901822ccdd6d317baf8605c0c0204
Author: Icecream95 <ixn at disroot.org>
Date: Fri Jan 1 01:18:51 2021 +1300
pan/mdg: Replace zext with a type enum
The index type is actually a two-bit field, with support for both sign
and zero extension.
What was previously labelled as `zext` actually does sign-extension,
but we want that in most cases anyway.
Reviewed-by: Alyssa Rosenzweig <alyssa.rosenzweig at collabora.com>
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/8264>
---
src/panfrost/midgard/midgard_address.c | 31 ++++++++++++++++++++++---------
1 file changed, 22 insertions(+), 9 deletions(-)
diff --git a/src/panfrost/midgard/midgard_address.c b/src/panfrost/midgard/midgard_address.c
index 2f57c3464a7..22411c3606e 100644
--- a/src/panfrost/midgard/midgard_address.c
+++ b/src/panfrost/midgard/midgard_address.c
@@ -36,11 +36,17 @@
* This allows for fast indexing into arrays. This file tries to pattern match the offset in NIR with this form to reduce pressure on the ALU pipe.
*/
+enum index_type {
+ ITYPE_U64 = 1 << 6,
+ ITYPE_U32 = 2 << 6, // zero-extend
+ ITYPE_I32 = 3 << 6, // sign-extend
+};
+
struct mir_address {
nir_ssa_scalar A;
nir_ssa_scalar B;
- bool zext;
+ enum index_type type;
unsigned shift;
unsigned bias;
};
@@ -105,7 +111,7 @@ mir_match_iadd(struct mir_address *address, bool first_free)
}
}
-/* Matches u2u64 and sets zext */
+/* Matches u2u64 and sets type */
static void
mir_match_u2u64(struct mir_address *address)
@@ -121,7 +127,7 @@ mir_match_u2u64(struct mir_address *address)
nir_ssa_scalar arg = nir_ssa_scalar_chase_alu_src(address->B, 0);
address->B = arg;
- address->zext = true;
+ address->type = ITYPE_U32;
}
/* Matches ishl to shift */
@@ -175,7 +181,8 @@ static struct mir_address
mir_match_offset(nir_ssa_def *offset, bool first_free)
{
struct mir_address address = {
- .B = { .def = offset }
+ .B = { .def = offset },
+ .type = ITYPE_U64,
};
mir_match_mov(&address);
@@ -198,15 +205,19 @@ mir_set_offset(compiler_context *ctx, midgard_instruction *ins, nir_src *offset,
ins->swizzle[2][i] = 0;
}
- bool force_zext = (nir_src_bit_size(*offset) < 64);
+ /* Sign extend instead of zero extend in case the address is something
+ * like `base + offset + 20`, where offset could be negative. */
+ bool force_sext = (nir_src_bit_size(*offset) < 64);
if (!offset->is_ssa) {
ins->load_store.arg_1 |= is_shared ? 0x6E : 0x7E;
ins->src[2] = nir_src_index(ctx, offset);
ins->src_types[2] = nir_type_uint | nir_src_bit_size(*offset);
- if (force_zext)
- ins->load_store.arg_1 |= 0x80;
+ if (force_sext)
+ ins->load_store.arg_1 |= ITYPE_I32;
+ else
+ ins->load_store.arg_1 |= ITYPE_U64;
return;
}
@@ -227,8 +238,10 @@ mir_set_offset(compiler_context *ctx, midgard_instruction *ins, nir_src *offset,
} else
ins->load_store.arg_2 = 0x1E;
- if (match.zext || force_zext)
- ins->load_store.arg_1 |= 0x80;
+ if (force_sext)
+ match.type = ITYPE_I32;
+
+ ins->load_store.arg_1 |= match.type;
assert(match.shift <= 7);
ins->load_store.arg_2 |= (match.shift) << 5;
More information about the mesa-commit
mailing list