Add vector_to_matrix function

.buildkite/pipeline.yml

@@ -22,7 +22,7 @@ steps:
     agents:
       queue: "juliagpu"
       rocm: "*"
-      rocmgpu: "gfx908"
+      rocmgpu: "*"
     env:
       JULIA_AMDGPU_CORE_MUST_LOAD: "1"
       JULIA_AMDGPU_HIP_MUST_LOAD: "1"

docs/src/api.md

@@ -60,4 +60,5 @@ Krylov.vec2str
 Krylov.ktypeof
 Krylov.kzeros
 Krylov.kones
+Krylov.vector_to_matrix
 ```

src/krylov_utils.jl

@@ -247,6 +247,24 @@ function ktypeof(v::S) where S <: SubArray
   return ktypeof(v.parent)
 end
 
+"""
+    M = vector_to_matrix(S)
+
+Return the dense matrix storage type `M` related to the dense vector storage type `S`.
+"""
+function vector_to_matrix(::Type{S}) where S <: DenseVector
+  V = hasproperty(S, :body) ? S.body : S
+  par = V.parameters
+  npar = length(par)
+  (2 ≤ npar ≤ 3) || error("Type $S is not supported.")
+  if npar == 2
+    M = V.name.wrapper{par[1], 2}
+  else
+    M = V.name.wrapper{par[1], 2, par[3]}
+  end
+  return M
+end
+
 """
     v = kzeros(S, n)
 

test/gpu/amd.jl

@@ -68,6 +68,12 @@ include("../test_utils.jl")
     #   Krylov.@kref!(n, x, y, c, s)
     # end
 
+    @testset "vector_to_matrix" begin
+      S = ROCVector{FC}
+      M = Krylov.vector_to_matrix(S)
+      @test M == ROCMatrix{FC}
+    end
+
     ε = eps(T)
     atol = √ε
     rtol = √ε

test/gpu/intel.jl

@@ -76,6 +76,12 @@ end
     #   Krylov.@kref!(n, x, y, c, s)
     # end
 
+    @testset "vector_to_matrix" begin
+      S = oneVector{FC}
+      M = Krylov.vector_to_matrix(S)
+      @test M == oneMatrix{FC}
+    end
+
     ε = eps(T)
     atol = √ε
     rtol = √ε

test/gpu/metal.jl

@@ -80,6 +80,12 @@ end
     #   Krylov.@kref!(n, x, y, c, s)
     # end
 
+    @testset "vector_to_matrix" begin
+      S = MtlVector{FC}
+      M = Krylov.vector_to_matrix(S)
+      @test M == MtlMatrix{FC}
+    end
+
     ε = eps(T)
     atol = √ε
     rtol = √ε

test/gpu/nvidia.jl

@@ -3,8 +3,6 @@ using LinearOperators, Krylov, CUDA, CUDA.CUSPARSE, CUDA.CUSOLVER
 
 include("../test_utils.jl")
 
-include("../test_utils.jl")
-
 @testset "Nvidia -- CUDA.jl" begin
 
   @test CUDA.functional()
@@ -146,6 +144,12 @@ include("../test_utils.jl")
       Krylov.@kref!(n, x, y, c, s)
     end
 
+    @testset "vector_to_matrix" begin
+      S = CuVector{FC}
+      M = Krylov.vector_to_matrix(S)
+      @test M == CuMatrix{FC}
+    end
+
     ε = eps(T)
     atol = √ε
     rtol = √ε

test/test_aux.jl

@@ -134,6 +134,15 @@
     @test Krylov.ktypeof(b) == Vector{Float64}
   end
 
+  @testset "vector_to_matrix" begin
+    # test vector_to_matrix
+    for FC in (Float32, Float64, ComplexF32, ComplexF64)
+      S = Vector{FC}
+      M = Krylov.vector_to_matrix(S)
+      @test M == Matrix{FC}
+    end
+  end
+
   @testset "macros" begin
     # test macros
     for FC ∈ (Float16, Float32, Float64, ComplexF16, ComplexF32, ComplexF64)