| Package | Description |
|---|---|
| jcuda.utils | |
| org.nd4j.linalg.jcublas.kernel |
| Modifier and Type | Method and Description |
|---|---|
static KernelLauncher |
KernelLauncher.compile(String sourceCode,
String functionName,
String... nvccArguments)
Create a new KernelLauncher for the function with the given
name, that is defined in the given source code.
|
static KernelLauncher |
KernelLauncher.create(String cuFileName,
String functionName,
boolean forceRebuild,
String... nvccArguments)
Create a new KernelLauncher for the function with the given
name, that is contained in the .CU CUDA source file with the
given name.
|
static KernelLauncher |
KernelLauncher.create(String cuFileName,
String functionName,
String... nvccArguments)
Create a new KernelLauncher for the function with the given
name, that is contained in the .CU CUDA source file with the
given name.
|
KernelLauncher |
KernelLauncher.forFunction(String functionName)
Create a new KernelLauncher which uses the same module as
this KernelLauncher, but may be used to execute a different
function.
|
static KernelLauncher |
KernelLauncher.load(InputStream moduleInputStream,
String functionName)
Create a new KernelLauncher which may be used to execute the
specified function which is loaded from the PTX- or CUBIN
data that is read from the given input stream.
|
static KernelLauncher |
KernelLauncher.load(String functionName,
String type,
CUmodule module)
Create a new KernelLauncher which may be used to execute the
specified function which is loaded from the PTX- or CUBIN
(CUDA binary) file with the given name.
|
static KernelLauncher |
KernelLauncher.load(String moduleFileName,
String functionName,
String type)
Create a new KernelLauncher which may be used to execute the
specified function which is loaded from the PTX- or CUBIN
(CUDA binary) file with the given name.
|
KernelLauncher |
KernelLauncher.setBlockSize(int x,
int y,
int z)
Set the block size (number of threads per block) for the function
call.
This corresponds to the second parameter in the runtime call:
kernel<<<gridSize, blockSize,
sharedMemSize, stream>>>(...);
The default block size is (1,1,1) |
KernelLauncher |
KernelLauncher.setGridSize(int x,
int y)
Set the grid size (number of blocks per grid) for the function
call.
This corresponds to the first parameter in the runtime call:
kernel<<<gridSize, blockSize,
sharedMemSize, stream>>>(...);
The default grid size is (1,1,1) |
KernelLauncher |
KernelLauncher.setGridSize(int x,
int y,
int z)
Set the grid size (number of blocks per grid) for the function
call.
This corresponds to the first parameter in the runtime call:
kernel<<<gridSize, blockSize,
sharedMemSize, stream>>>(...);
The default grid size is (1,1,1) |
KernelLauncher |
KernelLauncher.setSharedMemSize(int sharedMemSize)
Set the size of the shared memory for the function
call.
This corresponds to the third parameter in the runtime call:
kernel<<<gridSize, blockSize,
sharedMemSize, stream>>>(...);
The default shared memory size is 0. |
KernelLauncher |
KernelLauncher.setStream(CUstream stream)
Set the stream for the function call.
This corresponds to the fourth parameter in the runtime call:
kernel<<<gridSize, blockSize,
sharedMemSize, stream>>>(...);
The default stream is null (0). |
KernelLauncher |
KernelLauncher.setup(dim3 gridSize,
dim3 blockSize)
Set the given grid size and block size for this KernelLauncher.
|
KernelLauncher |
KernelLauncher.setup(dim3 gridSize,
dim3 blockSize,
int sharedMemSize)
Set the given grid size and block size and shared memory size
for this KernelLauncher.
|
KernelLauncher |
KernelLauncher.setup(dim3 gridSize,
dim3 blockSize,
int sharedMemSize,
CUstream stream)
Set the given grid size and block size, shared memory size
and stream for this KernelLauncher.
|
| Modifier and Type | Method and Description |
|---|---|
KernelLauncher |
KernelFunctionLoader.get(String functionName,
String dataType)
Gets a kernel launcher
for a given function name and data type
|
static KernelLauncher |
KernelFunctionLoader.launcher(String functionName,
String dataType)
Get the launcher for a function
|
Copyright © 2015. All Rights Reserved.