mirrors/ZLUDA
0
0
Fork 0
mirror of https://github.com/vosen/ZLUDA.git synced 2025-09-26 11:19:06 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

Set modes

Browse source
This commit is contained in:
Andrzej Janik 2025-09-23 17:43:26 +00:00
commit 7b007074bd
1 changed files with 12 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

22
zluda_trace_common/src/replay.rs
View file

@ -33,8 +33,7 @@ impl Manifest {
fn serialize(&self) -> std::io::Result<(Header, Vec<u8>)> {
let vec = serde_json::to_vec(self)?;
let mut header = Header::new_gnu();
header.set_size(vec.len() as u64);
let header = tar_header(vec.len());
Ok((header, vec))
}
}
@ -76,25 +75,21 @@ pub fn save(
}
.serialize()?;
builder.append_data(&mut header, Manifest::PATH, &*manifest)?;
let mut header = Header::new_gnu();
header.set_size(source.len() as u64);
let mut header = tar_header(source.len());
builder.append_data(&mut header, "source.ptx", source.as_bytes())?;
for (i, param) in kernel_params.into_iter().enumerate() {
let path = format!("param_{i}.bin");
let mut header = Header::new_gnu();
header.set_size(param.data.len() as u64);
let mut header = tar_header(param.data.len());
builder.append_data(&mut header, &*path, &*param.data)?;
for (offset_in_param, _, data_before, data_after) in param.device_ptrs {
let path = format!("param_{i}_ptr_{offset_in_param}_pre.bin");
let mut header = Header::new_gnu();
header.set_size(data_before.len() as u64);
let mut header = tar_header(data_before.len());
builder.append_data(&mut header, &*path, &*data_before)?;
if !has_outputs {
continue;
}
let path = format!("param_{i}_ptr_{offset_in_param}_post.bin");
let mut header = Header::new_gnu();
header.set_size(data_after.len() as u64);
let mut header = tar_header(data_after.len());
builder.append_data(&mut header, &*path, &*data_after)?;
}
}
@ -103,6 +98,13 @@ pub fn save(
Ok(())
}
fn tar_header(size: usize) -> Header {
let mut header = Header::new_gnu();
header.set_mode(0o644);
header.set_size(size as u64);
header
}
pub fn load(reader: impl Read) -> (Manifest, String, FxHashMap<String, Vec<u8>>) {
let archive = zstd::Decoder::new(reader).unwrap();
let mut archive = tar::Archive::new(archive);