public class CudaEngine extends Object
Modifier and Type | Field and Description |
---|---|
protected jcuda.driver.CUcontext |
context |
protected jcuda.driver.CUfunction |
f |
Modifier and Type | Method and Description |
---|---|
void |
addNeighborsPtr(String string,
jcuda.driver.CUdeviceptr neighborsPtr) |
void |
cuCtxSynchronize() |
static void |
cuCtxSynchronizeAll() |
int |
cuDeviceGetCount() |
void |
freeCUObjectsMemory()
Free memory from the currently registered CUObjects
|
static void |
freeMemory()
Stop the executors and clean memory on registered CUObject
|
static CudaEngine |
getCudaEngine(CudaObject co) |
jcuda.driver.CUfunction |
getKernelFunction(turtlekit.cuda.CudaEngine.Kernel f) |
int |
getMaxThreads() |
jcuda.driver.CUdeviceptr |
getNeighborsPtr(String string) |
static ByteBuffer |
getUnifiedByteBuffer(jcuda.Pointer pinnedMemory,
jcuda.driver.CUdeviceptr devicePtr,
int size) |
static int[] |
getUnifiedIntArray(jcuda.Pointer pinnedMemory,
jcuda.driver.CUdeviceptr devicePtr,
int size) |
static IntBuffer |
getUnifiedIntBuffer(jcuda.Pointer pinnedMemory,
jcuda.driver.CUdeviceptr devicePtr,
int size) |
static boolean |
init() |
static boolean |
isCudaAvailable() |
static void |
main(String[] args) |
static void |
stop()
Stop the executors and clean memory on registered CUObject
|
Future<?> |
submit(Runnable runnable) |
public static void main(String[] args)
public static boolean init()
public int cuDeviceGetCount()
public static boolean isCudaAvailable()
public static CudaEngine getCudaEngine(CudaObject co)
public static IntBuffer getUnifiedIntBuffer(jcuda.Pointer pinnedMemory, jcuda.driver.CUdeviceptr devicePtr, int size)
public static int[] getUnifiedIntArray(jcuda.Pointer pinnedMemory, jcuda.driver.CUdeviceptr devicePtr, int size)
public static ByteBuffer getUnifiedByteBuffer(jcuda.Pointer pinnedMemory, jcuda.driver.CUdeviceptr devicePtr, int size)
public static void stop()
public static void freeMemory()
public void freeCUObjectsMemory()
public int getMaxThreads()
public jcuda.driver.CUfunction getKernelFunction(turtlekit.cuda.CudaEngine.Kernel f)
public static void cuCtxSynchronizeAll()
public void cuCtxSynchronize()
public jcuda.driver.CUdeviceptr getNeighborsPtr(String string)
public void addNeighborsPtr(String string, jcuda.driver.CUdeviceptr neighborsPtr)