Update for HPCBackend type parameter refactoring

- Adapt to HPCBackend{T,Ti,Device,Comm,Solver} API changes
- Update backend instance mapping for new parameterization
- Update all tests and tools for new backend constructor

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>

Author: Sebastien Loisel

Project.toml

@@ -20,7 +20,7 @@ StaticArrays = "90137ffa-7385-5640-81b9-e52037218182"
 Statistics = "10745b16-79ce-11e8-11f9-7d13ad32a3b2"
 
 [sources]
-HPCLinearAlgebra = {path = "../LinearAlgebraMPI.jl"}
+HPCLinearAlgebra = {path = "../HPCLinearAlgebra.jl"}
 
 [compat]
 BenchmarkTools = "1.6"

examples/basic_solve.jl

@@ -15,7 +15,7 @@ using MPI
 MPI.Init()
 
 using MultiGridBarrierMPI
-using LinearAlgebraMPI
+using HPCLinearAlgebra
 using MultiGridBarrier
 using LinearAlgebra
 

examples/roundtrip_conversion.jl

@@ -16,7 +16,7 @@ using MPI
 MPI.Init()
 
 using MultiGridBarrierMPI
-using LinearAlgebraMPI
+using HPCLinearAlgebra
 using MultiGridBarrier
 using LinearAlgebra
 using SparseArrays

src/MultiGridBarrierMPI.jl

@@ -9,8 +9,8 @@ end
 using MultiGridBarrierMPI
 MultiGridBarrierMPI.Init()
 
-using LinearAlgebraMPI
-using LinearAlgebraMPI: VectorMPI, MatrixMPI, SparseMatrixMPI, io0
+using HPCLinearAlgebra
+using HPCLinearAlgebra: HPCVector, HPCMatrix, HPCSparseMatrix, io0
 using LinearAlgebra
 using SparseArrays
 using MultiGridBarrier
@@ -86,7 +86,7 @@ println(io0(), "[DEBUG] R_finest size: $(size(R_finest))")
 
 # Create a test vector with size ncols(R_finest)
 n_coarse = size(R_finest, 2)
-z_test = VectorMPI(ones(n_coarse))
+z_test = HPCVector(ones(n_coarse))
 println(io0(), "[DEBUG] z_test partition: $(z_test.partition)")
 
 # Compute R * z
@@ -105,7 +105,7 @@ println(io0(), "[DEBUG] \n--- Testing full Hessian assembly ---")
 # Let's simulate this
 w = g.w
 w_scaled = 2.0 .* Vector(w)  # Example scaling
-W_diag = spdiagm(length(w), length(w), 0 => VectorMPI(w_scaled))
+W_diag = spdiagm(length(w), length(w), 0 => HPCVector(w_scaled))
 
 println(io0(), "[DEBUG] W_diag size: $(size(W_diag))")
 println(io0(), "[DEBUG] W_diag row_partition: $(W_diag.row_partition)")
@@ -148,7 +148,7 @@ end
 
 # Try factorization
 println(io0(), "[DEBUG] Attempting to factorize...")
-b_test = VectorMPI(ones(size(Rt_Dt_W_D_R, 1)))
+b_test = HPCVector(ones(size(Rt_Dt_W_D_R, 1)))
 try
     x_test = Rt_Dt_W_D_R \ b_test
     println(io0(), "[DEBUG] Factorization succeeded!")

test/test_amg_structure.jl

@@ -9,8 +9,8 @@ end
 using MultiGridBarrierMPI
 MultiGridBarrierMPI.Init()
 
-using LinearAlgebraMPI
-using LinearAlgebraMPI: VectorMPI, MatrixMPI, SparseMatrixMPI, io0
+using HPCLinearAlgebra
+using HPCLinearAlgebra: HPCVector, HPCMatrix, HPCSparseMatrix, io0
 using LinearAlgebra
 using SparseArrays
 using MultiGridBarrier
@@ -30,7 +30,7 @@ D_op = [g.operators[:dx], g.operators[:id]]  # dx and id operators
 # Create a test vector (should be size 16)
 n = length(g.w)
 z_native = sin.(range(0, π, length=n))
-z_mpi = VectorMPI(z_native)
+z_mpi = HPCVector(z_native)
 
 println(io0(), "[DEBUG] z size: $(length(z_mpi))")
 println(io0(), "[DEBUG] z partition: $(z_mpi.partition)")

test/test_apply_d.jl

