diff --git a/crates/cust_core/src/lib.rs b/crates/cust_core/src/lib.rs index 81298bcf..19e7c0c9 100644 --- a/crates/cust_core/src/lib.rs +++ b/crates/cust_core/src/lib.rs @@ -6,6 +6,7 @@ pub use cust_derive::DeviceCopyCore as DeviceCopy; #[doc(hidden)] pub mod _hidden { use core::marker::PhantomData; + use core::mem::MaybeUninit; use core::num::*; /// Marker trait for types which can safely be copied to or from a CUDA device. @@ -87,6 +88,7 @@ pub mod _hidden { NonZeroU8 NonZeroU16 NonZeroU32 NonZeroU64 NonZeroU128 ); + unsafe impl DeviceCopy for MaybeUninit {} unsafe impl DeviceCopy for Option {} unsafe impl DeviceCopy for Result {} unsafe impl DeviceCopy for PhantomData {}