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	allowCrossDeviceAccess(boolean reallyAllow) This method allows to block cross-device access.
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
void	ensureLocation(INDArray array, AffinityManager.Location location) This method propagates given INDArray to specified location
AffinityManager.Location	getActiveLocation(INDArray array) This method returns last-updated location for the given INDArray
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
boolean	isCrossDeviceAccessSupported() This method returns TRUE if cross-device access is allowed on this system
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	unsafeSetDevice(Integer deviceId) This method forces specific device for current thread.

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

unsafeSetDevice

public void unsafeSetDevice(Integer deviceId)

Description copied from interface: AffinityManager

This method forces specific device for current thread. PLEASE NOTE: This method is UNSAFE and should NOT be used with 100% clearance about it.

Specified by:

unsafeSetDevice in interface AffinityManager

ensureLocation

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

Description copied from interface: AffinityManager

This method propagates given INDArray to specified location

Specified by:

ensureLocation in interface AffinityManager

isCrossDeviceAccessSupported

public boolean isCrossDeviceAccessSupported()

Description copied from interface: AffinityManager

This method returns TRUE if cross-device access is allowed on this system

Specified by:

isCrossDeviceAccessSupported in interface AffinityManager

allowCrossDeviceAccess

public void allowCrossDeviceAccess(boolean reallyAllow)

Description copied from interface: AffinityManager

This method allows to block cross-device access. Mostly suitable for debugging/testing purposes

Specified by:

allowCrossDeviceAccess in interface AffinityManager

getActiveLocation

public AffinityManager.Location getActiveLocation(INDArray array)

Description copied from interface: AffinityManager

This method returns last-updated location for the given INDArray

Specified by:

getActiveLocation in interface AffinityManager

Returns: