OpenHarmony/third_party_mesa3d
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

ac/llvm: add helpers to get pointer types of ac_arg

Browse Source 
Reviewed-by: Qiang Yu <yuq825@gmail.com>
Reviewed-by: Marek Olšák <marek.olsak@amd.com>
Acked-by: Samuel Pitoiset <samuel.pitoiset@gmail.com>
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/19035>
This commit is contained in:
Pierre-Eric Pelloux-Prayer
2022-10-05 17:41:04 +02:00
committed by Marge Bot
parent 5dcc2c216b
commit fab476bd11
2 changed files with 51 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

63
src/amd/llvm/ac_llvm_build.c
View File

@@ -2828,6 +2828,29 @@ LLVMValueRef ac_find_lsb(struct ac_llvm_context *ctx, LLVMTypeRef dst_type, LLVM
LLVMConstInt(ctx->i32, -1, 0), lsb, "");
}
LLVMTypeRef ac_arg_type_to_pointee_type(struct ac_llvm_context *ctx, enum ac_arg_type type) {
switch (type) {
case AC_ARG_CONST_PTR:
return ctx->i8;
break;
case AC_ARG_CONST_FLOAT_PTR:
return ctx->f32;
break;
case AC_ARG_CONST_PTR_PTR:
return ac_array_in_const32_addr_space(ctx->i8);
break;
case AC_ARG_CONST_DESC_PTR:
return ctx->v4i32;
break;
case AC_ARG_CONST_IMAGE_PTR:
return ctx->v8i32;
default:
/* Other ac_arg_type values aren't pointers. */
assert(false);
return NULL;
}
}
LLVMTypeRef ac_array_in_const_addr_space(LLVMTypeRef elem_type)
{
return LLVMPointerType(elem_type, AC_ADDR_SPACE_CONST);
@@ -4493,37 +4516,37 @@ void ac_build_export_prim(struct ac_llvm_context *ctx, const struct ac_ngg_prim
static LLVMTypeRef arg_llvm_type(enum ac_arg_type type, unsigned size, struct ac_llvm_context *ctx)
{
if (type == AC_ARG_FLOAT) {
return size == 1 ? ctx->f32 : LLVMVectorType(ctx->f32, size);
} else if (type == AC_ARG_INT) {
return size == 1 ? ctx->i32 : LLVMVectorType(ctx->i32, size);
} else {
LLVMTypeRef ptr_type;
switch (type) {
LLVMTypeRef base;
switch (type) {
case AC_ARG_FLOAT:
return size == 1 ? ctx->f32 : LLVMVectorType(ctx->f32, size);
case AC_ARG_INT:
return size == 1 ? ctx->i32 : LLVMVectorType(ctx->i32, size);
case AC_ARG_CONST_PTR:
ptr_type = ctx->i8;
base = ctx->i8;
break;
case AC_ARG_CONST_FLOAT_PTR:
ptr_type = ctx->f32;
base = ctx->f32;
break;
case AC_ARG_CONST_PTR_PTR:
ptr_type = ac_array_in_const32_addr_space(ctx->i8);
base = ac_array_in_const32_addr_space(ctx->i8);
break;
case AC_ARG_CONST_DESC_PTR:
ptr_type = ctx->v4i32;
base = ctx->v4i32;
break;
case AC_ARG_CONST_IMAGE_PTR:
ptr_type = ctx->v8i32;
base = ctx->v8i32;
break;
default:
unreachable("unknown arg type");
}
if (size == 1) {
return ac_array_in_const32_addr_space(ptr_type);
} else {
assert(size == 2);
return ac_array_in_const_addr_space(ptr_type);
}
return NULL;
}
assert(base);
if (size == 1) {
return ac_array_in_const32_addr_space(base);
} else {
assert(size == 2);
return ac_array_in_const_addr_space(base);
}
}

8
src/amd/llvm/ac_llvm_build.h
View File

@@ -575,12 +575,20 @@ LLVMValueRef ac_pack_edgeflags_for_export(struct ac_llvm_context *ctx,
LLVMValueRef ac_pack_prim_export(struct ac_llvm_context *ctx, const struct ac_ngg_prim *prim);
void ac_build_export_prim(struct ac_llvm_context *ctx, const struct ac_ngg_prim *prim);
LLVMTypeRef ac_arg_type_to_pointee_type(struct ac_llvm_context *ctx, enum ac_arg_type type);
static inline LLVMValueRef ac_get_arg(struct ac_llvm_context *ctx, struct ac_arg arg)
{
assert(arg.used);
return LLVMGetParam(ctx->main_function, arg.arg_index);
}
static inline LLVMTypeRef ac_get_arg_pointee_type(struct ac_llvm_context *ctx, const struct ac_shader_args *args, struct ac_arg arg)
{
assert(arg.used);
return ac_arg_type_to_pointee_type(ctx, args->args[arg.arg_index].type);
}
enum ac_llvm_calling_convention
{
AC_LLVM_AMDGPU_VS = 87,