Skip to content
Permalink
Branch: master
Find file Copy path
Find file Copy path
Fetching contributors…
Cannot retrieve contributors at this time
79 lines (55 sloc) 1.97 KB

Channels

Channels are streamed connections or pipes between a pair of chares. They simplify the expression of sends/receives from inside a coroutine without having to exit the coroutine.

Creation

To create a Channel:

  • Channel(chare, remote):

    Establish a channel between chare and remote, returning a channel object.

    Note that chare is an actual chare object, not a proxy. remote is a proxy to the remote chare.

    Channels do not have to be created from coroutines, but they can only be used from coroutines.

    There is no restriction on the number of channels that a chare can establish, and it can establish multiple channels with the same remote.

Methods

Channel objects have the following methods:

  • send(self, *args):

    Send the arguments through the channel to the remote chare.

  • recv(self):

    Receives arguments (unpacked) from the channel. Messages are received in order.

Example

from charm4py import charm, Chare, Array, coro, Channel

NUM_ITER = 100

class A(Chare):

    def __init__(self, numchares):
        myidx = self.thisIndex[0]
        neighbors = []
        neighbors.append(self.thisProxy[(myidx + 1) % numchares])
        neighbors.append(self.thisProxy[(myidx - 1) % numchares])
        self.channels = []
        for nb in neighbors:
            self.channels.append(Channel(self, remote=nb))

    @coro
    def work(self):
        for i in range(NUM_ITER):
            for ch in self.channels:
                ch.send(x, y, z)

            for ch in charm.iwait(self.channels):
                x, y, z = ch.recv()
                # ... do something with data ...

def main(args):
    numchares = charm.numPes() * 8
    array = Array(A, numchares, args=[numchares])
    future = array.work(awaitable=True)
    future.get()
    exit()

charm.start(main)
You can’t perform that action at this time.