org.nd4j.nativeblas

Class NativeOps

java.lang.Object

org.bytedeco.javacpp.Pointer

org.nd4j.nativeblas.NativeOps

All Implemented Interfaces:

AutoCloseable

@Platform(include="NativeOps.h",
          compiler="cpp11",
          link="nd4j",
          library="jnind4j")
public class NativeOps
extends org.bytedeco.javacpp.Pointer

Native interface for op execution on cpu

Author:

Adam Gibson

Nested Class Summary

Nested classes/interfaces inherited from class org.bytedeco.javacpp.Pointer

org.bytedeco.javacpp.Pointer.CustomDeallocator, org.bytedeco.javacpp.Pointer.Deallocator, org.bytedeco.javacpp.Pointer.NativeDeallocator

Field Summary

Fields inherited from class org.bytedeco.javacpp.Pointer

address, capacity, limit, position

Constructor Summary

Constructors

Constructor and Description
NativeOps()

Method Summary

Modifier and Type	Method and Description
void	concatDouble(long[] extraPointers, int dimension, int numArrays, long[] data, long[] inputShapeInfo, long result, long resultShapeInfo, long[] tadPointers, long[] tadOffsets)
void	concatFloat(long[] extraPointers, int dimension, int numArrays, long[] data, long[] inputShapeInfo, long result, long resultShapeInfo, long[] tadPointers, long[] tadOffsets)
long	createBlasHandle()
long	createContext()
long	createEvent()
long	createStream()
long	destroyEvent(long event)
void	enableDebugMode(boolean reallyEnable)
void	enableVerboseMode(boolean reallyEnable)
long	eventSynchronize(long event)
void	execBroadcastDouble(long[] extraPointers, int opNum, long x, long xShapeInfo, long y, long yShapeInfo, long result, long resultShapeInfo, long dimension, int dimensionLength)
void	execBroadcastFloat(long[] extraPointers, int opNum, long x, long xShapeInfo, long y, long yShapeInfo, long result, long resultShapeInfo, long dimension, int dimensionLength)
void	execIndexReduceDouble(long[] extraPointers, int opNum, long x, long xShapeInfo, long extraParams, long result, long resultShapeInfoBuffer, long dimension, int dimensionLength)
void	execIndexReduceFloat(long[] extraPointers, int opNum, long x, long xShapeInfo, long extraParams, long result, long resultShapeInfoBuffer, long dimension, int dimensionLength)
double	execIndexReduceScalarDouble(long[] extraPointers, int opNum, long x, long xShapeInfo, long extraParams)
float	execIndexReduceScalarFloat(long[] extraPointers, int opNum, long x, long xShapeInfo, long extraParams)
void	execPairwiseTransformDouble(long[] extraPointers, int opNum, long dx, int xStride, long y, int yStride, long result, int resultStride, long extraParams, long n)
void	execPairwiseTransformDouble(long[] extraPointers, int opNum, long dx, long xShapeInfo, long y, long yShapeInfo, long result, long resultShapeInfo, long extraParams)
void	execPairwiseTransformDouble(long[] extraPointers, int opNum, long dx, long xShapeInfo, long y, long yShapeInfo, long result, long resultShapeInfo, long extraParams, long xIndexes, long yIndexes, long resultIndexes)
void	execPairwiseTransformFloat(long[] extraPointers, int opNum, long dx, int xStride, long y, int yStride, long result, int resultStride, long extraParams, long n)
void	execPairwiseTransformFloat(long[] extraPointers, int opNum, long dx, long xShapeInfo, long y, long yShapeInfo, long result, long resultShapeInfo, long extraParams)
void	execPairwiseTransformFloat(long[] extraPointers, int opNum, long dx, long xShapeInfo, long y, long yShapeInfo, long result, long resultShapeInfo, long extraParams, long xIndexes, long yIndexes, long resultIndexes)
void	execReduce3Double(long[] extraPointers, int opNum, long x, long xShapeInfo, long extraParamsVals, long y, long yShapeInfo, long result, long resultShapeInfo)
void	execReduce3Double(long[] extraPointers, int opNum, long x, long xShapeInfo, long extraParamsVals, long y, long yShapeInfo, long result, long resultShapeInfoBuffer, long dimension, int dimensionLength)
void	execReduce3Float(long[] extraPointers, int opNum, long x, long xShapeInfo, long extraParamsVals, long y, long yShapeInfo, long result, long resultShapeInfo)
void	execReduce3Float(long[] extraPointers, int opNum, long x, long xShapeInfo, long extraParamsVals, long y, long yShapeInfo, long result, long resultShapeInfoBuffer, long dimension, int dimensionLength)
double	execReduce3ScalarDouble(long[] extraPointers, int opNum, long x, long xShapeInfo, long extraParamsVals, long y, long yShapeInfo)
float	execReduce3ScalarFloat(long[] extraPointers, int opNum, long x, long xShapeInfo, long extraParamsVals, long y, long yShapeInfo)
void	execReduceDouble(long[] extraPointers, int opNum, long x, long xShapeInfo, long extraParams, long result, long resultShapeInfo)
void	execReduceDouble(long[] extraPointers, int opNum, long x, long xShapeInfo, long extraParams, long result, long resultShapeInfo, long dimension, int dimensionLength)
void	execReduceFloat(long[] extraPointers, int opNum, long x, long xShapeInfo, long extraParams, long result, long resultShapeInfo)
void	execReduceFloat(long[] extraPointers, int opNum, long x, long xShapeInfo, long extraParams, long result, long resultShapeInfo, long dimension, int dimensionLength)
double	execReduceScalarDouble(long[] extraPointers, int opNum, long x, long xShapeInfo, long extraParams)
float	execReduceScalarFloat(long[] extraPointers, int opNum, long x, long xShapeInfo, long extraParams)
void	execScalarDouble(long[] extraPointers, int opNum, long x, int xStride, long result, int resultStride, double scalar, long extraParams, long n)
void	execScalarDouble(long[] extraPointers, int opNum, long x, long xShapeInfo, long result, long resultShapeInfo, double scalar, long extraParams)
void	execScalarDouble(long[] extraPointers, int opNum, long x, long xShapeInfo, long result, long resultShapeInfo, double scalar, long extraParams, long n, long xIndexes, long resultIndexes)
void	execScalarFloat(long[] extraPointers, int opNum, long x, int xStride, long result, int resultStride, double scalar, long extraParams, long n)
void	execScalarFloat(long[] extraPointers, int opNum, long x, long xShapeInfo, long result, long resultShapeInfo, double scalar, long extraParams, long xIndexes, long resultIndexes)
void	execScalarFloat(long[] extraPointers, int opNum, long x, long xShapeInfo, long result, long resultShapeInfo, float scalar, long extraParams)
void	execSummaryStatsDouble(long[] extraPointers, int opNum, long x, long xShapeInfo, long extraParams, long result, long resultShapeInfo, boolean biasCorrected)
void	execSummaryStatsDouble(long[] extraPointers, int opNum, long x, long xShapeInfo, long extraParams, long result, long resultShapeInfoBuffer, long dimension, int dimensionLength, boolean biasCorrected)
void	execSummaryStatsFloat(long[] extraPointers, int opNum, long x, long xShapeInfo, long extraParams, long result, long resultShapeInfo, boolean biasCorrected)
void	execSummaryStatsFloat(long[] extraPointers, int opNum, long x, long xShapeInfo, long extraParams, long result, long resultShapeInfoBuffer, long dimension, int dimensionLength, boolean biasCorrected)
double	execSummaryStatsScalarDouble(long[] extraPointers, int opNum, long x, long xShapeInfo, long extraParams, boolean biasCorrected)
float	execSummaryStatsScalarFloat(long[] extraPointers, int opNum, long x, long xShapeInfo, long extraParams, boolean biasCorrected)
void	execTransformDouble(long[] extraPointers, int opNum, long dx, int xStride, long result, int resultStride, long extraParams, long n)
void	execTransformDouble(long[] extraPointers, int opNum, long dx, long xShapeInfo, long result, long resultShapeInfo, long extraParams)
void	execTransformDouble(long[] extraPointers, int opNum, long dx, long xShapeInfo, long result, long resultShapeInfo, long extraParams, long xIndexes, long resultIndexes)
void	execTransformFloat(long[] extraPointers, int opNum, long dx, int xStride, long result, int resultStride, long extraParams, long n)
void	execTransformFloat(long[] extraPointers, int opNum, long dx, long xShapeInfo, long result, long resultShapeInfo, long extraParams)
void	execTransformFloat(long[] extraPointers, int opNum, long dx, long xShapeInfo, long result, long resultShapeInfo, long extraParams, long xIndexes, long resultIndexes)
void	flattenDouble(long[] extraPointers, int offset, char order, long result, long resultShapeInfo, long input, long inputShapeInfo) Append an input array to the end of a flat array in a particular order
void	flattenFloat(long[] extraPointers, int offset, char order, long result, long resultShapeInfo, long input, long inputShapeInfo) Append an input array to the end of a flat array in a particular order
long	freeDevice(long pointer, long deviceId)
long	freeHost(long pointer)
long	getAvailableDevices()
long	getConstantSpace()
long	getDeviceFreeMemory(long ptrToDeviceId)
long	getDeviceTotalMemory(long ptrToDeviceId)
void	initializeDevicesAndFunctions() NEVER EVER USE THIS METHOD OUTSIDE OF CUDA
long	mallocDevice(long memorySize, long ptrToDeviceId, int flags)
long	mallocHost(long memorySize, int flags)
long	memcpy(long dst, long src, long size, int flags, long reserved)
long	memcpyAsync(long dst, long src, long size, int flags, long reserved)
long	memcpyConstantAsync(long dst, long src, long size, int flags, long reserved)
long	memset(long dst, int value, long size, int flags, long reserved)
long	memsetAsync(long dst, int value, long size, int flags, long reserved)
int	ompGetNumThreads() Gets the number of open mp threads
long	registerEvent(long event, long stream)
long	setBlasStream(long handle, long stream)
long	setDevice(long ptrToDeviceId)
void	setGridLimit(int gridSize)
void	setOmpNumThreads(int threads) Sets the number of openmp threads
long	streamSynchronize(long stream)
void	tadOnlyShapeInfo(long shapeInfo, long dimension, int dimensionLength, long targetBuffer, long offsetsBuffer)

