org.nd4j.linalg.api.concurrency.BasicAffinityManager 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 abstract class BasicAffinityManager implements AffinityManager {
@Override
public Integer getDeviceForCurrentThread() {
return 0;
}
@Override
public Integer getDeviceForThread(Thread thread) {
return 0;
}
@Override
public Integer getDeviceForThread(long threadId) {
return 0;
}
@Override
public void attachThreadToDevice(Thread thread, Integer deviceId) {
// no-op
}
@Override
public void attachThreadToDevice(long threadId, Integer deviceId) {
// no-op
}
@Override
public int getNumberOfDevices() {
return 1;
}
/**
* This method replicates given INDArray, and places it to target device.
*
* @param deviceId target deviceId
* @param array INDArray to replicate
* @return
*/
@Override
public INDArray replicateToDevice(Integer deviceId, INDArray array) {
return null;
}
/**
* This method replicates given DataBuffer, and places it to target device.
*
* @param deviceId target deviceId
* @param buffer
* @return
*/
@Override
public DataBuffer replicateToDevice(Integer deviceId, DataBuffer buffer) {
return null;
}
@Override
public void tagLocation(INDArray array, Location location) {
// no-op
}
@Override
public void tagLocation(DataBuffer buffer, Location location) {
// no-op
}
}