org.apache.sysml.runtime.matrix.data

Class LibMatrixCuMatMult

java.lang.Object

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

org.apache.sysml.runtime.matrix.data.LibMatrixCuMatMult

public class LibMatrixCuMatMult
extends LibMatrixCUDA

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
LibMatrixCuMatMult()

Method Summary

Modifier and Type

Method and Description

static org.apache.sysml.runtime.controlprogram.caching.MatrixObject

matmult(org.apache.sysml.runtime.controlprogram.context.ExecutionContext ec, GPUContext gCtx, String instName, org.apache.sysml.runtime.controlprogram.caching.MatrixObject left, org.apache.sysml.runtime.controlprogram.caching.MatrixObject right, String outputName, boolean isLeftTransposed, boolean isRightTransposed) Matrix multiply on GPU Examines sparsity and shapes and routes call to appropriate method from cuBLAS or cuSparse C = op(A) x op(B) The user is expected to call ec.releaseMatrixOutputForGPUInstruction(outputName);

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

LibMatrixCuMatMult

public LibMatrixCuMatMult()

Method Detail

matmult

public static org.apache.sysml.runtime.controlprogram.caching.MatrixObject matmult(org.apache.sysml.runtime.controlprogram.context.ExecutionContext ec,
                                                                                   GPUContext gCtx,
                                                                                   String instName,
                                                                                   org.apache.sysml.runtime.controlprogram.caching.MatrixObject left,
                                                                                   org.apache.sysml.runtime.controlprogram.caching.MatrixObject right,
                                                                                   String outputName,
                                                                                   boolean isLeftTransposed,
                                                                                   boolean isRightTransposed)

Matrix multiply on GPU Examines sparsity and shapes and routes call to appropriate method from cuBLAS or cuSparse C = op(A) x op(B) The user is expected to call ec.releaseMatrixOutputForGPUInstruction(outputName);

Parameters:

ec - Current ExecutionContext instance

gCtx - a valid GPUContext

instName - name of the invoking instruction to recordStatistics.

left - Matrix A

right - Matrix B

outputName - Name of the output matrix C (in code generated after LOP layer)

isLeftTransposed - op for A, transposed or not

isRightTransposed - op for B, tranposed or not

Returns:

output of matrix multiply