Brutex
/
cavis
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Actually export functions from NativeOps.h

master
Samuel Audet 2019-07-23 14:36:48 +09:00 committed by AlexDBlack
parent dcc72e23b2
commit 780ae628a9
4 changed files with 157 additions and 157 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

290
libnd4j/blas/NativeOps.h
View File

@ -84,19 +84,19 @@ extern "C" {
* @param p * @param p
* @param len * @param len
*/ */
void tryPointer(Nd4jPointer extra, Nd4jPointer p, int len); ND4J_EXPORT void tryPointer(Nd4jPointer extra, Nd4jPointer p, int len);
/** /**
* *
* @param num * @param num
*/ */
void setElementThreshold(int num); ND4J_EXPORT void setElementThreshold(int num);
/** /**
* *
* @param num * @param num
*/ */
void setTADThreshold(int num); ND4J_EXPORT void setTADThreshold(int num);
/** /**
* *
@ -105,7 +105,7 @@ void setTADThreshold(int num);
* @param xShapeInfo * @param xShapeInfo
* @param extraParams * @param extraParams
*/ */
void execIndexReduceScalar(Nd4jPointer *extraPointers, ND4J_EXPORT void execIndexReduceScalar(Nd4jPointer *extraPointers,
int opNum, int opNum,
void *hX, Nd4jLong *hXShapeInfo, void *hX, Nd4jLong *hXShapeInfo,
void *dX, Nd4jLong *dXShapeInfo, void *dX, Nd4jLong *dXShapeInfo,
@ -124,7 +124,7 @@ void execIndexReduceScalar(Nd4jPointer *extraPointers,
* @param dimension * @param dimension
* @param dimensionLength * @param dimensionLength
*/ */
void execIndexReduce(Nd4jPointer *extraPointers, ND4J_EXPORT void execIndexReduce(Nd4jPointer *extraPointers,
int opNum, int opNum,
void *hX, Nd4jLong *hXShapeInfo, void *hX, Nd4jLong *hXShapeInfo,
void *dX, Nd4jLong *dXShapeInfo, void *dX, Nd4jLong *dXShapeInfo,
@ -146,7 +146,7 @@ void execIndexReduce(Nd4jPointer *extraPointers,
* @param dimension * @param dimension
* @param dimensionLength * @param dimensionLength
*/ */
void execBroadcast( ND4J_EXPORT void execBroadcast(
Nd4jPointer *extraPointers, Nd4jPointer *extraPointers,
int opNum, int opNum,
void *hX, Nd4jLong *hXShapeInfo, void *hX, Nd4jLong *hXShapeInfo,
@ -159,7 +159,7 @@ void execBroadcast(
void *dDimension, Nd4jLong *dDimensionShape); void *dDimension, Nd4jLong *dDimensionShape);
void execBroadcastBool( ND4J_EXPORT void execBroadcastBool(
Nd4jPointer *extraPointers, Nd4jPointer *extraPointers,
int opNum, int opNum,
void *hX, Nd4jLong *hXShapeInfo, void *hX, Nd4jLong *hXShapeInfo,
@ -183,7 +183,7 @@ void execBroadcastBool(
* @param extraParams * @param extraParams
* @param n * @param n
*/ */
void execPairwiseTransform( ND4J_EXPORT void execPairwiseTransform(
Nd4jPointer *extraPointers, Nd4jPointer *extraPointers,
int opNum, int opNum,
void *hX, Nd4jLong *hXShapeInfo, void *hX, Nd4jLong *hXShapeInfo,
@ -194,7 +194,7 @@ void execPairwiseTransform(
void *dZ, Nd4jLong *dZShapeInfo, void *dZ, Nd4jLong *dZShapeInfo,
void *extraParams); void *extraParams);
void execPairwiseTransformBool( ND4J_EXPORT void execPairwiseTransformBool(
Nd4jPointer *extraPointers, Nd4jPointer *extraPointers,
int opNum, int opNum,
void *hX, Nd4jLong *hXShapeInfo, void *hX, Nd4jLong *hXShapeInfo,
@ -214,7 +214,7 @@ void execPairwiseTransformBool(
* @param result * @param result
* @param resultShapeInfo * @param resultShapeInfo
*/ */
void execReduceFloat(Nd4jPointer *extraPointers, ND4J_EXPORT void execReduceFloat(Nd4jPointer *extraPointers,
int opNum, int opNum,
void *hX, Nd4jLong *hXShapeInfo, void *hX, Nd4jLong *hXShapeInfo,
void *dX, Nd4jLong *dXShapeInfo, void *dX, Nd4jLong *dXShapeInfo,
@ -222,7 +222,7 @@ void execReduceFloat(Nd4jPointer *extraPointers,
void *hZ, Nd4jLong *hZShapeInfo, void *hZ, Nd4jLong *hZShapeInfo,
void *dZ, Nd4jLong *dZShapeInfo); void *dZ, Nd4jLong *dZShapeInfo);
void execReduceSame(Nd4jPointer *extraPointers, ND4J_EXPORT void execReduceSame(Nd4jPointer *extraPointers,
int opNum, int opNum,
void *hX, Nd4jLong *hXShapeInfo, void *hX, Nd4jLong *hXShapeInfo,
void *dX, Nd4jLong *dXShapeInfo, void *dX, Nd4jLong *dXShapeInfo,
@ -230,7 +230,7 @@ void execReduceSame(Nd4jPointer *extraPointers,
void *hZ, Nd4jLong *hZShapeInfo, void *hZ, Nd4jLong *hZShapeInfo,
void *dZ, Nd4jLong *dZShapeInfo); void *dZ, Nd4jLong *dZShapeInfo);
void execReduceBool(Nd4jPointer *extraPointers, ND4J_EXPORT void execReduceBool(Nd4jPointer *extraPointers,
int opNum, int opNum,
void *hX, Nd4jLong *hXShapeInfo, void *hX, Nd4jLong *hXShapeInfo,
void *dX, Nd4jLong *dXShapeInfo, void *dX, Nd4jLong *dXShapeInfo,
@ -239,7 +239,7 @@ void execReduceBool(Nd4jPointer *extraPointers,
void *dZ, Nd4jLong *dZShapeInfo); void *dZ, Nd4jLong *dZShapeInfo);
void execReduceLong(Nd4jPointer *extraPointers, ND4J_EXPORT void execReduceLong(Nd4jPointer *extraPointers,
int opNum, int opNum,
void *hX, Nd4jLong *hXShapeInfo, void *hX, Nd4jLong *hXShapeInfo,
void *dX, Nd4jLong *dXShapeInfo, void *dX, Nd4jLong *dXShapeInfo,
@ -256,7 +256,7 @@ void execReduceLong(Nd4jPointer *extraPointers,
* @param result * @param result
* @param resultShapeInfo * @param resultShapeInfo
*/ */
void execReduceFloat2(Nd4jPointer *extraPointers, ND4J_EXPORT void execReduceFloat2(Nd4jPointer *extraPointers,
int opNum, int opNum,
void *hX, Nd4jLong *hXShapeInfo, void *hX, Nd4jLong *hXShapeInfo,
void *dX, Nd4jLong *dXShapeInfo, void *dX, Nd4jLong *dXShapeInfo,
@ -267,7 +267,7 @@ void execReduceFloat2(Nd4jPointer *extraPointers,
void *dDimension, Nd4jLong *dDimensionShape); void *dDimension, Nd4jLong *dDimensionShape);
void execReduceSame2(Nd4jPointer *extraPointers, ND4J_EXPORT void execReduceSame2(Nd4jPointer *extraPointers,
int opNum, int opNum,
void *hX, Nd4jLong *hXShapeInfo, void *hX, Nd4jLong *hXShapeInfo,
void *dX, Nd4jLong *dXShapeInfo, void *dX, Nd4jLong *dXShapeInfo,
@ -278,7 +278,7 @@ void execReduceSame2(Nd4jPointer *extraPointers,
void *dDimension, Nd4jLong *dDimensionShape); void *dDimension, Nd4jLong *dDimensionShape);
void execReduceBool2(Nd4jPointer *extraPointers, ND4J_EXPORT void execReduceBool2(Nd4jPointer *extraPointers,
int opNum, int opNum,
void *hX, Nd4jLong *hXShapeInfo, void *hX, Nd4jLong *hXShapeInfo,
void *dX, Nd4jLong *dXShapeInfo, void *dX, Nd4jLong *dXShapeInfo,
@ -289,7 +289,7 @@ void execReduceBool2(Nd4jPointer *extraPointers,
void *dDimension, Nd4jLong *dDimensionShape); void *dDimension, Nd4jLong *dDimensionShape);
void execReduceLong2(Nd4jPointer *extraPointers, ND4J_EXPORT void execReduceLong2(Nd4jPointer *extraPointers,
int opNum, int opNum,
void *hX, Nd4jLong *hXShapeInfo, void *hX, Nd4jLong *hXShapeInfo,
void *dX, Nd4jLong *dXShapeInfo, void *dX, Nd4jLong *dXShapeInfo,
@ -310,7 +310,7 @@ void execReduceLong2(Nd4jPointer *extraPointers,
* @param result * @param result
* @param resultShapeInfo * @param resultShapeInfo
*/ */
void execReduce3(Nd4jPointer *extraPointers, ND4J_EXPORT void execReduce3(Nd4jPointer *extraPointers,
int opNum, int opNum,
void *hX, Nd4jLong *hXShapeInfo, void *hX, Nd4jLong *hXShapeInfo,
void *dX, Nd4jLong *dXShapeInfo, void *dX, Nd4jLong *dXShapeInfo,
@ -329,7 +329,7 @@ void execReduce3(Nd4jPointer *extraPointers,
* @param y * @param y
* @param yShapeInfo * @param yShapeInfo
*/ */
void execReduce3Scalar(Nd4jPointer *extraPointers, ND4J_EXPORT void execReduce3Scalar(Nd4jPointer *extraPointers,
int opNum, int opNum,
void *hX, Nd4jLong *hXShapeInfo, void *hX, Nd4jLong *hXShapeInfo,
void *dX, Nd4jLong *dXShapeInfo, void *dX, Nd4jLong *dXShapeInfo,
@ -351,7 +351,7 @@ void execReduce3Scalar(Nd4jPointer *extraPointers,
* @param dimension * @param dimension
* @param dimensionLength * @param dimensionLength
*/ */
void execReduce3Tad(Nd4jPointer *extraPointers, ND4J_EXPORT void execReduce3Tad(Nd4jPointer *extraPointers,
int opNum, int opNum,
void *hX, Nd4jLong *hXShapeInfo, void *hX, Nd4jLong *hXShapeInfo,
void *dX, Nd4jLong *dXShapeInfo, void *dX, Nd4jLong *dXShapeInfo,
@ -366,7 +366,7 @@ void execReduce3Tad(Nd4jPointer *extraPointers,
Nd4jLong *yTadOnlyShapeInfo, Nd4jLong *yTadOffsets); Nd4jLong *yTadOnlyShapeInfo, Nd4jLong *yTadOffsets);
void execReduce3All(Nd4jPointer *extraPointers, ND4J_EXPORT void execReduce3All(Nd4jPointer *extraPointers,
int opNum, int opNum,
void *hX, Nd4jLong *hXShapeInfo, void *hX, Nd4jLong *hXShapeInfo,
void *dX, Nd4jLong *dXShapeInfo, void *dX, Nd4jLong *dXShapeInfo,
@ -391,7 +391,7 @@ void execReduce3All(Nd4jPointer *extraPointers,
* @param extraParams * @param extraParams
* @param n * @param n
*/ */
void execScalar(Nd4jPointer *extraPointers, ND4J_EXPORT void execScalar(Nd4jPointer *extraPointers,
int opNum, int opNum,
void *hX, Nd4jLong *hXShapeInfo, void *hX, Nd4jLong *hXShapeInfo,
void *dX, Nd4jLong *dXShapeInfo, void *dX, Nd4jLong *dXShapeInfo,
@ -401,7 +401,7 @@ void execScalar(Nd4jPointer *extraPointers,
void *dScalar, Nd4jLong *dSscalarShapeInfo, void *dScalar, Nd4jLong *dSscalarShapeInfo,
void *extraParams); void *extraParams);
void execScalarBool(Nd4jPointer *extraPointers, ND4J_EXPORT void execScalarBool(Nd4jPointer *extraPointers,
int opNum, int opNum,
void *hX, Nd4jLong *hXShapeInfo, void *hX, Nd4jLong *hXShapeInfo,
void *dX, Nd4jLong *dXShapeInfo, void *dX, Nd4jLong *dXShapeInfo,
@ -418,7 +418,7 @@ void execScalarBool(Nd4jPointer *extraPointers,
* @param xShapeInfo * @param xShapeInfo
* @param extraParams * @param extraParams
*/ */
void execSummaryStatsScalar(Nd4jPointer *extraPointers, ND4J_EXPORT void execSummaryStatsScalar(Nd4jPointer *extraPointers,
int opNum, int opNum,
void *hX, Nd4jLong *hXShapeInfo, void *hX, Nd4jLong *hXShapeInfo,
void *dX, Nd4jLong *dXShapeInfo, void *dX, Nd4jLong *dXShapeInfo,
@ -435,7 +435,7 @@ void execSummaryStatsScalar(Nd4jPointer *extraPointers,
* @param result * @param result
* @param resultShapeInfo * @param resultShapeInfo
*/ */
void execSummaryStats(Nd4jPointer *extraPointers, ND4J_EXPORT void execSummaryStats(Nd4jPointer *extraPointers,
int opNum, int opNum,
void *hX, Nd4jLong *hXShapeInfo, void *hX, Nd4jLong *hXShapeInfo,
void *dX, Nd4jLong *dXShapeInfo, void *dX, Nd4jLong *dXShapeInfo,
@ -454,7 +454,7 @@ void execSummaryStats(Nd4jPointer *extraPointers,
* @param dimension * @param dimension
* @param dimensionLength * @param dimensionLength
*/ */
void execSummaryStatsTad(Nd4jPointer *extraPointers, ND4J_EXPORT void execSummaryStatsTad(Nd4jPointer *extraPointers,
int opNum, int opNum,
void *hX, Nd4jLong *hXShapeInfo, void *hX, Nd4jLong *hXShapeInfo,
void *dX, Nd4jLong *dXShapeInfo, void *dX, Nd4jLong *dXShapeInfo,
@ -476,7 +476,7 @@ void execSummaryStatsTad(Nd4jPointer *extraPointers,
* @param extraParams * @param extraParams
* @param n * @param n
*/ */
void execTransformFloat(Nd4jPointer *extraPointers, ND4J_EXPORT void execTransformFloat(Nd4jPointer *extraPointers,
int opNum, int opNum,
void *hX, Nd4jLong *hXShapeInfo, void *hX, Nd4jLong *hXShapeInfo,
void *dX, Nd4jLong *dXShapeInfo, void *dX, Nd4jLong *dXShapeInfo,
@ -484,7 +484,7 @@ void execTransformFloat(Nd4jPointer *extraPointers,
void *dZ, Nd4jLong *dZShapeInfo, void *dZ, Nd4jLong *dZShapeInfo,
void *extraParams); void *extraParams);
void execTransformSame(Nd4jPointer *extraPointers, ND4J_EXPORT void execTransformSame(Nd4jPointer *extraPointers,
int opNum, int opNum,
void *hX, Nd4jLong *hXShapeInfo, void *hX, Nd4jLong *hXShapeInfo,
void *dX, Nd4jLong *dXShapeInfo, void *dX, Nd4jLong *dXShapeInfo,
@ -492,7 +492,7 @@ void execTransformSame(Nd4jPointer *extraPointers,
void *dZ, Nd4jLong *dZShapeInfo, void *dZ, Nd4jLong *dZShapeInfo,
void *extraParams); void *extraParams);
void execTransformBool(Nd4jPointer *extraPointers, ND4J_EXPORT void execTransformBool(Nd4jPointer *extraPointers,
int opNum, int opNum,
void *hX, Nd4jLong *hXShapeInfo, void *hX, Nd4jLong *hXShapeInfo,
void *dX, Nd4jLong *dXShapeInfo, void *dX, Nd4jLong *dXShapeInfo,
@ -500,7 +500,7 @@ void execTransformBool(Nd4jPointer *extraPointers,
void *dZ, Nd4jLong *dZShapeInfo, void *dZ, Nd4jLong *dZShapeInfo,
void *extraParams); void *extraParams);
void execTransformAny(Nd4jPointer *extraPointers, ND4J_EXPORT void execTransformAny(Nd4jPointer *extraPointers,
int opNum, int opNum,
void *hX, Nd4jLong *hXShapeInfo, void *hX, Nd4jLong *hXShapeInfo,
void *dX, Nd4jLong *dXShapeInfo, void *dX, Nd4jLong *dXShapeInfo,
@ -508,7 +508,7 @@ void execTransformAny(Nd4jPointer *extraPointers,
void *dZ, Nd4jLong *dZShapeInfo, void *dZ, Nd4jLong *dZShapeInfo,
void *extraParams); void *extraParams);
void execTransformStrict(Nd4jPointer *extraPointers, ND4J_EXPORT void execTransformStrict(Nd4jPointer *extraPointers,
int opNum, int opNum,
void *hX, Nd4jLong *hXShapeInfo, void *hX, Nd4jLong *hXShapeInfo,
void *dX, Nd4jLong *dXShapeInfo, void *dX, Nd4jLong *dXShapeInfo,
@ -529,7 +529,7 @@ void execTransformStrict(Nd4jPointer *extraPointers,
* @param dimension * @param dimension
* @param dimensionLength * @param dimensionLength
*/ */
void execScalarTad(Nd4jPointer *extraPointers, ND4J_EXPORT void execScalarTad(Nd4jPointer *extraPointers,
int opNum, int opNum,
void *hX, Nd4jLong *hXShapeInfo, void *hX, Nd4jLong *hXShapeInfo,
void *dX, Nd4jLong *dXShapeInfo, void *dX, Nd4jLong *dXShapeInfo,
@ -543,7 +543,7 @@ void execScalarTad(Nd4jPointer *extraPointers,
Nd4jLong *tadShapeInfo, Nd4jLong *tadOffsets, Nd4jLong *tadShapeInfo, Nd4jLong *tadOffsets,
Nd4jLong *tadShapeInfoZ, Nd4jLong *tadOffsetsZ); Nd4jLong *tadShapeInfoZ, Nd4jLong *tadOffsetsZ);
void execScalarBoolTad(Nd4jPointer *extraPointers, ND4J_EXPORT void execScalarBoolTad(Nd4jPointer *extraPointers,
int opNum, int opNum,
void *hX, Nd4jLong *hXShapeInfo, void *hX, Nd4jLong *hXShapeInfo,
void *dX, Nd4jLong *dXShapeInfo, void *dX, Nd4jLong *dXShapeInfo,
@ -569,7 +569,7 @@ void execScalarBoolTad(Nd4jPointer *extraPointers,
* @param input the input for the array * @param input the input for the array
* @param inputShapeInfo the shape information for that array * @param inputShapeInfo the shape information for that array
*/ */
void flatten( ND4J_EXPORT void flatten(
Nd4jPointer *extraPointers, Nd4jPointer *extraPointers,
int offset, int offset,
char order, char order,
@ -578,7 +578,7 @@ void flatten(
void *input, Nd4jLong *inputShapeInfo, void *input, Nd4jLong *inputShapeInfo,
void *dinput, Nd4jLong *dinputShapeInfo); void *dinput, Nd4jLong *dinputShapeInfo);
void concat( ND4J_EXPORT void concat(
Nd4jPointer *extraPointers, Nd4jPointer *extraPointers,
int dimension, int dimension,
int numArrays, int numArrays,
@ -589,7 +589,7 @@ void concat(
Nd4jPointer *tadPointers, Nd4jPointer *offsetPointers); Nd4jPointer *tadPointers, Nd4jPointer *offsetPointers);
void specialConcat ( ND4J_EXPORT void specialConcat (
Nd4jPointer *extraPointers, Nd4jPointer *extraPointers,
int dimension, int dimension,
int numArrays, int numArrays,
@ -604,9 +604,9 @@ void specialConcat (
* This method implementation exists only for cuda. * This method implementation exists only for cuda.
* The other backends should have dummy method for JNI compatibility reasons. * The other backends should have dummy method for JNI compatibility reasons.
*/ */
void initializeDevicesAndFunctions(); ND4J_EXPORT void initializeDevicesAndFunctions();
void initializeFunctions(Nd4jPointer *functions); ND4J_EXPORT void initializeFunctions(Nd4jPointer *functions);
/** /**
* This method acquires memory chunk of requested size on host side * This method acquires memory chunk of requested size on host side
@ -615,7 +615,7 @@ void initializeFunctions(Nd4jPointer *functions);
* @param memorySize memory size, in bytes * @param memorySize memory size, in bytes
* @param flags optional parameter * @param flags optional parameter
*/ */
Nd4jPointer mallocHost(Nd4jLong memorySize, int flags); ND4J_EXPORT Nd4jPointer mallocHost(Nd4jLong memorySize, int flags);
/** /**
* This method acquires memory chunk of requested size on specified device * This method acquires memory chunk of requested size on specified device
@ -625,14 +625,14 @@ Nd4jPointer mallocHost(Nd4jLong memorySize, int flags);
* @param ptrToDeviceId pointer to deviceId. For cuda that's just and int, for OpenCL that's pointer to device_id, etc * @param ptrToDeviceId pointer to deviceId. For cuda that's just and int, for OpenCL that's pointer to device_id, etc
* @param flags optional parameter * @param flags optional parameter
*/ */
Nd4jPointer mallocDevice(Nd4jLong memorySize, int deviceId, int flags); ND4J_EXPORT Nd4jPointer mallocDevice(Nd4jLong memorySize, int deviceId, int flags);
/** /**
* This method releases previously allocated host memory space * This method releases previously allocated host memory space
* *
* @param pointer pointer that'll be freed * @param pointer pointer that'll be freed
*/ */
int freeHost(Nd4jPointer pointer); ND4J_EXPORT int freeHost(Nd4jPointer pointer);
/** /**
* This method releases previously allocated memory space on device * This method releases previously allocated memory space on device
@ -640,31 +640,31 @@ int freeHost(Nd4jPointer pointer);
* @param pointer pointer that'll be freed * @param pointer pointer that'll be freed
* @param ptrToDeviceId pointer to deviceId. * @param ptrToDeviceId pointer to deviceId.
*/ */
int freeDevice(Nd4jPointer pointer, int deviceId); ND4J_EXPORT int freeDevice(Nd4jPointer pointer, int deviceId);
/** /**
* *
* @return * @return
*/ */
int ompGetMaxThreads(); ND4J_EXPORT int ompGetMaxThreads();
/** /**
* *
* @return * @return
*/ */
int ompGetNumThreads(); ND4J_EXPORT int ompGetNumThreads();
/** /**
* *
* @param threads * @param threads
*/ */
void setOmpNumThreads(int threads); ND4J_EXPORT void setOmpNumThreads(int threads);
/** /**
* *
* @param threads * @param threads
*/ */
void setOmpMinThreads(int threads); ND4J_EXPORT void setOmpMinThreads(int threads);
@ -673,19 +673,19 @@ void setOmpMinThreads(int threads);
* *
* @return * @return
*/ */
Nd4jPointer createContext(); ND4J_EXPORT Nd4jPointer createContext();
/** /**
* *
* @return * @return
*/ */
Nd4jPointer createStream(); ND4J_EXPORT Nd4jPointer createStream();
/** /**
* *
* @return * @return
*/ */
Nd4jPointer createEvent(); ND4J_EXPORT Nd4jPointer createEvent();
/** /**
* *
@ -693,89 +693,89 @@ Nd4jPointer createEvent();
* @param stream * @param stream
* @return * @return
*/ */
int registerEvent(Nd4jPointer event, Nd4jPointer stream); ND4J_EXPORT int registerEvent(Nd4jPointer event, Nd4jPointer stream);
/** /**
* *
* @param event * @param event
* @return * @return
*/ */
int destroyEvent(Nd4jPointer event); ND4J_EXPORT int destroyEvent(Nd4jPointer event);
/** /**
* *
* @param ptrToDeviceId * @param ptrToDeviceId
* @return * @return
*/ */
int setDevice(int deviceId); ND4J_EXPORT int setDevice(int deviceId);
/** /**
* *
* @return * @return
*/ */
int getDevice(); ND4J_EXPORT int getDevice();
/** /**
* *
* @param stream * @param stream
* @return * @return
*/ */
int streamSynchronize(Nd4jPointer stream); ND4J_EXPORT int streamSynchronize(Nd4jPointer stream);
/** /**
* *
* @param event * @param event
* @return * @return
*/ */
int eventSynchronize(Nd4jPointer event); ND4J_EXPORT int eventSynchronize(Nd4jPointer event);
/** /**
* *
* @param ptrToDeviceId * @param ptrToDeviceId
* @return * @return
*/ */
Nd4jLong getDeviceFreeMemory(int deviceId); ND4J_EXPORT Nd4jLong getDeviceFreeMemory(int deviceId);
/** /**
* Returns amount of free memory for current device * Returns amount of free memory for current device
* @return * @return
*/ */
Nd4jLong getDeviceFreeMemoryDefault(); ND4J_EXPORT Nd4jLong getDeviceFreeMemoryDefault();
/** /**
* *
* @param ptrToDeviceId * @param ptrToDeviceId
* @return * @return
*/ */
Nd4jLong getDeviceTotalMemory(int deviceId); ND4J_EXPORT Nd4jLong getDeviceTotalMemory(int deviceId);
/** /**
* *
* @param ptrToDeviceId * @param ptrToDeviceId
* @return * @return
*/ */
int getDeviceMajor(int deviceId); ND4J_EXPORT int getDeviceMajor(int deviceId);
/** /**
* This method returns amount of cached memory * This method returns amount of cached memory
* @param deviceId * @param deviceId
* @return * @return
*/ */
Nd4jLong getCachedMemory(int deviceId); ND4J_EXPORT Nd4jLong getCachedMemory(int deviceId);
/** /**
* *
* @param ptrToDeviceId * @param ptrToDeviceId
* @return * @return
*/ */
int getDeviceMinor(int deviceId); ND4J_EXPORT int getDeviceMinor(int deviceId);
/** /**
* *
* @param ptrToDeviceId * @param ptrToDeviceId
* @return * @return
*/ */
const char * getDeviceName(int deviceId); ND4J_EXPORT const char * getDeviceName(int deviceId);
/** /**
* *
@ -786,7 +786,7 @@ const char * getDeviceName(int deviceId);
* @param reserved * @param reserved
* @return * @return
*/ */
int memcpySync(Nd4jPointer dst, ND4J_EXPORT int memcpySync(Nd4jPointer dst,
Nd4jPointer src, Nd4jPointer src,
Nd4jLong size, Nd4jLong size,
int flags, int flags,
@ -801,7 +801,7 @@ int memcpySync(Nd4jPointer dst,
* @param reserved * @param reserved
* @return * @return
*/ */
int memcpyAsync(Nd4jPointer dst, ND4J_EXPORT int memcpyAsync(Nd4jPointer dst,
Nd4jPointer src, Nd4jPointer src,
Nd4jLong size, Nd4jLong size,
int flags, int flags,
@ -816,7 +816,7 @@ int memcpyAsync(Nd4jPointer dst,
* @param reserved * @param reserved
* @return * @return
*/ */
int memsetSync(Nd4jPointer dst, ND4J_EXPORT int memsetSync(Nd4jPointer dst,
int value, int value,
Nd4jLong size, Nd4jLong size,
int flags, int flags,
@ -831,7 +831,7 @@ int memsetSync(Nd4jPointer dst,
* @param reserved * @param reserved
* @return * @return
*/ */
int memsetAsync(Nd4jPointer dst, ND4J_EXPORT int memsetAsync(Nd4jPointer dst,
int value, int value,
Nd4jLong size, Nd4jLong size,
int flags, int flags,
@ -846,7 +846,7 @@ int memsetAsync(Nd4jPointer dst,
* @param reserved * @param reserved
* @return * @return
*/ */
int memcpyConstantAsync(Nd4jLong dst, ND4J_EXPORT int memcpyConstantAsync(Nd4jLong dst,
Nd4jPointer src, Nd4jPointer src,
Nd4jLong size, Nd4jLong size,
int flags, int flags,
@ -856,31 +856,31 @@ int memcpyConstantAsync(Nd4jLong dst,
* *
* @return * @return
*/ */
Nd4jPointer getConstantSpace(); ND4J_EXPORT Nd4jPointer getConstantSpace();
/** /**
* *
* @return * @return
*/ */
int getAvailableDevices(); ND4J_EXPORT int getAvailableDevices();
/** /**
* *
* @param reallyEnable * @param reallyEnable
*/ */
void enableDebugMode(bool reallyEnable); ND4J_EXPORT void enableDebugMode(bool reallyEnable);
/** /**
* *
* @param reallyEnable * @param reallyEnable
*/ */
void enableVerboseMode(bool reallyEnable); ND4J_EXPORT void enableVerboseMode(bool reallyEnable);
/** /**
* *
* @param gridSize * @param gridSize
*/ */
void setGridLimit(int gridSize); ND4J_EXPORT void setGridLimit(int gridSize);
/** /**
* *
@ -890,7 +890,7 @@ void setGridLimit(int gridSize);
* @param targetBuffer * @param targetBuffer
* @param offsetsBuffer * @param offsetsBuffer
*/ */
nd4j::TadPack* tadOnlyShapeInfo(Nd4jLong *xShapeInfo, ND4J_EXPORT nd4j::TadPack* tadOnlyShapeInfo(Nd4jLong *xShapeInfo,
int *dimension, int *dimension,
int dimensionLength); int dimensionLength);
@ -912,7 +912,7 @@ nd4j::TadPack* tadOnlyShapeInfo(Nd4jLong *xShapeInfo,
* @param zTadShapeInfo * @param zTadShapeInfo
* @param zTadOffsets * @param zTadOffsets
*/ */
void pullRows(Nd4jPointer *extraPointers, ND4J_EXPORT void pullRows(Nd4jPointer *extraPointers,
void *x, Nd4jLong *xShapeInfo, void *x, Nd4jLong *xShapeInfo,
void *dx, Nd4jLong *dxShapeInfo, void *dx, Nd4jLong *dxShapeInfo,
void *z, Nd4jLong *zShapeInfo, void *z, Nd4jLong *zShapeInfo,
@ -933,7 +933,7 @@ void pullRows(Nd4jPointer *extraPointers,
* @param length * @param length
* @param propagate * @param propagate
*/ */
void average(Nd4jPointer *extras, ND4J_EXPORT void average(Nd4jPointer *extras,
Nd4jPointer *x, Nd4jLong *xShapeInfo, Nd4jPointer *x, Nd4jLong *xShapeInfo,
Nd4jPointer *dx, Nd4jLong *dxShapeInfo, Nd4jPointer *dx, Nd4jLong *dxShapeInfo,
void *z, Nd4jLong *zShapeInfo, void *z, Nd4jLong *zShapeInfo,
@ -943,7 +943,7 @@ void average(Nd4jPointer *extras,
bool propagate); bool propagate);
void accumulate(Nd4jPointer *extras, ND4J_EXPORT void accumulate(Nd4jPointer *extras,
Nd4jPointer *x, Nd4jLong *xShapeInfo, Nd4jPointer *x, Nd4jLong *xShapeInfo,
Nd4jPointer *dx, Nd4jLong *dxShapeInfo, Nd4jPointer *dx, Nd4jLong *dxShapeInfo,
void *z, Nd4jLong *zShapeInfo, void *z, Nd4jLong *zShapeInfo,
@ -959,18 +959,18 @@ void accumulate(Nd4jPointer *extras,
* *
* @param enable * @param enable
*/ */
void enableP2P(bool enable); ND4J_EXPORT void enableP2P(bool enable);
/** /**
* *
*/ */
void checkP2P(); ND4J_EXPORT void checkP2P();
/** /**
* *
* @return * @return
*/ */
bool isP2PAvailable(); ND4J_EXPORT bool isP2PAvailable();
/** /**
* Shuffle methods * Shuffle methods
@ -988,7 +988,7 @@ bool isP2PAvailable();
* @param tadShapeInfo * @param tadShapeInfo
* @param tadOffsets * @param tadOffsets
*/ */
void shuffle(Nd4jPointer *extras, ND4J_EXPORT void shuffle(Nd4jPointer *extras,
Nd4jPointer *x, Nd4jPointer *xShapeInfo, Nd4jPointer *x, Nd4jPointer *xShapeInfo,
Nd4jPointer *dx, Nd4jPointer *dxShapeInfo, Nd4jPointer *dx, Nd4jPointer *dxShapeInfo,
Nd4jPointer *z, Nd4jPointer *zShapeInfo, Nd4jPointer *z, Nd4jPointer *zShapeInfo,
@ -1012,14 +1012,14 @@ void shuffle(Nd4jPointer *extras,
* @param dstType * @param dstType
* @param z * @param z
*/ */
void convertTypes(Nd4jPointer *extras, int srcType, Nd4jPointer x, Nd4jLong N, int dstType, Nd4jPointer z); ND4J_EXPORT void convertTypes(Nd4jPointer *extras, int srcType, Nd4jPointer x, Nd4jLong N, int dstType, Nd4jPointer z);
/** /**
* *
* @return * @return
*/ */
bool isExperimentalEnabled(); ND4J_EXPORT bool isExperimentalEnabled();
/** /**
* Aggregate * Aggregate
@ -1040,7 +1040,7 @@ bool isExperimentalEnabled();
* @param realArguments * @param realArguments
* @param numRealArguments * @param numRealArguments
*/ */
void execAggregate(Nd4jPointer *extraPointers, ND4J_EXPORT void execAggregate(Nd4jPointer *extraPointers,
int opNum, int opNum,
void **arguments, void **arguments,
int numArguments, int numArguments,
@ -1055,7 +1055,7 @@ void execAggregate(Nd4jPointer *extraPointers,
nd4j::DataType dtype); nd4j::DataType dtype);
void batchExecutor(Nd4jPointer *extraPointers, ND4J_EXPORT void batchExecutor(Nd4jPointer *extraPointers,
int numAggregates, int numAggregates,
int opNum, int opNum,
int maxArgs, int maxArgs,
@ -1067,7 +1067,7 @@ void batchExecutor(Nd4jPointer *extraPointers,
void *ptrToArguments, void *ptrToArguments,
nd4j::DataType dtype); nd4j::DataType dtype);
void execAggregateBatch(Nd4jPointer *extraPointers, ND4J_EXPORT void execAggregateBatch(Nd4jPointer *extraPointers,
int numAggregates, int numAggregates,
int opNum, int opNum,
int maxArgs, int maxArgs,
@ -1092,7 +1092,7 @@ void execAggregateBatch(Nd4jPointer *extraPointers,
* @param zShapeBuffer * @param zShapeBuffer
* @param extraArguments * @param extraArguments
*/ */
void execRandom(Nd4jPointer *extraPointers, ND4J_EXPORT void execRandom(Nd4jPointer *extraPointers,
int opNum, int opNum,
Nd4jPointer state, Nd4jPointer state,
void *hZ, Nd4jLong *hZShapeBuffer, void *hZ, Nd4jLong *hZShapeBuffer,
@ -1112,7 +1112,7 @@ void execRandom(Nd4jPointer *extraPointers,
* @param zShapeBuffer * @param zShapeBuffer
* @param extraArguments * @param extraArguments
*/ */
void execRandom3(Nd4jPointer *extraPointers, ND4J_EXPORT void execRandom3(Nd4jPointer *extraPointers,
int opNum, int opNum,
Nd4jPointer state, Nd4jPointer state,
void *hX, Nd4jLong *hXShapeBuffer, void *hX, Nd4jLong *hXShapeBuffer,
@ -1134,7 +1134,7 @@ void execRandom3(Nd4jPointer *extraPointers,
* @param zShapeBuffer * @param zShapeBuffer
* @param extraArguments * @param extraArguments
*/ */
void execRandom2(Nd4jPointer *extraPointers, ND4J_EXPORT void execRandom2(Nd4jPointer *extraPointers,
int opNum, int opNum,
Nd4jPointer state, Nd4jPointer state,
void *hX, Nd4jLong *hXShapeBuffer, void *hX, Nd4jLong *hXShapeBuffer,
@ -1152,7 +1152,7 @@ void execRandom2(Nd4jPointer *extraPointers,
* @param ptrToBuffer * @param ptrToBuffer
* @return * @return
*/ */
Nd4jPointer initRandom(Nd4jPointer *extraPointers, ND4J_EXPORT Nd4jPointer initRandom(Nd4jPointer *extraPointers,
long seed, long seed,
long bufferSize, long bufferSize,
Nd4jPointer ptrToBuffer); Nd4jPointer ptrToBuffer);
@ -1163,7 +1163,7 @@ Nd4jPointer initRandom(Nd4jPointer *extraPointers,
* @param seed * @param seed
* @param ptrRandom * @param ptrRandom
*/ */
void refreshBuffer(Nd4jPointer *extraPointers, ND4J_EXPORT void refreshBuffer(Nd4jPointer *extraPointers,
long seed, long seed,
Nd4jPointer ptrRandom); Nd4jPointer ptrRandom);
@ -1173,7 +1173,7 @@ void refreshBuffer(Nd4jPointer *extraPointers,
* @param seed * @param seed
* @param ptrRandom * @param ptrRandom
*/ */
void reSeedBuffer(Nd4jPointer *extraPointers, ND4J_EXPORT void reSeedBuffer(Nd4jPointer *extraPointers,
long seed, long seed,
Nd4jPointer ptrRandom); Nd4jPointer ptrRandom);
@ -1181,7 +1181,7 @@ void reSeedBuffer(Nd4jPointer *extraPointers,
* *
* @param ptrRandom * @param ptrRandom
*/ */
void destroyRandom(Nd4jPointer ptrRandom); ND4J_EXPORT void destroyRandom(Nd4jPointer ptrRandom);
/** /**
* Grid operations * Grid operations
@ -1210,7 +1210,7 @@ void destroyRandom(Nd4jPointer ptrRandom);
* @param scalarB * @param scalarB
*/ */
/* /*
void execMetaPredicateShape(Nd4jPointer *extras, ND4J_EXPORT void execMetaPredicateShape(Nd4jPointer *extras,
const int opTypeA, const int opTypeA,
const int opNumA, const int opNumA,
const int opTypeB, const int opTypeB,
@ -1345,7 +1345,7 @@ static Nd4jPointer numpyFromNd4j(Nd4jPointer data,Nd4jPointer shapeBuffer,Nd4jLo
* @param npyArray * @param npyArray
* @return * @return
*/ */
Nd4jPointer shapeBufferForNumpy(Nd4jPointer npyArray); ND4J_EXPORT Nd4jPointer shapeBufferForNumpy(Nd4jPointer npyArray);
/** /**
@ -1460,7 +1460,7 @@ static void* getNpyArrayFromMap(void *map, int index){
throw std::runtime_error("No array at index."); throw std::runtime_error("No array at index.");
} }
int dataTypeFromNpyHeader(void *header); ND4J_EXPORT int dataTypeFromNpyHeader(void *header);
static void* getNpyArrayData(void *npArray){ static void* getNpyArrayData(void *npArray){
cnpy::NpyArray* npyArray2 = reinterpret_cast<cnpy::NpyArray*>(npArray); cnpy::NpyArray* npyArray2 = reinterpret_cast<cnpy::NpyArray*>(npArray);
@ -1544,7 +1544,7 @@ static void releaseNumpy(Nd4jPointer npyArray) {
* @param buffer the buffer pointer to check * @param buffer the buffer pointer to check
* @return * @return
*/ */
int lengthForShapeBufferPointer(Nd4jPointer buffer); ND4J_EXPORT int lengthForShapeBufferPointer(Nd4jPointer buffer);
/** /**
@ -1554,7 +1554,7 @@ int lengthForShapeBufferPointer(Nd4jPointer buffer);
* @return the pointer for the given address * @return the pointer for the given address
*/ */
Nd4jPointer pointerForAddress(Nd4jLong address); ND4J_EXPORT Nd4jPointer pointerForAddress(Nd4jLong address);
/** /**
* This method takes single N-dimensional tensor, and copies its TADs to target arrays * This method takes single N-dimensional tensor, and copies its TADs to target arrays
@ -1565,45 +1565,45 @@ Nd4jPointer pointerForAddress(Nd4jLong address);
* @param zShapeInfo * @param zShapeInfo
* @return * @return
*/ */
void tear(Nd4jPointer *extraPointers, ND4J_EXPORT void tear(Nd4jPointer *extraPointers,
void *x, Nd4jLong *xShapeInfo, void *x, Nd4jLong *xShapeInfo,
void *dx, Nd4jLong *dxShapeInfo, void *dx, Nd4jLong *dxShapeInfo,
Nd4jPointer *targets, Nd4jLong *zShapeInfo, Nd4jPointer *targets, Nd4jLong *zShapeInfo,
Nd4jLong *tadShapeInfo, Nd4jLong *tadShapeInfo,
Nd4jLong *tadOffsets); Nd4jLong *tadOffsets);
Nd4jLong encodeBitmap(Nd4jPointer *extraPointers, void *dx, Nd4jLong *xShapeInfo, Nd4jLong N, int *dz, float threshold); ND4J_EXPORT Nd4jLong encodeBitmap(Nd4jPointer *extraPointers, void *dx, Nd4jLong *xShapeInfo, Nd4jLong N, int *dz, float threshold);
void decodeBitmap(Nd4jPointer *extraPointers, void *dx, Nd4jLong N, void *dz, Nd4jLong *zShapeInfo); ND4J_EXPORT void decodeBitmap(Nd4jPointer *extraPointers, void *dx, Nd4jLong N, void *dz, Nd4jLong *zShapeInfo);
void encodeThresholdP1(Nd4jPointer *extraPointers, void *dx, Nd4jLong *xShapeInfo, Nd4jLong N, int *dz, float threshold); ND4J_EXPORT void encodeThresholdP1(Nd4jPointer *extraPointers, void *dx, Nd4jLong *xShapeInfo, Nd4jLong N, int *dz, float threshold);
void encodeThresholdP2Int(Nd4jPointer *extraPointers, int *dx, Nd4jLong N, int *dz); ND4J_EXPORT void encodeThresholdP2Int(Nd4jPointer *extraPointers, int *dx, Nd4jLong N, int *dz);
void encodeThresholdP3(Nd4jPointer *extraPointers, void *dx, Nd4jLong *xShapeInfo, int *offsets, Nd4jLong N, int *dz); ND4J_EXPORT void encodeThresholdP3(Nd4jPointer *extraPointers, void *dx, Nd4jLong *xShapeInfo, int *offsets, Nd4jLong N, int *dz);
void decodeThreshold(Nd4jPointer *extraPointers, void *dx, Nd4jLong N, void *dz, Nd4jLong *zShapeInfo); ND4J_EXPORT void decodeThreshold(Nd4jPointer *extraPointers, void *dx, Nd4jLong N, void *dz, Nd4jLong *zShapeInfo);
void sort(Nd4jPointer *extraPointers, ND4J_EXPORT void sort(Nd4jPointer *extraPointers,
void *x, Nd4jLong *xShapeInfo, void *x, Nd4jLong *xShapeInfo,
void *dx, Nd4jLong *dxShapeInfo, void *dx, Nd4jLong *dxShapeInfo,
bool descending); bool descending);
void sortByKey(Nd4jPointer *extraPointers, ND4J_EXPORT void sortByKey(Nd4jPointer *extraPointers,
void *x, Nd4jLong *xShapeInfo, void *x, Nd4jLong *xShapeInfo,
void *dx, Nd4jLong *dxShapeInfo, void *dx, Nd4jLong *dxShapeInfo,
void *y, Nd4jLong *yShapeInfo, void *y, Nd4jLong *yShapeInfo,
void *dy, Nd4jLong *dyShapeInfo, void *dy, Nd4jLong *dyShapeInfo,
bool descending); bool descending);
void sortByValue(Nd4jPointer *extraPointers, ND4J_EXPORT void sortByValue(Nd4jPointer *extraPointers,
void *x, Nd4jLong *xShapeInfo, void *x, Nd4jLong *xShapeInfo,
void *dx, Nd4jLong *dxShapeInfo, void *dx, Nd4jLong *dxShapeInfo,
void *y, Nd4jLong *yShapeInfo, void *y, Nd4jLong *yShapeInfo,
void *dy, Nd4jLong *dyShapeInfo, void *dy, Nd4jLong *dyShapeInfo,
bool descending); bool descending);
void sortTad(Nd4jPointer *extraPointers, ND4J_EXPORT void sortTad(Nd4jPointer *extraPointers,
void *x, Nd4jLong *xShapeInfo, void *x, Nd4jLong *xShapeInfo,
void *dx, Nd4jLong *dxShapeInfo, void *dx, Nd4jLong *dxShapeInfo,
int *dimension, int *dimension,
@ -1612,7 +1612,7 @@ void sortTad(Nd4jPointer *extraPointers,
Nd4jLong *tadOffsets, Nd4jLong *tadOffsets,
bool descending); bool descending);
void sortTadByKey(Nd4jPointer *extraPointers, ND4J_EXPORT void sortTadByKey(Nd4jPointer *extraPointers,
void *x, Nd4jLong *xShapeInfo, void *x, Nd4jLong *xShapeInfo,
void *dx, Nd4jLong *dxShapeInfo, void *dx, Nd4jLong *dxShapeInfo,
void *y, Nd4jLong *yShapeInfo, void *y, Nd4jLong *yShapeInfo,
@ -1621,7 +1621,7 @@ void sortTadByKey(Nd4jPointer *extraPointers,
int dimensionLength, int dimensionLength,
bool descending); bool descending);
void sortTadByValue(Nd4jPointer *extraPointers, ND4J_EXPORT void sortTadByValue(Nd4jPointer *extraPointers,
void *x, Nd4jLong *xShapeInfo, void *x, Nd4jLong *xShapeInfo,
void *dx, Nd4jLong *dxShapeInfo, void *dx, Nd4jLong *dxShapeInfo,
void *y, Nd4jLong *yShapeInfo, void *y, Nd4jLong *yShapeInfo,
@ -1632,82 +1632,82 @@ void sortTadByValue(Nd4jPointer *extraPointers,
// special sort impl for sorting out COO indices and values // special sort impl for sorting out COO indices and values
void sortCooIndices(Nd4jPointer *extraPointers, Nd4jLong *indices, void *values, Nd4jLong length, int rank); ND4J_EXPORT void sortCooIndices(Nd4jPointer *extraPointers, Nd4jLong *indices, void *values, Nd4jLong length, int rank);
Nd4jLong* mmapFile(Nd4jPointer *extraPointers, const char *fileName, Nd4jLong length); ND4J_EXPORT Nd4jLong* mmapFile(Nd4jPointer *extraPointers, const char *fileName, Nd4jLong length);
void munmapFile(Nd4jPointer *extraPointers, Nd4jLong* ptrMap, Nd4jLong length); ND4J_EXPORT void munmapFile(Nd4jPointer *extraPointers, Nd4jLong* ptrMap, Nd4jLong length);
// flatbuffers execution // flatbuffers execution
nd4j::graph::ResultWrapper* executeFlatGraph(Nd4jPointer *extraPointers, Nd4jPointer flatBufferPointer); ND4J_EXPORT nd4j::graph::ResultWrapper* executeFlatGraph(Nd4jPointer *extraPointers, Nd4jPointer flatBufferPointer);
const char* getAllCustomOps(); ND4J_EXPORT const char* getAllCustomOps();
const char* getAllOperations(); ND4J_EXPORT const char* getAllOperations();
// customOp executioner // customOp executioner
int execCustomOp(Nd4jPointer* extraPointers, Nd4jLong hash, Nd4jPointer* inputBuffers, Nd4jPointer* inputShapes, int numInputs, Nd4jPointer* outputBuffers, Nd4jPointer* outputShapes, int numOutputs, double* tArgs, int numTArgs, Nd4jLong *iArgs, int numIArgs, bool* bArgs, int numBArgs, bool isInplace); ND4J_EXPORT int execCustomOp(Nd4jPointer* extraPointers, Nd4jLong hash, Nd4jPointer* inputBuffers, Nd4jPointer* inputShapes, int numInputs, Nd4jPointer* outputBuffers, Nd4jPointer* outputShapes, int numOutputs, double* tArgs, int numTArgs, Nd4jLong *iArgs, int numIArgs, bool* bArgs, int numBArgs, bool isInplace);
int execCustomOp2(Nd4jPointer* extraPointers, Nd4jLong hash, Nd4jPointer opContext); ND4J_EXPORT int execCustomOp2(Nd4jPointer* extraPointers, Nd4jLong hash, Nd4jPointer opContext);
nd4j::ShapeList* calculateOutputShapes(Nd4jPointer* extraPointers, Nd4jLong hash, Nd4jPointer* inputShapes, int numInputShapes, double* tArgs, int numTArgs, Nd4jLong *iArgs, int numIArgs); ND4J_EXPORT nd4j::ShapeList* calculateOutputShapes(Nd4jPointer* extraPointers, Nd4jLong hash, Nd4jPointer* inputShapes, int numInputShapes, double* tArgs, int numTArgs, Nd4jLong *iArgs, int numIArgs);
nd4j::ShapeList* calculateOutputShapes2(Nd4jPointer* extraPointers, Nd4jLong hash, Nd4jPointer* inputBuffers, Nd4jPointer* inputShapes, int numInputShapes, double* tArgs, int numTArgs, Nd4jLong *iArgs, int numIArgs, bool *bArgs, int numBArgs); ND4J_EXPORT nd4j::ShapeList* calculateOutputShapes2(Nd4jPointer* extraPointers, Nd4jLong hash, Nd4jPointer* inputBuffers, Nd4jPointer* inputShapes, int numInputShapes, double* tArgs, int numTArgs, Nd4jLong *iArgs, int numIArgs, bool *bArgs, int numBArgs);
void deleteShapeList(Nd4jPointer shapeList); ND4J_EXPORT void deleteShapeList(Nd4jPointer shapeList);
int registerGraph(Nd4jPointer *extraPointers, Nd4jLong graphId, Nd4jPointer flatBufferPointer); ND4J_EXPORT int registerGraph(Nd4jPointer *extraPointers, Nd4jLong graphId, Nd4jPointer flatBufferPointer);
nd4j::graph::VariablesSet *executeStoredGraph(Nd4jPointer *extraPointers, Nd4jLong graphId, Nd4jPointer *inputBuffers, Nd4jPointer *inputShapes, int* inputIndices, int numInputs); ND4J_EXPORT nd4j::graph::VariablesSet *executeStoredGraph(Nd4jPointer *extraPointers, Nd4jLong graphId, Nd4jPointer *inputBuffers, Nd4jPointer *inputShapes, int* inputIndices, int numInputs);
int unregisterGraph(Nd4jPointer *extraPointers, Nd4jLong graphId); ND4J_EXPORT int unregisterGraph(Nd4jPointer *extraPointers, Nd4jLong graphId);
void deleteCharArray(Nd4jPointer pointer); ND4J_EXPORT void deleteCharArray(Nd4jPointer pointer);
void deleteIntArray(Nd4jPointer pointer); ND4J_EXPORT void deleteIntArray(Nd4jPointer pointer);
void deleteLongArray(Nd4jPointer pointer); ND4J_EXPORT void deleteLongArray(Nd4jPointer pointer);
void deletePointerArray(Nd4jPointer pointer); ND4J_EXPORT void deletePointerArray(Nd4jPointer pointer);
void deleteVariablesSet(Nd4jPointer pointer); ND4J_EXPORT void deleteVariablesSet(Nd4jPointer pointer);
// GraphState creation // GraphState creation
Nd4jPointer getGraphState(Nd4jLong id); ND4J_EXPORT Nd4jPointer getGraphState(Nd4jLong id);
void deleteGraphState(Nd4jPointer state); ND4J_EXPORT void deleteGraphState(Nd4jPointer state);
void deleteResultWrapper(Nd4jPointer ptr); ND4J_EXPORT void deleteResultWrapper(Nd4jPointer ptr);
int estimateThreshold(Nd4jPointer *extraPointers, Nd4jPointer x, Nd4jLong *xShapeInfo, int N, float threshold); ND4J_EXPORT int estimateThreshold(Nd4jPointer *extraPointers, Nd4jPointer x, Nd4jLong *xShapeInfo, int N, float threshold);
// this method executes op that requires scope to be present: if/while/cond/whatever // this method executes op that requires scope to be present: if/while/cond/whatever
Nd4jStatus execCustomOpWithScope(Nd4jPointer *extraPointers, Nd4jPointer state, Nd4jLong opHash, Nd4jLong *scopes, int numScopes, Nd4jPointer *inputBuffers, Nd4jPointer *inputShapes, int numInputs, Nd4jPointer *outputBuffers, Nd4jPointer *outputShapes, int numOutputs); ND4J_EXPORT Nd4jStatus execCustomOpWithScope(Nd4jPointer *extraPointers, Nd4jPointer state, Nd4jLong opHash, Nd4jLong *scopes, int numScopes, Nd4jPointer *inputBuffers, Nd4jPointer *inputShapes, int numInputs, Nd4jPointer *outputBuffers, Nd4jPointer *outputShapes, int numOutputs);
//void fillUtf8String(Nd4jPointer *extraPointers, const char **string, int numStrings, Nd4jPointer buffer); //void fillUtf8String(Nd4jPointer *extraPointers, const char **string, int numStrings, Nd4jPointer buffer);
Nd4jPointer createUtf8String(Nd4jPointer *extraPointers, const char *string, int length); ND4J_EXPORT Nd4jPointer createUtf8String(Nd4jPointer *extraPointers, const char *string, int length);
void deleteUtf8String(Nd4jPointer *extraPointers, Nd4jPointer ptr); void deleteUtf8String(Nd4jPointer *extraPointers, Nd4jPointer ptr);
void scatterUpdate(Nd4jPointer *extraPointers, int opCode, int numOfSubArrs, ND4J_EXPORT void scatterUpdate(Nd4jPointer *extraPointers, int opCode, int numOfSubArrs,
void* hX, Nd4jLong* hXShapeInfo, Nd4jLong* hXOffsets, void* hX, Nd4jLong* hXShapeInfo, Nd4jLong* hXOffsets,
void* dX, Nd4jLong* dXShapeInfo, Nd4jLong* dXOffsets, void* dX, Nd4jLong* dXShapeInfo, Nd4jLong* dXOffsets,
void* hY, Nd4jLong* hYShapeInfo, Nd4jLong* hYOffsets, void* hY, Nd4jLong* hYShapeInfo, Nd4jLong* hYOffsets,
void* dY, Nd4jLong* dYShapeInfo, Nd4jLong* dYOffsets, void* dY, Nd4jLong* dYShapeInfo, Nd4jLong* dYOffsets,
int* hIindexes, int* dIindexes); int* hIindexes, int* dIindexes);
void deleteShapeBuffer(Nd4jPointer ptr); ND4J_EXPORT void deleteShapeBuffer(Nd4jPointer ptr);
void deleteTadPack(Nd4jPointer ptr); ND4J_EXPORT void deleteTadPack(Nd4jPointer ptr);
void inspectArray(Nd4jPointer *extraPointers, Nd4jPointer buffer, Nd4jLong *shapeInfo, Nd4jPointer specialBuffer, Nd4jLong *specialShapeInfo, Nd4jPointer debugInfo); ND4J_EXPORT void inspectArray(Nd4jPointer *extraPointers, Nd4jPointer buffer, Nd4jLong *shapeInfo, Nd4jPointer specialBuffer, Nd4jLong *specialShapeInfo, Nd4jPointer debugInfo);
nd4j::ConstantDataBuffer* shapeBuffer(int rank, Nd4jLong *shape, Nd4jLong *strides, nd4j::DataType dtype, char order, Nd4jLong ews, bool empty); ND4J_EXPORT nd4j::ConstantDataBuffer* shapeBuffer(int rank, Nd4jLong *shape, Nd4jLong *strides, nd4j::DataType dtype, char order, Nd4jLong ews, bool empty);
nd4j::ConstantDataBuffer* constantBufferLong(nd4j::DataType dtype, Nd4jLong *data, int length); ND4J_EXPORT nd4j::ConstantDataBuffer* constantBufferLong(nd4j::DataType dtype, Nd4jLong *data, int length);
nd4j::ConstantDataBuffer* constantBufferDouble(nd4j::DataType dtype, double *data, int length); ND4J_EXPORT nd4j::ConstantDataBuffer* constantBufferDouble(nd4j::DataType dtype, double *data, int length);
nd4j::ConstantDataBuffer* constantBuffer(nd4j::DataType dtype, nd4j::ConstantDescriptor *descriptor); ND4J_EXPORT nd4j::ConstantDataBuffer* constantBuffer(nd4j::DataType dtype, nd4j::ConstantDescriptor *descriptor);
const char* runLightBenchmarkSuit(bool printOut); ND4J_EXPORT const char* runLightBenchmarkSuit(bool printOut);
const char* runFullBenchmarkSuit(bool printOut); ND4J_EXPORT const char* runFullBenchmarkSuit(bool printOut);
} }

10
libnd4j/blas/cuda/NativeOps.cu
View File

@ -479,7 +479,7 @@ void execReduceLong(Nd4jPointer *extraPointers,
} }
//////////////////////////////////////////////////////////////////////// ////////////////////////////////////////////////////////////////////////
void execReduceBool(Nd4jPointer *extraPointers, void execReduceBool2(Nd4jPointer *extraPointers,
int opNum, int opNum,
void *hX, Nd4jLong *hXShapeInfo, void *hX, Nd4jLong *hXShapeInfo,
void *dX, Nd4jLong *dXShapeInfo, void *dX, Nd4jLong *dXShapeInfo,
@ -1000,7 +1000,7 @@ int setDevice(int deviceId) {
return 1; return 1;
} }
Nd4jLong getDeviceFreeMemory() { Nd4jLong getDeviceFreeMemoryDefault() {
size_t memFree = 0; size_t memFree = 0;
size_t memTotal = 0; size_t memTotal = 0;
@ -1820,7 +1820,7 @@ void execScalarBool(Nd4jPointer *extraPointers,
} }
//////////////////////////////////////////////////////////////////////// ////////////////////////////////////////////////////////////////////////
void execScalarBool(Nd4jPointer *extraPointers, void execScalarBoolTad(Nd4jPointer *extraPointers,
int opNum, int opNum,
void *hX, Nd4jLong *hXShapeInfo, void *hX, Nd4jLong *hXShapeInfo,
void *dX, Nd4jLong *dXShapeInfo, void *dX, Nd4jLong *dXShapeInfo,
@ -3247,11 +3247,11 @@ void deleteTadPack(Nd4jPointer ptr) {
delete buffer; delete buffer;
} }
nd4j::ConstantDataBuffer* constantBuffer(nd4j::DataType dtype, Nd4jLong *data, int length) { nd4j::ConstantDataBuffer* constantBufferLong(nd4j::DataType dtype, Nd4jLong *data, int length) {
return nd4j::ConstantHelper::getInstance()->constantBuffer(ConstantDescriptor(data, length), dtype); return nd4j::ConstantHelper::getInstance()->constantBuffer(ConstantDescriptor(data, length), dtype);
} }
nd4j::ConstantDataBuffer* constantBuffer(nd4j::DataType dtype, double *data, int length) { nd4j::ConstantDataBuffer* constantBufferDouble(nd4j::DataType dtype, double *data, int length) {
return nd4j::ConstantHelper::getInstance()->constantBuffer(ConstantDescriptor(data, length), dtype); return nd4j::ConstantHelper::getInstance()->constantBuffer(ConstantDescriptor(data, length), dtype);
} }

9
libnd4j/include/helpers/OpTracker.h
View File

@ -29,9 +29,6 @@
#include <ops/declarable/OpDescriptor.h> #include <ops/declarable/OpDescriptor.h>
#include <dll.h> #include <dll.h>
using namespace nd4j::ops;
using namespace nd4j::graph;
namespace nd4j { namespace nd4j {
class ND4J_EXPORT OpTracker { class ND4J_EXPORT OpTracker {
private: private:
@ -40,7 +37,7 @@ namespace nd4j {
std::string _export; std::string _export;
int _operations = 0; int _operations = 0;
std::map<OpType, std::vector<OpDescriptor>> _map; std::map<nd4j::graph::OpType, std::vector<nd4j::ops::OpDescriptor>> _map;
OpTracker() = default; OpTracker() = default;
~OpTracker() = default; ~OpTracker() = default;
@ -53,11 +50,11 @@ namespace nd4j {
int totalGroups(); int totalGroups();
int totalOperations(); int totalOperations();
void storeOperation(nd4j::graph::OpType opType, const OpDescriptor& descriptor); void storeOperation(nd4j::graph::OpType opType, const nd4j::ops::OpDescriptor& descriptor);
void storeOperation(nd4j::graph::OpType opType, const char* opName, const Nd4jLong opNum); void storeOperation(nd4j::graph::OpType opType, const char* opName, const Nd4jLong opNum);
const char* exportOperations(); const char* exportOperations();
}; };
} }
#endif #endif

5
libnd4j/include/helpers/impl/OpTracker.cpp
View File

@ -22,6 +22,9 @@
#include <sstream> #include <sstream>
#include <helpers/logger.h> #include <helpers/logger.h>
using namespace nd4j::ops;
using namespace nd4j::graph;
namespace nd4j { namespace nd4j {
OpTracker* OpTracker::getInstance() { OpTracker* OpTracker::getInstance() {
@ -90,4 +93,4 @@ namespace nd4j {
} }
nd4j::OpTracker* nd4j::OpTracker::_INSTANCE = 0; nd4j::OpTracker* nd4j::OpTracker::_INSTANCE = 0;
} }