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, createGPUObject
public 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
DMLRuntimeException
public long getAvailableMemory()
GPUContext
getAvailableMemory
in class GPUContext
public void ensureComputeCapability() throws DMLRuntimeException
GPUContext
JCudaContext
.ensureComputeCapability
in class GPUContext
DMLRuntimeException
- if DMLRuntimeException occurspublic void destroy() throws DMLRuntimeException
destroy
in class GPUContext
DMLRuntimeException
Copyright © 2017 The Apache Software Foundation. All rights reserved.