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]

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	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

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

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

getNumberOfDevices

public int getNumberOfDevices()

Description copied from interface: AffinityManager

This method returns number of available devices

Specified by:

getNumberOfDevices in interface AffinityManager

Returns: