org.nd4j.linalg.api.buffer.allocation

Interface MemoryStrategy

public interface MemoryStrategy

An allocation strategy handles allocating and freeing memory for the gpu (usually relative to the compute capabilities of the gpu)

Author:

Adam Gibson

Method Summary

Modifier and Type	Method and Description
java.lang.Object	alloc(DataBuffer buffer, int stride, int offset, int length) Allocate memory for the given buffer
java.lang.Object	copyToHost(DataBuffer copy, int offset) Copy data to native or gpu
void	free(DataBuffer buffer, int offset, int length) Free the buffer wrt the allocation strategy
void	setData(DataBuffer buffer, int offset)
void	setData(DataBuffer buffer, int offset, int stride, int length) Set the data for the buffer

Method Detail

setData

void setData(DataBuffer buffer,
             int offset,
             int stride,
             int length)

Set the data for the buffer

Parameters:

buffer - the buffer to set

offset - the offset to start at

stride - the stride to sue

length - the length to go till

setData

void setData(DataBuffer buffer,
             int offset)

Parameters:

buffer -

offset -

copyToHost

java.lang.Object copyToHost(DataBuffer copy,
                            int offset)

Copy data to native or gpu

Parameters:

copy - the buffer to copy

Returns:

a pointer representing the copied data

alloc

java.lang.Object alloc(DataBuffer buffer,
                       int stride,
                       int offset,
                       int length)

Allocate memory for the given buffer

Parameters:

buffer - the buffer to allocate for

stride - the stride

offset - the offset used for the buffer on allocation

length - length

free

void free(DataBuffer buffer,
          int offset,
          int length)

Free the buffer wrt the allocation strategy

Parameters:

buffer - the buffer to free

offset - the offset to free

length - the length to free