Package deepnetts.cudnn

Class CudaTensorBridge

java.lang.Object

deepnetts.cudnn.CudaTensorBridge

All Implemented Interfaces:

TensorAcceleratorBridge

public final class CudaTensorBridge
extends Object
implements TensorAcceleratorBridge

Bridge to tensor on cuda/gpu device. Acts as a bridge between tensor in JVM memory and on-device GPU memory.

Constructor Summary

Constructors

Constructor	Description
CudaTensorBridge(Tensor2D tensor, boolean isBatch)
CudaTensorBridge(TensorBase tensor)	Alocate tensor on cuda device, copy from jvm memory to cuda and back

Method Summary

Modifier and Type	Method	Description
jcuda.Pointer	allocateAndCopyToDev()
void	allocateOnDev()
void	copyToDev()
void	copyToHost()
void	freeOnDevice()
jcuda.Pointer	getHostPointer()
CudaTensorLayout	getLayout()
jcuda.Pointer	getOnDevPointer()
TensorBase	getTensor()
jcuda.jcudnn.cudnnTensorDescriptor	getTensorDescriptor()
boolean	isAllocatedOnDev()
void	printOnDevice()
int	setTensorDesc(jcuda.jcudnn.cudnnTensorDescriptor tensorDesc, int tensorFormat, int dataType, CudaTensorLayout t)

Methods inherited from class java.lang.Object

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

Constructor Details

CudaTensorBridge

public CudaTensorBridge(TensorBase tensor)

Alocate tensor on cuda device, copy from jvm memory to cuda and back

Parameters:

tensor -

CudaTensorBridge

public CudaTensorBridge(Tensor2D tensor,
 boolean isBatch)

Method Details

getLayout

public CudaTensorLayout getLayout()

allocateAndCopyToDev

public jcuda.Pointer allocateAndCopyToDev()

copyToDev

public void copyToDev()

Specified by:

copyToDev in interface TensorAcceleratorBridge

allocateOnDev

public void allocateOnDev()

isAllocatedOnDev

public boolean isAllocatedOnDev()

copyToHost

public void copyToHost()

Specified by:

copyToHost in interface TensorAcceleratorBridge

freeOnDevice

public void freeOnDevice()

getOnDevPointer

public jcuda.Pointer getOnDevPointer()

getHostPointer

public jcuda.Pointer getHostPointer()

getTensor

public TensorBase getTensor()

getTensorDescriptor

public jcuda.jcudnn.cudnnTensorDescriptor getTensorDescriptor()

setTensorDesc

public int setTensorDesc(jcuda.jcudnn.cudnnTensorDescriptor tensorDesc,
 int tensorFormat,
 int dataType,
 CudaTensorLayout t)

printOnDevice

public void printOnDevice()