org.nd4j.linalg.jcublas.kernel

Class KernelFunctions

java.lang.Object

org.nd4j.linalg.jcublas.kernel.KernelFunctions

public class KernelFunctions
extends Object

Kernel functions.

Derived from: http://www.jcuda.org/samples/JCudaVectorAdd.java

Author:

Adam Gibson

Field Summary

Fields

Modifier and Type	Field and Description
static int	BLOCKS
static String	BLOCKS_KEY
static String	DOUBLE
static String	FLOAT
static String	NAME_SPACE
static String	REDUCE
static int	SHARED_MEM
static String	SHARED_MEM_KEY
static int	THREADS
static String	THREADS_KEY

Method Summary

Modifier and Type	Method and Description
static JCudaBuffer	alloc(double[] data) Allocate a pointer of a given data type
static JCudaBuffer	alloc(float[] data) Allocate a pointer of a given data type
static JCudaBuffer	alloc(int[] data) Allocate a pointer of a given data type
static void	invoke(GpuMetrics metrics, boolean sync, String functionName, String dataType, CudaContext cudaContext, Object... kernelParameters) Invoke a function
static void	register() Called at initialization in the static context.

Methods inherited from class java.lang.Object

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

Field Detail

NAME_SPACE

public static final String NAME_SPACE

See Also:

Constant Field Values

DOUBLE

public static final String DOUBLE

See Also:

Constant Field Values

FLOAT

public static final String FLOAT

See Also:

Constant Field Values

REDUCE

public static final String REDUCE

See Also:

Constant Field Values

SHARED_MEM_KEY

public static final String SHARED_MEM_KEY

See Also:

Constant Field Values

THREADS_KEY

public static final String THREADS_KEY

See Also:

Constant Field Values

BLOCKS_KEY

public static final String BLOCKS_KEY

See Also:

Constant Field Values

SHARED_MEM

public static int SHARED_MEM

THREADS

public static int THREADS

BLOCKS

public static int BLOCKS

Method Detail

register

public static void register()
                     throws Exception

Called at initialization in the static context. Registers cuda functions based on the cudafunctions.properties in the classpath

Throws:

IOException

Exception

invoke

public static void invoke(GpuMetrics metrics,
                          boolean sync,
                          String functionName,
                          String dataType,
                          CudaContext cudaContext,
                          Object... kernelParameters)

Invoke a function

Parameters:

metrics -

functionName -

dataType -

cudaContext -

kernelParameters -

alloc

public static JCudaBuffer alloc(int[] data)

Allocate a pointer of a given data type

Parameters:

data - the data for the pointer

Returns:

the pointer

alloc

public static JCudaBuffer alloc(double[] data)

Allocate a pointer of a given data type

Parameters:

data - the data for the pointer

Returns:

the pointer

alloc

public static JCudaBuffer alloc(float[] data)

Allocate a pointer of a given data type

Parameters:

data - the data for the pointer

Returns:

the pointer