ac: fix ac_build_umsb() for 16-bit integer type

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

diff --git a/src/amd/common/ac_llvm_build.c b/src/amd/common/ac_llvm_build.c
index 20c722e..fe280b3 100644 (file)
--- a/src/amd/common/ac_llvm_build.c
+++ b/src/amd/common/ac_llvm_build.c
@@ -1970,7 +1970,12 @@ ac_build_umsb(struct ac_llvm_context *ctx,
        /* The HW returns the last bit index from MSB, but TGSI/NIR wants
         * the index from LSB. Invert it by doing "31 - msb". */
        msb = LLVMBuildSub(ctx->builder, highest_bit, msb, "");
-       msb = LLVMBuildTruncOrBitCast(ctx->builder, msb, ctx->i32, "");
+
+       if (bitsize == 64) {
+               msb = LLVMBuildTrunc(ctx->builder, msb, ctx->i32, "");
+       } else if (bitsize == 16) {
+               msb = LLVMBuildSExt(ctx->builder, msb, ctx->i32, "");
+       }
 
        /* check for zero */
        return LLVMBuildSelect(ctx->builder,