edu.iu.dsc.tws.comms.mpi

Class TWSMPIChannel

java.lang.Object

edu.iu.dsc.tws.comms.mpi.TWSMPIChannel

All Implemented Interfaces:

TWSChannel

public class TWSMPIChannel
extends java.lang.Object
implements TWSChannel

We are going to use a byte based messaging protocol.

The transport threads doesn't handle the message serialization and it is left to the application level.

Constructor Summary

Constructors

Constructor and Description
TWSMPIChannel(Config config, IWorkerController wController) Create the mpi channel

Method Summary

Modifier and Type	Method and Description
void	close() Close the channel
java.nio.ByteBuffer	createBuffer(int capacity) Create a buffer
boolean	isComplete() Check weather we have any pending sends
void	progress() Progress the communications that are pending
void	progressReceives(int receiveGroupIndex) Progress only receives belonging to a group
void	progressSends() Progress only the sends
boolean	receiveMessage(int group, int wId, int e, ChannelListener callback, java.util.Queue<DataBuffer> receiveBuffers) Register our interest to receive messages from particular rank using a stream
void	reInit(java.util.List<JobMasterAPI.WorkerInfo> restartedWorkers) re-initialize the channel with restarted workers
void	releaseBuffers(int wId, int e) Close a worker id with edge
boolean	sendMessage(int wId, ChannelMessage message, ChannelListener callback) Send messages to the particular id

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

TWSMPIChannel

public TWSMPIChannel(Config config,
                     IWorkerController wController)

Create the mpi channel

Parameters:

config - configuration

wController - controller

Method Detail

sendMessage

public boolean sendMessage(int wId,
                           ChannelMessage message,
                           ChannelListener callback)

Send messages to the particular id

Specified by:

sendMessage in interface TWSChannel

Parameters:

wId - id to be used for sending messages

message - the message

callback - callback for message completions

Returns:

true if the message is accepted to be sent

receiveMessage

public boolean receiveMessage(int group,
                              int wId,
                              int e,
                              ChannelListener callback,
                              java.util.Queue<DataBuffer> receiveBuffers)

Register our interest to receive messages from particular rank using a stream

Specified by:

receiveMessage in interface TWSChannel

Parameters:

wId - worker id to listen to

e - the graph edge to receive from

callback - callback for message completions

receiveBuffers - the list of receive buffers

Returns:

true if the message is accepted

close

public void close()

Description copied from interface: TWSChannel

Close the channel

Specified by:

close in interface TWSChannel

isComplete

public boolean isComplete()

Description copied from interface: TWSChannel

Check weather we have any pending sends

Specified by:

isComplete in interface TWSChannel

progressSends

public void progressSends()

Description copied from interface: TWSChannel

Progress only the sends

Specified by:

progressSends in interface TWSChannel

progressReceives

public void progressReceives(int receiveGroupIndex)

Description copied from interface: TWSChannel

Progress only receives belonging to a group

Specified by:

progressReceives in interface TWSChannel

progress

public void progress()

Progress the communications that are pending

Specified by:

progress in interface TWSChannel

createBuffer

public java.nio.ByteBuffer createBuffer(int capacity)

Description copied from interface: TWSChannel

Create a buffer

Specified by:

createBuffer in interface TWSChannel

Parameters:

capacity - capacity

Returns:

the byte buffer

releaseBuffers

public void releaseBuffers(int wId,
                           int e)

Close a worker id with edge

Specified by:

releaseBuffers in interface TWSChannel

Parameters:

wId - worker id

e - edge

reInit

public void reInit(java.util.List<JobMasterAPI.WorkerInfo> restartedWorkers)

Description copied from interface: TWSChannel

re-initialize the channel with restarted workers

Specified by:

reInit in interface TWSChannel