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.