org.nd4j.linalg.jcublas.parallel

Class GpuParallelExecutioner

java.lang.Object

org.nd4j.linalg.api.parallel.DefaultParallelExecutioner

org.nd4j.linalg.jcublas.parallel.GpuParallelExecutioner

All Implemented Interfaces:

ParallelExecutioner

public class GpuParallelExecutioner
extends DefaultParallelExecutioner

Created by agibsonccc on 10/3/15.

Field Summary

Fields inherited from class org.nd4j.linalg.api.parallel.DefaultParallelExecutioner

ENABLED

Constructor Summary

Constructors

Constructor and Description
GpuParallelExecutioner()

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)

Methods inherited from class org.nd4j.linalg.api.parallel.DefaultParallelExecutioner

getEnabled, parallelEnabled, setParallelEnabled

Methods inherited from class java.lang.Object

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

Constructor Detail

GpuParallelExecutioner

public GpuParallelExecutioner()

Method Detail

execBasedOnArraysAlongDimension

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

Specified by:

execBasedOnArraysAlongDimension in interface ParallelExecutioner

Overrides:

execBasedOnArraysAlongDimension in class DefaultParallelExecutioner

execBasedOnArraysAlongDimension

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

Specified by:

execBasedOnArraysAlongDimension in interface ParallelExecutioner

Overrides:

execBasedOnArraysAlongDimension in class DefaultParallelExecutioner

execBasedOnSlices

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

Specified by:

execBasedOnSlices in interface ParallelExecutioner

Overrides:

execBasedOnSlices in class DefaultParallelExecutioner

execBasedOnArraysAlongDimension

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

Specified by:

execBasedOnArraysAlongDimension in interface ParallelExecutioner

Overrides:

execBasedOnArraysAlongDimension in class DefaultParallelExecutioner

execBasedOnArraysAlongDimension

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

Specified by:

execBasedOnArraysAlongDimension in interface ParallelExecutioner

Overrides:

execBasedOnArraysAlongDimension in class DefaultParallelExecutioner

execBasedOnSlices

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

Specified by:

execBasedOnSlices in interface ParallelExecutioner

Overrides:

execBasedOnSlices in class DefaultParallelExecutioner

exec

public Future exec(Runnable runnable)

Specified by:

exec in interface ParallelExecutioner

Overrides:

exec in class DefaultParallelExecutioner

exec

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

Specified by:

exec in interface ParallelExecutioner

Overrides:

exec in class DefaultParallelExecutioner