Visible nrhs in solve_triangular

Author: termoshtt

src/solve.rs

@@ -22,7 +22,8 @@ pub trait ImplSolve: Sized {
                       uplo: u8,
                       size: usize,
                       a: &[Self],
-                      b: Vec<Self>)
+                      b: Vec<Self>,
+                      nrhs: i32)
                       -> Result<Vec<Self>, LapackError>;
 }
 
@@ -65,14 +66,14 @@ impl ImplSolve for $scalar {
             Err(From::from(info))
         }
     }
-    fn solve_triangle(layout: Layout, uplo: u8, size: usize, a: &[Self], mut b: Vec<Self>) -> Result<Vec<Self>, LapackError> {
+    fn solve_triangle(layout: Layout, uplo: u8, size: usize, a: &[Self], mut b: Vec<Self>, nrhs: i32) -> Result<Vec<Self>, LapackError> {
         let n = size as i32;
         let lda = n;
         let ldb = match layout {
             Layout::ColumnMajor => n,
             Layout::RowMajor => 1,
         };
-        let info = $trtrs(layout, uplo, 'N' as u8, 'N' as u8, n, 1, a, lda, &mut b, ldb);
+        let info = $trtrs(layout, uplo, 'N' as u8, 'N' as u8, n, nrhs, a, lda, &mut b, ldb);
         if info == 0 {
             Ok(b)
         } else {

src/triangular.rs

@@ -19,15 +19,15 @@ impl<A: MFloat> TriangularMatrix for Array<A, Ix2> {
         let (n, _) = self.size();
         let layout = self.layout()?;
         let a = self.as_slice_memory_order().unwrap();
-        let x = ImplSolve::solve_triangle(layout, 'U' as u8, n, a, b.into_raw_vec())?;
+        let x = ImplSolve::solve_triangle(layout, 'U' as u8, n, a, b.into_raw_vec(), 1)?;
         Ok(Array::from_vec(x))
     }
     fn solve_lower(&self, b: Self::Vector) -> Result<Self::Vector, LinalgError> {
         self.check_square()?;
         let (n, _) = self.size();
         let layout = self.layout()?;
         let a = self.as_slice_memory_order().unwrap();
-        let x = ImplSolve::solve_triangle(layout, 'L' as u8, n, a, b.into_raw_vec())?;
+        let x = ImplSolve::solve_triangle(layout, 'L' as u8, n, a, b.into_raw_vec(), 1)?;
         Ok(Array::from_vec(x))
     }
 }