Use vec_uninit2 in eigh.rs

Author: termoshtt

lax/src/eigh.rs

@@ -42,10 +42,10 @@ macro_rules! impl_eigh {
                 assert_eq!(layout.len(), layout.lda());
                 let n = layout.len();
                 let jobz = if calc_v { EigenVectorFlag::Calc } else { EigenVectorFlag::Not };
-                let mut eigs = unsafe { vec_uninit(n as usize) };
+                let mut eigs: Vec<MaybeUninit<Self::Real>> = unsafe { vec_uninit2(n as usize) };
 
                 $(
-                let mut $rwork_ident: Vec<Self::Real> = unsafe { vec_uninit(3 * n as usize - 2 as usize) };
+                let mut $rwork_ident: Vec<MaybeUninit<Self::Real>> = unsafe { vec_uninit2(3 * n as usize - 2 as usize) };
                 )*
 
                 // calc work size
@@ -69,7 +69,7 @@ macro_rules! impl_eigh {
 
                 // actual ev
                 let lwork = work_size[0].to_usize().unwrap();
-                let mut work: Vec<Self> = unsafe { vec_uninit(lwork) };
+                let mut work: Vec<MaybeUninit<Self>> = unsafe { vec_uninit2(lwork) };
                 let lwork = lwork as i32;
                 unsafe {
                     $ev(
@@ -86,6 +86,8 @@ macro_rules! impl_eigh {
                     );
                 }
                 info.as_lapack_result()?;
+
+                let eigs = unsafe { eigs.assume_init() };
                 Ok(eigs)
             }
 
@@ -99,10 +101,10 @@ macro_rules! impl_eigh {
                 assert_eq!(layout.len(), layout.lda());
                 let n = layout.len();
                 let jobz = if calc_v { EigenVectorFlag::Calc } else { EigenVectorFlag::Not };
-                let mut eigs = unsafe { vec_uninit(n as usize) };
+                let mut eigs: Vec<MaybeUninit<Self::Real>> = unsafe { vec_uninit2(n as usize) };
 
                 $(
-                let mut $rwork_ident: Vec<Self::Real> = unsafe { vec_uninit(3 * n as usize - 2) };
+                let mut $rwork_ident: Vec<MaybeUninit<Self::Real>> = unsafe { vec_uninit2(3 * n as usize - 2) };
                 )*
 
                 // calc work size
@@ -129,7 +131,7 @@ macro_rules! impl_eigh {
 
                 // actual evg
                 let lwork = work_size[0].to_usize().unwrap();
-                let mut work: Vec<Self> = unsafe { vec_uninit(lwork) };
+                let mut work: Vec<MaybeUninit<Self>> = unsafe { vec_uninit2(lwork) };
                 let lwork = lwork as i32;
                 unsafe {
                     $evg(
@@ -149,6 +151,7 @@ macro_rules! impl_eigh {
                     );
                 }
                 info.as_lapack_result()?;
+                let eigs = unsafe { eigs.assume_init() };
                 Ok(eigs)
             }
         }