Modifier and Type | Method and Description |
---|---|
static void |
DMLScript.initHadoopExecution(org.apache.sysml.conf.DMLConfig config) |
Modifier and Type | Class and Description |
---|---|
class |
DMLScriptException
This exception should be thrown to flag DML Script errors.
|
Modifier and Type | Method and Description |
---|---|
void |
DnnGPUInstruction.processBatchNorm2dBackwardInstruction(org.apache.sysml.runtime.controlprogram.context.ExecutionContext ec) |
Constructor and Description |
---|
DnnGPUInstruction(org.apache.sysml.runtime.instructions.cp.CPOperand in1,
org.apache.sysml.runtime.instructions.cp.CPOperand in2,
org.apache.sysml.runtime.instructions.cp.CPOperand in3,
org.apache.sysml.runtime.instructions.cp.CPOperand in4,
org.apache.sysml.runtime.instructions.cp.CPOperand in5,
org.apache.sysml.runtime.instructions.cp.CPOperand in6,
org.apache.sysml.runtime.instructions.cp.CPOperand in7,
org.apache.sysml.runtime.instructions.cp.CPOperand in8,
org.apache.sysml.runtime.instructions.cp.CPOperand out,
org.apache.sysml.runtime.instructions.cp.CPOperand out2,
org.apache.sysml.runtime.instructions.cp.CPOperand out3,
org.apache.sysml.runtime.instructions.cp.CPOperand out4,
org.apache.sysml.runtime.instructions.cp.CPOperand out5,
String opcode,
String istr,
double intermediateMemoryBudget) |
DnnGPUInstruction(org.apache.sysml.runtime.instructions.cp.CPOperand in1,
org.apache.sysml.runtime.instructions.cp.CPOperand in2,
org.apache.sysml.runtime.instructions.cp.CPOperand in3,
org.apache.sysml.runtime.instructions.cp.CPOperand in4,
org.apache.sysml.runtime.instructions.cp.CPOperand in5,
org.apache.sysml.runtime.instructions.cp.CPOperand in6,
org.apache.sysml.runtime.instructions.cp.CPOperand out,
org.apache.sysml.runtime.instructions.cp.CPOperand out2,
String opcode,
String istr,
double intermediateMemoryBudget) |
DnnGPUInstruction(org.apache.sysml.runtime.instructions.cp.CPOperand in1,
org.apache.sysml.runtime.instructions.cp.CPOperand in2,
org.apache.sysml.runtime.instructions.cp.CPOperand in3,
org.apache.sysml.runtime.instructions.cp.CPOperand in4,
org.apache.sysml.runtime.instructions.cp.CPOperand out,
String opcode,
String istr,
double intermediateMemoryBudget) |
DnnGPUInstruction(org.apache.sysml.runtime.instructions.cp.CPOperand in1,
org.apache.sysml.runtime.instructions.cp.CPOperand in2,
org.apache.sysml.runtime.instructions.cp.CPOperand in3,
org.apache.sysml.runtime.instructions.cp.CPOperand out,
String opcode,
String istr,
double intermediateMemoryBudget) |
Modifier and Type | Method and Description |
---|---|
void |
GPUObject.clearData(String opcode,
boolean eager)
Clears the data associated with this
GPUObject instance |
protected void |
GPUObject.copyFromDeviceToHost(String instName,
boolean isEviction,
boolean eagerDelete)
Copies the data from device to host.
|
void |
GPUMemoryManager.free(String opcode,
jcuda.Pointer toFree,
boolean eager)
Deallocate the pointer
|
jcuda.Pointer |
GPULazyCudaFreeMemoryManager.getRmvarPointerMinSize(String opcode,
long minSize) |
Modifier and Type | Method and Description |
---|---|
static void |
LibMatrixCuDNN.batchNormalizationBackward(GPUContext gCtx,
String instName,
org.apache.sysml.runtime.controlprogram.caching.MatrixObject image,
org.apache.sysml.runtime.controlprogram.caching.MatrixObject dout,
org.apache.sysml.runtime.controlprogram.caching.MatrixObject scale,
org.apache.sysml.runtime.controlprogram.caching.MatrixObject dX,
org.apache.sysml.runtime.controlprogram.caching.MatrixObject dScale,
org.apache.sysml.runtime.controlprogram.caching.MatrixObject dBias,
double epsilon,
org.apache.sysml.runtime.controlprogram.caching.MatrixObject resultSaveMean,
org.apache.sysml.runtime.controlprogram.caching.MatrixObject resultSaveInvVariance)
This method computes the backpropagation errors for image, scale and bias of batch normalization layer
|
static void |
LibMatrixCuDNN.batchNormalizationForwardInference(GPUContext gCtx,
String instName,
org.apache.sysml.runtime.controlprogram.caching.MatrixObject image,
org.apache.sysml.runtime.controlprogram.caching.MatrixObject scale,
org.apache.sysml.runtime.controlprogram.caching.MatrixObject bias,
org.apache.sysml.runtime.controlprogram.caching.MatrixObject runningMean,
org.apache.sysml.runtime.controlprogram.caching.MatrixObject runningVar,
org.apache.sysml.runtime.controlprogram.caching.MatrixObject ret,
double epsilon)
Performs the forward BatchNormalization layer computation for inference
|
static void |
LibMatrixCuDNN.batchNormalizationForwardTraining(GPUContext gCtx,
String instName,
org.apache.sysml.runtime.controlprogram.caching.MatrixObject image,
org.apache.sysml.runtime.controlprogram.caching.MatrixObject scale,
org.apache.sysml.runtime.controlprogram.caching.MatrixObject bias,
org.apache.sysml.runtime.controlprogram.caching.MatrixObject runningMean,
org.apache.sysml.runtime.controlprogram.caching.MatrixObject runningVar,
org.apache.sysml.runtime.controlprogram.caching.MatrixObject ret,
org.apache.sysml.runtime.controlprogram.caching.MatrixObject retRunningMean,
org.apache.sysml.runtime.controlprogram.caching.MatrixObject retRunningVar,
double epsilon,
double exponentialAverageFactor,
org.apache.sysml.runtime.controlprogram.caching.MatrixObject resultSaveMean,
org.apache.sysml.runtime.controlprogram.caching.MatrixObject resultSaveInvVariance)
Performs the forward BatchNormalization layer computation for training
|
static void |
LibMatrixCUDA.denseTranspose(org.apache.sysml.runtime.controlprogram.context.ExecutionContext ec,
GPUContext gCtx,
String instName,
jcuda.Pointer A,
jcuda.Pointer C,
long numRowsA,
long numColsA)
Computes C = t(A)
|
static JCudaKernels |
LibMatrixCUDA.getCudaKernels(GPUContext gCtx) |
static jcuda.Pointer |
LibMatrixCUDA.getDensePointer(GPUContext gCtx,
org.apache.sysml.runtime.controlprogram.caching.MatrixObject input,
String instName)
Convenience method to get jcudaDenseMatrixPtr.
|
static jcuda.Pointer |
LibMatrixCuDNN.getDensePointerForCuDNN(GPUContext gCtx,
org.apache.sysml.runtime.controlprogram.caching.MatrixObject image,
String instName,
int numRows,
int numCols)
Convenience method to get jcudaDenseMatrixPtr.
|
static void |
LibMatrixCuDNN.lstm(org.apache.sysml.runtime.controlprogram.context.ExecutionContext ec,
GPUContext gCtx,
String instName,
jcuda.Pointer X,
jcuda.Pointer wPointer,
jcuda.Pointer out0,
jcuda.Pointer c0,
boolean return_sequences,
String outputName,
String cyName,
int N,
int M,
int D,
int T)
Computes the forward pass for an LSTM layer with M neurons.
|
static void |
LibMatrixCuDNN.lstmBackward(org.apache.sysml.runtime.controlprogram.context.ExecutionContext ec,
GPUContext gCtx,
String instName,
jcuda.Pointer x,
jcuda.Pointer hx,
jcuda.Pointer cx,
jcuda.Pointer wPointer,
String doutName,
String dcyName,
String dxName,
String dwName,
String dbName,
String dhxName,
String dcxName,
boolean return_sequences,
int N,
int M,
int D,
int T) |
Constructor and Description |
---|
LibMatrixCuDNNRnnAlgorithm(org.apache.sysml.runtime.controlprogram.context.ExecutionContext ec,
GPUContext gCtx,
String instName,
String rnnMode,
int N,
int T,
int M,
int D,
boolean isTraining,
jcuda.Pointer w) |
Copyright © 2018 The Apache Software Foundation. All rights reserved.