Domain Map Standard Interface¶
Contents
Introduction¶
This document defines DSI – the Domain map Standard Interface. It is the interface that a domain map must implement. This interface is currently in place and is implemented by the standard domain maps provided with this Chapel distribution.
The specification is split into “phases”. The intention is to suggest an implementation order. The requirements of first phase are relied upon by the more essential operations on domains and arrays. Later phases support additional functionality and/or improved performance. The phasing is approximate and is not required.
Status of this document¶
This document, as well as the DSI interface itself, are work in progress. We solicit feedback on both.
The following aspects are not documented yet:
a summary of the six descriptor classes
the requirements for a domain map to support associative or sparse domains;
the significance of dsiClone();
the “local descriptor” classes;
some required methods.
Additional Resources¶
Implementations of the standard domain maps may be consulted as examples. They are available in:
$CHPL_HOME/modules/dists $CHPL_HOME/modules/layouts
In addition, the following publication provide more background information on domain maps in general.
“User-Defined Distributions and Layouts in Chapel: Philosophy and Framework.” Bradford L. Chamberlain, Steven J. Deitz, David Iten, Sung-Eun Choi. 2nd USENIX Workshop on Hot Topics in Parallelism, June 2010.
The high level motivation and overview of DSI.
“Authoring User-Defined Domain Maps in Chapel.” Bradford L. Chamberlain, Sung-Eun Choi, Steven J. Deitz, David Iten, Vassily Litvinov. CUG 2011, May 2011.
A follow-on paper that outlines how to write a domain map.
Overview¶
A domain map must provide three (or more) “global descriptor” classes.
Here they are called GlobalDistribution
, GlobalDomain
,
and GlobalArray
.
Their required fields and methods, as well as suggested naming,
are specified below.
A domain map is allowed to contain/provide features, data structures and functionality (internally to facilitate implementation and/or externally for users’ benefit) beyond the DSI requirements. For example, a distribution (i.e. a domain map that distributes data across multiple locales) will typically define “local descriptor” classes as well.
A domain map author can provide additional functionality to users simply by implementing methods on “global descriptor” classes. When user code contains a method call on an array, domain, or distribution that is not handled by one of the predefined methods, that call is forwarded to the corresponding global descriptor.
Phase 1: The Essentials¶
class GlobalDistribution
¶
This class is visible to the users of the domain map: the
dmap
wrapper in Chapel’s dmapped clauses wraps instances of this class. This class must be a subclass ofBaseDist
.
- proc GlobalDistribution.GlobalDistribution() // or with arguments
Constructor(s) These are not regulated by DSI - their specifics are at the domain map implementor’s discretion.
We suggest providing constructor(s) that accept, as an argument, an array of locales over which to distribute, with
Locales
as the default value.
- proc GlobalDistribution.dsiClone(): GlobalDistribution
Returns a duplicate of this.
- proc GlobalDistribution.dsiDisplayRepresentation(): void
A debugging method. It implements displayRepresentation() on the dmap wrapper.
- proc GlobalDistribution.dsiEqualDMaps(that: /*some other GlobalDistribution*/): bool
Return whether or not the two domain maps are “equal” (specify the same distribution). This is invoked when
==
is applied to two domain maps.
class GlobalDomain
¶
An instance of this class is created whenever a
GlobalDistribution
-mapped Chapel domain is created (e.g. when a domain variable is declared). There is a one-to-one correspondence at run time between a (conceptual) Chapel domain value and an instance ofGlobalDomain
. For presentation simplicity,GlobalDomain
’s properties refer to the properties of the corresponding Chapel domain value, when clear from the context.Depending on the domain kind, this class is returned by the following method on
GlobalDistribution
and must be a subclass of the following class:
domain kind
creating method
required superclass
rectangular
dsiNewRectangularDom
BaseRectangularDom
associative
dsiNewAssociativeDom
BaseAssociativeDom
sparse
dsiNewSparseDom
BaseSparseDom
It is legal for these methods to return instances of different classes in different circumstances. (For example, due to the absence of multiple inheritance, different domain kinds require different classes.)
The DSI requirements apply to each of those classes.
GlobalDomain
is used in this document to refer to each such class.The DSI requirements depend on the domain kind that is supported. The following requirements apply when supporting rectangular domains.
- class GlobalDomain¶
class GlobalDomain ... { param rank: int; type idxType; param stridable: bool; var dist; ... }
The fields
rank
,idxType
,stridable
are the attributes of the corresponding Chapel domain. (They could be replaced with parentheses-less functions of the same names and param/type intents.)The field
dist
must contain a reference to theGlobalDistribution
object that created thisGlobalDomain
. It is used when creating the (runtime) type for this domain, which contains the domain’s domain map.
- proc GlobalDomain.dsiMyDist() return dist;
Returns this domain’s domain map. This procedure should be provided as shown. (Exception: see
dsiLinksDistribution()
.)
- proc GlobalDistribution.dsiNewRectangularDom(param rank: int, type idxType, param stridable: bool, inds) : GlobalDomain(rank, idxType, stridable)
This method is invoked when the Chapel program is creating a domain value of the type domain(rank, idxType, stridable) mapped using the domain map this with initial indices inds.
This method returns a new
GlobalDomain
instance that will correspond to that Chapel domain value, i.e., be that value’s runtime representation. The fielddist
of the returnedGlobalDomain
must point to this.
- proc GlobalDomain.dsiGetIndices(): rank * range(idxType, BoundedRangeType.bounded, stridable)
Returns a tuple of ranges describing the dimensions of this domain.
dsiDims()
anddsiGetIndices()
have the same specification and so may be implemented in terms of one another.
- proc GlobalDomain.dsiSetIndices(dom: domain(rank, idxType, stridable)): void
Updates the internal representation of this to match the index set of
dom
.dsiSetIndices
anddsiGetIndices
are used to assign a domain value to a domain variable (or any domain l-value). Note: the arrays associated with this domain will be updated outside these method.
- proc GlobalDomain.dsiSetIndices(ranges: rank * range(idxType)): void
The same as the other
dsiSetIndices
. Could be implemented like this:{ dsiSetIndices([(...rangesArg)]); }
It is used to initialize the index set of the object returned by
dsiNewRectangularDom()
to the index set of the corresponding Chapel domain value.
- proc GlobalDomain.dsiAssignDomain(rhs: domain, lhsPrivate:bool): void
Set one domain to another. The receiver is the domain being set. The rhs could be any domain, but this function should raise a compilation error if the types do not match sufficiently.
In some cases when this method is called, the LHS can’t be shared or have any arrays declared over it. In that case,
lhsPrivate=true
is passed.Note that the method
chpl_assignDomainWithGetSetIndices
can be called to usedsiReallocate
/getIndices
/dsiSetIndices
/dsiPostReallocate
to accomplish the assignment.
- iter GlobalDomain.these()
The serial iterator over the indices of this domain. Yields values of the type
idxType
ifrank==1
, otherwiserank*idxType
.
- iter GlobalDomain.these(param tag) where tag == iterKind.leader
- iter GlobalDomain.these(param tag, followThis) where tag == iterKind.follower
The “leader” and “follower” iterators (defined below). They are invoked implicitly to implement parallel (“forall”) loops over this domain. Currently the serial iterator must also be defined for “forall” loops to work.
The presentation below is tailored to the use of leader/follower iterators in domain maps. For a more general discussion, see:
$CHPL_HOME/examples/primers/parIters.chpl
The job of the leader iterator is to:
subdivide the given domain into subdomains,
introduce parallelism between these subdomains, and
place computations for each subdomain on the desired locale, as appropriate for this domain map.
The leader iterator must yield, for each subdomain it chooses, a description of that subdomain, created as defined below. Parallelism and placement are achieved, for example, by placing the yield statements within parallel loops and on statements. The leader is invoked (implicitly) once per parallel loop or expression over the corresponding domain.
The job of the follower iterator is simply to iterate sequentially over a subdomain whose description is yielded by the leader iterator, yielding all indices in that subdomain. Each time the leader yields a description, the follower is invoked (implicitly) with that description passed to its
followThis
argument. (The argument name must be exactly “followThis”.)For a zippered loop, only the leader for the first of the zippered items is invoked. Each time that leader yields a description, all followers are invoked in a zippered manner, with that description passed to their
followThis
arguments.In general, it is up to the implementer of the leaders/followers whether and how to support their interoperability in this situation. That is, whether and how the description from the leader of one zippered item is handled by the follower of another item. For example, one could support zippering of similar items and generate a compile-time or run-time error when the items are not “similar”.
DSI requires interoperability between domain maps only for rectangular domains, by prescribing:
how the leader builds the description of a subdomain, and
in what order the follower yields the indices of that subdomain.
A. Subdomain Description¶
Let
Dwhole
be the Chapel domain corresponding to thisGlobalDomain
.Let
Dsub
be a subdomain chosen by the leader iterator.Let
DD
be the number of dimensions in Dwhole and Dsub.The description of
Dsub
is its “densification” w.r.t.Dwhole
:// densification of a domain is a tuple of densifications // of ranges for each dimension densify(Dsub, Dwhole) = ( densify(Dsub.dim(1), Dwhole.dim(1)), ..., densify(Dsub.dim(DD), Dwhole.dim(DD)) ) // densification of a range is the sequence of densifications // of its elements, preserving their order if Rsub describes the sequence i_1, ..., i_N then densify(Rsub, Rwhole) describes the sequence densify(i_1, Rwhole), ..., densify(i_N, Rwhole) For example, densify(Rwhole, Rwhole) == 0.. #(Rwhole.length) // densification of an index is its indexOrder densify(i, Rwhole) = Rwhole.indexOrder(i)The optional module DSIUtil provides the following implementations of densifications and the reverse transformation:
densify(Dsub, Dwhole) densify(RSsub, RSwhole) // on tuples of ranges densify(Rsub, Rwhole) unDensify(Dsub, Dwhole) unDensify(RSsub, RSwhole) // on tuples of ranges unDensify(Rsub, Rwhole)Hint: when developing a new domain map, skip densification and de-densification, but ensure it is used in zippering only with domains/arrays of the same domain map.
B. Follower Yield Order¶
The follower iterator must traverse the subdomain for which the above description is created in the order given by the default domain map. (That order can be observed, e.g., by assigning
Rsub
to a domain variable whose type does not specify a domain map explicitly, then invokingwrite()
on that variable.)(Interoperability of domain maps implies that the follower must accept a densification of any subdomain of
Dwhole
, not just the ones that can be generated by its leader.)
- iter GlobalDomain.these(param tag) where tag == iterKind.standalone
A “standalone” parallel iterator. It is optional. If it is provided, it is used in non-zippered “forall” loops instead of a combination of leader+follower iterators.
- proc GlobalDomain.dsiSerialWrite(f: Writer): void
Writes out the domain to the given Writer (e.g. a file or stdout) serially. Is used to implement write() on the corresponding domain.
This method will typically invoke f.write() on
GlobalDomain
’s components (e.g. bounds) and strings (e.g. “[” and “]”). Such invocations, whether direct or indirect, must occur within the same task thatdsiSerialWrite()
is invoked in, and not within any on statements, direct or indirect. Otherwise a deadlock may occur. (Invokingwrite()
onWriters
other thanf
is not affected by this.)dsiSerialWrite()
will always be invoked onLocales(0)
.
- proc GlobalDomain.dsiDisplayRepresentation(): void
A debugging method. It implements
displayRepresentation()
on the corresponding Chapel domain value.
class GlobalArray
¶
An instance of this class is created whenever a Chapel array is created over a domain represented by a
GlobalDomain
. There is a one-to-one correspondence at run time between a (conceptual) Chapel array value and an instance ofGlobalArray
.This class is returned by
GlobalDomain.dsiBuildArray
. It must be a subclass ofBaseArr
.It is legal for
GlobalDomain.dsiBuildArray
to return instances of different classes in different circumstances. The DSI requirements apply to each of those classes. HereGlobalArray
refers to each such class.class GlobalArray ... { type eltType; var dom; ... }The field
eltType
gives the type of the array elements.The field
dom
must contain a reference to theGlobalDomain
object that created thisGlobalArray
. This is used when creating the (runtime) type for this array, which contains the array’s domain.
- proc GlobalArray.GlobalArray() // or with arguments
Constructor(s). These are not regulated by DSI – their specifics are at the domain map implementor’s discretion.
- proc GlobalDomain.dsiBuildArray(type eltType) : GlobalArray(eltType, this.type)
This method is invoked when the Chapel program is creating an array value over the domain this with the element type
eltType
. This method returns a newGlobalArray
instance that will correspond to that Chapel array value, i.e., be that value’s runtime representation. The fielddom
of the returnedGlobalDomain
must point to this.
- proc GlobalArray.dsiGetBaseDom() return dom;
Returns this array’s
GlobalDomain
. This procedure should be provided as shown.
- proc GlobalArray.dsiAccess(indexx) var: eltType
Given an index, returns the corresponding array element (as an l-value).
The domain map implementer is allowed to restrict the type of indexx that this method accepts.
- proc GlobalArray.dsiSerialWrite(f: Writer): void
Writes out the array to the given
Writer
(e.g. afile
orstdout
) serially. Is used to implementwrite()
on the corresponding array.The restrictions on this method are the same as on
GlobalDomain.dsiSerialWrite()
.
- iter GlobalArray.these() var: eltType
The serial iterator over the elements of this array.
- iter GlobalArray.these(param tag) where tag == iterKind.leader
- iter GlobalArray.these(param tag, followThis) var where tag == iterKind.follower
- iter GlobalArray.these(param tag) where tag == iterKind.standalone
The leader, follower, and standalone iterators.
These are defined in the same way as the leader and follower for
GlobalDomain
, except the follower must yield array locations.Given a subdomain description passed to the
followThis
argument, theGlobalArray
follower iterator must yield array locations corresponding to the indices yielded by theGlobalDomain
follower, in the same order.
- proc GlobalArray.dsiReallocate(d: domain): void
When this array’s domain is assigned a new value, say
newDom
, firstdsiReallocate(newDom)
is invoked on this array. Then,dsiSetIndices(newDom)
is invoked on this array’sdom
.Correspondingly,
dsiReallocate
needs to adjust everything that won’t be taken care of indsiSetIndices
.NOTE: the formal’s name must be exactly
d
at present (due to compiler specifics).NOTE: this method can/should be a no-op if:
GlobalArray
stores its array elements in Chapel array(s), andthe domain(s) of those array(s) are updated by
dsiSetIndices
,
as in the following simplified example:
class GlobalDomain : BaseRectangularDom { // required param rank: int; type idxType; param stridable: bool; const dist; // for example, store indices as a single Chapel domain var myIndices: domain(rank, idxType, stridable); } proc GlobalDomain.dsiSetIndices(dom: domain(rank,idxType,stridable)): void { myIndices = dom; } class GlobalArray : BaseArr { // required type eltType; const dom; // for example, store elements as a single Chapel array var myElements: [dom.myIndices] eltType; }
Given one instance of each class, say
gd
andga
, wherega.dom == gd
the domain ofga.myElements
isgd.myIndices
. A callgd.dsiSetIndices(newDom)
updatesgd.myIndices
, which, in turn, resizesga.myElements
, according to Chapel’s array semantics. Nothing remains to do indsiReallocate()
.(Behind the scene, resizing of
ga.myElements
is implemented bydsiReallocate
of the default domain map. That’s the domain map thatmyIndices
is distributed with, sincemyIndices
’s type provides no explicit domain map.)
- proc GlobalArray.dsiDisplayRepresentation(): void
A debugging method. It implements
displayRepresentation()
on the corresponding Chapel array value.
Naming¶
Once you have chosen the name for your domain map, say, MyMap, we suggest naming the descriptor classes as follows:
in this document
in your code
GlobalDistribution
MyMapDist
GlobalDomain
MyMapDom
GlobalArray
MyMapArr
LocalDistribution
LocMyMapDist
LocalDomain
LocMyMapDom
LocalArray
LocMyMapArr
Some domain maps in this Chapel distribution use just the domain map
name for their GlobalDistribution
classes, e.g. Block and Cyclic.
Phase 2: Additional Operations¶
The operations in this phase are required by DSI. However, if a domain map is in use by a limited set of applications, these operations do not need to be implemented up front. Instead, each of them could be implemented later, when the need arises. The “unresolved call” compilation errors could be used as an indication of what procedure(s) need to be defined.
- proc GlobalDistribution.dsiIndexToLocale(indexx): locale
Given an index
indexx
, returns the locale that “owns” that index, i.e. on which the corresponding data is located. This is used to implementidxToLocale()
on thedmap
wrapper.The domain map implementer is allowed to restrict the type of
indexx
that this method accepts.
- proc GlobalDomain.dsiDim(dim: int): range(idxType, BoundedRangeType.bounded, stridable)
- proc GlobalDomain.dsiDims(): rank * range(idxType, BoundedRangeType.bounded, stridable)
- proc GlobalDomain.dsiLow
- proc GlobalDomain.dsiHigh
- proc GlobalDomain.dsiStride
- proc GlobalDomain.dsiNumIndices
- proc GlobalDomain.dsiMember(indexx)
- proc GlobalDomain.dsiIndexOrder(indexx)
These methods implement the corresponding queries (
dim
,dims
,low
, etc.) of the domain value for which thisGlobalDomain
instance was created. For example,dsiDim(d)
returns the range describing the domain’sd
-th dimension.dsiDims()
anddsiGetIndices()
have the same specification and so may be implemented in terms of one another.
- proc GlobalDomain.linksDistribution() param
- proc GlobalDomain.dsiLinksDistribution()
Typically these should not be defined.
If the domains mapped using
GlobalDistribution
do NOT need to be tracked and theGlobalDistribution
itself does NOT need to be reference counted, these two methods should be defined to return false; Also in this casedsiMyDist()
does not need to be defined.
- proc GlobalDomain.doiToString()
This is an optional method that defines how the domain should be cast to a string. It can be useful for reducing the size of the generated code in the event that the domain representation is fairly straightforward (say, a dense rectangular domain) in order to avoid the more general stringify() routine, which is used otherwise.
Phase 3: Privatization¶
What is privatization?¶
Privatization of an object X
means providing a local copy of X
on
each locale. Such a copy is called the “privatized copy”.
On X.locale
, X
itself serves as the privatized copy.
We refer to X
as the “original object”.
Privatization aims at reducing communication between locales.
When X
needs to be accessed (by reading its fields or invoking
its methods) from another locale, its privatized copy
on the current locale is used instead.
Therefore, communication to X.locale
is eliminated.
How does privatization work with DSI?¶
Privatization is optional in DSI. Each of the global descriptor classes
can support privatization independently of the others. A class indicates
to the Chapel implementation whether it supports privatization via
its method dsiSupportsPrivatization()
(see below).
Layouts (i.e. the domain maps that do not distribute domains across locales) can potentially benefit from privatization. However, the standard layouts do not support it.
The remainder of DSI privatization requirements must be implemented by each global descriptor class that chooses to support privatization.
The domain map implementation must provide methods to:
create a privatized copy given the original object, and
update a privatized copy when some other privatized copy changes (see “reprivatization” below).
The Chapel implementation:
invokes DSI privatization methods to create or update privatized copies, and
redirects original object accesses to its privatized copies.
The Chapel implementation creates privatized copies (over all locales) greedily as follows (if that class supports privatization):
of a
GlobalDistribution
- when it is wrapped innew dmap()
and when that wrapper is copied;of a
GlobalDomain
orGlobalArray
- when the corresponding Chapel domain or array is created.
What is reprivatization?¶
Should any privatized copy be modified, the changes need to be propagated to all the other privatized copies. This propagation is called reprivatization.
The domain map implementation provides methods to update a privatized copy. The Chapel implementation invokes these methods when necessary.
DSI privatization requirements¶
The following requirements apply individually to each global descriptor
class that chooses to support privatization.
Global
denotes such a class.
- proc Global.dsiSupportsPrivatization() param return true;
Returns true to indicate that privatization is supported. NOTE: do not specify the return type (due to a bug in the compiler).
class Global ... {
...
var pid = -1;
...
}
The field pid
should be provided as shown.
It should not be accessed by the DSI implementation except
in conjunction with chpl_getPrivatizedCopy()
as discussed later.
- proc Global.dsiGetPrivatizeData()
Returns the data to be used as the argument to
dsiPrivatize()
. It can be a tuple of values or any other type, at the implementor’s discretion. SeedsiPrivatize()
for explanation.
- proc Global.dsiPrivatize(privatizeData): Global
Returns a privatized copy of this. The Chapel implementation invokes this method as follows:
on the locale where the privatized copy is to be located,
this is either the original object or its privatized copy created by
dsiPrivatize
on some other locale,privatizeData
is the result of invoking dsiGetPrivatizeData() on the original object.
Typically
dsiPrivatize()
will need to copy and/or privatize some information from the original object into the privatized copy being created. This information could be obtained by querying this directly. Alternatively, it could be passed from the original object viaprivatizeData
. This approach could allow the needed information to be bundled into the original active message, rather than requiring additional communications when accessing this.
- proc Global.dsiGetReprivatizeData()
Similar do
dsiGetPrivatizeData()
, except the result is used as the argument todsiReprivatize()
.
- proc Global.dsiReprivatize(other: Global, reprivatizeData): void
Updates a privatized copy.
this is the object to be updated as part of reprivatization.
other
is either the object whose modification originally necessitated reprivatization or one of the privatized copies that have already been updated for that modification bydsiReprivatize()
.reprivatizeData
is the result of invoking dsiGetReprivatizeData() on the originally-modified object.Q: What modifications must
dsiReprivatize()
reflect?A: Any changes that the domain map implementation may perform on an instance of
Global
(or its privatized copy) after that instance has been created and privatized.For example,
GlobalDomain.dsiReprivatize()
must reflect any changes thatGlobalDomain.dsiSetIndices()
may perform.
Tip: testing privatization¶
In addition to the usual correctness testing, it may be useful to check whether array access is purely local, i.e., results in no communication. This can be done using local statements.
(A local statement performs runtime checks that report an error whenever any communication occurs within the its body.)
Here is a simple example:
// declare an array that uses the domain map to be tested var A: ...; // initialize it A = value1; forall loc in (locales that A is distributed over) do // run the check on each locale on loc { // value2 to be different from value1 var valTemp = value2; const idxTemp = (an index (tuple) that is mapped to the locale 'loc'); local { // Access the array. // This statement succeeds if there is no communication. valTemp = A[idxTemp]; } // reference the read value assert(valTemp == value1); }
Tip: privatized copies should reference privatized copies¶
The global descriptor classes are required to reference each other
(e.g. GlobalArray.dom
references GlobalDomain
; GlobalDomain.dist
references GlobalDistribution
). Therefore, if it is desired to
eliminate communication completely upon array references,
all the three global descriptor classes may have to be privatized.
When one descriptor object references another, a privatized copy of
the former needs to reference a privatized copy of the latter.
To obtain a privatized copy of an object, e.g. for use within
dsiPrivatize()
, use the following procedure:
proc chpl_getPrivatizedCopy(type objectType, objectPid:int): objectType
Notes:
The first argument is type of the object being privatized.
The second argument is the original object’s
pid
field.The procedure returns a privatized copy of the original object.
The procedure can be applied only to objects that have already been privatized.
A
GlobalArray
is privatized after itsGlobalDomain
, which is privatized after itsGlobalDistribution
.The
pid
field is set to a different value than-1
when an object is privatized.
Here is an example of using it for privatizing GlobalDomain
,
assuming that GlobalDistribution
also supports privatization:
proc GlobalDomain.dsiGetPrivatizeData() { // include the desired 'pid' return (this.dist.pid, ... other data as needed ...); } proc GlobalDomain.dsiPrivatize(privatizeData) { // extract the 'pid' provided above const distPid = privatizeData(1); // obtain the privatized copy of the GlobalDistribution object const privatizedDMap = chpl_getPrivatizedCopy(this.dist.type, distPid); return new GlobalDomain(dist = privatizedDMap, ... other fields as needed ...) }
Tip: “privatize” the domains and arrays used in the implementation¶
If a global descriptor class to be privatized uses domains and arrays, those need to be “privatized”, too, to reduce communication. In the case of domains and arrays mapped using the default layout, their “privatization” is achieved by copying.
Careful consideration is needed to eliminate all communication. For example:
Copying an array preserves the source’s domain - unless the destination’s domain is declared explicitly.
Copying a domain preserves the source’s domain map - unless the destination’s type is declared explicitly.
If an array’s domain or a domain’s domain map is declared explicitly, ensure that privatized copies of those are used.
There is no need to privatize the default layout. Specifically, copying a domain declared without an explicit domain map is sufficient to privatize that domain.
The same considerations are valid for reprivatization.
Here is an example of “privatizing” domains/arrays within GlobalDomain
.
The key insight here is that auxArrayG
’s domain needs to be “privatized”.
Cf. the domains of auxArrayED1
and auxArrayED2
are created
implicitly for each GlobalDomain
object and so will be local in any case.
class GlobalDomain ... { ... // this field is generic var auxArrayG; // domain is specified explicitly var auxArrayED1: [1..10] int; // similar var auxDomain: domain(1); var auxArrayED2: [auxDomain] int; } proc GlobalDomain.dsiGetPrivatizeData() { return (this.dist.pid, auxArrayG, auxArrayED1, auxDomain, auxArrayED2, ...); } proc GlobalDomain.dsiPrivatize(privatizeData) { // To privatize auxArrayG, we must "privatize" its domain first. // No need to declare privDom's type if auxArrayG uses the default layout. var privDom = privatizeData(2).domain; var privArr: [privDom] privatizeData(2).eltType = privatizeData(2); // If the following were used, privArr.domain would be the same // as privatizeData(2).domain, i.e., it would not be privatized. //var privArr = privatizeData(2); return new GlobalDomain(..., auxArrayG = privArr, // the other fields can be simply copied auxArrayED1 = privatizeData(3), auxDomain = privatizeData(4), auxArrayED2 = privatizeData(5)); }
Phase 4: Bulk-Transfer Interface¶
What is the bulk-transfer interface?¶
While the Chapel language does not require a specific implementation for array assignment, in practice the values are assigned individually:
for[all] (d, s) in zip(dest, src) { d = s; }
When assignment occurs between arrays on different locales, this can result in significant communication overhead. The bulk-transfer interface aims at reducing communication between locales by enabling domain maps to handle certain kinds of assignments themselves.
How does the bulk-transfer interface work with DSI?¶
The bulk-transfer interface is an optional interface supported only on array
descriptors. The following methods are prefixed with doi
for “Domain map
Optional Interface.” An array descriptor can implement one or more of these
methods to opt-in to support for bulk-transfers:
- proc GlobalArray.doiBulkTransferToKnown(myDom:domain, otherClass, otherDom:domain) : bool
Perform an array assignment from
this
tootherClass
.
- proc GlobalArray.doiBulkTransferFromKnown(myDom:domain, otherClass, otherDom:domain) : bool
Perform an array assignment from
otherClass
tothis
- proc GlobalArray.doiBulkTransferToAny(myDom:domain, otherClass, otherDom:domain) : bool
Perform an array assignment from
this
tootherClass
.
- proc GlobalArray.doiBulkTransferFromAny(myDom:domain, otherClass, otherDom:domain) : bool
Perform an array assignment from
otherClass
tothis
Each method shares these arguments:
myDom:domain
- the indices to be transferred to/fromthis
otherClass
- a class inheriting fromBaseArr
that represents the other array in the assignmentotherDom:domain
- the indices to be transferred to/fromotherClass
Callers of these methods must satisfy the following guarantees for these arguments:
this.rank == myDom.rank
otherClass.rank == otherDom.rank
myDom.size == otherDom.size
Note
myDom
and otherDom
are not required to be distributed, even if the
corresponding arrays are distributed. They simply represent the indices
involved in the transfer.
Each method returns a bool
indicating whether the assignment occurred. This
allows domain maps to resort to the Chapel implementation of array assignment
if they are incapable or unwilling to perform the assignment themselves.
Known and Any methods¶
There are two kinds of transfers accounted for in this interface.
The first kind of transfer is a Known
transfer. These are transfers that
the implementer believes to be optimal. Typically this means that the
implementer knows the type and internals of the other domain map involved in
the transfer.
The second kind of transfer is an Any
transfer. These are transfers to/from
domain maps whose type and internals are not known. These transfers
typically rely on attempting to have the unknown domain map transfer to/from
the internal data of the domain map on which the method is implemented. See
Implementing ‘Any’ Methods for more detail.
This interface recognizes these kinds of transfers because otherwise it is difficult to determine which distribution implements the best transfer.
For example, the author of PopularDist
could implement Any
methods for
better performance with unknown third-party domain maps. The author of
NewDist
could implement Known
methods to PopularDist
due to its
popularity. By doing so the author of NewDist
has informed Chapel that it
knows better than PopularDist
and a more optimal transfer occurs.
Chapel will call Known
methods before Any
methods if they can be
resolved. A recommended convention is to implement Known
methods with a
where-clause that constrains the type of the otherClass
argument. This
prevents the Known
method from being called with unknown types.
Note
There is no defined preference for To
and From
methods. Domain map
authors should not currently rely on a particular direction being attempted
first.
When Does a Bulk-Transfer Occur?¶
There are a number of conditions that need to be true for a bulk-transfer to occur:
The config param
useBulkTransfer
must be trueThe element types of the arrays must be identical
The element type must be one of the following:
integral
real
complex
records or tuples that are “Plain Old Data”
The element type is restricted in order to preserve copy semantics of each
element. For example, integral
types can be bit-copied and so have no
side-effects. A record with a copy-initializer may have side effects that
should not be eliminated by the bulk-transfer optimization. In order to avoid
eliminating these side-effects, only records or tuples that are “Plain Old
Data” are considered valid for bulk-transfers. The (currently undocumented)
function isPODType(t)
returns a param boolean indicating whether the given
type is “Plain Old Data”.
The useBulkTransfer
config param is true
by default but can be disabled
at compilation time. For example:
chpl myProgram.chpl -suseBulkTransfer=false
Implementing ‘Any’ Methods¶
Any methods are generally implemented by
transferring data of an internal type of this
to/from a subset of indices
in otherClass
. For example, a distribution can be implemented in terms of
Chapel’s default rectangular arrays. An Any
method on that distribution
could be implemented by attempting to transfer data to/from otherClass
to
a default rectangular array, like so (in pseudo-chapel):
// For a single locale's data
var localDom = localData.domain[myDom]; // local subset of indices for ``this``
var locOther = <translate localDom to otherDom's coordinates>;
otherClass.doiBulkTransfer[To|From][Any|Known](locOther, localData, localDom);
Implementers who call these doiBulkTransfer
methods should note that they
can return false
, meaning the caller needs to complete the transfer.
Warning
Beware of implementing bulk-transfers by manually calling other
doiBulkTransfer
methods. For example:
proc GlobalArray.doiBulkTransferFromAny(myDom:domain, otherClass, otherDom:domain) {
otherClass.doiBulkTransferToAny(otherDom, this, myDom);
}
This pattern can lead to potential recursion depending on the implementation
of the other domain map’s bulk-transfer method. Instead of implementing the
example method above, the author could have not implemented it at all and
let the Chapel internals invoke the doiBulkTransferToAny
directly.
Transfers for Aliasing Domain Maps¶
It is valid to write a domain map that aliases another domain map’s data. For example, a user could write their own variant of a Chapel array slice in terms of the Domain Map Standard Interface.
When aliasing other domain maps it is recommended to only implement Known transfers that forward to the underlying domain map’s bulk-transfer methods. By implementing Known transfers the underlying domain map will have a chance to perform a more optimal transfer before the Any transfers are attempted.
Transfers for Rank-changes¶
Implementing a bulk-transfer when rank-changed can be tricky. In the interest of simplicity, Chapel by default will not attempt to call bulk-transfer methods on arrays that have been rank-changed.
Domain map authors can opt-in to handling rank changes by implementing the following method:
- proc GlobalArray.doiCanBulkTransferRankChange() param : bool
If this method is resolvable and returns true
then Chapel will attempt
to call bulk-transfer methods on the array.
Domain map authors should note that this means that this.rank
may not be
equal to otherClass.rank
when rank-changes are involved. They should also
note that while myDom
and otherDom
are of the same size, their
dimensions may not match in size. For example:
var A, B : [1..10, 1..10] int;
A[.., 1] = B[1, ..];
The call to a bulk-transfer method could look like:
classA.doiBulkTransferFromKnown({1..10, 1..1}, classB, {1..1, 1..10});
The implementer of this method is left with the task of determining which dimensions were rank-changed.