org.nd4j.linalg.api.concurrency

Interface AffinityManager

All Known Implementing Classes:

BasicAffinityManager

public interface AffinityManager

Author:

[email protected]

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

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 -