Methods inherited from class org.bytedeco.javacpp.Pointer

address, asBuffer, asByteBuffer, capacity, capacity, close, deallocate, deallocate, deallocateReferences, deallocator, deallocator, equals, fill, hashCode, isNull, limit, limit, maxBytes, memchr, memcmp, memcpy, memmove, memset, offsetof, position, position, put, setNull, sizeof, toString, totalBytes, withDeallocator, zero

Methods inherited from class java.lang.Object

clone, finalize, getClass, notify, notifyAll, wait, wait, wait

Constructor Detail

NativeOps

public NativeOps()

Method Detail

execIndexReduceScalarDouble

public double execIndexReduceScalarDouble(long[] extraPointers,
                                          int opNum,
                                          long x,
                                          long xShapeInfo,
                                          long extraParams)

Parameters:

opNum -

x -

xShapeInfo -

extraParams -

execIndexReduceDouble

public void execIndexReduceDouble(long[] extraPointers,
                                  int opNum,
                                  long x,
                                  long xShapeInfo,
                                  long extraParams,
                                  long result,
                                  long resultShapeInfoBuffer,
                                  long dimension,
                                  int dimensionLength)

Parameters:

opNum -

x -

xShapeInfo -

extraParams -

result -

resultShapeInfoBuffer -

