fix svd

fortran-lang · Oct 18, 2024 · e244e13 · e244e13
1 parent 91dcc50
commit e244e13
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/stdlib_linalg_svd.fypp b/src/stdlib_linalg_svd.fypp
@@ -269,7 +269,11 @@ submodule(stdlib_linalg) stdlib_linalg_svd
          if (info==0) then
 
             !> Prepare working storage
-            lwork = nint(real(work_dummy(1),kind=${rk}$), kind=ilp)
+            ! Check if the returned working storage space is smaller than the largest value
+            ! allowed by lwork
+            lwork = merge(nint(real(work_dummy(1),kind=${rk}$), kind=ilp) &
+                          , huge(lwork) &
+                          , real(work_dummy(1),kind=${rk}$) < real(huge(lwork),kind=${rk}$) )
             allocate(work(lwork))
 
             !> Compute SVD