Skip to content

Commit 57ad819

Browse files
committed
Renamed lids -> partition
1 parent e82bc60 commit 57ad819

File tree

4 files changed

+106
-106
lines changed

4 files changed

+106
-106
lines changed

src/Interfaces.jl

+59-59
Original file line numberDiff line numberDiff line change
@@ -848,20 +848,20 @@ end
848848
# mutable is needed to correctly implement add_gid!
849849
mutable struct PRange{A,B} <: AbstractUnitRange{Int}
850850
ngids::Int
851-
lids::A
851+
partition::A
852852
ghost::Bool
853853
exchanger::B
854854
function PRange(
855855
ngids::Integer,
856-
lids::PData{<:IndexSet},
856+
partition::PData{<:IndexSet},
857857
ghost::Bool,
858858
exchanger::Exchanger)
859859

860-
A = typeof(lids)
860+
A = typeof(partition)
861861
B = typeof(exchanger)
862862
new{A,B}(
863863
ngids,
864-
lids,
864+
partition,
865865
ghost,
866866
exchanger)
867867
end
@@ -870,29 +870,29 @@ end
870870
function Base.copy(a::PRange)
871871
PRange(
872872
copy(a.ngids),
873-
copy(a.lids),
873+
copy(a.partition),
874874
copy(a.ghost),
875875
copy(a.exchanger))
876876
end
877877

878878
function PRange(
879879
ngids::Integer,
880-
lids::PData{<:IndexSet},
880+
partition::PData{<:IndexSet},
881881
ghost::Bool=true)
882882

883-
exchanger = ghost ? Exchanger(lids) : empty_exchanger(lids)
884-
PRange(ngids,lids,ghost,exchanger)
883+
exchanger = ghost ? Exchanger(partition) : empty_exchanger(partition)
884+
PRange(ngids,partition,ghost,exchanger)
885885
end
886886

887887
Base.first(a::PRange) = 1
888888
Base.last(a::PRange) = a.ngids
889889

890890
num_gids(a::PRange) = a.ngids
891-
num_parts(a::PRange) = num_parts(a.lids)
891+
num_parts(a::PRange) = num_parts(a.partition)
892892

893893
function PRange(parts::PData{<:Integer},ngids::Integer)
894894
np = num_parts(parts)
895-
lids = map_parts(parts) do part
895+
partition = map_parts(parts) do part
896896
gids = _oid_to_gid(ngids,np,part)
897897
lid_to_gid = collect(gids)
898898
lid_to_part = fill(part,length(gids))
@@ -910,7 +910,7 @@ function PRange(parts::PData{<:Integer},ngids::Integer)
910910
hid_to_lid)
911911
end
912912
ghost = false
913-
PRange(ngids,lids,ghost)
913+
PRange(ngids,partition,ghost)
914914
end
915915

916916
function PRange(
@@ -922,7 +922,7 @@ end
922922
function PRange(parts::PData{<:Integer},noids::PData{<:Integer})
923923
ngids = reduce(+,noids,init=0)
924924
firstgids = xscan_all(+,noids,init=1)
925-
lids = map_parts(parts,noids,firstgids) do part,noids,firstgids
925+
partition = map_parts(parts,noids,firstgids) do part,noids,firstgids
926926
lid_to_gid = collect(Int,(1:noids) .+ (firstgids[part]-1))
927927
lid_to_part = fill(part,length(lid_to_gid))
928928
gid_to_part = GidToPart(ngids,firstgids)
@@ -938,15 +938,15 @@ function PRange(parts::PData{<:Integer},noids::PData{<:Integer})
938938
hid_to_lid)
939939
end
940940
ghost = false
941-
PRange(ngids,lids,ghost)
941+
PRange(ngids,partition,ghost)
942942
end
943943

944944
function PRange(
945945
parts::PData{<:Integer,N},
946946
ngids::NTuple{N,<:Integer}) where N
947947

948948
np = size(parts)
949-
lids = map_parts(parts) do part
949+
partition = map_parts(parts) do part
950950
gids = _oid_to_gid(ngids,np,part)
951951
lid_to_gid = gids
952952
lid_to_part = fill(part,length(gids))
@@ -964,7 +964,7 @@ function PRange(
964964
hid_to_lid)
965965
end
966966
ghost = false
967-
PRange(prod(ngids),lids,ghost)
967+
PRange(prod(ngids),partition,ghost)
968968
end
969969

