org.nd4j.linalg.jcublas.gpumetrics

Class GpuMetrics

java.lang.Object

org.nd4j.linalg.jcublas.gpumetrics.GpuMetrics

public class GpuMetrics
extends Object

See: http://devblogs.nvidia.com/parallelforall/cuda-pro-tip-occupancy-api-simplifies-launch-configuration/

Author:

Adam Gibson

Field Summary

Fields

Modifier and Type	Field and Description
static int	MAX_BLOCKS
static int	MAX_THREADS

Constructor Summary

Constructors

Constructor and Description
GpuMetrics()

Method Summary

Modifier and Type	Method and Description
static GpuMetrics	blockAndThreads(String dataType, int n) Get the blocks and threads used for a kernel launch
static GpuMetrics	blocksAndThreadsOccupancy(String functionName, String dataType, int n)
int	getBlockSize()
int[]	getGpuDefinitionInfo() Outputs the expected gpu information to send to the gpu for cuda kernel metadata.
int	getGridSize()
int	getSharedMemory()
static int[]	getThreadsAndBlocks(int n, int maxThreads, int maxBlocks) Given n, max threads
void	setBlockSizeNotOverMax(int blockSize) Special setter that queries the block size to ensure not over the max possible block size is specified.
void	setGridSizeNotOverMax(int gridSize) Special setter that queries the maximum amount of shared memory per block allowed
void	setSharedMemoryNotOverMax(int sharedMemory) Special setter that queries the maximum amount of shared memory per block allowed
void	validate() Validates the current configuration against the gpu's hardware constraints.

Methods inherited from class java.lang.Object

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

Field Detail

MAX_THREADS

public static final int MAX_THREADS

See Also:

Constant Field Values

MAX_BLOCKS

public static final int MAX_BLOCKS

See Also:

Constant Field Values

Constructor Detail

GpuMetrics

public GpuMetrics()

Method Detail

getGpuDefinitionInfo

public int[] getGpuDefinitionInfo()

Outputs the expected gpu information to send to the gpu for cuda kernel metadata. The first entry is the block size The second entry is the grid size The third entry is the shared memory

Returns:

a 3 length array representing the gpu information

getGridSize

public int getGridSize()

getBlockSize

public int getBlockSize()

getSharedMemory

public int getSharedMemory()

getThreadsAndBlocks

public static int[] getThreadsAndBlocks(int n,
                                        int maxThreads,
                                        int maxBlocks)

Given n, max threads

Parameters:

n - the number of elements to process

maxThreads - the max number of threads

maxBlocks - the max number of blocks

Returns:

an array with the number of threads as the first entry and number of blocks as the second entry

blockAndThreads

public static GpuMetrics blockAndThreads(String dataType,
                                         int n)

Get the blocks and threads used for a kernel launch

Parameters:

dataType - the data type

n - the number of elements

Returns:

the information used for launching a kernel

blocksAndThreadsOccupancy

public static GpuMetrics blocksAndThreadsOccupancy(String functionName,
                                                   String dataType,
                                                   int n)

Parameters:

functionName -

dataType -

n -

Returns:

validate

public void validate()

Validates the current configuration against the gpu's hardware constraints. Throws an IllegalArgumentException if any of the values surpass the GPU's built in hardware constraints

setSharedMemoryNotOverMax

public void setSharedMemoryNotOverMax(int sharedMemory)

Special setter that queries the maximum amount of shared memory per block allowed

Parameters:

sharedMemory -

setGridSizeNotOverMax

public void setGridSizeNotOverMax(int gridSize)

Special setter that queries the maximum amount of shared memory per block allowed

Parameters:

gridSize -

setBlockSizeNotOverMax

public void setBlockSizeNotOverMax(int blockSize)

Special setter that queries the block size to ensure not over the max possible block size is specified.

Parameters:

blockSize - the block size to attempt to set