org.apache.sysds.hops.rewrite

Class IPAPassRewriteFederatedPlan

java.lang.Object

org.apache.sysds.hops.ipa.IPAPass

org.apache.sysds.hops.rewrite.IPAPassRewriteFederatedPlan

public class IPAPassRewriteFederatedPlan
extends IPAPass

This rewrite generates a federated execution plan by estimating and setting costs and the FederatedOutput values of all relevant hops in the DML program. The rewrite is only applied if federated compilation is activated in OptimizerUtils.

Constructor Summary

Constructors

Constructor and Description
IPAPassRewriteFederatedPlan()

Method Summary

Modifier and Type	Method and Description
boolean	isApplicable(FunctionCallGraph fgraph) Indicates if an IPA pass is applicable for the current configuration.
boolean	rewriteProgram(DMLProgram prog, FunctionCallGraph fgraph, FunctionCallSizeInfo fcallSizes) Estimates cost and selects a federated execution plan by setting the federated output value of each hop in the program.
ArrayList<StatementBlock>	rewriteStatementBlock(StatementBlock sb) Estimates cost and selects a federated execution plan by setting the federated output value of each hop in the statement blocks.
ArrayList<StatementBlock>	rewriteStatementBlocks(List<StatementBlock> sbs) Estimates cost and selects a federated execution plan by setting the federated output value of each hop in the statement blocks.

Methods inherited from class java.lang.Object

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

Constructor Detail

IPAPassRewriteFederatedPlan

public IPAPassRewriteFederatedPlan()

Method Detail

isApplicable

public boolean isApplicable(FunctionCallGraph fgraph)

Indicates if an IPA pass is applicable for the current configuration. The configuration depends on OptimizerUtils.FEDERATED_COMPILATION.

Specified by:

isApplicable in class IPAPass

Parameters:

fgraph - function call graph

Returns:

true if federated compilation is activated.

rewriteProgram

public boolean rewriteProgram(DMLProgram prog,
                              FunctionCallGraph fgraph,
                              FunctionCallSizeInfo fcallSizes)

Estimates cost and selects a federated execution plan by setting the federated output value of each hop in the program.

Specified by:

rewriteProgram in class IPAPass

Parameters:

prog - dml program

fgraph - function call graph

fcallSizes - function call size infos

Returns:

false since the function call graph never has to be rebuilt

rewriteStatementBlocks

public ArrayList<StatementBlock> rewriteStatementBlocks(List<StatementBlock> sbs)

Estimates cost and selects a federated execution plan by setting the federated output value of each hop in the statement blocks. The method calls the contained statement blocks recursively.

Parameters:

sbs - list of statement blocks

Returns:

list of statement blocks with the federated output value updated for each hop

rewriteStatementBlock

public ArrayList<StatementBlock> rewriteStatementBlock(StatementBlock sb)

Estimates cost and selects a federated execution plan by setting the federated output value of each hop in the statement blocks. The method calls the contained statement blocks recursively.

Parameters:

sb - statement block

Returns:

list of statement blocks with the federated output value updated for each hop