org.nd4j.linalg.jcublas.device.conf

Class DeviceConfiguration

java.lang.Object

org.nd4j.linalg.jcublas.device.conf.DeviceConfiguration

public class DeviceConfiguration
extends Object

Configuration for a device. This includes the memory allocation strategy among other things.

Author:

Adam Gibson

Field Summary

Fields

Modifier and Type	Field and Description
static String	MEMORY
static String	NAME_SPACE

Constructor Summary

Constructors

Constructor and Description
DeviceConfiguration(int deviceNumber) Initialize the device configuration based on the device information
DeviceConfiguration(int deviceNumber, Properties props) Initialize the configuration with the given properties

Method Summary

Modifier and Type	Method and Description
int	getDeviceNumber()
MemoryStrategy	getMemoryStrategy()
void	setDeviceNumber(int deviceNumber)
void	setMemoryStrategy(MemoryStrategy memoryStrategy)

Methods inherited from class java.lang.Object

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

Field Detail

NAME_SPACE

public static final String NAME_SPACE

MEMORY

public static final String MEMORY

Constructor Detail

DeviceConfiguration

public DeviceConfiguration(int deviceNumber,
                           Properties props)

Initialize the configuration with the given properties

Parameters:

deviceNumber - the device number to initialize

props - the properties to initialize with

DeviceConfiguration

public DeviceConfiguration(int deviceNumber)

Initialize the device configuration based on the device information

Parameters:

deviceNumber - the device number

Method Detail

getMemoryStrategy

public MemoryStrategy getMemoryStrategy()

setMemoryStrategy

public void setMemoryStrategy(MemoryStrategy memoryStrategy)

getDeviceNumber

public int getDeviceNumber()

setDeviceNumber

public void setDeviceNumber(int deviceNumber)