org.nd4j.linalg.api.concurrency

Class BasicAffinityManager

java.lang.Object

org.nd4j.linalg.api.concurrency.BasicAffinityManager

All Implemented Interfaces:

AffinityManager

public abstract class BasicAffinityManager
extends Object
implements AffinityManager

Author:

[email protected]

Nested Class Summary

Nested classes/interfaces inherited from interface org.nd4j.linalg.api.concurrency.AffinityManager

AffinityManager.Location

Constructor Summary

Constructors

Constructor and Description
BasicAffinityManager()

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	getDeviceForArray(INDArray array) This method returns id of current device for a given INDArray
Integer	getDeviceForCurrentThread() This method returns deviceId for current thread
Integer	getDeviceForThread(long threadId) This method returns deviceId for specified threadId
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

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Methods inherited from interface org.nd4j.linalg.api.concurrency.AffinityManager

touch, touch

Constructor Detail

BasicAffinityManager

public BasicAffinityManager()

Method Detail

getDeviceForCurrentThread

public Integer getDeviceForCurrentThread()

Description copied from interface: AffinityManager

This method returns deviceId for current thread

Specified by:

getDeviceForCurrentThread in interface AffinityManager

Returns:

getDeviceForThread

public Integer getDeviceForThread(Thread thread)

Description copied from interface: AffinityManager

This method returns deviceId for specified thread

Specified by:

getDeviceForThread in interface AffinityManager

Returns:

getDeviceForThread

public Integer getDeviceForThread(long threadId)

Description copied from interface: AffinityManager

This method returns deviceId for specified threadId

Specified by:

getDeviceForThread in interface AffinityManager

Returns:

attachThreadToDevice

public void attachThreadToDevice(Thread thread,
                                 Integer deviceId)

Description copied from interface: AffinityManager

This method attaches specified thread to specified device

Specified by:

attachThreadToDevice in interface AffinityManager

attachThreadToDevice

public void attachThreadToDevice(long threadId,
                                 Integer deviceId)

Description copied from interface: AffinityManager

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

Specified by:

attachThreadToDevice in interface AffinityManager

Parameters:

threadId - java ID of the thread

getDeviceForArray

public Integer getDeviceForArray(INDArray array)

Description copied from interface: AffinityManager

This method returns id of current device for a given INDArray

Specified by:

getDeviceForArray in interface AffinityManager

Returns:

getNumberOfDevices

public int getNumberOfDevices()

Description copied from interface: AffinityManager

This method returns number of available devices

Specified by:

getNumberOfDevices in interface AffinityManager

Returns:

replicateToDevice

public INDArray replicateToDevice(Integer deviceId,
                                  INDArray array)

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

Specified by:

replicateToDevice in interface AffinityManager

Parameters:

deviceId - target deviceId

array - INDArray to replicate

Returns:

replicateToDevice

public DataBuffer replicateToDevice(Integer deviceId,
                                    DataBuffer buffer)

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

Specified by:

replicateToDevice in interface AffinityManager

Parameters:

deviceId - target deviceId

buffer -

Returns:

tagLocation

public void tagLocation(INDArray array,
                        AffinityManager.Location location)

Description copied from interface: AffinityManager

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

Specified by:

tagLocation in interface AffinityManager

tagLocation

public void tagLocation(DataBuffer buffer,
                        AffinityManager.Location location)

Description copied from interface: AffinityManager

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

Specified by:

tagLocation in interface AffinityManager