OpenHarmony/third_party_mesa3d
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

spirv: Add a vtn_get_mem_operands() helper

Browse Source 
Add a vtn_get_mem_operands() helper to extract memory operand attached
to load/store operations.

Signed-off-by: Boris Brezillon <boris.brezillon@collabora.com>
Reviewed-by: Jason Ekstrand <jason@jlekstrand.net>
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/6090>
This commit is contained in:
Boris Brezillon
2020-08-17 19:38:47 +02:00
committed by Marge Bot
parent 959539fbbd
commit 857b9c5027
1 changed files with 50 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

58
src/compiler/spirv/vtn_variables.c
View File

@@ -2447,6 +2447,37 @@ nir_sloppy_bitcast(nir_builder *b, nir_ssa_def *val,
return nir_shrink_zero_pad_vec(b, val, num_components);
}
static bool
vtn_get_mem_operands(struct vtn_builder *b, const uint32_t *w, unsigned count,
unsigned *idx, SpvMemoryAccessMask *access, unsigned *alignment,
SpvScope *dest_scope, SpvScope *src_scope)
{
*access = 0;
*alignment = 0;
if (*idx >= count)
return false;
*access = w[(*idx)++];
if (*access & SpvMemoryAccessAlignedMask) {
vtn_assert(*idx < count);
*alignment = w[(*idx)++];
}
if (*access & SpvMemoryAccessMakePointerAvailableMask) {
vtn_assert(*idx < count);
vtn_assert(dest_scope);
*dest_scope = vtn_constant_uint(b, w[(*idx)++]);
}
if (*access & SpvMemoryAccessMakePointerVisibleMask) {
vtn_assert(*idx < count);
vtn_assert(src_scope);
*src_scope = vtn_constant_uint(b, w[(*idx)++]);
}
return true;
}
void
vtn_handle_variables(struct vtn_builder *b, SpvOp opcode,
const uint32_t *w, unsigned count)
@@ -2560,21 +2591,16 @@ vtn_handle_variables(struct vtn_builder *b, SpvOp opcode,
vtn_assert_types_equal(b, opcode, res_type, src_val->type->deref);
if (count > 4) {
unsigned idx = 5;
SpvMemoryAccessMask access = w[4];
if (access & SpvMemoryAccessAlignedMask)
idx++;
unsigned idx = 4, alignment;
SpvMemoryAccessMask access;
SpvScope scope;
vtn_get_mem_operands(b, w, count, &idx, &access, &alignment, NULL, &scope);
if (access & SpvMemoryAccessMakePointerVisibleMask) {
SpvMemorySemanticsMask semantics =
SpvMemorySemanticsMakeVisibleMask |
vtn_storage_class_to_memory_semantics(src->ptr_type->storage_class);
SpvScope scope = vtn_constant_uint(b, w[idx]);
vtn_emit_memory_barrier(b, scope, semantics);
}
}
vtn_push_ssa_value(b, w[2], vtn_variable_load(b, src));
break;
@@ -2610,24 +2636,20 @@ vtn_handle_variables(struct vtn_builder *b, SpvOp opcode,
vtn_assert_types_equal(b, opcode, dest_val->type->deref, src_val->type);
unsigned idx = 3, alignment;
SpvMemoryAccessMask access;
SpvScope scope;
vtn_get_mem_operands(b, w, count, &idx, &access, &alignment, &scope, NULL);
struct vtn_ssa_value *src = vtn_ssa_value(b, w[2]);
vtn_variable_store(b, src, dest);
if (count > 3) {
unsigned idx = 4;
SpvMemoryAccessMask access = w[3];
if (access & SpvMemoryAccessAlignedMask)
idx++;
if (access & SpvMemoryAccessMakePointerAvailableMask) {
SpvMemorySemanticsMask semantics =
SpvMemorySemanticsMakeAvailableMask |
vtn_storage_class_to_memory_semantics(dest->ptr_type->storage_class);
SpvScope scope = vtn_constant_uint(b, w[idx]);
vtn_emit_memory_barrier(b, scope, semantics);
}
}
break;
}