public interface GPUMemoryAllocator
| Modifier and Type | Method and Description |
|---|---|
void |
allocate(jcuda.Pointer devPtr,
long size)
Allocate memory on the device.
|
boolean |
canAllocate(long size)
Check if there is enough memory to allocate a pointer of given size
|
void |
free(jcuda.Pointer devPtr)
Frees memory on the device
|
long |
getAvailableMemory()
Gets the available memory on GPU that SystemDS can use.
|
void allocate(jcuda.Pointer devPtr,
long size)
throws jcuda.CudaException
devPtr - Pointer to allocated device memorysize - size in bytesjcuda.CudaException - if unable to allocatevoid free(jcuda.Pointer devPtr) throws jcuda.CudaException
devPtr - Device pointer to memory to freejcuda.CudaException - if error occursboolean canAllocate(long size)
size - size in byteslong getAvailableMemory()
Copyright © 2020 The Apache Software Foundation. All rights reserved.