diff --git a/src/util.rs b/src/util.rs index b29d387..fc7ef3c 100644 --- a/src/util.rs +++ b/src/util.rs @@ -138,7 +138,11 @@ pub unsafe fn alloc_aligned_box_slice(len: usize) -> Box< #[inline(always)] pub unsafe fn alloc_aligned(len: usize, align: usize) -> *mut u8 { let layout = std::alloc::Layout::from_size_align_unchecked(len, align); - std::alloc::alloc(layout) + let ptr = std::alloc::alloc(layout); + if ptr.is_null() { + std::alloc::handle_alloc_error(layout); + } + ptr } #[macro_export]