| Package | Description |
|---|---|
| jcuda.driver |
Contains the classes related to the JCuda driver API.
|
| jcuda.jcublas |
Contains the classes of JCublas.
|
| jcuda.jcufft |
Contains the classes of JCufft.
|
| jcuda.jcurand |
Contains the classes of JCurand.
|
| jcuda.jcusparse |
Contains the classes of JCusparse.
|
| jcuda.runtime |
Contains the classes related to the JCuda runtime API.
|
| org.nd4j.linalg.jcublas.context | |
| org.nd4j.linalg.jcublas.context.pool | |
| org.nd4j.linalg.jcublas.context.pool.factory |
| Constructor and Description |
|---|
CUstream(cudaStream_t stream)
Creates a CUstream for the given
cudaStream_t. |
| Modifier and Type | Method and Description |
|---|---|
static int |
JCublas2.cublasGetMatrixAsync(int rows,
int cols,
int elemSize,
Pointer A,
int lda,
Pointer B,
int ldb,
cudaStream_t stream)
cublasStatus_t
cublasGetMatrixAsync (int rows, int cols, int elemSize, const void *A,
int lda, void *B, int ldb, cudaStream_t stream)
cublasGetMatrixAsync has the same functionnality as cublasGetMatrix
but the transfer is done asynchronously within the CUDA stream passed
in parameter.
|
static int |
JCublas.cublasGetMatrixAsync(int rows,
int cols,
int elemSize,
Pointer A,
int lda,
Pointer B,
int ldb,
cudaStream_t stream) |
static int |
JCublas2.cublasGetStream(cublasHandle handle,
cudaStream_t streamId) |
static int |
JCublas2.cublasGetVectorAsync(int n,
int elemSize,
Pointer devicePtr,
int incx,
Pointer hostPtr,
int incy,
cudaStream_t stream)
cublasStatus
cublasGetVectorAsync( int n, int elemSize, const void *x, int incx,
void *y, int incy, cudaStream_t stream)
cublasGetVectorAsync has the same functionnality as cublasGetVector
but the transfer is done asynchronously within the CUDA stream passed
in parameter.
|
static int |
JCublas.cublasGetVectorAsync(int n,
int elemSize,
Pointer devicePtr,
int incx,
Pointer hostPtr,
int incy,
cudaStream_t stream) |
static int |
JCublas.cublasSetKernelStream(cudaStream_t stream) |
static int |
JCublas2.cublasSetMatrixAsync(int rows,
int cols,
int elemSize,
Pointer A,
int lda,
Pointer B,
int ldb,
cudaStream_t stream)
cublasStatus_t
cublasSetMatrixAsync (int rows, int cols, int elemSize, const void *A,
int lda, void *B, int ldb, cudaStream_t stream)
cublasSetMatrixAsync has the same functionnality as cublasSetMatrix
but the transfer is done asynchronously within the CUDA stream passed
in parameter.
|
static int |
JCublas.cublasSetMatrixAsync(int rows,
int cols,
int elemSize,
Pointer A,
int lda,
Pointer B,
int ldb,
cudaStream_t stream) |
static int |
JCublas2.cublasSetStream(cublasHandle handle,
cudaStream_t streamId) |
static int |
JCublas2.cublasSetVectorAsync(int n,
int elemSize,
Pointer hostPtr,
int incx,
Pointer devicePtr,
int incy,
cudaStream_t stream)
cublasStatus
cublasSetVectorAsync ( int n, int elemSize, const void *x, int incx,
void *y, int incy, cudaStream_t stream );
cublasSetVectorAsync has the same functionnality as cublasSetVector
but the transfer is done asynchronously within the CUDA stream passed
in parameter.
|
static int |
JCublas.cublasSetVectorAsync(int n,
int elemSize,
Pointer hostPtr,
int incx,
Pointer devicePtr,
int incy,
cudaStream_t stream) |
| Modifier and Type | Method and Description |
|---|---|
static int |
JCufft.cufftSetStream(cufftHandle plan,
cudaStream_t stream)
Associates a CUDA stream with a CUFFT plan.
|
| Modifier and Type | Method and Description |
|---|---|
static int |
JCurand.curandSetStream(curandGenerator generator,
cudaStream_t stream)
Set the current stream for CURAND kernel launches.
|
| Modifier and Type | Method and Description |
|---|---|
static int |
JCusparse.cusparseSetStream(cusparseHandle handle,
cudaStream_t streamId) |
| Modifier and Type | Field and Description |
|---|---|
static cudaStream_t |
JCuda.cudaStreamLegacy
Stream handle that can be passed as a cudaStream_t to use an implicit stream
with legacy synchronization behavior.
|
static cudaStream_t |
JCuda.cudaStreamPerThread
Stream handle that can be passed as a cudaStream_t to use an implicit stream
with per-thread synchronization behavior.
|
| Modifier and Type | Method and Description |
|---|---|
void |
cudaStreamCallback.call(cudaStream_t stream,
int status,
Object userData)
The function that will be called
|
static int |
JCuda.cudaConfigureCall(dim3 gridDim,
dim3 blockDim,
long sharedMem,
cudaStream_t stream)
Deprecated.
This function is deprecated as of CUDA 7.0
|
static int |
JCuda.cudaEventRecord(cudaEvent_t event,
cudaStream_t stream)
Records an event.
|
static int |
JCuda.cudaGLMapBufferObjectAsync(Pointer devPtr,
int bufObj,
cudaStream_t stream)
Deprecated.
Deprecated as of CUDA 3.0
|
static int |
JCuda.cudaGLUnmapBufferObjectAsync(int bufObj,
cudaStream_t stream)
Deprecated.
Deprecated as of CUDA 3.0
|
static int |
JCuda.cudaGraphicsMapResources(int count,
cudaGraphicsResource[] resources,
cudaStream_t stream)
Map graphics resources for access by CUDA.
|
static int |
JCuda.cudaGraphicsUnmapResources(int count,
cudaGraphicsResource[] resources,
cudaStream_t stream)
Unmap graphics resources.
|
static int |
JCuda.cudaMemcpy2DAsync(Pointer dst,
long dpitch,
Pointer src,
long spitch,
long width,
long height,
int cudaMemcpyKind_kind,
cudaStream_t stream)
Copies data between host and device.
|
static int |
JCuda.cudaMemcpy2DFromArrayAsync(Pointer dst,
long dpitch,
cudaArray src,
long wOffset,
long hOffset,
long width,
long height,
int cudaMemcpyKind_kind,
cudaStream_t stream)
Copies data between host and device.
|
static int |
JCuda.cudaMemcpy2DToArrayAsync(cudaArray dst,
long wOffset,
long hOffset,
Pointer src,
long spitch,
long width,
long height,
int cudaMemcpyKind_kind,
cudaStream_t stream)
Copies data between host and device.
|
static int |
JCuda.cudaMemcpy3DAsync(cudaMemcpy3DParms p,
cudaStream_t stream)
Copies data between 3D objects.
|
static int |
JCuda.cudaMemcpy3DPeerAsync(cudaMemcpy3DPeerParms p,
cudaStream_t stream)
Copies memory between devices asynchronously.
|
static int |
JCuda.cudaMemcpyAsync(Pointer dst,
Pointer src,
long count,
int cudaMemcpyKind_kind,
cudaStream_t stream)
Copies data between host and device.
|
static int |
JCuda.cudaMemcpyFromArrayAsync(Pointer dst,
cudaArray src,
long wOffset,
long hOffset,
long count,
int cudaMemcpyKind_kind,
cudaStream_t stream)
Copies data between host and device.
|
static int |
JCuda.cudaMemcpyFromSymbolAsync(Pointer dst,
String symbol,
long count,
long offset,
int cudaMemcpyKind_kind,
cudaStream_t stream)
[C++ API] Copies data from the given symbol on the device
|
static int |
JCuda.cudaMemcpyPeerAsync(Pointer dst,
int dstDevice,
Pointer src,
int srcDevice,
long count,
cudaStream_t stream)
Copies memory between two devices asynchronously.
|
static int |
JCuda.cudaMemcpyToArrayAsync(cudaArray dst,
long wOffset,
long hOffset,
Pointer src,
long count,
int cudaMemcpyKind_kind,
cudaStream_t stream)
Copies data between host and device.
|
static int |
JCuda.cudaMemcpyToSymbolAsync(String symbol,
Pointer src,
long count,
long offset,
int cudaMemcpyKind_kind,
cudaStream_t stream)
[C++ API] Copies data to the given symbol on the device
|
static int |
JCuda.cudaMemset2DAsync(Pointer devPtr,
long pitch,
int value,
long width,
long height,
cudaStream_t stream)
Initializes or sets device memory to a value.
|
static int |
JCuda.cudaMemset3DAsync(cudaPitchedPtr pitchedDevPtr,
int value,
cudaExtent extent,
cudaStream_t stream)
Initializes or sets device memory to a value.
|
static int |
JCuda.cudaMemsetAsync(Pointer devPtr,
int value,
long count,
cudaStream_t stream)
Initializes or sets device memory to a value.
|
static int |
JCuda.cudaStreamAddCallback(cudaStream_t stream,
cudaStreamCallback callback,
Object userData,
int flags)
Add a callback to a compute stream.
|
static int |
JCuda.cudaStreamAttachMemAsync(cudaStream_t stream,
Pointer devPtr,
long length,
int flags) |
static int |
JCuda.cudaStreamCreate(cudaStream_t stream)
Create an asynchronous stream.
|
static int |
JCuda.cudaStreamCreateWithFlags(cudaStream_t pStream,
int flags)
Create an asynchronous stream.
|
static int |
JCuda.cudaStreamCreateWithPriority(cudaStream_t pStream,
int flags,
int priority) |
static int |
JCuda.cudaStreamDestroy(cudaStream_t stream)
Destroys and cleans up an asynchronous stream.
|
static int |
JCuda.cudaStreamGetFlags(cudaStream_t hStream,
int[] flags) |
static int |
JCuda.cudaStreamGetPriority(cudaStream_t hStream,
int[] priority) |
static int |
JCuda.cudaStreamQuery(cudaStream_t stream)
Queries an asynchronous stream for completion status.
|
static int |
JCuda.cudaStreamSynchronize(cudaStream_t stream)
Waits for stream tasks to complete.
|
static int |
JCuda.cudaStreamWaitEvent(cudaStream_t stream,
cudaEvent_t event,
int flags)
Make a compute stream wait on an event.
|
| Modifier and Type | Method and Description |
|---|---|
cudaStream_t |
ContextHolder.getCudaStream()
Get the stream for the current thread
based on the device for the thread
|
| Modifier and Type | Method and Description |
|---|---|
org.apache.commons.pool2.ObjectPool<cudaStream_t> |
ContextHolder.getOldStreamPool() |
| Constructor and Description |
|---|
OldStreamPool(org.apache.commons.pool2.PooledObjectFactory<cudaStream_t> factory) |
OldStreamPool(org.apache.commons.pool2.PooledObjectFactory<cudaStream_t> factory,
org.apache.commons.pool2.impl.GenericObjectPoolConfig config) |
OldStreamPool(org.apache.commons.pool2.PooledObjectFactory<cudaStream_t> factory,
org.apache.commons.pool2.impl.GenericObjectPoolConfig config,
org.apache.commons.pool2.impl.AbandonedConfig abandonedConfig) |
| Modifier and Type | Method and Description |
|---|---|
cudaStream_t |
OldStreamItemFactory.create() |
| Modifier and Type | Method and Description |
|---|---|
org.apache.commons.pool2.PooledObject<cudaStream_t> |
OldStreamItemFactory.wrap(cudaStream_t cudaStream_t) |
| Modifier and Type | Method and Description |
|---|---|
org.apache.commons.pool2.PooledObject<cudaStream_t> |
OldStreamItemFactory.wrap(cudaStream_t cudaStream_t) |
| Modifier and Type | Method and Description |
|---|---|
void |
OldStreamItemFactory.destroyObject(org.apache.commons.pool2.PooledObject<cudaStream_t> p) |
Copyright © 2015. All Rights Reserved.