org.nd4j.linalg.api.concurrency.AffinityManager Maven / Gradle / Ivy
package org.nd4j.linalg.api.concurrency;
import org.nd4j.linalg.api.buffer.DataBuffer;
import org.nd4j.linalg.api.ndarray.INDArray;
/**
* @author [email protected]
*/
public interface AffinityManager {
/**
* This method returns deviceId for current thread
* @return
*/
Integer getDeviceForCurrentThread();
/**
* This method returns deviceId for specified thread
* @param thread
* @return
*/
Integer getDeviceForThread(Thread thread);
/**
* This method returns
*
* @param threadId
* @return
*/
Integer getDeviceForThread(long threadId);
/**
* This method attaches specified thread to specified device
*
* @param thread
* @param deviceId
*/
void attachThreadToDevice(Thread thread, Integer deviceId);
/**
* This method attaches specified thread (by Id) to specified device
*
* @param threadId java ID of the thread
* @param deviceId
*/
void attachThreadToDevice(long threadId, Integer deviceId);
/**
* This method returns number of available devices
* @return
*/
int getNumberOfDevices();
/**
* Utility method, to associate INDArray with specific device (backend-specific)
*
* @param array
*/
void touch(INDArray array);
/**
* Utility method, to associate INDArray with specific device (backend-specific)
*
* @param buffer
*/
void touch(DataBuffer buffer);
}