From: Daniel Schürmann <daniel.schuerm...@campus.tu-berlin.de>

Co-authored-by: Dave Airlie <airl...@redhat.com>
Signed-off-by: Daniel Schürmann <daniel.schuerm...@campus.tu-berlin.de>
---
 src/amd/common/ac_nir_to_llvm.c | 28 ++++++++++++++++++++++++++++
 1 file changed, 28 insertions(+)

diff --git a/src/amd/common/ac_nir_to_llvm.c b/src/amd/common/ac_nir_to_llvm.c
index cccc687157..1743c0bbe1 100644
--- a/src/amd/common/ac_nir_to_llvm.c
+++ b/src/amd/common/ac_nir_to_llvm.c
@@ -1675,6 +1675,10 @@ static void visit_alu(struct ac_nir_context *ctx, const 
nir_alu_instr *instr)
        case nir_op_unpack_half_2x16:
                src_components = 1;
                break;
+       case nir_op_cube_face_coord:
+       case nir_op_cube_face_index:
+               src_components = 3;
+               break;
        default:
                src_components = num_components;
                break;
@@ -2076,6 +2080,30 @@ static void visit_alu(struct ac_nir_context *ctx, const 
nir_alu_instr *instr)
                break;
        }
 
+       case nir_op_cube_face_coord: {
+               src[0] = ac_to_float(&ctx->ac, src[0]);
+               LLVMValueRef results[2];
+               LLVMValueRef in[3];
+               for (unsigned chan = 0; chan < 3; chan++)
+                       in[chan] = ac_llvm_extract_elem(&ctx->ac, src[0], chan);
+               results[0] = ac_build_intrinsic(&ctx->ac, "llvm.amdgcn.cubetc",
+                                               ctx->ac.f32, in, 3, 
AC_FUNC_ATTR_READNONE);
+               results[1] = ac_build_intrinsic(&ctx->ac, "llvm.amdgcn.cubesc",
+                                               ctx->ac.f32, in, 3, 
AC_FUNC_ATTR_READNONE);
+               result = ac_build_gather_values(&ctx->ac, results, 2);
+               break;
+       }
+
+       case nir_op_cube_face_index: {
+               src[0] = ac_to_float(&ctx->ac, src[0]);
+               LLVMValueRef in[3];
+               for (unsigned chan = 0; chan < 3; chan++)
+                       in[chan] = ac_llvm_extract_elem(&ctx->ac, src[0], chan);
+               result = ac_build_intrinsic(&ctx->ac,  "llvm.amdgcn.cubeid",
+                                               ctx->ac.f32, in, 3, 
AC_FUNC_ATTR_READNONE);
+               break;
+       }
+
        default:
                fprintf(stderr, "Unknown NIR alu instr: ");
                nir_print_instr(&instr->instr, stderr);
-- 
2.14.1

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

Reply via email to