dimension -

dimensionLength -

execBroadcastDouble

public void execBroadcastDouble(long[] extraPointers,
                                int opNum,
                                long x,
                                long xShapeInfo,
                                long y,
                                long yShapeInfo,
                                long result,
                                long resultShapeInfo,
                                long dimension,
                                int dimensionLength)

Parameters:

opNum -

x -

xShapeInfo -

y -

yShapeInfo -

result -

resultShapeInfo -

dimension -

dimensionLength -

execPairwiseTransformDouble

public void execPairwiseTransformDouble(long[] extraPointers,
                                        int opNum,
                                        long dx,
                                        int xStride,
                                        long y,
                                        int yStride,
                                        long result,
                                        int resultStride,
                                        long extraParams,
                                        long n)

Parameters:

opNum -

dx -

xStride -

y -

yStride -

result -

resultStride -

extraParams -

n -

execPairwiseTransformDouble

public void execPairwiseTransformDouble(long[] extraPointers,
                                        int opNum,
                                        long dx,
                                        long xShapeInfo,
                                        long y,
                                        long yShapeInfo,
                                        long result,
                                        long resultShapeInfo,
                                        long extraParams,
                                        long xIndexes,
                                        long yIndexes,
                                        long resultIndexes)

Parameters:

opNum -

