ac/nir: fix nir_op_b2i16

Signed-off-by: Samuel Pitoiset <samuel.pitoiset@gmail.com>
Reviewed-by: Bas Nieuwenhuizen <bas@basnieuwenhuizen.nl>

diff --git a/src/amd/common/ac_nir_to_llvm.c b/src/amd/common/ac_nir_to_llvm.c
index 7b457015e16e6934f927e55b3dbf531b02a94788..4e9af58538f488a5d433ee964c94f77efaf4ea72 100644 (file)
--- a/src/amd/common/ac_nir_to_llvm.c
+++ b/src/amd/common/ac_nir_to_llvm.c
@@ -344,10 +344,16 @@ static LLVMValueRef emit_b2i(struct ac_llvm_context *ctx,
 {
        LLVMValueRef result = LLVMBuildAnd(ctx->builder, src0, ctx->i32_1, "");
 
-       if (bitsize == 32)
+       switch (bitsize) {
+       case 16:
+               return LLVMBuildTrunc(ctx->builder, result, ctx->i16, "");
+       case 32:
                return result;
-
-       return LLVMBuildZExt(ctx->builder, result, ctx->i64, "");
+       case 64:
+               return LLVMBuildZExt(ctx->builder, result, ctx->i64, "");
+       default:
+               unreachable("Unsupported bit size.");
+       }
 }
 
 static LLVMValueRef emit_i2b(struct ac_llvm_context *ctx,