mirror of
https://gitlab.freedesktop.org/mesa/mesa.git
synced 2026-01-06 13:10:10 +01:00
ac: add 16bit conversion operations
Reviewed-by: Bas Nieuwenhuizen <bas@basnieuwenhuizen.nl>
This commit is contained in:
parent
d73f1026b4
commit
f582367d49
2 changed files with 33 additions and 11 deletions
|
|
@ -175,6 +175,8 @@ ac_get_type_size(LLVMTypeRef type)
|
|||
switch (kind) {
|
||||
case LLVMIntegerTypeKind:
|
||||
return LLVMGetIntTypeWidth(type) / 8;
|
||||
case LLVMHalfTypeKind:
|
||||
return 2;
|
||||
case LLVMFloatTypeKind:
|
||||
return 4;
|
||||
case LLVMDoubleTypeKind:
|
||||
|
|
@ -320,6 +322,9 @@ void ac_build_type_name_for_intr(LLVMTypeRef type, char *buf, unsigned bufsize)
|
|||
case LLVMIntegerTypeKind:
|
||||
snprintf(buf, bufsize, "i%d", LLVMGetIntTypeWidth(elem_type));
|
||||
break;
|
||||
case LLVMHalfTypeKind:
|
||||
snprintf(buf, bufsize, "f16");
|
||||
break;
|
||||
case LLVMFloatTypeKind:
|
||||
snprintf(buf, bufsize, "f32");
|
||||
break;
|
||||
|
|
@ -1819,11 +1824,9 @@ LLVMValueRef ac_build_cvt_pkrtz_f16(struct ac_llvm_context *ctx,
|
|||
{
|
||||
LLVMTypeRef v2f16 =
|
||||
LLVMVectorType(LLVMHalfTypeInContext(ctx->context), 2);
|
||||
LLVMValueRef res =
|
||||
ac_build_intrinsic(ctx, "llvm.amdgcn.cvt.pkrtz",
|
||||
v2f16, args, 2,
|
||||
AC_FUNC_ATTR_READNONE);
|
||||
return LLVMBuildBitCast(ctx->builder, res, ctx->i32, "");
|
||||
|
||||
return ac_build_intrinsic(ctx, "llvm.amdgcn.cvt.pkrtz", v2f16,
|
||||
args, 2, AC_FUNC_ATTR_READNONE);
|
||||
}
|
||||
|
||||
/* Upper 16 bits must be zero. */
|
||||
|
|
|
|||
|
|
@ -478,7 +478,8 @@ static LLVMValueRef emit_pack_half_2x16(struct ac_llvm_context *ctx,
|
|||
comp[0] = LLVMBuildExtractElement(ctx->builder, src0, ctx->i32_0, "");
|
||||
comp[1] = LLVMBuildExtractElement(ctx->builder, src0, ctx->i32_1, "");
|
||||
|
||||
return ac_build_cvt_pkrtz_f16(ctx, comp);
|
||||
return LLVMBuildBitCast(ctx->builder, ac_build_cvt_pkrtz_f16(ctx, comp),
|
||||
ctx->i32, "");
|
||||
}
|
||||
|
||||
static LLVMValueRef emit_unpack_half_2x16(struct ac_llvm_context *ctx,
|
||||
|
|
@ -857,34 +858,47 @@ static void visit_alu(struct ac_nir_context *ctx, const nir_alu_instr *instr)
|
|||
src[i] = ac_to_integer(&ctx->ac, src[i]);
|
||||
result = ac_build_gather_values(&ctx->ac, src, num_components);
|
||||
break;
|
||||
case nir_op_f2i16:
|
||||
case nir_op_f2i32:
|
||||
case nir_op_f2i64:
|
||||
src[0] = ac_to_float(&ctx->ac, src[0]);
|
||||
result = LLVMBuildFPToSI(ctx->ac.builder, src[0], def_type, "");
|
||||
break;
|
||||
case nir_op_f2u16:
|
||||
case nir_op_f2u32:
|
||||
case nir_op_f2u64:
|
||||
src[0] = ac_to_float(&ctx->ac, src[0]);
|
||||
result = LLVMBuildFPToUI(ctx->ac.builder, src[0], def_type, "");
|
||||
break;
|
||||
case nir_op_i2f16:
|
||||
case nir_op_i2f32:
|
||||
case nir_op_i2f64:
|
||||
src[0] = ac_to_integer(&ctx->ac, src[0]);
|
||||
result = LLVMBuildSIToFP(ctx->ac.builder, src[0], ac_to_float_type(&ctx->ac, def_type), "");
|
||||
break;
|
||||
case nir_op_u2f16:
|
||||
case nir_op_u2f32:
|
||||
case nir_op_u2f64:
|
||||
src[0] = ac_to_integer(&ctx->ac, src[0]);
|
||||
result = LLVMBuildUIToFP(ctx->ac.builder, src[0], ac_to_float_type(&ctx->ac, def_type), "");
|
||||
break;
|
||||
case nir_op_f2f16_rtz:
|
||||
src[0] = ac_to_float(&ctx->ac, src[0]);
|
||||
LLVMValueRef param[2] = { src[0], ctx->ac.f32_0 };
|
||||
result = ac_build_cvt_pkrtz_f16(&ctx->ac, param);
|
||||
result = LLVMBuildExtractElement(ctx->ac.builder, result, ctx->ac.i32_0, "");
|
||||
break;
|
||||
case nir_op_f2f16_rtne:
|
||||
case nir_op_f2f16_undef:
|
||||
case nir_op_f2f32:
|
||||
case nir_op_f2f64:
|
||||
src[0] = ac_to_float(&ctx->ac, src[0]);
|
||||
result = LLVMBuildFPExt(ctx->ac.builder, src[0], ac_to_float_type(&ctx->ac, def_type), "");
|
||||
break;
|
||||
case nir_op_f2f32:
|
||||
src[0] = ac_to_float(&ctx->ac, src[0]);
|
||||
result = LLVMBuildFPTrunc(ctx->ac.builder, src[0], ac_to_float_type(&ctx->ac, def_type), "");
|
||||
if (ac_get_elem_bits(&ctx->ac, LLVMTypeOf(src[0])) < ac_get_elem_bits(&ctx->ac, def_type))
|
||||
result = LLVMBuildFPExt(ctx->ac.builder, src[0], ac_to_float_type(&ctx->ac, def_type), "");
|
||||
else
|
||||
result = LLVMBuildFPTrunc(ctx->ac.builder, src[0], ac_to_float_type(&ctx->ac, def_type), "");
|
||||
break;
|
||||
case nir_op_u2u16:
|
||||
case nir_op_u2u32:
|
||||
case nir_op_u2u64:
|
||||
src[0] = ac_to_integer(&ctx->ac, src[0]);
|
||||
|
|
@ -893,6 +907,7 @@ static void visit_alu(struct ac_nir_context *ctx, const nir_alu_instr *instr)
|
|||
else
|
||||
result = LLVMBuildTrunc(ctx->ac.builder, src[0], def_type, "");
|
||||
break;
|
||||
case nir_op_i2i16:
|
||||
case nir_op_i2i32:
|
||||
case nir_op_i2i64:
|
||||
src[0] = ac_to_integer(&ctx->ac, src[0]);
|
||||
|
|
@ -1098,6 +1113,10 @@ static void visit_load_const(struct ac_nir_context *ctx,
|
|||
|
||||
for (unsigned i = 0; i < instr->def.num_components; ++i) {
|
||||
switch (instr->def.bit_size) {
|
||||
case 16:
|
||||
values[i] = LLVMConstInt(element_type,
|
||||
instr->value.u16[i], false);
|
||||
break;
|
||||
case 32:
|
||||
values[i] = LLVMConstInt(element_type,
|
||||
instr->value.u32[i], false);
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue