AlexIndustrial/mesa
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

nak: Implement load/store_shared

Browse Source 
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/24998>
This commit is contained in:
Faith Ekstrand
2023-04-23 03:31:31 -05:00
committed by Marge Bot
parent 73976c3f31
commit 2a2103d73b
1 changed files with 32 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/nouveau/compiler/nak_from_nir.rs
+32
View File
@@ -921,6 +921,22 @@ impl<'a> ShaderFromNir<'a> {
let dst = self.get_dst(&intrin.def);
self.instrs.push(Instr::new_ld(dst, access, addr, offset));
}
nir_intrinsic_load_shared => {
let size_B =
(intrin.def.bit_size() / 8) * intrin.def.num_components();
assert!(u32::from(size_B) <= intrin.align());
let access = MemAccess {
addr_type: MemAddrType::A32,
mem_type: MemType::from_size(size_B, false),
space: MemSpace::Shared,
order: MemOrder::Strong,
scope: MemScope::CTA,
};
let (addr, offset) = self.get_io_addr_offset(&srcs[0], 24);
let offset = offset + intrin.base();
let dst = self.get_dst(&intrin.def);
self.instrs.push(Instr::new_ld(dst, access, addr, offset));
}
nir_intrinsic_load_sysval_nv => {
let idx = u8::try_from(intrin.base()).unwrap();
let dst = self.get_dst(&intrin.def);
@@ -1015,6 +1031,22 @@ impl<'a> ShaderFromNir<'a> {
let (addr, offset) = self.get_io_addr_offset(&srcs[1], 24);
self.instrs.push(Instr::new_st(access, addr, offset, data));
}
nir_intrinsic_store_shared => {
let data = self.get_src(&srcs[0]);
let size_B =
(srcs[0].bit_size() / 8) * srcs[0].num_components();
assert!(u32::from(size_B) <= intrin.align());
let access = MemAccess {
addr_type: MemAddrType::A32,
mem_type: MemType::from_size(size_B, false),
space: MemSpace::Shared,
order: MemOrder::Strong,
scope: MemScope::CTA,
};
let (addr, offset) = self.get_io_addr_offset(&srcs[1], 24);
let offset = offset + intrin.base();
self.instrs.push(Instr::new_st(access, addr, offset, data));
}
_ => panic!(
"Unsupported intrinsic instruction: {}",
intrin.info().name()