Use vec_uninit2 in qr.rs, opnorm.rs

Author: termoshtt

lax/src/opnorm.rs

@@ -18,8 +18,8 @@ macro_rules! impl_opnorm {
                     MatrixLayout::F { .. } => t,
                     MatrixLayout::C { .. } => t.transpose(),
                 };
-                let mut work: Vec<Self::Real> = if matches!(t, NormType::Infinity) {
-                    unsafe { vec_uninit(m as usize) }
+                let mut work: Vec<MaybeUninit<Self::Real>> = if matches!(t, NormType::Infinity) {
+                    unsafe { vec_uninit2(m as usize) }
                 } else {
                     Vec::new()
                 };

lax/src/qr.rs

@@ -25,7 +25,7 @@ macro_rules! impl_qr {
                 let m = l.lda();
                 let n = l.len();
                 let k = m.min(n);
-                let mut tau = unsafe { vec_uninit(k as usize) };
+                let mut tau = unsafe { vec_uninit2(k as usize) };
 
                 // eval work size
                 let mut info = 0;
@@ -62,7 +62,7 @@ macro_rules! impl_qr {
 
                 // calc
                 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) };
                 unsafe {
                     match l {
                         MatrixLayout::F { .. } => {
@@ -93,6 +93,8 @@ macro_rules! impl_qr {
                 }
                 info.as_lapack_result()?;
 
+                let tau = unsafe { tau.assume_init() };
+
                 Ok(tau)
             }
 
@@ -134,7 +136,7 @@ macro_rules! impl_qr {
 
                 // calc
                 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) };
                 unsafe {
                     match l {
                         MatrixLayout::F { .. } => $gqr(