public class SpoofCUDACellwise extends SpoofCellwise implements SpoofCUDAOperator
SpoofCellwise.AggOp, SpoofCellwise.CellType
SpoofOperator.SideInput, SpoofOperator.SideInputSparseCell, SpoofOperator.SideInputSparseRow
SpoofCUDAOperator.PrecisionProxy
JNI_MAT_ENTRY_SIZE
Constructor and Description |
---|
SpoofCUDACellwise(SpoofCellwise.CellType type,
boolean sparseSafe,
boolean containsSeq,
SpoofCellwise.AggOp aggOp,
int id,
SpoofCUDAOperator.PrecisionProxy ep,
SpoofCellwise fallback) |
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) |
containsSeq, createCUDAInstrcution, execute, execute, execute, execute, execute, getAggOp, getCellType, getSpoofType, isSparseSafe
execute, getDenseMatrices, getTotalInputNnz, getTotalInputSize
equals, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait
prepareInputPointers, prepareOutputPointers, prepareSideInputPointers
public SpoofCUDACellwise(SpoofCellwise.CellType type, boolean sparseSafe, boolean containsSeq, SpoofCellwise.AggOp aggOp, int id, SpoofCUDAOperator.PrecisionProxy ep, SpoofCellwise fallback)
public ScalarObject execute(ExecutionContext ec, ArrayList<MatrixObject> inputs, ArrayList<ScalarObject> scalarObjects)
execute
in interface SpoofCUDAOperator
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 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.