org.nd4j.linalg.api.concurrency

Interface AffinityManager

All Known Implementing Classes:

BasicAffinityManager

public interface AffinityManager

Author:

[email protected]

Nested Class Summary

Nested Classes

Modifier and Type	Interface and Description
static class	AffinityManager.Location

Method Summary

Modifier and Type	Method and Description
void	attachThreadToDevice(long threadId, Integer deviceId) This method attaches specified thread (by Id) to specified device
void	attachThreadToDevice(Thread thread, Integer deviceId) This method attaches specified thread to specified device
Integer	getDeviceForCurrentThread() This method returns deviceId for current thread
Integer	getDeviceForThread(long threadId) This method returns
Integer	getDeviceForThread(Thread thread) This method returns deviceId for specified thread
int	getNumberOfDevices() This method returns number of available devices
DataBuffer	replicateToDevice(Integer deviceId, DataBuffer buffer) This method replicates given DataBuffer, and places it to target device.
INDArray	replicateToDevice(Integer deviceId, INDArray array) This method replicates given INDArray, and places it to target device.
void	tagLocation(DataBuffer buffer, AffinityManager.Location location) This method tags specific DataBuffer as "recent" on specified location
void	tagLocation(INDArray array, AffinityManager.Location location) This method tags specific INDArray as "recent" on specified location
void	touch(DataBuffer buffer) Utility method, to associate INDArray with specific device (backend-specific)
void	touch(INDArray array) Utility method, to associate INDArray with specific device (backend-specific)

Method Detail

getDeviceForCurrentThread

Integer getDeviceForCurrentThread()

This method returns deviceId for current thread

Returns:

getDeviceForThread

Integer getDeviceForThread(Thread thread)

This method returns deviceId for specified thread

Parameters:

thread -

Returns:

getDeviceForThread

Integer getDeviceForThread(long threadId)

This method returns

Parameters:

threadId -

Returns:

attachThreadToDevice

void attachThreadToDevice(Thread thread,
                          Integer deviceId)

This method attaches specified thread to specified device

Parameters:

thread -

deviceId -

attachThreadToDevice

void attachThreadToDevice(long threadId,
                          Integer deviceId)

This method attaches specified thread (by Id) to specified device

Parameters:

threadId - java ID of the thread

deviceId -

getNumberOfDevices

int getNumberOfDevices()

This method returns number of available devices

Returns:

touch

void touch(INDArray array)

Utility method, to associate INDArray with specific device (backend-specific)

Parameters:

array -

touch

void touch(DataBuffer buffer)

Utility method, to associate INDArray with specific device (backend-specific)

Parameters:

buffer -

replicateToDevice

INDArray replicateToDevice(Integer deviceId,
                           INDArray array)

This method replicates given INDArray, and places it to target device.

Parameters:

deviceId - target deviceId

array - INDArray to replicate

Returns:

replicateToDevice

DataBuffer replicateToDevice(Integer deviceId,
                             DataBuffer buffer)

This method replicates given DataBuffer, and places it to target device.

Parameters:

deviceId - target deviceId

buffer -

Returns:

tagLocation

void tagLocation(INDArray array,
                 AffinityManager.Location location)

This method tags specific INDArray as "recent" on specified location

Parameters:

location -

tagLocation

void tagLocation(DataBuffer buffer,
                 AffinityManager.Location location)

This method tags specific DataBuffer as "recent" on specified location

Parameters:

location -