Skip to content
Snippets Groups Projects
Commit 008fd95d authored by Russel Arbore's avatar Russel Arbore
Browse files

fix lifetime

parent 12b5e54f
No related branches found
No related tags found
1 merge request!115GPU backend
Pipeline #201301 passed
Stage: test
Hide whitespace changes
Inline Side-by-side
Showing
with 4 additions and 4 deletions
hercules_rt/src/lib.rs
+ 4
− 4
View file @ 008fd95d
...@@ -152,12 +152,12 @@ impl<'a> HerculesCPURefMut<'a> { ...@@ -152,12 +152,12 @@ impl<'a> HerculesCPURefMut<'a> {
#[cfg(feature = "cuda")] #[cfg(feature = "cuda")]
impl<'a> HerculesCUDARef<'a> { impl<'a> HerculesCUDARef<'a> {
pub fn to_cpu_ref<T>(self, dst: &mut [T]) -> HerculesCPURef<'a> { pub fn to_cpu_ref<'b, T>(self, dst: &'b mut [T]) -> HerculesCPURefMut<'b> {
unsafe { unsafe {
let size = self.size; let size = self.size;
let ptr = NonNull::new(dst.as_ptr() as *mut u8).unwrap(); let ptr = NonNull::new(dst.as_ptr() as *mut u8).unwrap();
__copy_cuda_to_cpu(ptr.as_ptr(), self.ptr.as_ptr(), size); __copy_cuda_to_cpu(ptr.as_ptr(), self.ptr.as_ptr(), size);
HerculesCPURef { HerculesCPURefMut {
ptr, ptr,
size, size,
_phantom: PhantomData, _phantom: PhantomData,
...@@ -192,12 +192,12 @@ impl<'a> HerculesCUDARefMut<'a> { ...@@ -192,12 +192,12 @@ impl<'a> HerculesCUDARefMut<'a> {
} }
} }
pub fn to_cpu_ref<T>(self, dst: &mut [T]) -> HerculesCPURef<'a> { pub fn to_cpu_ref<'b, T>(self, dst: &mut [T]) -> HerculesCPURefMut<'b> {
unsafe { unsafe {
let size = self.size; let size = self.size;
let ptr = NonNull::new(dst.as_ptr() as *mut u8).unwrap(); let ptr = NonNull::new(dst.as_ptr() as *mut u8).unwrap();
__copy_cuda_to_cpu(ptr.as_ptr(), self.ptr.as_ptr(), size); __copy_cuda_to_cpu(ptr.as_ptr(), self.ptr.as_ptr(), size);
HerculesCPURef { HerculesCPURefMut {
ptr, ptr,
size, size,
_phantom: PhantomData, _phantom: PhantomData,
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment