Class NemoOptimizer

  • All Implemented Interfaces:
    Optimizer

    @NotThreadSafe
    public final class NemoOptimizer
    extends Object
    implements Optimizer
    An interface for optimizer, which manages the optimization over submitted IR DAGs through Policys. The instance of this class will reside in driver.
    • Method Detail

      • optimizeAtCompileTime

        public IRDAG optimizeAtCompileTime​(IRDAG dag)
        Description copied from interface: Optimizer
        Optimize the submitted DAG at compile time.
        Specified by:
        optimizeAtCompileTime in interface Optimizer
        Parameters:
        dag - the input DAG to optimize.
        Returns:
        optimized DAG, reshaped or tagged with execution properties.
      • optimizeAtRunTime

        public IRDAG optimizeAtRunTime​(IRDAG dag,
                                       Message message)
        Description copied from interface: Optimizer
        Optimize the submitted DAG at run time.
        Specified by:
        optimizeAtRunTime in interface Optimizer
        Parameters:
        dag - input.
        message - for optimization.
        Returns:
        optimized DAG.