public class JCudaContext extends GPUContext
| Modifier and Type | Field and Description |
|---|---|
static AtomicLong |
availableNumBytesWithoutUtilFactor |
static boolean |
DEBUG |
double |
GPU_MEMORY_UTILIZATION_FACTOR |
boolean |
REFRESH_AVAILABLE_MEMORY_EVERY_TIME |
static long |
totalNumBytes |
allocatedPointers, currContext, isGPUContextCreated| Constructor and Description |
|---|
JCudaContext() |
| Modifier and Type | Method and Description |
|---|---|
void |
destroy() |
void |
ensureComputeCapability()
Ensures that all the CUDA cards on the current system are
of the minimum required compute capability.
|
long |
getAvailableMemory()
Gets device memory available for SystemML operations
|
createGPUContext, createGPUObjectpublic static boolean DEBUG
public static long totalNumBytes
public static AtomicLong availableNumBytesWithoutUtilFactor
public boolean REFRESH_AVAILABLE_MEMORY_EVERY_TIME
public double GPU_MEMORY_UTILIZATION_FACTOR
public JCudaContext()
throws DMLRuntimeException
DMLRuntimeExceptionpublic long getAvailableMemory()
GPUContextgetAvailableMemory in class GPUContextpublic void ensureComputeCapability()
throws DMLRuntimeException
GPUContextJCudaContext.ensureComputeCapability in class GPUContextDMLRuntimeException - if DMLRuntimeException occurspublic void destroy()
throws DMLRuntimeException
destroy in class GPUContextDMLRuntimeExceptionCopyright © 2017 The Apache Software Foundation. All rights reserved.