dx -

xShapeInfo -

y -

yShapeInfo -

result -

resultShapeInfo -

extraParams -

xIndexes -

yIndexes -

resultIndexes -

execPairwiseTransformDouble

public void execPairwiseTransformDouble(long[] extraPointers,
                                        int opNum,
                                        long dx,
                                        long xShapeInfo,
                                        long y,
                                        long yShapeInfo,
                                        long result,
                                        long resultShapeInfo,
                                        long extraParams)

Parameters:

opNum -

dx -

xShapeInfo -

y -

yShapeInfo -

result -

resultShapeInfo -

extraParams -

execReduceDouble

public void execReduceDouble(long[] extraPointers,
                             int opNum,
                             long x,
                             long xShapeInfo,
                             long extraParams,
                             long result,
                             long resultShapeInfo)

Parameters:

opNum -

x -

xShapeInfo -

extraParams -

result -

resultShapeInfo -

execReduceDouble

public void execReduceDouble(long[] extraPointers,
                             int opNum,
                             long x,
                             long xShapeInfo,
                             long extraParams,
                             long result,
                             long resultShapeInfo,
                             long dimension,
                             int dimensionLength)

Parameters:

opNum -

x -

xShapeInfo -

extraParams -

result -

resultShapeInfo -

execReduceScalarDouble

public double execReduceScalarDouble(long[] extraPointers,
                                     int opNum,
                                     long x,
                                     long xShapeInfo,
                                     long extraParams)

Parameters:

opNum -

x -

xShapeInfo -

extraParams -

Returns:

execReduce3Double

public void execReduce3Double(long[] extraPointers,
                              int opNum,
                              long x,
                              long xShapeInfo,
                              long extraParamsVals,
                              long y,
                              long yShapeInfo,
                              long result,
                              long resultShapeInfo)

Parameters:

opNum -

x -

xShapeInfo -

extraParamsVals -

y -

yShapeInfo -

result -

resultShapeInfo -

execReduce3ScalarDouble

public double execReduce3ScalarDouble(long[] extraPointers,
                                      int opNum,
                                      long x,
                                      long xShapeInfo,
                                      long extraParamsVals,
                                      long y,
                                      long yShapeInfo)

Parameters:

opNum -

x -

xShapeInfo -

extraParamsVals -

y -

yShapeInfo -

execReduce3Double

public void execReduce3Double(long[] extraPointers,
                              int opNum,
                              long x,
                              long xShapeInfo,
                              long extraParamsVals,
                              long y,
                              long yShapeInfo,
                              long result,
                              long resultShapeInfoBuffer,
                              long dimension,
                              int dimensionLength)

