amd/common: Use correct writemask for shared memory stores.

The check was for 1 bit being set, which is clearly not what we want.

CC: <mesa-stable@lists.freedesktop.org>
Reviewed-by: Samuel Pitoiset <samuel.pitoiset@gmail.com>

diff --git a/src/amd/common/ac_nir_to_llvm.c b/src/amd/common/ac_nir_to_llvm.c
index efd3e260af1a48154fc14b963cb498bd0307cdc2..6ba0ee74f11c4865debf08a79c4ae2786a230a21 100644 (file)
--- a/src/amd/common/ac_nir_to_llvm.c
+++ b/src/amd/common/ac_nir_to_llvm.c
@@ -2107,7 +2107,7 @@ visit_store_var(struct ac_nir_context *ctx,
                int writemask = instr->const_index[0];
                LLVMValueRef address = get_src(ctx, instr->src[0]);
                LLVMValueRef val = get_src(ctx, instr->src[1]);
-               if (util_is_power_of_two_nonzero(writemask)) {
+               if (writemask == (1u << ac_get_llvm_num_components(val)) - 1) {
                        val = LLVMBuildBitCast(
                           ctx->ac.builder, val,
                           LLVMGetElementType(LLVMTypeOf(address)), "");