Package org.apache.sysds.hops.recompile
Class Recompiler
- java.lang.Object
-
- org.apache.sysds.hops.recompile.Recompiler
-
public class Recompiler extends Object
Dynamic recompilation of hop dags to runtime instructions, which includes the following substeps: (1) deep copy hop dag, (2) refresh matrix characteristics, (3) apply dynamic rewrites, (4) refresh memory estimates, (5) construct lops (incl operator selection), and (6) generate runtime program (incl piggybacking).
-
-
Nested Class Summary
Nested Classes Modifier and Type Class Description static classRecompiler.ResetType
-
Constructor Summary
Constructors Constructor Description Recompiler()
-
Method Summary
All Methods Static Methods Concrete Methods Modifier and Type Method Description static booleancheckCPCheckpoint(DataCharacteristics dc)static booleancheckCPReblock(ExecutionContext ec, String varin)CP Reblock check for spark instructions; in contrast to MR, we can not rely on the input file sizes because inputs might be passed via rdds.static ArrayList<Hop>deepCopyHopsDag(List<Hop> hops)Deep copy of hops dags for parallel recompilation.static HopdeepCopyHopsDag(Hop hops)Deep copy of hops dags for parallel recompilation.static voidexecuteInMemoryReblock(ExecutionContext ec, String varin, String varout)static voidexecuteInMemoryReblock(ExecutionContext ec, String varin, String varout, LineageItem litem)static voidextractDAGOutputStatistics(ArrayList<Hop> hops, LocalVariableMap vars)static voidextractDAGOutputStatistics(ArrayList<Hop> hops, LocalVariableMap vars, boolean overwrite)static voidextractDAGOutputStatistics(Hop hop, LocalVariableMap vars, boolean overwrite)static voidrClearLops(Hop hop)Clearing lops for a given hops includes to (1) remove the reference to constructed lops and (2) clear the exec type (for consistency).static ArrayList<Instruction>recompileHopsDag(Hop hop, LocalVariableMap vars, RecompileStatus status, boolean inplace, boolean replaceLit, long tid)static ArrayList<Instruction>recompileHopsDag(StatementBlock sb, ArrayList<Hop> hops, ExecutionContext ec, RecompileStatus status, boolean inplace, boolean replaceLit, long tid)static ArrayList<Instruction>recompileHopsDag(StatementBlock sb, ArrayList<Hop> hops, LocalVariableMap vars, RecompileStatus status, boolean inplace, boolean replaceLit, long tid)static ArrayList<Instruction>recompileHopsDag2Forced(Hop hop, long tid, Types.ExecType et)static ArrayList<Instruction>recompileHopsDag2Forced(StatementBlock sb, ArrayList<Hop> hops, long tid, Types.ExecType et)static ArrayList<Instruction>recompileHopsDagInstructions(Hop hop)static ArrayList<Instruction>recompileHopsDagInstructions(StatementBlock sb, ArrayList<Hop> hops)static voidrecompileProgramBlockHierarchy(ArrayList<ProgramBlock> pbs, LocalVariableMap vars, long tid, boolean inplace, Recompiler.ResetType resetRecompile)static voidrecompileProgramBlockHierarchy2Forced(ArrayList<ProgramBlock> pbs, long tid, HashSet<String> fnStack, Types.ExecType et)Method to recompile program block hierarchy to forced execution time.static voidrecompileProgramBlockInstructions(ProgramBlock pb)This method does NO full program block recompile (no stats update, no rewrites, no recursion) but only regenerates lops and instructions.static RecompileStatusreconcileUpdatedCallVarsIf(RecompileStatus oldStatus, RecompileStatus callStatusIf, RecompileStatus callStatusElse, StatementBlock sb)static LocalVariableMapreconcileUpdatedCallVarsIf(LocalVariableMap oldCallVars, LocalVariableMap callVarsIf, LocalVariableMap callVarsElse, StatementBlock sb)static booleanreconcileUpdatedCallVarsLoops(RecompileStatus oldCallStatus, RecompileStatus callStatus, StatementBlock sb)static booleanreconcileUpdatedCallVarsLoops(LocalVariableMap oldCallVars, LocalVariableMap callVars, StatementBlock sb)static voidreinitRecompiler()Re-initializes the recompiler according to the current optimizer flags.static voidremoveUpdatedScalars(LocalVariableMap callVars, StatementBlock sb)Remove any scalar variables from the variable map if the variable is updated in this block.static booleanrequiresRecompilation(ArrayList<Hop> hops)static booleanrequiresRecompilation(Hop hop)static intrGetMaxParallelism(List<Hop> hops)static intrGetMaxParallelism(Hop hop)static voidrRecompileProgramBlock2Forced(ProgramBlock pb, long tid, HashSet<String> fnStack, Types.ExecType et)static voidrReplaceLiterals(Hop hop, ExecutionContext ec, boolean scalarsOnly)public interface to package local literal replacementstatic voidrReplaceLiterals(Hop hop, LocalVariableMap vars, boolean scalarsOnly)static voidrSetExecType(Hop hop, Types.ExecType etype)static voidrSetMaxParallelism(List<Hop> hops, int k)static voidrSetMaxParallelism(Hop hop, int k)static voidrUpdateFunctionNames(Hop hop, long pid)static voidrUpdateStatistics(Hop hop, LocalVariableMap vars)static voidupdateFunctionNames(ArrayList<Hop> hops, long pid)
-
-
-
Method Detail
-
reinitRecompiler
public static void reinitRecompiler()
Re-initializes the recompiler according to the current optimizer flags.
-
recompileHopsDag
public static ArrayList<Instruction> recompileHopsDag(StatementBlock sb, ArrayList<Hop> hops, ExecutionContext ec, RecompileStatus status, boolean inplace, boolean replaceLit, long tid)
-
recompileHopsDag
public static ArrayList<Instruction> recompileHopsDag(StatementBlock sb, ArrayList<Hop> hops, LocalVariableMap vars, RecompileStatus status, boolean inplace, boolean replaceLit, long tid)
-
recompileHopsDag
public static ArrayList<Instruction> recompileHopsDag(Hop hop, LocalVariableMap vars, RecompileStatus status, boolean inplace, boolean replaceLit, long tid)
-
recompileHopsDag2Forced
public static ArrayList<Instruction> recompileHopsDag2Forced(StatementBlock sb, ArrayList<Hop> hops, long tid, Types.ExecType et)
-
recompileHopsDag2Forced
public static ArrayList<Instruction> recompileHopsDag2Forced(Hop hop, long tid, Types.ExecType et)
-
recompileHopsDagInstructions
public static ArrayList<Instruction> recompileHopsDagInstructions(StatementBlock sb, ArrayList<Hop> hops)
-
recompileHopsDagInstructions
public static ArrayList<Instruction> recompileHopsDagInstructions(Hop hop)
-
recompileProgramBlockHierarchy
public static void recompileProgramBlockHierarchy(ArrayList<ProgramBlock> pbs, LocalVariableMap vars, long tid, boolean inplace, Recompiler.ResetType resetRecompile)
-
recompileProgramBlockHierarchy2Forced
public static void recompileProgramBlockHierarchy2Forced(ArrayList<ProgramBlock> pbs, long tid, HashSet<String> fnStack, Types.ExecType et)
Method to recompile program block hierarchy to forced execution time. This affects also referenced functions and chains of functions. Use et==null in order to release the forced exec type.- Parameters:
pbs- list of program blockstid- thread idfnStack- function stacket- execution type
-
recompileProgramBlockInstructions
public static void recompileProgramBlockInstructions(ProgramBlock pb) throws IOException
This method does NO full program block recompile (no stats update, no rewrites, no recursion) but only regenerates lops and instructions. The primary use case is recompilation after are hop configuration changes which allows to preserve statistics (e.g., propagated worst case stats from other program blocks) and better performance for recompiling individual program blocks.- Parameters:
pb- program block- Throws:
IOException- if IOException occurs
-
requiresRecompilation
public static boolean requiresRecompilation(Hop hop)
-
deepCopyHopsDag
public static ArrayList<Hop> deepCopyHopsDag(List<Hop> hops)
Deep copy of hops dags for parallel recompilation.- Parameters:
hops- list of high-level operators- Returns:
- list of high-level operators
-
deepCopyHopsDag
public static Hop deepCopyHopsDag(Hop hops)
Deep copy of hops dags for parallel recompilation.- Parameters:
hops- high-level operator- Returns:
- high-level operator
-
rUpdateFunctionNames
public static void rUpdateFunctionNames(Hop hop, long pid)
-
reconcileUpdatedCallVarsLoops
public static boolean reconcileUpdatedCallVarsLoops(LocalVariableMap oldCallVars, LocalVariableMap callVars, StatementBlock sb)
-
reconcileUpdatedCallVarsLoops
public static boolean reconcileUpdatedCallVarsLoops(RecompileStatus oldCallStatus, RecompileStatus callStatus, StatementBlock sb)
-
reconcileUpdatedCallVarsIf
public static LocalVariableMap reconcileUpdatedCallVarsIf(LocalVariableMap oldCallVars, LocalVariableMap callVarsIf, LocalVariableMap callVarsElse, StatementBlock sb)
-
reconcileUpdatedCallVarsIf
public static RecompileStatus reconcileUpdatedCallVarsIf(RecompileStatus oldStatus, RecompileStatus callStatusIf, RecompileStatus callStatusElse, StatementBlock sb)
-
rRecompileProgramBlock2Forced
public static void rRecompileProgramBlock2Forced(ProgramBlock pb, long tid, HashSet<String> fnStack, Types.ExecType et)
-
removeUpdatedScalars
public static void removeUpdatedScalars(LocalVariableMap callVars, StatementBlock sb)
Remove any scalar variables from the variable map if the variable is updated in this block.- Parameters:
callVars- Map of variables eligible for propagation.sb- DML statement block.
-
extractDAGOutputStatistics
public static void extractDAGOutputStatistics(ArrayList<Hop> hops, LocalVariableMap vars)
-
extractDAGOutputStatistics
public static void extractDAGOutputStatistics(ArrayList<Hop> hops, LocalVariableMap vars, boolean overwrite)
-
extractDAGOutputStatistics
public static void extractDAGOutputStatistics(Hop hop, LocalVariableMap vars, boolean overwrite)
-
rClearLops
public static void rClearLops(Hop hop)
Clearing lops for a given hops includes to (1) remove the reference to constructed lops and (2) clear the exec type (for consistency). The latter is important for advanced optimizers like parfor; otherwise subtle side-effects of program recompilation and hop-lop rewrites possible (e.g., see indexingop hop-lop rewrite in combination parfor rewrite set exec type that eventuelly might lead to unnecessary remote_parfor jobs).- Parameters:
hop- high-level operator
-
rUpdateStatistics
public static void rUpdateStatistics(Hop hop, LocalVariableMap vars)
-
rReplaceLiterals
public static void rReplaceLiterals(Hop hop, ExecutionContext ec, boolean scalarsOnly)
public interface to package local literal replacement- Parameters:
hop- high-level operatorec- Execution contextscalarsOnly- if true, replace only scalar variables but no matrix operations; if false, apply full literal replacement
-
rReplaceLiterals
public static void rReplaceLiterals(Hop hop, LocalVariableMap vars, boolean scalarsOnly)
-
rSetExecType
public static void rSetExecType(Hop hop, Types.ExecType etype)
-
rGetMaxParallelism
public static int rGetMaxParallelism(Hop hop)
-
rSetMaxParallelism
public static void rSetMaxParallelism(Hop hop, int k)
-
checkCPReblock
public static boolean checkCPReblock(ExecutionContext ec, String varin)
CP Reblock check for spark instructions; in contrast to MR, we can not rely on the input file sizes because inputs might be passed via rdds.- Parameters:
ec- execution contextvarin- variable- Returns:
- true if CP reblock?
-
checkCPCheckpoint
public static boolean checkCPCheckpoint(DataCharacteristics dc)
-
executeInMemoryReblock
public static void executeInMemoryReblock(ExecutionContext ec, String varin, String varout)
-
executeInMemoryReblock
public static void executeInMemoryReblock(ExecutionContext ec, String varin, String varout, LineageItem litem)
-
-