Parameters:

opNum -

x -

xShapeInfo -

extraParamsVals -

y -

yShapeInfo -

result -

resultShapeInfoBuffer -

dimension -

dimensionLength -

execScalarDouble

public void execScalarDouble(long[] extraPointers,
                             int opNum,
                             long x,
                             int xStride,
                             long result,
                             int resultStride,
                             double scalar,
                             long extraParams,
                             long n)

Parameters:

opNum -

x -

xStride -

result -

resultStride -

scalar -

extraParams -

n -

execScalarDouble

public void execScalarDouble(long[] extraPointers,
                             int opNum,
                             long x,
                             long xShapeInfo,
                             long result,
                             long resultShapeInfo,
                             double scalar,
                             long extraParams)

Parameters:

opNum -

x -

xShapeInfo -

result -

resultShapeInfo -

scalar -

extraParams -

execScalarDouble

public void execScalarDouble(long[] extraPointers,
                             int opNum,
                             long x,
                             long xShapeInfo,
                             long result,
                             long resultShapeInfo,
                             double scalar,
                             long extraParams,
                             long n,
                             long xIndexes,
                             long resultIndexes)

Parameters:

opNum -

x -

xShapeInfo -

result -

resultShapeInfo -

scalar -

extraParams -

n -

xIndexes -

resultIndexes -

execSummaryStatsScalarDouble

public double execSummaryStatsScalarDouble(long[] extraPointers,
                                           int opNum,
                                           long x,
                                           long xShapeInfo,
                                           long extraParams,
                                           boolean biasCorrected)

Parameters:

opNum -

x -

xShapeInfo -

extraParams -

biasCorrected -

execSummaryStatsDouble

public void execSummaryStatsDouble(long[] extraPointers,
                                   int opNum,
                                   long x,
                                   long xShapeInfo,
                                   long extraParams,
                                   long result,
                                   long resultShapeInfo,
                                   boolean biasCorrected)

Parameters:

opNum -

x -

xShapeInfo -

extraParams -

result -

resultShapeInfo -

biasCorrected -

execSummaryStatsDouble

public void execSummaryStatsDouble(long[] extraPointers,
                                   int opNum,
                                   long x,
                                   long xShapeInfo,
                                   long extraParams,
                                   long result,
                                   long resultShapeInfoBuffer,
                                   long dimension,
                                   int dimensionLength,
                                   boolean biasCorrected)

Parameters:

opNum -

x -

xShapeInfo -

extraParams -

result -

resultShapeInfoBuffer -

dimension -

dimensionLength -

execTransformDouble

public void execTransformDouble(long[] extraPointers,
                                int opNum,
                                long dx,
                                int xStride,
                                long result,
                                int resultStride,
                                long extraParams,
                                long n)

Parameters:

opNum -

dx -

xStride -

result -

resultStride -

extraParams -

n -

execTransformDouble

public void execTransformDouble(long[] extraPointers,
                                int opNum,
                                long dx,
                                long xShapeInfo,
                                long result,
                                long resultShapeInfo,
                                long extraParams)

Parameters:

opNum -

dx -

xShapeInfo -

result -

resultShapeInfo -

extraParams -

n -

execTransformDouble

public void execTransformDouble(long[] extraPointers,
                                int opNum,
                                long dx,
                                long xShapeInfo,
                                long result,
                                long resultShapeInfo,
                                long extraParams,
                                long xIndexes,
                                long resultIndexes)

Parameters:

opNum -

dx -

xShapeInfo -

result -

resultShapeInfo -

extraParams -

n -

execIndexReduceScalarFloat

public float execIndexReduceScalarFloat(long[] extraPointers,
                                        int opNum,
                                        long x,
                                        long xShapeInfo,
                                        long extraParams)

Parameters:

opNum -

x -

xShapeInfo -

extraParams -

execIndexReduceFloat

