| Package | Description |
|---|---|
| org.nd4j.linalg.jcublas | |
| org.nd4j.linalg.jcublas.buffer | |
| org.nd4j.linalg.jcublas.context | |
| org.nd4j.linalg.jcublas.kernel | |
| org.nd4j.linalg.jcublas.util |
| Constructor and Description |
|---|
CublasPointer(INDArray array,
CudaContext context)
Creates a CublasPointer for a given INDArray.
|
CublasPointer(JCudaBuffer buffer,
CudaContext context)
Creates a CublasPointer
for a given JCudaBuffer
|
| Modifier and Type | Method and Description |
|---|---|
void |
JCudaBuffer.copyToHost(CudaContext context,
int offset,
int length)
Copy to the host synchronizing
using the given context
|
void |
BaseCudaDataBuffer.copyToHost(CudaContext context,
int offset,
int length) |
| Modifier and Type | Method and Description |
|---|---|
static CudaContext |
CudaContext.getBlasContext()
Sets up a context with an old stream
and a blas handle
|
| Modifier and Type | Method and Description |
|---|---|
static void |
KernelFunctions.invoke(GpuMetrics metrics,
boolean sync,
String functionName,
String dataType,
CudaContext cudaContext,
Object... kernelParameters)
Invoke a function
|
| Modifier and Type | Method and Description |
|---|---|
CudaContext |
KernelParamsWrapper.getContext() |
Copyright © 2015. All Rights Reserved.