From 498ff6a054684a97fcb901320e23533073a5baea Mon Sep 17 00:00:00 2001
From: DrMeepster <19316085+DrMeepster@users.noreply.github.com>
Date: Fri, 11 Feb 2022 21:26:46 -0800
Subject: [PATCH] Implement DeviceCopy for MaybeUninit

---
 crates/cust_core/src/lib.rs | 2 ++
 1 file changed, 2 insertions(+)

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<T: DeviceCopy> DeviceCopy for MaybeUninit<T> {}
     unsafe impl<T: DeviceCopy> DeviceCopy for Option<T> {}
     unsafe impl<L: DeviceCopy, R: DeviceCopy> DeviceCopy for Result<L, R> {}
     unsafe impl<T: ?Sized + DeviceCopy> DeviceCopy for PhantomData<T> {}