public void execIndexReduceFloat(long[] extraPointers,
                                 int opNum,
                                 long x,
                                 long xShapeInfo,
                                 long extraParams,
                                 long result,
                                 long resultShapeInfoBuffer,
                                 long dimension,
                                 int dimensionLength)

Parameters:

opNum -

x -

xShapeInfo -

extraParams -

result -

resultShapeInfoBuffer -

dimension -

dimensionLength -

execBroadcastFloat

public void execBroadcastFloat(long[] extraPointers,
                               int opNum,
                               long x,
                               long xShapeInfo,
                               long y,
                               long yShapeInfo,
                               long result,
                               long resultShapeInfo,
                               long dimension,
                               int dimensionLength)

Parameters:

opNum -

x -

xShapeInfo -

y -

yShapeInfo -

result -

resultShapeInfo -

dimension -

dimensionLength -

execPairwiseTransformFloat

public void execPairwiseTransformFloat(long[] extraPointers,
                                       int opNum,
                                       long dx,
                                       int xStride,
                                       long y,
                                       int yStride,
                                       long result,
                                       int resultStride,
                                       long extraParams,
                                       long n)

Parameters:

opNum -

dx -

xStride -

y -

yStride -

result -

resultStride -

extraParams -

n -

execPairwiseTransformFloat

public void execPairwiseTransformFloat(long[] extraPointers,
                                       int opNum,
                                       long dx,
                                       long xShapeInfo,
                                       long y,
                                       long yShapeInfo,
                                       long result,
                                       long resultShapeInfo,
                                       long extraParams,
                                       long xIndexes,
                                       long yIndexes,
                                       long resultIndexes)

Parameters:

opNum -

dx -

xShapeInfo -

y -

yShapeInfo -

result -

resultShapeInfo -

extraParams -

xIndexes -

yIndexes -

resultIndexes -

execPairwiseTransformFloat

public void execPairwiseTransformFloat(long[] extraPointers,
                                       int opNum,
                                       long dx,
                                       long xShapeInfo,
                                       long y,
                                       long yShapeInfo,
                                       long result,
                                       long resultShapeInfo,
                                       long extraParams)

Parameters:

opNum -

dx -

xShapeInfo -

y -

yShapeInfo -

result -

resultShapeInfo -

extraParams -

execReduceFloat

public void execReduceFloat(long[] extraPointers,
                            int opNum,
                            long x,
                            long xShapeInfo,
                            long extraParams,
                            long result,
                            long resultShapeInfo)

Parameters:

opNum -

x -

xShapeInfo -

extraParams -

result -

resultShapeInfo -

execReduceFloat

public void execReduceFloat(long[] extraPointers,
                            int opNum,
                            long x,
                            long xShapeInfo,
                            long extraParams,
                            long result,
                            long resultShapeInfo,
                            long dimension,
                            int dimensionLength)

Parameters:

opNum -

x -

xShapeInfo -

extraParams -

result -

resultShapeInfo -

execReduceScalarFloat

public float execReduceScalarFloat(long[] extraPointers,
                                   int opNum,
                                   long x,
                                   long xShapeInfo,
                                   long extraParams)

Parameters:

opNum -

x -

xShapeInfo -

extraParams -

Returns:

execReduce3Float

public void execReduce3Float(long[] extraPointers,
                             int opNum,
                             long x,
                             long xShapeInfo,
                             long extraParamsVals,
                             long y,
                             long yShapeInfo,
                             long result,
                             long resultShapeInfo)

Parameters:

opNum -

x -

xShapeInfo -

extraParamsVals -

y -

yShapeInfo -

result -

resultShapeInfo -

execReduce3ScalarFloat

public float execReduce3ScalarFloat(long[] extraPointers,
                                    int opNum,
                                    long x,
                                    long xShapeInfo,
                                    long extraParamsVals,
                                    long y,
                                    long yShapeInfo)

Parameters:

opNum -

x -

xShapeInfo -

extraParamsVals -

y -

yShapeInfo -

execReduce3Float

