public class SpoofCUDARowwise extends SpoofRowwise implements SpoofCUDAOperator
SpoofRowwise.RowType
SpoofOperator.SideInput, SpoofOperator.SideInputSparseCell, SpoofOperator.SideInputSparseRow
SpoofCUDAOperator.PrecisionProxy
JNI_MAT_ENTRY_SIZE
Constructor and Description |
---|
SpoofCUDARowwise(SpoofRowwise.RowType type,
long constDim2,
boolean tB1,
int reqVectMem,
int id,
SpoofCUDAOperator.PrecisionProxy ep) |
Modifier and Type | Method and Description |
---|---|
static int |
execute_d(long ctx,
long[] meta,
long[] in,
long[] sides,
long[] out,
long scalars) |
int |
execute_dp(long ctx,
long[] meta,
long[] in,
long[] sides,
long[] out,
long scalars) |
static int |
execute_f(long ctx,
long[] meta,
long[] in,
long[] sides,
long[] out,
long scalars) |
int |
execute_sp(long ctx,
long[] meta,
long[] in,
long[] sides,
long[] out,
long scalars) |
ScalarObject |
execute(ExecutionContext ec,
ArrayList<MatrixObject> inputs,
ArrayList<ScalarObject> scalarObjects) |
MatrixObject |
execute(ExecutionContext ec,
ArrayList<MatrixObject> inputs,
ArrayList<ScalarObject> scalarObjects,
String outputName) |
String |
getName() |
jcuda.Pointer |
getScalarPtr() |
void |
releaseScalarGPUMemory(ExecutionContext ec) |
void |
setScalarPtr(jcuda.Pointer _ptr) |
createCUDAInstrcution, execute, execute, execute, execute, getConstDim2, getNumIntermediates, getRowType, getSpoofType, hasMatrixObjectSideInput, hasMatrixSideInput
execute, getDenseMatrices, getTotalInputNnz, getTotalInputSize
equals, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait
prepareInputPointers, prepareOutputPointers, prepareSideInputPointers
public SpoofCUDARowwise(SpoofRowwise.RowType type, long constDim2, boolean tB1, int reqVectMem, int id, SpoofCUDAOperator.PrecisionProxy ep)
public String getName()
getName
in interface SpoofCUDAOperator
public void setScalarPtr(jcuda.Pointer _ptr)
setScalarPtr
in interface SpoofCUDAOperator
public jcuda.Pointer getScalarPtr()
getScalarPtr
in interface SpoofCUDAOperator
public void releaseScalarGPUMemory(ExecutionContext ec)
releaseScalarGPUMemory
in interface SpoofCUDAOperator
public ScalarObject execute(ExecutionContext ec, ArrayList<MatrixObject> inputs, ArrayList<ScalarObject> scalarObjects)
execute
in interface SpoofCUDAOperator
public MatrixObject execute(ExecutionContext ec, ArrayList<MatrixObject> inputs, ArrayList<ScalarObject> scalarObjects, String outputName)
execute
in interface SpoofCUDAOperator
public int execute_sp(long ctx, long[] meta, long[] in, long[] sides, long[] out, long scalars)
execute_sp
in interface SpoofCUDAOperator
public int execute_dp(long ctx, long[] meta, long[] in, long[] sides, long[] out, long scalars)
execute_dp
in interface SpoofCUDAOperator
public static int execute_f(long ctx, long[] meta, long[] in, long[] sides, long[] out, long scalars)
public static int execute_d(long ctx, long[] meta, long[] in, long[] sides, long[] out, long scalars)
Copyright © 2021 The Apache Software Foundation. All rights reserved.