fdo-mirrors/mesa
1
0
Fork 0
mirror of https://gitlab.freedesktop.org/mesa/mesa.git synced 2025-12-28 19:00:13 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

nak: Add a lowering pass for VEC and SPLIT instructions

Browse source 
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/24998>
This commit is contained in:
Faith Ekstrand 2023-01-30 20:53:16 -06:00 • committed by Marge Bot
parent d44b31f3bf
commit f694f1f8a8
2 changed files with 61 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
src/nouveau/compiler/nak.rs
View file

@ -48,6 +48,7 @@ pub extern "C" fn nak_compile_shader(
println!("NAK IR:\n{}", &s);
s.assign_regs_trivial();
s.lower_vec_split();
println!("NAK IR:\n{}", &s);

60
src/nouveau/compiler/nak_ir.rs
View file

@ -778,6 +778,66 @@ impl Shader {
f.map_instrs(map);
}
}
pub fn lower_vec_split(&mut self) {
self.map_instrs(&|instr: Instr| -> Instr {
match instr.op {
Opcode::VEC => {
let comps = u8::try_from(instr.num_srcs()).unwrap();
if comps == 1 {
let src = instr.src(0);
let dst = instr.dst(0);
Instr::new_mov(*dst, *src)
} else {
let mut instrs = Vec::new();
let vec_dst = instr.dst(0).as_reg().unwrap();
assert!(comps == vec_dst.comps());
for i in 0..comps {
let src = instr.src(i.into());
let dst = Dst::Reg(vec_dst.as_comp(i).unwrap());
instrs.push(Instr::new_mov(dst, *src));
}
Instr::new_meta(instrs)
}
}
Opcode::SPLIT => {
let comps = u8::try_from(instr.num_dsts()).unwrap();
if instr.num_dsts() == 1 {
let src = instr.src(0);
let dst = instr.dst(0);
Instr::new_mov(*dst, *src)
} else {
let mut instrs = Vec::new();
let vec_src = instr.src(0).as_reg().unwrap();
assert!(comps == vec_src.comps());
for i in 0..comps {
let src = Dst::Reg(vec_src.as_comp(i).unwrap());
let dst = instr.dst(i.into());
if let Dst::Zero = dst {
continue;
}
instrs.push(Instr::new_mov(*dst, src));
}
Instr::new_meta(instrs)
}
}
Opcode::FS_OUT => {
let mut instrs = Vec::new();
for i in 0..instr.num_srcs() {
let dst = Ref::new_reg(
RegFile::GPR,
i.try_into().unwrap(),
1,
);
let src = instr.src(i);
instrs.push(Instr::new_mov(dst, *src));
}
Instr::new_meta(instrs)
}
_ => instr,
}
})
}
}
impl fmt::Display for Shader {