From 0cb26f31feb954e7a1c49d54930f83c002329c7d Mon Sep 17 00:00:00 2001 From: Rhys Perry Date: Thu, 13 Jul 2023 16:37:11 +0100 Subject: [PATCH] ac/llvm: skip ballot zext for 32-bit dest with wave32-as-wave64 MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit NGG lowering can ballots with a 32-bit destination even if the ballot bit size is 64. Signed-off-by: Rhys Perry Reviewed-by: Timur Kristóf Reviewed-by: Marek Olšák Cc: mesa-stable Part-of: --- src/amd/llvm/ac_nir_to_llvm.c | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/src/amd/llvm/ac_nir_to_llvm.c b/src/amd/llvm/ac_nir_to_llvm.c index 0f63dc5c1fa..03b74cb9515 100644 --- a/src/amd/llvm/ac_nir_to_llvm.c +++ b/src/amd/llvm/ac_nir_to_llvm.c @@ -3043,8 +3043,10 @@ static bool visit_intrinsic(struct ac_nir_context *ctx, nir_intrinsic_instr *ins switch (instr->intrinsic) { case nir_intrinsic_ballot: result = ac_build_ballot(&ctx->ac, get_src(ctx, instr->src[0])); - if (ctx->ac.ballot_mask_bits > ctx->ac.wave_size) - result = LLVMBuildZExt(ctx->ac.builder, result, ctx->ac.iN_ballotmask, ""); + if (instr->dest.ssa.bit_size > ctx->ac.wave_size) { + LLVMTypeRef dest_type = LLVMIntTypeInContext(ctx->ac.context, instr->dest.ssa.bit_size); + result = LLVMBuildZExt(ctx->ac.builder, result, dest_type, ""); + } break; case nir_intrinsic_read_invocation: result =