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

Methods inherited from class java.lang.Object

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

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