@@ -9,8 +9,8 @@ end
 using MultiGridBarrierMPI
 MultiGridBarrierMPI.Init()
 
-using LinearAlgebraMPI
-using LinearAlgebraMPI: VectorMPI, MatrixMPI, SparseMatrixMPI, io0
+using HPCLinearAlgebra
+using HPCLinearAlgebra: HPCVector, HPCMatrix, HPCSparseMatrix, io0
 using LinearAlgebra
 using SparseArrays
 
@@ -23,13 +23,13 @@ if rank == 0
     flush(stdout)
 end
 
-# Test: Two SparseMatrixMPI multiplication
+# Test: Two HPCSparseMatrix multiplication
 A_native = sparse([1.0 0.0; 2.0 3.0; 0.0 4.0])  # 3x2
 B_native = sparse([1.0 2.0 3.0; 4.0 5.0 6.0])   # 2x3
 C_expected = A_native * B_native                  # 3x3
 
-A_mpi = SparseMatrixMPI{Float64}(A_native)
-B_mpi = SparseMatrixMPI{Float64}(B_native)
+A_mpi = HPCSparseMatrix{Float64}(A_native)
+B_mpi = HPCSparseMatrix{Float64}(B_native)
 
 if rank == 0
     println("[DEBUG] A size: $(size(A_mpi)), B size: $(size(B_mpi))")
@@ -71,10 +71,10 @@ end
 n = size(AtA_expected, 1)
 reg = 0.01 * sparse(I, n, n)
 AtA_reg = AtA_expected + reg
-AtA_reg_mpi = SparseMatrixMPI{Float64}(AtA_reg)
+AtA_reg_mpi = HPCSparseMatrix{Float64}(AtA_reg)
 
 b = ones(n)
-b_mpi = VectorMPI(b)
+b_mpi = HPCVector(b)
 
 if rank == 0
     println("[DEBUG] Attempting to solve (A'A + reg) \\ b...")

test/test_basic_ops.jl

@@ -9,8 +9,8 @@ end
 using MultiGridBarrierMPI
 MultiGridBarrierMPI.Init()
 
-using LinearAlgebraMPI
-using LinearAlgebraMPI: VectorMPI, MatrixMPI, SparseMatrixMPI, io0
+using HPCLinearAlgebra
+using HPCLinearAlgebra: HPCVector, HPCMatrix, HPCSparseMatrix, io0
 using LinearAlgebra
 using SparseArrays
 using MultiGridBarrier
@@ -28,7 +28,7 @@ g = fem1d_mpi(Float64; L=2)
 D_op = [g.operators[:dx], g.operators[:id]]
 n = length(g.w)
 z_native = sin.(range(0, π, length=n))
-z_mpi = VectorMPI(z_native)
+z_mpi = HPCVector(z_native)
 
 # Apply D operators to get matrix
 Dz_mpi = hcat([D * z_mpi for D in D_op]...)
@@ -50,10 +50,10 @@ println(io0(), "[DEBUG] Testing column extraction y[:, 1]...")
 col1 = y_mpi[:, 1]
 println(io0(), "[DEBUG] col1 type: $(typeof(col1))")
 println(io0(), "[DEBUG] col1 size: $(size(col1))")
-if col1 isa VectorMPI
+if col1 isa HPCVector
     println(io0(), "[DEBUG] col1 partition: $(col1.partition)")
 else
-    println(io0(), "[DEBUG] col1 is NOT a VectorMPI!")
+    println(io0(), "[DEBUG] col1 is NOT a HPCVector!")
 end
 
 # Test element-wise multiplication
@@ -64,7 +64,7 @@ println(io0(), "[DEBUG] w partition: $(w.partition)")
 try
     w_col = w .* col1
     println(io0(), "[DEBUG] w .* col1 type: $(typeof(w_col))")
-    if w_col isa VectorMPI
+    if w_col isa HPCVector
         println(io0(), "[DEBUG] w .* col1 partition: $(w_col.partition)")
     end
     println(io0(), "[DEBUG] Element-wise multiplication succeeded!")
@@ -80,7 +80,7 @@ try
     foo = MultiGridBarrier.amgb_diag(D, w .* col1)
     println(io0(), "[DEBUG] amgb_diag result type: $(typeof(foo))")
     println(io0(), "[DEBUG] amgb_diag result size: $(size(foo))")
