ac: add 8-bit support to ac_build_bit_count()

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 fe280b31368ee09c7fb1a06c00715e7f44a8fd44..6d26ab31a66dd9d3ed981810a0811c7f3ac91cdf 100644 (file)
--- a/src/amd/common/ac_llvm_build.c
+++ b/src/amd/common/ac_llvm_build.c
@@ -2553,6 +2553,13 @@ LLVMValueRef ac_build_bit_count(struct ac_llvm_context *ctx, LLVMValueRef src0)
                                            (LLVMValueRef []) { src0 }, 1,
                                            AC_FUNC_ATTR_READNONE);
 
+               result = LLVMBuildZExt(ctx->builder, result, ctx->i32, "");
+               break;
+       case 8:
+               result = ac_build_intrinsic(ctx, "llvm.ctpop.i8", ctx->i8,
+                                           (LLVMValueRef []) { src0 }, 1,
+                                           AC_FUNC_ATTR_READNONE);
+
                result = LLVMBuildZExt(ctx->builder, result, ctx->i32, "");
                break;
        default: