fdo-mirrors/mesa
1
0
Fork 0
mirror of https://gitlab.freedesktop.org/mesa/mesa.git synced 2026-06-21 01:38:23 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

kraid: Implement nir_op_extract_*

Browse source 
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/41841>
This commit is contained in:
Faith Ekstrand 2026-05-30 01:23:39 -04:00 • committed by Marge Bot
parent aa766b8f76
commit 0c51b7ffe9
1 changed files with 72 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

72
src/panfrost/compiler/kraid/nir.rs
View file

@ -316,6 +316,78 @@ impl<'a> ShaderFromNir<'a> {
};
match alu.op {
nir_op_extract_i8 | nir_op_extract_u8 => {
assert!(alu.def.bit_size >= 16);
assert!(alu.def.num_components <= 2);
let mut sel = [0_u8; 2];
for c in 0..alu.def.num_components {
sel[usize::from(c)] = alu
.get_src(1)
.comp_as_uint(c)
.expect("nir_op_extract.src[1] should be constant")
.try_into()
.unwrap();
}
if alu.def.num_components == 1 {
sel[1] = sel[0];
}
let (swz, num_type) = match alu.op {
nir_op_extract_i8 => {
let swz = match alu.def.bit_size {
16 => Swizzle::widen_v2s8(sel[0], sel[1]),
32 => Swizzle::widen_s8(sel[0]),
_ => panic!("Invalid 8-bit extract"),
};
(swz, NumericType::SignedInteger)
}
nir_op_extract_u8 => {
let swz = match alu.def.bit_size {
16 => Swizzle::widen_v2u8(sel[0], sel[1]),
32 => Swizzle::widen_u8(sel[0]),
_ => panic!("Invalid 8-bit extract"),
};
(swz, NumericType::UnsignedInteger)
}
_ => panic!("Invalid 8-bit extract"),
};
b.push_op(OpIAdd {
dst: dst.into(),
dst_type: dst_type(num_type),
saturate: false,
srcs: [srcs(0).swizzle(swz), 0.into()],
});
}
nir_op_extract_i16 | nir_op_extract_u16 => {
assert!(alu.def.bit_size == 32);
assert!(alu.def.num_components == 1);
let sel = alu
.get_src(1)
.comp_as_uint(0)
.expect("nir_op_extract.src[1] should be constant")
.try_into()
.unwrap();
let (swz, num_type) = match alu.op {
nir_op_extract_i8 => {
(Swizzle::widen_s16(sel), NumericType::SignedInteger)
}
nir_op_extract_u8 => {
(Swizzle::widen_u16(sel), NumericType::UnsignedInteger)
}
_ => panic!("Invalid 8-bit extract"),
};
b.push_op(OpIAdd {
dst: dst.into(),
dst_type: dst_type(num_type),
saturate: false,
srcs: [srcs(0).swizzle(swz), 0.into()],
});
}
nir_op_fabs => {
// TODO: Do we really want FAdd for this?
b.push_op(OpFAdd {