public void execReduce3Float(long[] extraPointers,
                             int opNum,
                             long x,
                             long xShapeInfo,
                             long extraParamsVals,
                             long y,
                             long yShapeInfo,
                             long result,
                             long resultShapeInfoBuffer,
                             long dimension,
                             int dimensionLength)

Parameters:

opNum -

x -

xShapeInfo -

extraParamsVals -

y -

yShapeInfo -

result -

resultShapeInfoBuffer -

dimension -

dimensionLength -

execScalarFloat

public void execScalarFloat(long[] extraPointers,
                            int opNum,
                            long x,
                            int xStride,
                            long result,
                            int resultStride,
                            double scalar,
                            long extraParams,
                            long n)

Parameters:

opNum -

x -

xStride -

result -

resultStride -

scalar -

extraParams -

n -

execScalarFloat

public void execScalarFloat(long[] extraPointers,
                            int opNum,
                            long x,
                            long xShapeInfo,
                            long result,
                            long resultShapeInfo,
                            float scalar,
                            long extraParams)

Parameters:

opNum -

x -

xShapeInfo -

result -

resultShapeInfo -

scalar -

extraParams -

n -

execScalarFloat

public void execScalarFloat(long[] extraPointers,
                            int opNum,
                            long x,
                            long xShapeInfo,
                            long result,
                            long resultShapeInfo,
                            double scalar,
                            long extraParams,
                            long xIndexes,
                            long resultIndexes)

Parameters:

opNum -

x -

xShapeInfo -

result -

resultShapeInfo -

scalar -

extraParams -

xIndexes -

resultIndexes -

execSummaryStatsScalarFloat

public float execSummaryStatsScalarFloat(long[] extraPointers,
                                         int opNum,
                                         long x,
                                         long xShapeInfo,
                                         long extraParams,
                                         boolean biasCorrected)

Parameters:

opNum -

x -

xShapeInfo -

extraParams -

execSummaryStatsFloat

public void execSummaryStatsFloat(long[] extraPointers,
                                  int opNum,
                                  long x,
                                  long xShapeInfo,
                                  long extraParams,
                                  long result,
                                  long resultShapeInfo,
                                  boolean biasCorrected)

Parameters:

opNum -

x -

xShapeInfo -

extraParams -

result -

resultShapeInfo -

execSummaryStatsFloat

public void execSummaryStatsFloat(long[] extraPointers,
                                  int opNum,
                                  long x,
                                  long xShapeInfo,
                                  long extraParams,
                                  long result,
                                  long resultShapeInfoBuffer,
                                  long dimension,
                                  int dimensionLength,
                                  boolean biasCorrected)

Parameters:

opNum -

x -

xShapeInfo -

extraParams -

result -

resultShapeInfoBuffer -

dimension -

dimensionLength -

execTransformFloat

public void execTransformFloat(long[] extraPointers,
                               int opNum,
                               long dx,
                               int xStride,
                               long result,
                               int resultStride,
                               long extraParams,
                               long n)

Parameters:

opNum -

dx -

xStride -

result -

resultStride -

extraParams -

n -

execTransformFloat

public void execTransformFloat(long[] extraPointers,
                               int opNum,
                               long dx,
                               long xShapeInfo,
                               long result,
                               long resultShapeInfo,
                               long extraParams)

Parameters:

opNum -

dx -

xShapeInfo -

result -

resultShapeInfo -

extraParams -

n -

execTransformFloat

public void execTransformFloat(long[] extraPointers,
                               int opNum,
                               long dx,
                               long xShapeInfo,
                               long result,
                               long resultShapeInfo,
                               long extraParams,
                               long xIndexes,
                               long resultIndexes)

Parameters:

extraPointers -

opNum -

dx -

xShapeInfo -

result -

resultShapeInfo -

extraParams -

xIndexes -

resultIndexes -

flattenFloat

public void flattenFloat(long[] extraPointers,
                         int offset,
                         char order,
                         long result,
                         long resultShapeInfo,
                         long input,
                         long inputShapeInfo)