-    if foo isa SparseMatrixMPI
+    if foo isa HPCSparseMatrix
         println(io0(), "[DEBUG] amgb_diag row_partition: $(foo.row_partition)")
         println(io0(), "[DEBUG] amgb_diag col_partition: $(foo.col_partition)")
     end
@@ -99,7 +99,7 @@ try
     foo = MultiGridBarrier.amgb_diag(D, w .* col1)
     bar = D' * foo * D
     println(io0(), "[DEBUG] D'*foo*D size: $(size(bar))")
-    if bar isa SparseMatrixMPI
+    if bar isa HPCSparseMatrix
         println(io0(), "[DEBUG] D'*foo*D row_partition: $(bar.row_partition)")
         println(io0(), "[DEBUG] D'*foo*D col_partition: $(bar.col_partition)")
     end
@@ -132,7 +132,7 @@ if rank == 0
     println("[DEBUG] y difference (MPI vs native): $(norm(y_mpi_native - y_native))")
 
     col1_native = y_native[:, 1]
-    col1_mpi_native = col1 isa VectorMPI ? Vector(col1) : col1
+    col1_mpi_native = col1 isa HPCVector ? Vector(col1) : col1
     println("[DEBUG] col1 difference: $(norm(col1_mpi_native - col1_native))")
 end
 

test/test_column_extract.jl

@@ -9,8 +9,8 @@ end
 using MultiGridBarrierMPI
 MultiGridBarrierMPI.Init()
 
-using LinearAlgebraMPI
-using LinearAlgebraMPI: VectorMPI, MatrixMPI, SparseMatrixMPI, io0
+using HPCLinearAlgebra
+using HPCLinearAlgebra: HPCVector, HPCMatrix, HPCSparseMatrix, io0
 using LinearAlgebra
 using SparseArrays
 using MultiGridBarrier
@@ -37,7 +37,7 @@ n = size(D_dx_mpi, 1)
 println(io0(), "[DEBUG] Testing hcat of sparse matrices...")
 
 # Create zero matrix
-Z_mpi = SparseMatrixMPI{Float64}(spzeros(Float64, n, n))
+Z_mpi = HPCSparseMatrix{Float64}(spzeros(Float64, n, n))
 Z_native = spzeros(Float64, n, n)
 
 # Test 1: hcat(Z, D_dx) - like foo = [Z; D_dx] in amg_helper
@@ -113,15 +113,15 @@ y22 = ones(n) * 0.3
 # MPI version
 H_mpi = nothing
 # j=1: dx'*diag(w*y11)*dx
-foo = spdiagm(n, n, 0 => w_mpi .* VectorMPI(y11))
+foo = spdiagm(n, n, 0 => w_mpi .* HPCVector(y11))
 H_mpi = D0_dx_mpi' * foo * D0_dx_mpi
 
 # j=2: id'*diag(w*y22)*id
-foo = spdiagm(n, n, 0 => w_mpi .* VectorMPI(y22))
+foo = spdiagm(n, n, 0 => w_mpi .* HPCVector(y22))
 H_mpi = H_mpi + D0_id_mpi' * foo * D0_id_mpi
 
 # Cross term: dx'*diag(w*y12)*id + id'*diag(w*y12)*dx
-foo = spdiagm(n, n, 0 => w_mpi .* VectorMPI(y12))
+foo = spdiagm(n, n, 0 => w_mpi .* HPCVector(y12))
 H_mpi = H_mpi + D0_dx_mpi' * foo * D0_id_mpi + D0_id_mpi' * foo * D0_dx_mpi
 
 # Native version

test/test_d0_construction.jl

@@ -9,8 +9,8 @@ end
 using MultiGridBarrierMPI
 MultiGridBarrierMPI.Init()
 
-using LinearAlgebraMPI
-using LinearAlgebraMPI: VectorMPI, MatrixMPI, SparseMatrixMPI, io0
+using HPCLinearAlgebra
+using HPCLinearAlgebra: HPCVector, HPCMatrix, HPCSparseMatrix, io0
 using LinearAlgebra
 using SparseArrays
 using MultiGridBarrier
@@ -26,7 +26,7 @@ end
 
 # Override solve to add diagnostics
 original_solve = MultiGridBarrier.solve
-function debug_solve(A::SparseMatrixMPI{T}, b::VectorMPI{T}) where T
+function debug_solve(A::HPCSparseMatrix{T}, b::HPCVector{T}) where T
     if rank == 0
         println("[DEBUG-SOLVE] Matrix size: $(size(A))")
         println("[DEBUG-SOLVE] Vector length: $(length(b))")