fdo-mirrors/mesa
1
0
Fork 0
mirror of https://gitlab.freedesktop.org/mesa/mesa.git synced 2026-05-05 09:38:07 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

radv/nir/opt_cmat: support bfloat16

Browse source 
Reviewed-by: Rhys Perry <pendingchaos02@gmail.com>
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/34768>
This commit is contained in:
Georg Lehmann 2025-04-30 11:35:45 +02:00 • committed by Marge Bot
parent 5ca98bf99e
commit 78524837c1
1 changed files with 10 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
src/amd/vulkan/nir/radv_nir_lower_cooperative_matrix.c
View file

@ -648,13 +648,21 @@ opt_cmat(nir_builder *b, nir_intrinsic_instr *intrin, void *data)
if (intrin->intrinsic != nir_intrinsic_cmat_muladd_amd)
return false;
enum glsl_base_type a_type = nir_intrinsic_src_base_type(intrin);
if (glsl_base_type_is_integer(a_type))
return false;
bool progress = false;
if (intrin->src[0].ssa->bit_size != 8) {
for (unsigned i = 0; i < 3; i++)
if (a_type == GLSL_TYPE_FLOAT16) {
for (unsigned i = 0; i < 2; i++)
progress |= opt_cmat_modifiers(b, intrin, gfx_level, i);
}
if (a_type == GLSL_TYPE_FLOAT16 || intrin->def.bit_size == 32)
progress |= opt_cmat_modifiers(b, intrin, gfx_level, 2);
return progress;
}