edu.iu.dsc.tws.comms.stream

Class SAllReduce

java.lang.Object

edu.iu.dsc.tws.api.comms.BaseOperation

edu.iu.dsc.tws.comms.stream.SAllReduce

public class SAllReduce
extends BaseOperation

Streaming ALLReduce Operation

Field Summary

Fields inherited from class edu.iu.dsc.tws.api.comms.BaseOperation

channel, op

Constructor Summary

Constructors

Constructor and Description
SAllReduce(Communicator comm, LogicalPlanBuilder logicalPlanBuilder, MessageType dataType, ReduceFunction fnc, SingularReceiver rcvr)
SAllReduce(Communicator comm, LogicalPlan plan, java.util.Set<java.lang.Integer> sources, java.util.Set<java.lang.Integer> targets, MessageType dataType, ReduceFunction fnc, SingularReceiver rcvr)
SAllReduce(Communicator comm, LogicalPlan plan, java.util.Set<java.lang.Integer> sources, java.util.Set<java.lang.Integer> targets, MessageType dataType, ReduceFunction fnc, SingularReceiver rcvr, int reduceEdgeId, int bcastEdgeId, MessageSchema messageSchema) Construct a Streaming AllReduce operation

Method Summary

Modifier and Type	Method and Description
boolean	reduce(int src, java.lang.Object message, int flags) Send a message to be reduced

Methods inherited from class edu.iu.dsc.tws.api.comms.BaseOperation

close, finish, isComplete, progress, progressChannel, reset, sendBarrier, waitForCompletion

Methods inherited from class java.lang.Object

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

Constructor Detail

SAllReduce

public SAllReduce(Communicator comm,
                  LogicalPlan plan,
                  java.util.Set<java.lang.Integer> sources,
                  java.util.Set<java.lang.Integer> targets,
                  MessageType dataType,
                  ReduceFunction fnc,
                  SingularReceiver rcvr,
                  int reduceEdgeId,
                  int bcastEdgeId,
                  MessageSchema messageSchema)

Construct a Streaming AllReduce operation

Parameters:

comm - the communicator

plan - task plan

sources - source tasks

targets - target tasks

rcvr - receiver

dataType - data type

SAllReduce

public SAllReduce(Communicator comm,
                  LogicalPlan plan,
                  java.util.Set<java.lang.Integer> sources,
                  java.util.Set<java.lang.Integer> targets,
                  MessageType dataType,
                  ReduceFunction fnc,
                  SingularReceiver rcvr)

SAllReduce

public SAllReduce(Communicator comm,
                  LogicalPlanBuilder logicalPlanBuilder,
                  MessageType dataType,
                  ReduceFunction fnc,
                  SingularReceiver rcvr)

Method Detail

reduce

public boolean reduce(int src,
                      java.lang.Object message,
                      int flags)

Send a message to be reduced

Parameters:

src - source

message - message

flags - message flag

Returns:

true if the message is accepted