org.nd4j.linalg.util

Class DeviceLocal<T>

java.lang.Object

org.nd4j.linalg.util.DeviceLocal<T>

Direct Known Subclasses:

DeviceLocalNDArray

public abstract class DeviceLocal<T>
extends Object

Class similar to Java ThreadLocal class, but locality is preserved with respect to device used

Author:

[email protected]

Field Summary

Fields

Modifier and Type	Field and Description
protected Map<Integer,T>	backingMap
protected INDArray	delayedArray
protected boolean	delayedMode
protected int	lastSettledDevice
protected List<ReentrantReadWriteLock>	locksMap
protected List<AtomicInteger>	updatesMap

Constructor Summary

Constructors

Constructor and Description
DeviceLocal(boolean delayedMode)

Method Summary

Modifier and Type	Method and Description
void	clear() This method removes object stored for current device
T	get() This method returns object local to current deviceId
T	get(int deviceId) This method returns object local to target device
void	set(int deviceId, T object) This method sets object for specific device
void	set(T object) This method sets object for current device

Methods inherited from class java.lang.Object

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

Field Detail

backingMap

protected Map<Integer,T> backingMap

locksMap

protected List<ReentrantReadWriteLock> locksMap

updatesMap

protected List<AtomicInteger> updatesMap

delayedMode

protected final boolean delayedMode

delayedArray

protected volatile INDArray delayedArray

lastSettledDevice

protected int lastSettledDevice

Constructor Detail

DeviceLocal

public DeviceLocal(boolean delayedMode)

Method Detail

get

@Nullable
public T get()

This method returns object local to current deviceId

Returns:

get

@Nullable
public T get(int deviceId)

This method returns object local to target device

Parameters:

deviceId -

Returns:

set

public void set(int deviceId,
                T object)

This method sets object for specific device

Parameters:

deviceId -

object -

set

public void set(T object)

This method sets object for current device

Parameters:

object -

clear

public void clear()

This method removes object stored for current device