Browse Source

Type.

pull/4/head
termi-official 3 years ago
parent
commit
1e26d4781b
  1. 6
      src/HYPRE.jl

6
src/HYPRE.jl

@ -3,7 +3,7 @@
module HYPRE module HYPRE
using MPI: MPI using MPI: MPI
using PartitionedArrays: IndexRange, MPIData, PSparseMatrix, PVector, PartitionedArrays, using PartitionedArrays: AbstractIndexSet, MPIData, PSparseMatrix, PVector, PartitionedArrays,
SequentialData, map_parts SequentialData, map_parts
using SparseArrays: SparseArrays, SparseMatrixCSC, nnz, nonzeros, nzrange, rowvals using SparseArrays: SparseArrays, SparseMatrixCSC, nnz, nonzeros, nzrange, rowvals
using SparseMatricesCSR: SparseMatrixCSR, colvals, getrowptr using SparseMatricesCSR: SparseMatrixCSR, colvals, getrowptr
@ -305,7 +305,7 @@ end
################################################## ##################################################
# TODO: This has some duplicated code with to_hypre_data(::SparseMatrixCSC, ilower, iupper) # TODO: This has some duplicated code with to_hypre_data(::SparseMatrixCSC, ilower, iupper)
function Internals.to_hypre_data(A::SparseMatrixCSC, r::PartitionedArrays.AbstractIndexSet, c::PartitionedArrays.AbstractIndexSet) function Internals.to_hypre_data(A::SparseMatrixCSC, r::AbstractIndexSet, c::AbstractIndexSet)
# @assert r.oid_to_lid isa UnitRange && r.oid_to_lid.start == 1 # @assert r.oid_to_lid isa UnitRange && r.oid_to_lid.start == 1
lidlower = minimum(r.oid_to_lid) lidlower = minimum(r.oid_to_lid)
@ -363,7 +363,7 @@ end
# TODO: Possibly this can be optimized if it is possible to pass overlong vectors to HYPRE. # TODO: Possibly this can be optimized if it is possible to pass overlong vectors to HYPRE.
# At least values should be possible to directly share, but cols needs to translated # At least values should be possible to directly share, but cols needs to translated
# to global ids. # to global ids.
function Internals.to_hypre_data(A::SparseMatrixCSR, r::IndexRange, c::IndexRange) function Internals.to_hypre_data(A::SparseMatrixCSR, r::AbstractIndexSet, c::AbstractIndexSet)
#@assert r.oid_to_lid isa UnitRange && r.oid_to_lid.start == 1 #@assert r.oid_to_lid isa UnitRange && r.oid_to_lid.start == 1
lidlower = minimum(r.oid_to_lid) lidlower = minimum(r.oid_to_lid)

Loading…
Cancel
Save