Merge pull request #881 from jvdp1/fix_svd_1

svd - check the size of storage space
fortran-lang · Oct 25, 2024 · 8050d1f · 8050d1f
2 parents 38e8e0e + e244e13
commit 8050d1f
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