org.nd4j.linalg.jcublas.util

Class KernelParamsWrapper

java.lang.Object

org.nd4j.linalg.jcublas.util.KernelParamsWrapper

All Implemented Interfaces:

AutoCloseable

public class KernelParamsWrapper
extends Object
implements AutoCloseable

Wraps the generation of kernel parameters , creating, copying and destroying any cuda device allocations

Author:

bam4d

Field Summary

Fields

Modifier and Type	Field and Description
Object[]	kernelParameters List of processed kernel parameters ready to be passed to the kernel

Constructor Summary

Constructors

Constructor and Description
KernelParamsWrapper(Op op, boolean closeContext, Object... kernelParams) Create a new wrapper for the kernel parameters.
KernelParamsWrapper(Op op, Object... kernelParams) Create a new wrapper for the kernel parameters.

Method Summary

Modifier and Type	Method and Description
void	close() Free all the buffers from this kernel's parameters
CudaContext	getContext()
Object[]	getKernelParameters() The list of processed kernel parameters, These should be get passed to the cuda kernel
KernelParamsWrapper	setResultArray(INDArray array) set the array that will contain the results, If the array is not set, then data from the device will not be copied to the host
KernelParamsWrapper	setResultOp(Accumulation op, INDArray result) set the Op that this result is for
void	sync() Sync the streams

Methods inherited from class java.lang.Object

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

Field Detail

kernelParameters

public final Object[] kernelParameters

List of processed kernel parameters ready to be passed to the kernel

Constructor Detail

KernelParamsWrapper

public KernelParamsWrapper(Op op,
                           Object... kernelParams)

Create a new wrapper for the kernel parameters. This wrapper manages the host - and device communication and. To set the result on a specific operation, use setResultOp() To set the array which is the result INDArray, use setResultArray()

Parameters:

kernelParams -

KernelParamsWrapper

public KernelParamsWrapper(Op op,
                           boolean closeContext,
                           Object... kernelParams)

Create a new wrapper for the kernel parameters. This wrapper manages the host - and device communication and. To set the result on a specific operation, use setResultOp() To set the array which is the result INDArray, use setResultArray()

Parameters:

kernelParams -

Method Detail

getKernelParameters

public Object[] getKernelParameters()

The list of processed kernel parameters, These should be get passed to the cuda kernel

Returns:

setResultArray

public KernelParamsWrapper setResultArray(INDArray array)

set the array that will contain the results, If the array is not set, then data from the device will not be copied to the host

Parameters:

array -

Returns:

setResultOp

public KernelParamsWrapper setResultOp(Accumulation op,
                                       INDArray result)

set the Op that this result is for

Parameters:

op -

result -

Returns:

close

public void close()
           throws Exception

Free all the buffers from this kernel's parameters

Specified by:

close in interface AutoCloseable

Throws:

Exception

getContext

public CudaContext getContext()

sync

public void sync()

Sync the streams