public class CudaMemoryManager extends BasicMemoryManager
Constructor and Description |
---|
CudaMemoryManager() |
Modifier and Type | Method and Description |
---|---|
org.bytedeco.javacpp.Pointer |
allocate(long bytes,
MemoryKind kind,
boolean initialize)
This method returns
PLEASE NOTE: Cache options depend on specific implementations
|
void |
collect(INDArray... arrays)
This method detaches off-heap memory from passed INDArray instances, and optionally stores them in cache for future reuse
PLEASE NOTE: Cache options depend on specific implementations
|
void |
memcpy(DataBuffer dstBuffer,
DataBuffer srcBuffer)
This method provides basic memcpy functionality with respect to target environment
|
void |
purgeCaches()
This method purges all cached memory chunks
PLEASE NOTE: This method SHOULD NOT EVER BE USED without being 146% clear of all consequences.
|
public org.bytedeco.javacpp.Pointer allocate(long bytes, MemoryKind kind, boolean initialize)
allocate
in interface MemoryManager
allocate
in class BasicMemoryManager
bytes
- kind
- initialize
- public void collect(INDArray... arrays)
collect
in interface MemoryManager
collect
in class BasicMemoryManager
arrays
- public void purgeCaches()
purgeCaches
in interface MemoryManager
purgeCaches
in class BasicMemoryManager
public void memcpy(DataBuffer dstBuffer, DataBuffer srcBuffer)
memcpy
in interface MemoryManager
memcpy
in class BasicMemoryManager
dstBuffer
- srcBuffer
- Copyright © 2016. All Rights Reserved.