@@ -848,20 +848,20 @@ end
848
848
# mutable is needed to correctly implement add_gid!
849
849
mutable struct PRange{A,B} <: AbstractUnitRange{Int}
850
850
ngids:: Int
851
- lids :: A
851
+ partition :: A
852
852
ghost:: Bool
853
853
exchanger:: B
854
854
function PRange (
855
855
ngids:: Integer ,
856
- lids :: PData{<:IndexSet} ,
856
+ partition :: PData{<:IndexSet} ,
857
857
ghost:: Bool ,
858
858
exchanger:: Exchanger )
859
859
860
- A = typeof (lids )
860
+ A = typeof (partition )
861
861
B = typeof (exchanger)
862
862
new {A,B} (
863
863
ngids,
864
- lids ,
864
+ partition ,
865
865
ghost,
866
866
exchanger)
867
867
end
@@ -870,29 +870,29 @@ end
870
870
function Base. copy (a:: PRange )
871
871
PRange (
872
872
copy (a. ngids),
873
- copy (a. lids ),
873
+ copy (a. partition ),
874
874
copy (a. ghost),
875
875
copy (a. exchanger))
876
876
end
877
877
878
878
function PRange (
879
879
ngids:: Integer ,
880
- lids :: PData{<:IndexSet} ,
880
+ partition :: PData{<:IndexSet} ,
881
881
ghost:: Bool = true )
882
882
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)
885
885
end
886
886
887
887
Base. first (a:: PRange ) = 1
888
888
Base. last (a:: PRange ) = a. ngids
889
889
890
890
num_gids (a:: PRange ) = a. ngids
891
- num_parts (a:: PRange ) = num_parts (a. lids )
891
+ num_parts (a:: PRange ) = num_parts (a. partition )
892
892
893
893
function PRange (parts:: PData{<:Integer} ,ngids:: Integer )
894
894
np = num_parts (parts)
895
- lids = map_parts (parts) do part
895
+ partition = map_parts (parts) do part
896
896
gids = _oid_to_gid (ngids,np,part)
897
897
lid_to_gid = collect (gids)
898
898
lid_to_part = fill (part,length (gids))
@@ -910,7 +910,7 @@ function PRange(parts::PData{<:Integer},ngids::Integer)
910
910
hid_to_lid)
911
911
end
912
912
ghost = false
913
- PRange (ngids,lids ,ghost)
913
+ PRange (ngids,partition ,ghost)
914
914
end
915
915
916
916
function PRange (
922
922
function PRange (parts:: PData{<:Integer} ,noids:: PData{<:Integer} )
923
923
ngids = reduce (+ ,noids,init= 0 )
924
924
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
926
926
lid_to_gid = collect (Int,(1 : noids) .+ (firstgids[part]- 1 ))
927
927
lid_to_part = fill (part,length (lid_to_gid))
928
928
gid_to_part = GidToPart (ngids,firstgids)
@@ -938,15 +938,15 @@ function PRange(parts::PData{<:Integer},noids::PData{<:Integer})
938
938
hid_to_lid)
939
939
end
940
940
ghost = false
941
- PRange (ngids,lids ,ghost)
941
+ PRange (ngids,partition ,ghost)
942
942
end
943
943
944
944
function PRange (
945
945
parts:: PData{<:Integer,N} ,
946
946
ngids:: NTuple{N,<:Integer} ) where N
947
947
948
948
np = size (parts)
949
- lids = map_parts (parts) do part
949
+ partition = map_parts (parts) do part
950
950
gids = _oid_to_gid (ngids,np,part)
951
951
lid_to_gid = gids
952
952
lid_to_part = fill (part,length (gids))
@@ -964,7 +964,7 @@ function PRange(
964
964
hid_to_lid)
965
965
end
966
966
ghost = false
967
- PRange (prod (ngids),lids ,ghost)
967
+ PRange (prod (ngids),partition ,ghost)
968
968
end
969
969
970
970
function PCartesianIndices (
@@ -993,7 +993,7 @@ function PRange(
993
993
:: WithGhost ) where N
994
994
995
995
np = size (parts)
996
- lids = map_parts (parts) do part
996
+ partition = map_parts (parts) do part
997
997
cp = Tuple (CartesianIndices (np)[part])
998
998
d_to_ldid_to_gdid = map (_lid_to_gid,ngids,np,cp)
999
999
lid_to_gid = _id_tensor_product (Int,d_to_ldid_to_gdid,ngids)
@@ -1013,7 +1013,7 @@ function PRange(
1013
1013
hid_to_lid)
1014
1014
end
1015
1015
ghost = true
1016
- PRange (prod (ngids),lids ,ghost)
1016
+ PRange (prod (ngids),partition ,ghost)
1017
1017
end
1018
1018
1019
1019
function PRange (
@@ -1208,54 +1208,54 @@ function _find_part_from_gid(
1208
1208
end
1209
1209
1210
1210
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
1212
1212
for gid in gids
1213
- add_gid! (lids ,gid)
1213
+ add_gid! (partition ,gid)
1214
1214
end
1215
1215
end
1216
- a. exchanger = Exchanger (a. lids )
1216
+ a. exchanger = Exchanger (a. partition )
1217
1217
a. ghost = true
1218
1218
a
1219
1219
end
1220
1220
1221
1221
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 )
1224
1224
add_gid! (b,gids)
1225
1225
b
1226
1226
end
1227
1227
1228
1228
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 )
1230
1230
end
1231
1231
1232
1232
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 )
1234
1234
end
1235
1235
1236
1236
function oids_are_equal (a:: PRange ,b:: PRange )
1237
- if a. lids === b. lids
1237
+ if a. partition === b. partition
1238
1238
true
1239
1239
else
1240
- c = map_parts (oids_are_equal,a. lids ,b. lids )
1240
+ c = map_parts (oids_are_equal,a. partition ,b. partition )
1241
1241
reduce (& ,c,init= true )
1242
1242
end
1243
1243
end
1244
1244
1245
1245
function hids_are_equal (a:: PRange ,b:: PRange )
1246
- if a. lids === b. lids
1246
+ if a. partition === b. partition
1247
1247
true
1248
1248
else
1249
- c = map_parts (hids_are_equal,a. lids ,b. lids )
1249
+ c = map_parts (hids_are_equal,a. partition ,b. partition )
1250
1250
reduce (& ,c,init= true )
1251
1251
end
1252
1252
end
1253
1253
1254
1254
function lids_are_equal (a:: PRange ,b:: PRange )
1255
- if a. lids === b. lids
1255
+ if a. partition === b. partition
1256
1256
true
1257
1257
else
1258
- c = map_parts (lids_are_equal,a. lids ,b. lids )
1258
+ c = map_parts (lids_are_equal,a. partition ,b. partition )
1259
1259
reduce (& ,c,init= true )
1260
1260
end
1261
1261
end
@@ -1275,11 +1275,11 @@ end
1275
1275
1276
1276
function Base. getproperty (x:: PVector , sym:: Symbol )
1277
1277
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
1279
1279
view (v,r. oid_to_lid)
1280
1280
end
1281
1281
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
1283
1283
view (v,r. hid_to_lid)
1284
1284
end
1285
1285
else
@@ -1313,8 +1313,8 @@ end
1313
1313
1314
1314
function Base. similar (a:: PVector ,:: Type{T} ,axes:: Tuple{<:PRange} ) where T
1315
1315
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 ))
1318
1318
end
1319
1319
PVector (values,rows)
1320
1320
end
@@ -1327,8 +1327,8 @@ end
1327
1327
function Base. similar (
1328
1328
:: Type{<:PVector{T,<:PData{A}}} ,axes:: Tuple{<:PRange} ) where {T,A}
1329
1329
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 ))
1332
1332
end
1333
1333
PVector (values,rows)
1334
1334
end
@@ -1439,8 +1439,8 @@ function PVector{T}(
1439
1439
:: UndefInitializer ,
1440
1440
rows:: PRange ) where T
1441
1441
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 )
1444
1444
Vector {T} (undef,nlids)
1445
1445
end
1446
1446
PVector (values,rows)
@@ -1465,8 +1465,8 @@ function PVector(
1465
1465
to_lid! (I,rows)
1466
1466
end
1467
1467
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 ))
1470
1470
fill! (values,zero (eltype (values)))
1471
1471
for i in 1 : length (I)
1472
1472
lid = I[i]
@@ -1536,8 +1536,8 @@ function Base.fill!(a::PVector,v)
1536
1536
end
1537
1537
1538
1538
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)
1541
1541
reduce (op,owned_values,init= init)
1542
1542
end
1543
1543
reduce (op,b,init= init)
@@ -1548,7 +1548,7 @@ function Base.sum(a::PVector)
1548
1548
end
1549
1549
1550
1550
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
1552
1552
a_owned = view (a,alids. oid_to_lid)
1553
1553
b_owned = view (b,blids. oid_to_lid)
1554
1554
dot (a_owned,b_owned)
@@ -1561,8 +1561,8 @@ function local_view(a::PVector)
1561
1561
end
1562
1562
1563
1563
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,))
1566
1566
end
1567
1567
end
1568
1568
@@ -1613,10 +1613,10 @@ function async_assemble!(
1613
1613
exchanger_rcv = a. rows. exchanger # receives data at ghost ids from remote parts
1614
1614
exchanger_snd = reverse (exchanger_rcv) # sends data at ghost ids to remote parts
1615
1615
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
1617
1617
@task begin
1618
1618
wait (schedule (t1))
1619
- values[lids . hid_to_lid] .= zero (eltype (values))
1619
+ values[partition . hid_to_lid] .= zero (eltype (values))
1620
1620
end
1621
1621
end
1622
1622
end
@@ -1662,37 +1662,37 @@ function PSparseMatrix(
1662
1662
cols:: PRange ) where T
1663
1663
1664
1664
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 )
1666
1666
else
1667
- exchanger = empty_exchanger (rows. lids )
1667
+ exchanger = empty_exchanger (rows. partition )
1668
1668
end
1669
1669
PSparseMatrix (values,rows,cols,exchanger)
1670
1670
end
1671
1671
1672
1672
function Base. getproperty (x:: PSparseMatrix , sym:: Symbol )
1673
1673
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
1675
1675
indices = (r. oid_to_lid,c. oid_to_lid)
1676
1676
inv_indices = (r. lid_to_ohid,c. lid_to_ohid)
1677
1677
flag = (1 ,1 )
1678
1678
SubSparseMatrix (v,indices,inv_indices,flag)
1679
1679
end
1680
1680
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
1682
1682
indices = (r. oid_to_lid,c. hid_to_lid)
1683
1683
inv_indices = (r. lid_to_ohid,c. lid_to_ohid)
1684
1684
flag = (1 ,- 1 )
1685
1685
SubSparseMatrix (v,indices,inv_indices,flag)
1686
1686
end
1687
1687
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
1689
1689
indices = (r. hid_to_lid,c. oid_to_lid)
1690
1690
inv_indices = (r. lid_to_ohid,c. lid_to_ohid)
1691
1691
flag = (- 1 ,1 )
1692
1692
SubSparseMatrix (v,indices,inv_indices,flag)
1693
1693
end
1694
1694
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
1696
1696
indices = (r. hid_to_lid,c. hid_to_lid)
1697
1697
inv_indices = (r. lid_to_ohid,c. lid_to_ohid)
1698
1698
flag = (- 1 ,- 1 )
@@ -1736,7 +1736,7 @@ function PSparseMatrix(
1736
1736
to_lid! (J,cols)
1737
1737
end
1738
1738
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
1740
1740
init (I,J,V,num_lids (rlids),num_lids (clids))
1741
1741
end
1742
1742
@@ -1877,7 +1877,7 @@ function local_view(a::PSparseMatrix)
1877
1877
end
1878
1878
1879
1879
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
1881
1881
GlobalView (values,(rlids. gid_to_lid,clids. gid_to_lid),(rlids. ngids,clids. ngids))
1882
1882
end
1883
1883
end
@@ -2054,7 +2054,7 @@ function async_assemble!(
2054
2054
2055
2055
map_parts (wait∘ schedule,t0)
2056
2056
2057
- part = get_part_ids (rows. lids )
2057
+ part = get_part_ids (rows. partition )
2058
2058
parts_rcv = rows. exchanger. parts_rcv
2059
2059
parts_snd = rows. exchanger. parts_snd
2060
2060
@@ -2098,7 +2098,7 @@ function async_assemble!(
2098
2098
gi_rcv, gj_rcv, v_rcv
2099
2099
end
2100
2100
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)
2102
2102
2103
2103
gi_snd, t1 = async_exchange (gi_rcv,parts_snd,parts_rcv)
2104
2104
gj_snd, t2 = async_exchange (gj_rcv,parts_snd,parts_rcv)
@@ -2123,7 +2123,7 @@ function async_assemble!(
2123
2123
end
2124
2124
end
2125
2125
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)
2127
2127
2128
2128
t4
2129
2129
end
@@ -2178,7 +2178,7 @@ struct Jacobi{T,A,B,C}
2178
2178
end
2179
2179
2180
2180
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
2182
2182
@assert num_oids (rlids) == num_oids (clids)
2183
2183
@notimplementedif rlids. oid_to_lid != clids. oid_to_lid
2184
2184
ldiag = collect (diag (values))
0 commit comments