org.apache.sysml.runtime.matrix.data

Class LibMatrixCuDNNInputRowFetcher

java.lang.Object

org.apache.sysml.runtime.matrix.data.LibMatrixCUDA

org.apache.sysml.runtime.matrix.data.LibMatrixCuDNNInputRowFetcher

All Implemented Interfaces:

AutoCloseable

public class LibMatrixCuDNNInputRowFetcher
extends LibMatrixCUDA
implements AutoCloseable

Performs a slice operation: out = in[(n+1):(n+1), 1:numColumns]

Field Summary

Fields inherited from class org.apache.sysml.runtime.matrix.data.LibMatrixCUDA

cudaSupportFunctions, CUDNN_DATA_TYPE, customKernelSuffix, maxNumElementsOfCuDNNTensor, sizeOfDataType

Constructor Summary

Constructors

Constructor and Description
LibMatrixCuDNNInputRowFetcher(GPUContext gCtx, String instName, org.apache.sysml.runtime.controlprogram.caching.MatrixObject image) Initialize the input fetcher

Method Summary

Modifier and Type	Method and Description
void	close() Deallocates temporary pointer
jcuda.Pointer	getNthRow(int n) Copy the nth row and return the dense pointer

Methods inherited from class org.apache.sysml.runtime.matrix.data.LibMatrixCUDA

abs, acos, asin, atan, axpy, biasAdd, biasMultiply, cbind, ceil, channelSums, computeNNZ, cos, cosh, dataTypePointerTo, denseTranspose, deviceCopy, double2float, exp, float2double, floor, getCublasHandle, getCudaKernels, getCusparseHandle, getDenseMatrixOutputForGPUInstruction, getDensePointer, getNnz, getSparsePointer, isInSparseFormat, log, matmultTSMM, matrixMatrixArithmetic, matrixMatrixRelational, matrixScalarArithmetic, matrixScalarOp, matrixScalarRelational, one, rbind, reluBackward, resetFloatingPointPrecision, round, sigmoid, sign, sin, sinh, sliceDenseDense, sliceOperations, sliceSparseDense, solve, sqrt, tan, tanh, toInt, transpose, unaryAggregate, zero

Methods inherited from class java.lang.Object

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

Constructor Detail

LibMatrixCuDNNInputRowFetcher

public LibMatrixCuDNNInputRowFetcher(GPUContext gCtx,
                                     String instName,
                                     org.apache.sysml.runtime.controlprogram.caching.MatrixObject image)

Initialize the input fetcher

Parameters:

gCtx - current gpu context

instName - name of the instruction

image - input matrix object.

Method Detail

getNthRow

public jcuda.Pointer getNthRow(int n)

Copy the nth row and return the dense pointer

Parameters:

n - zero-based row index

Returns:

dense pointer containing the nth row. This row is reused in the next iteration

close

public void close()

Deallocates temporary pointer

Specified by:

close in interface AutoCloseable