public interface SpoofCUDAOperator
Modifier and Type | Interface and Description |
---|---|
static class |
SpoofCUDAOperator.PrecisionProxy |
Modifier and Type | Field and Description |
---|---|
static int |
JNI_MAT_ENTRY_SIZE |
Modifier and Type | Method and Description |
---|---|
int |
execute_dp(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() |
default long[] |
prepareInputPointers(ExecutionContext ec,
ArrayList<MatrixObject> inputs,
int offset) |
default long[] |
prepareOutputPointers(ExecutionContext ec,
MatrixObject output,
boolean sparseOut) |
default long[] |
prepareSideInputPointers(ExecutionContext ec,
ArrayList<MatrixObject> inputs,
int offset,
boolean tB1) |
void |
releaseScalarGPUMemory(ExecutionContext ec) |
void |
setScalarPtr(jcuda.Pointer ptr) |
static final int JNI_MAT_ENTRY_SIZE
String getName()
void setScalarPtr(jcuda.Pointer ptr)
jcuda.Pointer getScalarPtr()
void releaseScalarGPUMemory(ExecutionContext ec)
default long[] prepareInputPointers(ExecutionContext ec, ArrayList<MatrixObject> inputs, int offset)
default long[] prepareSideInputPointers(ExecutionContext ec, ArrayList<MatrixObject> inputs, int offset, boolean tB1)
default long[] prepareOutputPointers(ExecutionContext ec, MatrixObject output, boolean sparseOut)
MatrixObject execute(ExecutionContext ec, ArrayList<MatrixObject> inputs, ArrayList<ScalarObject> scalarObjects, String outputName)
ScalarObject execute(ExecutionContext ec, ArrayList<MatrixObject> inputs, ArrayList<ScalarObject> scalarObjects)
int execute_sp(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)
Copyright © 2021 The Apache Software Foundation. All rights reserved.