Update ktypeof function to support BlockArrays and FillArrays

JuliaSmoothOptimizers · Aug 16, 2022 · b96ff11 · b96ff11
1 parent 56196d7
commit b96ff11
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/src/krylov_utils.jl b/src/krylov_utils.jl
@@ -180,13 +180,17 @@ end
 
 Return a dense storage type `S` based on the type of `v`.
 """
-function ktypeof(v :: AbstractVector)
+function ktypeof(v :: AbstractVector{T}) where T
   S = typeof(v)
   if S <: SubArray
-    S = S.types[1]  # SubArray
+    S = S.types[1]
   end
-  if S <: AbstractSparseVector
-    S = S <: SparseVector ? S.types[3] : S.types[2]  # SparseVector / CuSparseVector
+  if !(S <: DenseVector)
+    if S <: AbstractSparseVector
+      S = S <: SparseVector ? S.types[3] : S.types[2]  # SparseVector / CuSparseVector
+    else
+      S = Vector{T}
+    end
   end
   return S
 end