Add UPLO

Author: termoshtt

src/impl2/cholesky.rs

@@ -6,18 +6,18 @@ use types::*;
 use error::*;
 use layout::Layout;
 
-use super::into_result;
+use super::{into_result, UPLO};
 
 pub trait Cholesky_: Sized {
-    fn cholesky(Layout, a: &mut [Self]) -> Result<()>;
+    fn cholesky(Layout, UPLO, a: &mut [Self]) -> Result<()>;
 }
 
 macro_rules! impl_cholesky {
     ($scalar:ty, $potrf:path) => {
 impl Cholesky_ for $scalar {
-    fn cholesky(l: Layout, mut a: &mut [Self]) -> Result<()> {
+    fn cholesky(l: Layout, uplo: UPLO, mut a: &mut [Self]) -> Result<()> {
         let (n, _) = l.size();
-        let info = $potrf(l.lapacke_layout(), b'U', n, &mut a, n);
+        let info = $potrf(l.lapacke_layout(), uplo as u8, n, &mut a, n);
         into_result(info, ())
     }
 }

src/impl2/mod.rs

@@ -23,3 +23,10 @@ pub fn into_result<T>(info: i32, val: T) -> Result<T> {
         Err(LapackError::new(info).into())
     }
 }
+
+#[derive(Debug, Clone, Copy)]
+#[repr(u8)]
+pub enum UPLO {
+    Upper = b'U',
+    Lower = b'L',
+}