pan/va: Add support for SPV_ARM_core_builtins

Reviewed-by: Mary Guillemard <mary.guillemard@collabora.com>
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/36019>
This commit is contained in:
John Anthony 2025-07-07 22:03:57 +02:00
parent 000bd3046d
commit c46407de88
2 changed files with 24 additions and 0 deletions

View file

@ -2395,6 +2395,26 @@ bi_emit_intrinsic(bi_builder *b, nir_intrinsic_instr *instr)
bi_mov_i32_to(b, dst, bi_fau(BIR_FAU_SHADER_OUTPUT, false));
break;
case nir_intrinsic_load_core_id:
assert(b->shader->arch >= 9);
bi_mov_i32_to(b, dst, bi_fau(BIR_FAU_CORE_ID, false));
break;
case nir_intrinsic_load_core_count_arm:
assert(b->shader->arch >= 9);
bi_mov_i32_to(b, dst, bi_iadd_u32(b, bi_fau(BIR_FAU_CORE_ID, true), bi_imm_u32(1), false));
break;
case nir_intrinsic_load_warp_id_arm:
assert(b->shader->arch >= 9);
bi_mov_i32_to(b, dst, bi_fau(BIR_FAU_WARP_ID, false));
break;
case nir_intrinsic_load_warp_max_id_arm:
assert(b->shader->arch >= 9);
bi_mov_i32_to(b, dst, bi_fau(BIR_FAU_WARP_ID, true));
break;
default:
fprintf(stderr, "Unhandled intrinsic %s\n",
nir_intrinsic_infos[instr->intrinsic].name);

View file

@ -96,6 +96,8 @@ static unsigned
va_pack_fau_special(const bi_instr *I, enum bir_fau fau)
{
switch (fau) {
case BIR_FAU_WARP_ID:
return VA_FAU_SPECIAL_PAGE_0_WARP_ID;
case BIR_FAU_ATEST_PARAM:
return VA_FAU_SPECIAL_PAGE_0_ATEST_DATUM;
case BIR_FAU_TLS_PTR:
@ -104,6 +106,8 @@ va_pack_fau_special(const bi_instr *I, enum bir_fau fau)
return VA_FAU_SPECIAL_PAGE_1_WORKGROUP_LOCAL_POINTER;
case BIR_FAU_LANE_ID:
return VA_FAU_SPECIAL_PAGE_3_LANE_ID;
case BIR_FAU_CORE_ID:
return VA_FAU_SPECIAL_PAGE_3_CORE_ID;
case BIR_FAU_SHADER_OUTPUT:
return VA_FAU_SPECIAL_PAGE_3_SHADER_OUTPUT;
case BIR_FAU_PROGRAM_COUNTER: