[Mesa-dev] [PATCH 21/38] ac/nir: implement 16-bit ac_build_ddxy

[Rhys Perry]

Signed-off-by: Rhys Perry <pendingchao...@gmail.com>
---
 src/amd/common/ac_llvm_build.c | 17 ++++++++++++++---
 1 file changed, 14 insertions(+), 3 deletions(-)

diff --git a/src/amd/common/ac_llvm_build.c b/src/amd/common/ac_llvm_build.c
index e85c178f78..f394d16bc9 100644
--- a/src/amd/common/ac_llvm_build.c
+++ b/src/amd/common/ac_llvm_build.c
@@ -1519,6 +1519,11 @@ ac_build_ddxy(struct ac_llvm_context *ctx,
        LLVMValueRef tl, trbl, args[2];
        LLVMValueRef result;
 
+       int size = ac_get_type_size(LLVMTypeOf(val));
+
+       if (size == 2)
+               val = LLVMBuildZExt(ctx->builder, val, ctx->i32, "");
+
        if (HAVE_LLVM >= 0x0700) {
                unsigned tl_lanes[4], trbl_lanes[4];
 
@@ -1600,13 +1605,19 @@ ac_build_ddxy(struct ac_llvm_context *ctx,
                                        AC_FUNC_ATTR_CONVERGENT);
        }
 
-       tl = LLVMBuildBitCast(ctx->builder, tl, ctx->f32, "");
-       trbl = LLVMBuildBitCast(ctx->builder, trbl, ctx->f32, "");
+       if (size == 2) {
+               tl = LLVMBuildTrunc(ctx->builder, tl, ctx->i16, "");
+               trbl = LLVMBuildTrunc(ctx->builder, trbl, ctx->i16, "");
+       }
+
+       LLVMTypeRef type = ac_float_of_size(ctx, size * 8);
+       tl = LLVMBuildBitCast(ctx->builder, tl, type, "");
+       trbl = LLVMBuildBitCast(ctx->builder, trbl, type, "");
        result = LLVMBuildFSub(ctx->builder, trbl, tl, "");
 
        if (HAVE_LLVM >= 0x0700) {
                result = ac_build_intrinsic(ctx,
-                       "llvm.amdgcn.wqm.f32", ctx->f32,
+                       LLVMTypeOf(val) == ctx->f32 ? "llvm.amdgcn.wqm.f32" : 
"llvm.amdgcn.wqm.f16", type,
                        &result, 1, 0);
        }
 
-- 
2.19.2

_______________________________________________
mesa-dev mailing list
mesa-dev@lists.freedesktop.org
https://lists.freedesktop.org/mailman/listinfo/mesa-dev