org.nd4j.linalg.jcublas.buffer.allocation

Class MemoryStrategies

java.lang.Object

org.nd4j.linalg.jcublas.buffer.allocation.MemoryStrategies

public class MemoryStrategies
extends Object

Based on the jcuda memory bandwidth example: http://www.jcuda.org/samples/JCudaBandwidthTest.java

Author:

Adam Gibson

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	MemoryStrategies.MemoryMode Memory modes for the host memory (Credit to Marco from jcuda)

Constructor Summary

Constructors

Constructor and Description
MemoryStrategies()

Method Summary

Modifier and Type	Method and Description
static MemoryStrategies.MemoryMode	getMode(int device) Get the mode for the device based on the device information
static MemoryStrategy	getStrategy(int device) Get the configured memory strategy for the device
static MemoryStrategy	getStrategy(MemoryStrategies.MemoryMode mode) Get the strategy for a given mode

Methods inherited from class java.lang.Object

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

Constructor Detail

MemoryStrategies

public MemoryStrategies()

Method Detail

getMode

public static MemoryStrategies.MemoryMode getMode(int device)

Get the mode for the device based on the device information

Parameters:

device - the device to get the mode for

Returns:

the default mode for the device based on the device characteristics

getStrategy

public static MemoryStrategy getStrategy(MemoryStrategies.MemoryMode mode)

Get the strategy for a given mode

Parameters:

mode - the mode to get the strategy for

Returns:

the strategy for the given mode

getStrategy

public static MemoryStrategy getStrategy(int device)

Get the configured memory strategy for the device

Parameters:

device - the device to get the memory strategy for

Returns:

the memory strategy