Append an input array to the end of a flat array in a particular order

Parameters:

offset - the offset of the array to start at

order - the order

result - the result array

resultShapeInfo - the shape info for te array

input - the input for the array

inputShapeInfo - the shape information for that array

flattenDouble

public void flattenDouble(long[] extraPointers,
                          int offset,
                          char order,
                          long result,
                          long resultShapeInfo,
                          long input,
                          long inputShapeInfo)

Append an input array to the end of a flat array in a particular order

Parameters:

offset - the offset of the array to start at

order - the order

result - the result array

resultShapeInfo - the shape info for te array

input - the input for the array

inputShapeInfo - the shape information for that array

concatDouble

public void concatDouble(long[] extraPointers,
                         int dimension,
                         int numArrays,
                         long[] data,
                         long[] inputShapeInfo,
                         long result,
                         long resultShapeInfo,
                         long[] tadPointers,
                         long[] tadOffsets)

Parameters:

dimension -

data -

inputShapeInfo -

result -

resultShapeInfo -

concatFloat

public void concatFloat(long[] extraPointers,
                        int dimension,
                        int numArrays,
                        long[] data,
                        long[] inputShapeInfo,
                        long result,
                        long resultShapeInfo,
                        long[] tadPointers,
                        long[] tadOffsets)

Parameters:

dimension -

data -

inputShapeInfo -

result -

resultShapeInfo -

ompGetNumThreads

public int ompGetNumThreads()

Gets the number of open mp threads

Returns:

setOmpNumThreads

public void setOmpNumThreads(int threads)

Sets the number of openmp threads

Parameters:

threads -

initializeDevicesAndFunctions

public void initializeDevicesAndFunctions()

NEVER EVER USE THIS METHOD OUTSIDE OF CUDA

mallocHost

public long mallocHost(long memorySize,
                       int flags)

mallocDevice

public long mallocDevice(long memorySize,
                         long ptrToDeviceId,
                         int flags)

freeHost

public long freeHost(long pointer)

freeDevice

public long freeDevice(long pointer,
                       long deviceId)

createContext

public long createContext()

createStream

public long createStream()

createEvent

public long createEvent()

createBlasHandle

public long createBlasHandle()

registerEvent

public long registerEvent(long event,
                          long stream)

destroyEvent

public long destroyEvent(long event)

setBlasStream

public long setBlasStream(long handle,
                          long stream)

setDevice

public long setDevice(long ptrToDeviceId)

streamSynchronize

public long streamSynchronize(long stream)

eventSynchronize

public long eventSynchronize(long event)

getDeviceFreeMemory

public long getDeviceFreeMemory(long ptrToDeviceId)

getDeviceTotalMemory

public long getDeviceTotalMemory(long ptrToDeviceId)

memcpy

public long memcpy(long dst,
                   long src,
                   long size,
                   int flags,
                   long reserved)

memcpyAsync

public long memcpyAsync(long dst,
                        long src,
                        long size,
                        int flags,
                        long reserved)

memcpyConstantAsync

public long memcpyConstantAsync(long dst,
                                long src,
                                long size,
                                int flags,
                                long reserved)

memset

public long memset(long dst,
                   int value,
                   long size,
                   int flags,
                   long reserved)

memsetAsync

public long memsetAsync(long dst,
                        int value,
                        long size,
                        int flags,
                        long reserved)

getConstantSpace

public long getConstantSpace()

getAvailableDevices

public long getAvailableDevices()

enableDebugMode

public void enableDebugMode(boolean reallyEnable)

enableVerboseMode

public void enableVerboseMode(boolean reallyEnable)

setGridLimit

public void setGridLimit(int gridSize)

tadOnlyShapeInfo

public void tadOnlyShapeInfo(long shapeInfo,
                             long dimension,
                             int dimensionLength,
                             long targetBuffer,
                             long offsetsBuffer)