970970
function PCartesianIndices(
@@ -993,7 +993,7 @@ function PRange(
993993
::WithGhost) where N
994994

995995
np = size(parts)
996-
lids = map_parts(parts) do part
996+
partition = map_parts(parts) do part
997997
cp = Tuple(CartesianIndices(np)[part])
998998
d_to_ldid_to_gdid = map(_lid_to_gid,ngids,np,cp)
999999
lid_to_gid = _id_tensor_product(Int,d_to_ldid_to_gdid,ngids)
@@ -1013,7 +1013,7 @@ function PRange(
10131013
hid_to_lid)
10141014
end
10151015
ghost = true
1016-
PRange(prod(ngids),lids,ghost)
1016+
PRange(prod(ngids),partition,ghost)
10171017
end
10181018

10191019
function PRange(
@@ -1208,54 +1208,54 @@ function _find_part_from_gid(
12081208
end
12091209

12101210
function add_gid!(a::PRange,gids::PData{<:AbstractArray{<:Integer}})
1211-
map_parts(a.lids,gids) do lids,gids
1211+
map_parts(a.partition,gids) do partition,gids
12121212
for gid in gids
1213-
add_gid!(lids,gid)
1213+
add_gid!(partition,gid)
12141214
end
12151215
end
1216-
a.exchanger = Exchanger(a.lids)
1216+
a.exchanger = Exchanger(a.partition)
12171217
a.ghost = true
12181218
a
12191219
end
12201220

12211221
function add_gid(a::PRange,gids::PData{<:AbstractArray{<:Integer}})
1222-
lids = map_parts(copy,a.lids)
1223-
b = PRange(a.ngids,lids)
1222+
partition = map_parts(copy,a.partition)
1223+
b = PRange(a.ngids,partition)
12241224
add_gid!(b,gids)
12251225
b
12261226
end
12271227

12281228
function to_lid!(ids::PData{<:AbstractArray{<:Integer}},a::PRange)
1229-
map_parts(to_lid!,ids,a.lids)
1229+
map_parts(to_lid!,ids,a.partition)
12301230
end
12311231

12321232
function to_gid!(ids::PData{<:AbstractArray{<:Integer}},a::PRange)
1233-
map_parts(to_gid!,ids,a.lids)
1233+
map_parts(to_gid!,ids,a.partition)
12341234
end
12351235

12361236
function oids_are_equal(a::PRange,b::PRange)
1237-
if a.lids === b.lids
1237+
if a.partition === b.partition
12381238
true
12391239
else
1240-
c = map_parts(oids_are_equal,a.lids,b.lids)
1240+
c = map_parts(oids_are_equal,a.partition,b.partition)
12411241
reduce(&,c,init=true)
12421242
end
12431243
end
12441244

12451245
function hids_are_equal(a::PRange,b::PRange)
1246-
if a.lids === b.lids
1246+
if a.partition === b.partition
12471247
true
12481248
else
1249-
c = map_parts(hids_are_equal,a.lids,b.lids)
1249+
c = map_parts(hids_are_equal,a.partition,b.partition)
12501250
reduce(&,c,init=true)
12511251
end
12521252
end
12531253

12541254
function lids_are_equal(a::PRange,b::PRange)
1255-
if a.lids === b.lids
1255+
if a.partition === b.partition
12561256
true
12571257
else
1258-
c = map_parts(lids_are_equal,a.lids,b.lids)
1258+
c = map_parts(lids_are_equal,a.partition,b.partition)
12591259
reduce(&,c,init=true)
12601260
end
12611261
end
@@ -1275,11 +1275,11 @@ end
12751275

12761276
function Base.getproperty(x::PVector, sym::Symbol)
12771277
if sym == :owned_values
1278-
map_parts(x.values,x.rows.lids) do v,r
1278+
map_parts(x.values,x.rows.partition) do v,r
12791279
view(v,r.oid_to_lid)
12801280
end
12811281
elseif sym == :ghost_values
1282-
map_parts(x.values,x.rows.lids) do v,r
1282+
map_parts(x.values,x.rows.partition) do v,r
12831283
view(v,r.hid_to_lid)
12841284
end
12851285
else
@@ -1313,8 +1313,8 @@ end
13131313

13141314
function Base.similar(a::PVector,::Type{T},axes::Tuple{<:PRange}) where T
13151315
rows = axes[1]
1316-
values = map_parts(a.values,rows.lids) do values, lids
1317-
similar(values,T,num_lids(lids))
1316+
values = map_parts(a.values,rows.partition) do values, partition
1317+
similar(values,T,num_lids(partition))
13181318
end
13191319
PVector(values,rows)
13201320
end
@@ -1327,8 +1327,8 @@ end
13271327
function Base.similar(
13281328
::Type{<:PVector{T,<:PData{A}}},axes::Tuple{<:PRange}) where {T,A}
13291329
rows = axes[1]
1330-
values = map_parts(rows.lids) do lids
1331-
similar(A,num_lids(lids))
1330+
values = map_parts(rows.partition) do partition
1331+
similar(A,num_lids(partition))
13321332
end
13331333
PVector(values,rows)
13341334
end
@@ -1439,8 +1439,8 @@ function PVector{T}(
14391439
::UndefInitializer,
14401440
rows::PRange) where T
14411441

1442-
values = map_parts(rows.lids) do lids
1443-
nlids = num_lids(lids)
1442+
values = map_parts(rows.partition) do partition
1443+
nlids = num_lids(partition)
14441444
Vector{T}(undef,nlids)
14451445
end
14461446
PVector(values,rows)
@@ -1465,8 +1465,8 @@ function PVector(
14651465
to_lid!(I,rows)
14661466
end
14671467

1468-
values = map_parts(rows.lids,I,V) do lids,I,V
1469-
values = init(num_lids(lids))
1468+
values = map_parts(rows.partition,I,V) do partition,I,V
1469+
values = init(num_lids(partition))
14701470
fill!(values,zero(eltype(values)))
14711471
for i in 1:length(I)
14721472
lid = I[i]
@@ -1536,8 +1536,8 @@ function Base.fill!(a::PVector,v)
15361536
end
15371537

15381538
function Base.reduce(op,a::PVector;init)
1539-
b = map_parts(a.values,a.rows.lids) do values,lids
1540-
owned_values = view(values,lids.oid_to_lid)
1539+
b = map_parts(a.values,a.rows.partition) do values,partition
1540+
owned_values = view(values,partition.oid_to_lid)
15411541
reduce(op,owned_values,init=init)
15421542
end
15431543
reduce(op,b,init=init)
@@ -1548,7 +1548,7 @@ function Base.sum(a::PVector)
15481548
end
15491549

15501550
function LinearAlgebra.dot(a::PVector,b::PVector)
1551-
c = map_parts(a.values,b.values,a.rows.lids,b.rows.lids) do a,b,alids,blids
1551+
c = map_parts(a.values,b.values,a.rows.partition,b.rows.partition) do a,b,alids,blids
15521552
a_owned = view(a,alids.oid_to_lid)
15531553
b_owned = view(b,blids.oid_to_lid)
15541554
dot(a_owned,b_owned)
@@ -1561,8 +1561,8 @@ function local_view(a::PVector)
15611561
end
15621562

15631563
function global_view(a::PVector)
1564-
map_parts(a.values,a.rows.lids) do values, lids
1565-
GlobalView(values,(lids.gid_to_lid,),(lids.ngids,))
1564+
map_parts(a.values,a.rows.partition) do values, partition
1565+
GlobalView(values,(partition.gid_to_lid,),(partition.ngids,))
15661566
end
15671567
end
15681568

@@ -1613,10 +1613,10 @@ function async_assemble!(
16131613
exchanger_rcv = a.rows.exchanger # receives data at ghost ids from remote parts
16141614
exchanger_snd = reverse(exchanger_rcv) # sends data at ghost ids to remote parts
16151615
t1 = async_exchange!(combine_op,a.values,exchanger_snd,t0)
1616-
map_parts(t1,a.values,a.rows.lids) do t1,values,lids
1616+
map_parts(t1,a.values,a.rows.partition) do t1,values,partition
16171617
@task begin
16181618
wait(schedule(t1))
1619-
values[lids.hid_to_lid] .= zero(eltype(values))
1619+
values[partition.hid_to_lid] .= zero(eltype(values))
16201620
end
16211621
end
16221622
end
@@ -1662,37 +1662,37 @@ function PSparseMatrix(
16621662
cols::PRange) where T
16631663

16641664
if rows.ghost
1665-
exchanger = matrix_exchanger(values,rows.exchanger,rows.lids,cols.lids)
1665+
exchanger = matrix_exchanger(values,rows.exchanger,rows.partition,cols.partition)
16661666
else
1667-
exchanger = empty_exchanger(rows.lids)
1667+
exchanger = empty_exchanger(rows.partition)
16681668
end
16691669
PSparseMatrix(values,rows,cols,exchanger)
16701670
end
16711671

16721672
function Base.getproperty(x::PSparseMatrix, sym::Symbol)
16731673
if sym == :owned_owned_values
1674-
map_parts(x.values,x.rows.lids,x.cols.lids) do v,r,c
1674+
map_parts(x.values,x.rows.partition,x.cols.partition) do v,r,c
16751675
indices = (r.oid_to_lid,c.oid_to_lid)
16761676
inv_indices = (r.lid_to_ohid,c.lid_to_ohid)
16771677
flag = (1,1)
16781678
SubSparseMatrix(v,indices,inv_indices,flag)
16791679
end
16801680
elseif sym == :owned_ghost_values
1681-
map_parts(x.values,x.rows.lids,x.cols.lids) do v,r,c
1681+
map_parts(x.values,x.rows.partition,x.cols.partition) do v,r,c
16821682
indices = (r.oid_to_lid,c.hid_to_lid)
16831683
inv_indices = (r.lid_to_ohid,c.lid_to_ohid)
16841684
flag = (1,-1)
16851685
SubSparseMatrix(v,indices,inv_indices,flag)
16861686
end
16871687
elseif sym == :ghost_owned_values
1688-
map_parts(x.values,x.rows.lids,x.cols.lids) do v,r,c
1688+
map_parts(x.values,x.rows.partition,x.cols.partition) do v,r,c
16891689
indices = (r.hid_to_lid,c.oid_to_lid)
16901690
inv_indices = (r.lid_to_ohid,c.lid_to_ohid)
16911691
flag = (-1,1)
16921692
SubSparseMatrix(v,indices,inv_indices,flag)
16931693
end
16941694
elseif sym == :ghost_ghost_values
1695-
map_parts(x.values,x.rows.lids,x.cols.lids) do v,r,c
1695+
map_parts(x.values,x.rows.partition,x.cols.partition) do v,r,c
16961696
indices = (r.hid_to_lid,c.hid_to_lid)
16971697
inv_indices = (r.lid_to_ohid,c.lid_to_ohid)
16981698
flag = (-1,-1)
@@ -1736,7 +1736,7 @@ function PSparseMatrix(
17361736
to_lid!(J,cols)
17371737
end
17381738

1739-
values = map_parts(I,J,V,rows.lids,cols.lids) do I,J,V,rlids,clids
1739+
values = map_parts(I,J,V,rows.partition,cols.partition) do I,J,V,rlids,clids
17401740
init(I,J,V,num_lids(rlids),num_lids(clids))
17411741
end
17421742

@@ -1877,7 +1877,7 @@ function local_view(a::PSparseMatrix)
18771877
end
18781878

18791879
function global_view(a::PSparseMatrix)
1880-
map_parts(a.values,a.rows.lids,a.cols.lids) do values,rlids,clids
1880+
map_parts(a.values,a.rows.partition,a.cols.partition) do values,rlids,clids
18811881
GlobalView(values,(rlids.gid_to_lid,clids.gid_to_lid),(rlids.ngids,clids.ngids))
18821882
end
18831883
end
@@ -2054,7 +2054,7 @@ function async_assemble!(
20542054

20552055
map_parts(waitschedule,t0)
20562056

2057-
part = get_part_ids(rows.lids)
2057+
part = get_part_ids(rows.partition)
20582058
parts_rcv = rows.exchanger.parts_rcv
20592059
parts_snd = rows.exchanger.parts_snd
20602060

@@ -2098,7 +2098,7 @@ function async_assemble!(
20982098
gi_rcv, gj_rcv, v_rcv
20992099
end
21002100

2101-
gi_rcv, gj_rcv, v_rcv = map_parts(setup_rcv,part,parts_rcv,rows.lids,coo_values)
2101+
gi_rcv, gj_rcv, v_rcv = map_parts(setup_rcv,part,parts_rcv,rows.partition,coo_values)
21022102

21032103
gi_snd, t1 = async_exchange(gi_rcv,parts_snd,parts_rcv)
21042104
gj_snd, t2 = async_exchange(gj_rcv,parts_snd,parts_rcv)
@@ -2123,7 +2123,7 @@ function async_assemble!(
21232123
end
21242124
end
21252125

2126-
t4 = map_parts(setup_snd,t1,t2,t3,part,rows.lids,gi_snd,gj_snd,v_snd,coo_values)
2126+
t4 = map_parts(setup_snd,t1,t2,t3,part,rows.partition,gi_snd,gj_snd,v_snd,coo_values)
21272127

21282128
t4
21292129
end
@@ -2178,7 +2178,7 @@ struct Jacobi{T,A,B,C}
21782178
end
21792179

21802180
function Jacobi(a::PSparseMatrix)
2181-
diaginv = map_parts(a.values,a.rows.lids,a.cols.lids) do values, rlids, clids
2181+
diaginv = map_parts(a.values,a.rows.partition,a.cols.partition) do values, rlids, clids
21822182
@assert num_oids(rlids) == num_oids(clids)
21832183
@notimplementedif rlids.oid_to_lid != clids.oid_to_lid
21842184
ldiag = collect(diag(values))

0 commit comments

Comments
 (0)