ac: implement AMD_gcn_shader extended instructions

Co-authored-by: Dave Airlie <[email protected]>
Signed-off-by: Daniel Schürmann <[email protected]>
Reviewed-by: Bas Nieuwenhuizen <[email protected]>

1 files changed, 28 insertions, 0 deletions

diff --git a/src/amd/common/ac_nir_to_llvm.c b/src/amd/common/ac_nir_to_llvm.c
index d02238dba7d..0fbce5bbfd6 100644
--- a/src/amd/common/ac_nir_to_llvm.c
+++ b/src/amd/common/ac_nir_to_llvm.c
@@ -1603,6 +1603,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;
@@ -2015,6 +2019,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);