StaticArrays support

.buildkite/pipeline.yml

@@ -69,3 +69,17 @@ steps:
         Pkg.instantiate()
         include("test/gpu/metal.jl")'
     timeout_in_minutes: 30
+
+  - label: "CPUs -- StaticArrays.jl"
+    plugins:
+      - JuliaCI/julia#v1:
+          version: 1.9
+    agents:
+      queue: "juliaecosystem"
+    command: |
+      julia --color=yes --project -e '
+        using Pkg
+        Pkg.add("StaticArrays")
+        Pkg.instantiate()
+        include("test/test_extensions.jl")'
+    timeout_in_minutes: 30

src/krylov_utils.jl

@@ -210,9 +210,9 @@ function ktypeof(v::S) where S <: DenseVector
 end
 
 function ktypeof(v::S) where S <: AbstractVector
-  if S.name.name == :Zeros || S.name.name == :Ones
+  if S.name.name == :Zeros || S.name.name == :Ones || S.name.name == :SArray || S.name.name == :MArray || S.name.name == :SizedArray
     T = eltype(S)
-    return Vector{T}  # FillArrays
+    return Vector{T}  # FillArrays, StaticArrays
   else
     return S  # BlockArrays, PartitionedArrays, etc...
   end

test/test_extensions.jl

@@ -0,0 +1,25 @@
+using LinearAlgebra, SparseArrays, Test
+using Krylov, StaticArrays
+
+@testset "StaticArrays" begin
+  n = 5
+
+  for T in (Float32, Float64)
+    A = rand(T, n, n)
+
+    b = SVector{n}(rand(T, n))
+    @test Krylov.ktypeof(b) == Vector{T}
+    x, stats = gmres(A, b)
+    @test stats.solved
+
+    b = MVector{n}(rand(T, n))
+    @test Krylov.ktypeof(b) == Vector{T}
+    x, stats = gmres(A, b)
+    @test stats.solved
+
+    b = SizedVector{n}(rand(T, n))
+    @test Krylov.ktypeof(b) == Vector{T}
+    x, stats = gmres(A, b)
+    @test stats.solved
+  end
+end