mirrors/ZLUDA
0
0
Fork 0
mirror of https://github.com/vosen/ZLUDA.git synced 2025-08-04 15:19:49 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

zoc: Refactoring

Browse source
This commit is contained in:
Joëlle van Essen 2025-03-03 13:38:47 +01:00
commit 8c72eae6ba
No known key found for this signature in database
GPG key ID: 28D3B5CDD4B43882
1 changed files with 10 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

18
zoc/src/main.rs
View file

@ -26,14 +26,14 @@ fn main() -> Result<(), Box<dyn Error>> {
let ptx = fs::read(ptx_path)?; let ptx = fs::read(ptx_path)?;
let ptx = str::from_utf8(&ptx)?; let ptx = str::from_utf8(&ptx)?;
let llvm_artifacts = ptx_to_llvm(ptx)?; let llvm = ptx_to_llvm(ptx)?;
let ll_path = ptx_path.with_extension("ll"); let ll_path = ptx_path.with_extension("ll");
write_to_file(&llvm_artifacts.ll, &ll_path)?; write_to_file(&llvm.llvm_ir, &ll_path)?;
let elf_module = llvm_to_elf(&llvm_artifacts)?; let elf = llvm_to_elf(&llvm)?;
let elf_path = ptx_path.with_extension("elf"); let elf_path = ptx_path.with_extension("elf");
write_to_file(&elf_module, &elf_path)?; write_to_file(&elf, &elf_path)?;
Ok(()) Ok(())
} }
@ -48,18 +48,20 @@ fn ptx_to_llvm(ptx: &str) -> Result<LLVMArtifacts, Box<dyn Error>> {
let module = ptx::to_llvm_module(ast)?; let module = ptx::to_llvm_module(ast)?;
let bitcode = module.llvm_ir.write_bitcode_to_memory().to_vec(); let bitcode = module.llvm_ir.write_bitcode_to_memory().to_vec();
let linked_bitcode = module.linked_bitcode().to_vec(); let linked_bitcode = module.linked_bitcode().to_vec();
let ll = module.llvm_ir.print_module_to_string().to_bytes().to_vec(); let llvm_ir = module.llvm_ir.print_module_to_string().to_bytes().to_vec();
Ok(LLVMArtifacts { bitcode, linked_bitcode, ll }) Ok(LLVMArtifacts { bitcode, linked_bitcode, llvm_ir })
} }
fn llvm_to_elf(module: &LLVMArtifacts) -> Result<Vec<u8>, ElfError> { fn llvm_to_elf(module: &LLVMArtifacts) -> Result<Vec<u8>, ElfError> {
use hip_runtime_sys::*; use hip_runtime_sys::*;
unsafe { hipInit(0) }?;
let mut dev_props: MaybeUninit<hipDeviceProp_tR0600> = MaybeUninit::uninit(); let mut dev_props: MaybeUninit<hipDeviceProp_tR0600> = MaybeUninit::uninit();
unsafe { hipGetDevicePropertiesR0600(dev_props.as_mut_ptr(), 0) }?; unsafe { hipGetDevicePropertiesR0600(dev_props.as_mut_ptr(), 0) }?;
let dev_props = unsafe { dev_props.assume_init() }; let dev_props = unsafe { dev_props.assume_init() };
let gcn_arch = unsafe { CStr::from_ptr(dev_props.gcnArchName.as_ptr()) };
comgr::compile_bitcode( comgr::compile_bitcode(
unsafe { CStr::from_ptr(dev_props.gcnArchName.as_ptr()) }, gcn_arch,
&module.bitcode, &module.bitcode,
&module.linked_bitcode, &module.linked_bitcode,
).map_err(ElfError::from) ).map_err(ElfError::from)
@ -76,7 +78,7 @@ fn write_to_file(content: &[u8], path: &Path) -> io::Result<()> {
struct LLVMArtifacts { struct LLVMArtifacts {
bitcode: Vec<u8>, bitcode: Vec<u8>,
linked_bitcode: Vec<u8>, linked_bitcode: Vec<u8>,
ll: Vec<u8>, llvm_ir: Vec<u8>,
} }
#[derive(Debug, thiserror::Error)] #[derive(Debug, thiserror::Error)]