Skip to content

test: stabilize uninitialized replay allocation parity #194

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
shinaoka merged 1 commit into from
Jul 30, 2026
Merged

test: stabilize uninitialized replay allocation parity #194

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
25 changes: 18 additions & 7 deletions strided-kernel/tests/copy_plan_alloc.rs
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -185,6 +185,16 @@ fn execute_is_allocation_free_up_to_rank_limit() {
let mut dst = vec![0.0f64; 256];
let mut dest = ErasedRawStridedMut::from_slice_mut(&mut dst, &dims, &strides, 0).unwrap();

let mut typed_dst = vec![0.0f64; 256];
let lhs_view: StridedView<'_, f64, Identity> =
StridedView::new(&lhs, &dims, &strides, 0).unwrap();
let rhs_view: StridedView<'_, f64, Identity> =
StridedView::new(&rhs, &dims, &strides, 0).unwrap();
let mut typed_dest = StridedViewMut::new(&mut typed_dst, &dims, &strides, 0).unwrap();
// Warm each exact counted sequence symmetrically so TLS/coverage runtime
// initialization is outside both allocation windows.
zip_map2_into(&mut typed_dest, &lhs_view, &rhs_view, |lhs, rhs| lhs + rhs).unwrap();
map_into(&mut typed_dest, &lhs_view, |value| -value).unwrap();
erased_zip_into(
KernelDType::F64,
ErasedZipOp::Add,
Expand All @@ -194,13 +204,14 @@ fn execute_is_allocation_free_up_to_rank_limit() {
&ErasedRawStridedPtr::from_ref(&rhs_ref),
)
.unwrap();
let mut typed_dst = vec![0.0f64; 256];
let lhs_view: StridedView<'_, f64, Identity> =
StridedView::new(&lhs, &dims, &strides, 0).unwrap();
let rhs_view: StridedView<'_, f64, Identity> =
StridedView::new(&rhs, &dims, &strides, 0).unwrap();
let mut typed_dest = StridedViewMut::new(&mut typed_dst, &dims, &strides, 0).unwrap();
zip_map2_into(&mut typed_dest, &lhs_view, &rhs_view, |lhs, rhs| lhs + rhs).unwrap();
erased_map_into(
KernelDType::F64,
ErasedMapOp::Negate,
&ExecContext::serial(),
&mut dest,
&ErasedRawStridedPtr::from_ref(&lhs_ref),
)
.unwrap();
let kernel_allocations = count_allocations(|| {
for _ in 0..16 {
zip_map2_into(&mut typed_dest, &lhs_view, &rhs_view, |lhs, rhs| lhs + rhs).unwrap();
Expand Down
Loading