Write out dev pointers

This commit is contained in:
Andrzej Janik 2025-09-23 01:28:26 +00:00
commit 0efbbd002e

View file

@ -70,17 +70,23 @@ pub(crate) fn pre_kernel_launch(
) )
})?; })?;
if size != 0 { if size != 0 {
let mut pre_buffer = vec![0u8; size]; let i = all_params.len();
let post_buffer = vec![0u8; size];
fn_logger.try_cuda(|| {
libcuda.cuMemcpyDtoH_v2(
pre_buffer.as_mut_ptr().cast(),
CUdeviceptr_v2(start as _),
size,
)
})?;
let buffer_offset = maybe_ptr - start; let buffer_offset = maybe_ptr - start;
ptr_overrides.push((offset, buffer_offset, pre_buffer, post_buffer)); eprintln!("[{i}][{offset}] {start:#x}+{buffer_offset}");
//fn_logger.log(ErrorEntry::ErrorBox(
// format!("[{i}][{offset}] {start:#x}+{buffer_offset}").into(),
//));
// let mut pre_buffer = vec![0u8; size];
// let post_buffer = vec![0u8; size];
// fn_logger.try_cuda(|| {
// libcuda.cuMemcpyDtoH_v2(
// pre_buffer.as_mut_ptr().cast(),
// CUdeviceptr_v2(start as _),
// size,
// )
// })?;
// let buffer_offset = maybe_ptr - start;
// ptr_overrides.push((offset, buffer_offset, pre_buffer, post_buffer));
} }
offset += std::mem::size_of::<usize>(); offset += std::mem::size_of::<usize>();
} }
@ -90,6 +96,7 @@ pub(crate) fn pre_kernel_launch(
device_ptrs: ptr_overrides, device_ptrs: ptr_overrides,
}); });
} }
return None;
if state.kernel_no_output { if state.kernel_no_output {
let enqueue_counter = state.enqueue_counter; let enqueue_counter = state.enqueue_counter;
let kernel_name = name; let kernel_name = name;