Resolved bug in solver_lin

by ensuring correct dimensions for variable 'x'.
gha3mi · Dec 9, 2023 · 229b02f · 229b02f
1 parent 3b31a4f
commit 229b02f
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/forsolver.f90 b/src/forsolver.f90
@@ -57,7 +57,7 @@ pure function solver_lin(A, b, method) result(x)
       character(*), optional,                    intent(in) :: method
 
       ! outputs:
-      real(rk), dimension(size(b))                          :: x      ! solution matrix x
+      real(rk), dimension(size(A,2))                        :: x      ! solution matrix x
 
       if (present(method)) then
          select case (method)
@@ -86,7 +86,7 @@ pure function dgesv_rel(A, b) result(x)
       real(rk), dimension(:),    contiguous, intent(in)  :: b    ! right-hand side matrix b
 
       ! outputs:
-      real(rk), dimension(size(b))                       :: x    ! solution matrix x
+      real(rk), dimension(size(A,2))                     :: x    ! solution matrix x
 
       ! local variables
       integer                                  :: info ! result info
@@ -133,7 +133,7 @@ pure function dgels_rel(A, b) result(x)
       real(rk), dimension(:),    contiguous, intent(in)  :: b    ! right-hand side matrix b
 
       ! outputs:
-      real(rk), dimension(size(b))                        :: x    ! solution matrix x
+      real(rk), dimension(size(A,2))                     :: x    ! solution matrix x
 
       ! local variables
       integer                                  :: info ! result info