public class GPULazyCudaFreeMemoryManager extends Object
| Constructor and Description | 
|---|
| GPULazyCudaFreeMemoryManager(GPUMemoryManager gpuManager) | 
| Modifier and Type | Method and Description | 
|---|---|
| void | add(long size,
   jcuda.Pointer toFree)Add a pointer to the rmvar-ed list | 
| void | clearAll() | 
| Set<jcuda.Pointer> | getAllPointers() | 
| int | getNumPointers()Get total number of rmvared pointers | 
| jcuda.Pointer | getRmvarPointer(String opcode,
               long size)Get any pointer of the given size from rmvar-ed pointers (applicable if eager cudaFree is set to false) | 
| jcuda.Pointer | getRmvarPointerMinSize(String opcode,
                      long minSize) | 
| long | getTotalMemoryAllocated()Return the total memory in bytes used by this memory manager | 
| void | removeIfPresent(long size,
               jcuda.Pointer ptr)Remove a specific pointer if present in the internal hashmap | 
public GPULazyCudaFreeMemoryManager(GPUMemoryManager gpuManager)
public jcuda.Pointer getRmvarPointer(String opcode, long size)
opcode - opcodesize - size in bytespublic Set<jcuda.Pointer> getAllPointers()
public void clearAll()
public jcuda.Pointer getRmvarPointerMinSize(String opcode, long minSize) throws DMLRuntimeException
DMLRuntimeExceptionpublic long getTotalMemoryAllocated()
public int getNumPointers()
public void add(long size,
                jcuda.Pointer toFree)
size - size of the pointertoFree - pointerpublic void removeIfPresent(long size,
                            jcuda.Pointer ptr)
size - size in bytesptr - pointer to be removedCopyright © 2020 The Apache Software Foundation. All rights reserved.