Fixed allocation from source for complex types.

Author: loiseaujc

src/stdlib_specialmatrices.fypp

@@ -16,6 +16,23 @@ module stdlib_specialmatrices
     public :: dense, transpose, hermitian
     public :: operator(*), operator(+), operator(-)
 
+    real(sp), parameter :: zero_sp = 0._sp
+    real(sp), parameter :: one_sp = 1._sp
+    real(dp), parameter :: zero_dp = 0._dp
+    real(dp), parameter :: one_dp = 1._dp
+    real(xdp), parameter :: zero_xdp = 0._xdp
+    real(xdp), parameter :: one_xdp = 1._xdp
+    real(qp), parameter :: zero_qp = 0._qp
+    real(qp), parameter :: one_qp = 1._qp
+    complex(sp), parameter :: zero_csp = (0._sp,0._sp)
+    complex(sp), parameter :: one_csp = (1._sp,1._sp)
+    complex(dp), parameter :: zero_cdp = (0._dp,0._dp)
+    complex(dp), parameter :: one_cdp = (1._dp,1._dp)
+    complex(xdp), parameter :: zero_cxdp = (0._xdp,0._xdp)
+    complex(xdp), parameter :: one_cxdp = (1._xdp,1._xdp)
+    complex(qp), parameter :: zero_cqp = (0._qp,0._qp)
+    complex(qp), parameter :: one_cqp = (1._qp,1._qp)
+
     !--------------------------------------
     !-----                           ------
     !-----     TYPE DEFINITIONS      ------

src/stdlib_specialmatrices_tridiagonal.fypp

@@ -120,7 +120,11 @@ submodule (stdlib_specialmatrices) tridiagonal_matrices
         integer(ilp) :: i
 
         associate (n => A%n)
-        allocate(B(n, n)) ; B = 0.0_${k1}$
+        #:if t1.startswith('complex')
+        allocate(B(n, n), source=zero_c${k1}$)
+        #:else
+        allocate(B(n, n), source=zero_${k1}$)
+        #:endif
         B(1, 1) = A%dv(1) ; B(1, 2) = A%du(1)
         do concurrent (i=2:n-1)
             B(i, i-1) = A%dl(i-1)