org.nd4j.linalg.api.parallel

Class DefaultParallelExecutioner

java.lang.Object

org.nd4j.linalg.api.parallel.DefaultParallelExecutioner

All Implemented Interfaces:

ParallelExecutioner

public class DefaultParallelExecutioner
extends Object
implements ParallelExecutioner

Parallel executioner. Meant for multi threaded execution of vector and tensor operations such as dimension wise or slice wise operations.

Author:

Adam Gibson

Field Summary

Fields

Modifier and Type	Field and Description
static String	ENABLED

Constructor Summary

Constructors

Constructor and Description
DefaultParallelExecutioner()
DefaultParallelExecutioner(ExecutorService executorService)
DefaultParallelExecutioner(ForkJoinPool forkJoinPool)

Method Summary

Modifier and Type	Method and Description
<T> void	exec(ForkJoinTask<T> task)
Future	exec(Runnable runnable)
void	execBasedOnArraysAlongDimension(INDArray[] arr, TaskCreator.INDArrayTask task, int... dimension)
INDArray	execBasedOnArraysAlongDimension(INDArray arr, Accumulation task, OpExecutioner executioner, int... dimension)
void	execBasedOnArraysAlongDimension(INDArray arr, Op task, OpExecutioner executioner, int... dimension)
void	execBasedOnArraysAlongDimension(INDArray arr, TaskCreator.INDArrayTask task, int... dimension)
void	execBasedOnSlices(INDArray arr, Op task, OpExecutioner executioner)
void	execBasedOnSlices(INDArray arr, TaskCreator.INDArrayTask task)
static boolean	getEnabled()
boolean	parallelEnabled() Whether the parallel execution is enabled
void	setParallelEnabled(boolean parallelEnabled) Sets parallel enabled

Methods inherited from class java.lang.Object

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

Field Detail

ENABLED

public static final String ENABLED

See Also:

Constant Field Values

Constructor Detail

DefaultParallelExecutioner

public DefaultParallelExecutioner(ForkJoinPool forkJoinPool)

DefaultParallelExecutioner

public DefaultParallelExecutioner(ExecutorService executorService)

DefaultParallelExecutioner

public DefaultParallelExecutioner()

Method Detail

getEnabled

public static boolean getEnabled()

setParallelEnabled

public void setParallelEnabled(boolean parallelEnabled)

Description copied from interface: ParallelExecutioner

Sets parallel enabled

Specified by:

setParallelEnabled in interface ParallelExecutioner

parallelEnabled

public boolean parallelEnabled()

Description copied from interface: ParallelExecutioner

Whether the parallel execution is enabled

Specified by:

parallelEnabled in interface ParallelExecutioner

Returns:

true if the parallel execution is enabled false otherwise

execBasedOnArraysAlongDimension

public INDArray execBasedOnArraysAlongDimension(INDArray arr,
                                                Accumulation task,
                                                OpExecutioner executioner,
                                                int... dimension)

Specified by:

execBasedOnArraysAlongDimension in interface ParallelExecutioner

execBasedOnArraysAlongDimension

public void execBasedOnArraysAlongDimension(INDArray arr,
                                            Op task,
                                            OpExecutioner executioner,
                                            int... dimension)

Specified by:

execBasedOnArraysAlongDimension in interface ParallelExecutioner

execBasedOnSlices

public void execBasedOnSlices(INDArray arr,
                              Op task,
                              OpExecutioner executioner)

Specified by:

execBasedOnSlices in interface ParallelExecutioner

execBasedOnArraysAlongDimension

public void execBasedOnArraysAlongDimension(INDArray arr,
                                            TaskCreator.INDArrayTask task,
                                            int... dimension)

Specified by:

execBasedOnArraysAlongDimension in interface ParallelExecutioner

execBasedOnArraysAlongDimension

public void execBasedOnArraysAlongDimension(INDArray[] arr,
                                            TaskCreator.INDArrayTask task,
                                            int... dimension)

Specified by:

execBasedOnArraysAlongDimension in interface ParallelExecutioner

execBasedOnSlices

public void execBasedOnSlices(INDArray arr,
                              TaskCreator.INDArrayTask task)

Specified by:

execBasedOnSlices in interface ParallelExecutioner

exec

public Future exec(Runnable runnable)

Specified by:

exec in interface ParallelExecutioner

Returns:

exec

public <T> void exec(ForkJoinTask<T> task)

Specified by:

exec in interface ParallelExecutioner