make default inner_product point to a default dot_product add enum for workspace sizes

Author: jalvesz

src/stdlib_linalg_iterative_solvers.fypp

@@ -10,11 +10,17 @@ module stdlib_linalg_iterative_solvers
     implicit none
     private 
 
+    enum, bind(c)
+        enumerator :: size_wksp_cg = 3
+        enumerator :: size_wksp_pccg = 4
+    end enum
+    public :: size_wksp_cg, size_wksp_pccg
+
     #:for k, t, s in R_KINDS_TYPES
     type, public :: linop_${s}$
         procedure(vector_sub_${s}$), nopass, pointer    :: matvec => null()
         procedure(vector_sub_${s}$), nopass, pointer    :: preconditionner => null()
-        procedure(reduction_sub_${s}$), nopass, pointer :: inner_product => null()
+        procedure(reduction_sub_${s}$), nopass, pointer :: inner_product => default_dot_${s}$
     end type
     #:endfor
 
@@ -114,6 +120,22 @@ module stdlib_linalg_iterative_solvers
         #:endfor
         #:endfor
     end interface
-    public :: solve_pccg
+    public :: solve_pccg 
+
+
+contains
+
+    !------------------------------------------------------------------
+    ! defaults
+    !------------------------------------------------------------------
+    #:for k, t, s in R_KINDS_TYPES
+    pure ${t}$ function default_dot_${s}$(x,y) result(r)
+        use stdlib_intrinsics, only: stdlib_dot_product
+        ${t}$, intent(in) :: x(:)
+        ${t}$, intent(in) :: y(:)
+        r = stdlib_dot_product(x,y)
+    end function
+
+    #:endfor
 
 end module stdlib_linalg_iterative_solvers

src/stdlib_linalg_iterative_solvers_cg.fypp

@@ -8,7 +8,6 @@ submodule(stdlib_linalg_iterative_solvers) stdlib_linalg_iterative_cg
     use stdlib_kinds
     use stdlib_sparse
     use stdlib_constants
-    use stdlib_intrinsics, only: dot_product => stdlib_dot_product
     use stdlib_linalg_iterative_solvers
     implicit none
 
@@ -100,7 +99,7 @@ contains
         !-------------------------
         ! internal memory setup
         op%matvec => default_matvec
-        op%inner_product => default_dot
+        ! op%inner_product => default_dot
         if(present(di))then
             di_ => di
         else 
@@ -112,7 +111,7 @@ contains
         else
             allocate( workspace_ )
         end if
-        if(.not.allocated(workspace_%tmp)) allocate( workspace_%tmp(n,3), source = zero_${s}$ )
+        if(.not.allocated(workspace_%tmp)) allocate( workspace_%tmp(n,size_wksp_cg), source = zero_${s}$ )
         !-------------------------
         ! main call to the solver
         call solve_cg_generic(op,b,x,di_,tol_,maxiter_,restart_,workspace_)
@@ -138,11 +137,6 @@ contains
             call spmv( A , x, y )
             #:endif
         end subroutine
-        pure ${t}$ function default_dot(x,y) result(r)
-            ${t}$, intent(in) :: x(:)
-            ${t}$, intent(in) :: y(:)
-            r = dot_product(x,y)
-        end function
     end subroutine
 
     #:endfor

src/stdlib_linalg_iterative_solvers_pccg.fypp

@@ -8,7 +8,6 @@ submodule(stdlib_linalg_iterative_solvers) stdlib_linalg_iterative_pccg
     use stdlib_kinds
     use stdlib_sparse
     use stdlib_constants
-    use stdlib_intrinsics, only: dot_product => stdlib_dot_product
     use stdlib_linalg_iterative_solvers
     implicit none
 
@@ -110,7 +109,7 @@ contains
         !-------------------------
         ! internal memory setup
         op%matvec => default_matvec
-        op%inner_product => default_dot
+        ! op%inner_product => default_dot
         op%preconditionner => default_preconditionner
         if(present(di))then
             di_ => di
@@ -123,7 +122,7 @@ contains
         else
             allocate( workspace_ )
         end if
-        if(.not.allocated(workspace_%tmp)) allocate( workspace_%tmp(n,4) , source = zero_${s}$ )
+        if(.not.allocated(workspace_%tmp)) allocate( workspace_%tmp(n,size_wksp_pccg) , source = zero_${s}$ )
         !-------------------------
         ! main call to the solver
         call solve_pccg_generic(op,b,x,di_,tol_,maxiter_,restart_,workspace_)
@@ -154,11 +153,6 @@ contains
             ${t}$, intent(inout) :: y(:)
             y = x
         end subroutine
-        pure ${t}$ function default_dot(x,y) result(r)
-            ${t}$, intent(in) :: x(:)
-            ${t}$, intent(in) :: y(:)
-            r = dot_product(x,y)
-        end function
     end subroutine
 
     #:endfor