mirror of https://github.com/jlizier/jidt
Added final code to call gpuCMI functions from JNI.
This commit is contained in:
Pedro Martinez Mediano
parent
75a303875d
commit
fc40c279cd
|
|
@ -234,144 +234,137 @@ JNIEXPORT jdoubleArray JNICALL
|
|||
(*env)->ThrowNew(env, Exception, "Surrogates only supported for average MI with KSG1.");
|
||||
}
|
||||
|
||||
// // Copy variables from Java
|
||||
// // =====================
|
||||
// int N = j_N;
|
||||
// int k = j_k;
|
||||
// int dimx = j_dimx;
|
||||
// int dimy = j_dimy;
|
||||
// int theiler = j_theiler;
|
||||
// int returnLocals = j_returnLocals ? 1 : 0;
|
||||
// int useMaxNorm = j_useMaxNorm ? 1 : 0;
|
||||
// int isAlgorithm1 = j_isAlgorithm1 ? 1 : 0;
|
||||
// int nb_surrogates = j_nbSurrogates;
|
||||
// int reorderingsGiven = j_reorderingsGiven ? 1 : 0;
|
||||
//
|
||||
// CPerfTimer pt = startTimer("Java array copy");
|
||||
// Copy variables from Java
|
||||
// =====================
|
||||
int N = j_N;
|
||||
int k = j_k;
|
||||
int dimx = j_dimx;
|
||||
int dimy = j_dimy;
|
||||
int dimz = j_dimz;
|
||||
int theiler = j_theiler;
|
||||
int returnLocals = j_returnLocals ? 1 : 0;
|
||||
int useMaxNorm = j_useMaxNorm ? 1 : 0;
|
||||
int isAlgorithm1 = j_isAlgorithm1 ? 1 : 0;
|
||||
int nb_surrogates = j_nbSurrogates;
|
||||
int reorderingsGiven = j_reorderingsGiven ? 1 : 0;
|
||||
|
||||
CPerfTimer pt = startTimer("Java array copy");
|
||||
|
||||
// float *source = (float *) malloc(N * dimx * sizeof(float));
|
||||
// float *dest = (float *) malloc(N * dimy * sizeof(float));
|
||||
float *source = (float *) malloc(N * dimx * sizeof(float));
|
||||
float *dest = (float *) malloc(N * dimy * sizeof(float));
|
||||
float *cond = (float *) malloc(N * dimz * sizeof(float));
|
||||
|
||||
// if (NULL == source || NULL == dest) {
|
||||
// jclass Exception = (*env)->FindClass(env, "java/lang/Exception");
|
||||
// (*env)->ThrowNew(env, Exception, "Error allocating data.");
|
||||
// }
|
||||
if (NULL == source || NULL == dest || NULL == cond) {
|
||||
jclass Exception = (*env)->FindClass(env, "java/lang/Exception");
|
||||
(*env)->ThrowNew(env, Exception, "Error allocating data.");
|
||||
}
|
||||
|
||||
// for (int i = 0; i < N; i++) {
|
||||
// jdoubleArray j_sourceRow = (jdoubleArray) (*env)->GetObjectArrayElement(env, j_sourceArray, i);
|
||||
// jdoubleArray j_destRow = (jdoubleArray) (*env)->GetObjectArrayElement(env, j_destArray, i);
|
||||
for (int i = 0; i < N; i++) {
|
||||
jdoubleArray j_sourceRow = (jdoubleArray) (*env)->GetObjectArrayElement(env, j_sourceArray, i);
|
||||
jdoubleArray j_destRow = (jdoubleArray) (*env)->GetObjectArrayElement(env, j_destArray, i);
|
||||
jdoubleArray j_condRow = (jdoubleArray) (*env)->GetObjectArrayElement(env, j_condArray, i);
|
||||
|
||||
// jdouble *sourceRow = (*env)->GetDoubleArrayElements(env, j_sourceRow, NULL);
|
||||
// jdouble *destRow = (*env)->GetDoubleArrayElements(env, j_destRow, NULL);
|
||||
jdouble *sourceRow = (*env)->GetDoubleArrayElements(env, j_sourceRow, NULL);
|
||||
jdouble *destRow = (*env)->GetDoubleArrayElements(env, j_destRow, NULL);
|
||||
jdouble *condRow = (*env)->GetDoubleArrayElements(env, j_condRow, NULL);
|
||||
|
||||
// // Data in java are doubles, but GPUs need floats.
|
||||
// // We have to cast them manually, so we can't memcopy
|
||||
// Data in java are doubles, but GPUs need floats.
|
||||
// We have to cast them manually, so we can't memcopy
|
||||
|
||||
// // The following two for-loops get two matrices in T-by-D indexing (i.e.
|
||||
// // first dimension is time, second is variable) and return the data in
|
||||
// // column-major form
|
||||
// for (int j = 0; j < dimx; j++) {
|
||||
// source[N*j + i] = (float) sourceRow[j];
|
||||
// }
|
||||
// The following for-loops get two matrices in T-by-D indexing (i.e.
|
||||
// first dimension is time, second is variable) and return the data in
|
||||
// column-major form
|
||||
for (int j = 0; j < dimx; j++) {
|
||||
source[N*j + i] = (float) sourceRow[j];
|
||||
}
|
||||
|
||||
// for (int j = 0; j < dimy; j++) {
|
||||
// dest[N*j + i] = (float) destRow[j];
|
||||
// }
|
||||
for (int j = 0; j < dimy; j++) {
|
||||
dest[N*j + i] = (float) destRow[j];
|
||||
}
|
||||
|
||||
// // // The following two for-loops get the same matrices but return arrays
|
||||
// // // in row-major form. Code is left here for future development purposes.
|
||||
// // for (j = 0; j < dimx; j++) {
|
||||
// // register float x = (float) sourceRow[j];
|
||||
// // source[dimx*i + j] = x;
|
||||
// // pointset[dims*i + j] = x;
|
||||
// // }
|
||||
for (int j = 0; j < dimz; j++) {
|
||||
cond[N*j + i] = (float) condRow[j];
|
||||
}
|
||||
|
||||
// // for (j = 0; j < dimy; j++) {
|
||||
// // register float x = (float) destRow[j];
|
||||
// // dest[dimy*i + j] = x
|
||||
// // pointset[i*dims + dimx + j] = x;
|
||||
// // }
|
||||
(*env)->ReleaseDoubleArrayElements(env, j_sourceRow, sourceRow, 0);
|
||||
(*env)->ReleaseDoubleArrayElements(env, j_destRow, destRow, 0);
|
||||
(*env)->ReleaseDoubleArrayElements(env, j_condRow, condRow, 0);
|
||||
|
||||
// (*env)->ReleaseDoubleArrayElements(env, j_sourceRow, sourceRow, 0);
|
||||
// (*env)->ReleaseDoubleArrayElements(env, j_destRow, destRow, 0);
|
||||
(*env)->DeleteLocalRef(env, j_sourceRow);
|
||||
(*env)->DeleteLocalRef(env, j_destRow);
|
||||
(*env)->DeleteLocalRef(env, j_condRow);
|
||||
|
||||
// (*env)->DeleteLocalRef(env, j_sourceRow);
|
||||
// (*env)->DeleteLocalRef(env, j_destRow);
|
||||
// FIXME: I'm not entirely sure I'm freeing all the memory here. I should
|
||||
// check for memory leaks more carefully.
|
||||
|
||||
// // FIXME: I'm not entirely sure I'm freeing all the memory here. I should
|
||||
// // check for memory leaks more carefully.
|
||||
}
|
||||
|
||||
// }
|
||||
int **reorderings = NULL;
|
||||
if (reorderingsGiven) {
|
||||
reorderings = (int **) malloc(nb_surrogates * sizeof(int *));
|
||||
for (int i = 0; i < nb_surrogates; i++) {
|
||||
jintArray j_order = (jdoubleArray) (*env)->GetObjectArrayElement(env, j_orderings, i);
|
||||
jint *order = (*env)->GetIntArrayElements(env, j_order, NULL);
|
||||
|
||||
// int **reorderings = NULL;
|
||||
// if (reorderingsGiven) {
|
||||
// reorderings = (int **) malloc(nb_surrogates * sizeof(int *));
|
||||
// for (int i = 0; i < nb_surrogates; i++) {
|
||||
// jintArray j_order = (jdoubleArray) (*env)->GetObjectArrayElement(env, j_orderings, i);
|
||||
// jint *order = (*env)->GetIntArrayElements(env, j_order, NULL);
|
||||
reorderings[i] = (int *) malloc(N * sizeof(int));
|
||||
for (int j = 0; j < N; j++) {
|
||||
reorderings[i][j] = order[j];
|
||||
}
|
||||
|
||||
// reorderings[i] = (int *) malloc(N * sizeof(int));
|
||||
// for (int j = 0; j < N; j++) {
|
||||
// reorderings[i][j] = order[j];
|
||||
// }
|
||||
(*env)->ReleaseIntArrayElements(env, j_order, order, 0);
|
||||
(*env)->DeleteLocalRef(env, j_order);
|
||||
}
|
||||
}
|
||||
|
||||
// (*env)->ReleaseIntArrayElements(env, j_order, order, 0);
|
||||
// (*env)->DeleteLocalRef(env, j_order);
|
||||
// }
|
||||
// }
|
||||
|
||||
// stopTimer(pt);
|
||||
stopTimer(pt);
|
||||
|
||||
|
||||
// // Call C function
|
||||
// // =========================
|
||||
// int resultSize;
|
||||
// if (returnLocals) {
|
||||
// resultSize = N;
|
||||
// } else if (nb_surrogates > 0) {
|
||||
// resultSize = nb_surrogates + 1;
|
||||
// } else {
|
||||
// resultSize = 3;
|
||||
// }
|
||||
// float *result = (float *) malloc(resultSize * sizeof(float));
|
||||
// jidt_error_t ret;
|
||||
// if (!reorderingsGiven) {
|
||||
// ret = MIKraskov_C(N, source, dimx, dest, dimy,
|
||||
// k, theiler, nb_surrogates, returnLocals, useMaxNorm,
|
||||
// isAlgorithm1, result);
|
||||
// Call C function
|
||||
// =========================
|
||||
int resultSize;
|
||||
if (returnLocals) {
|
||||
resultSize = N;
|
||||
} else if (nb_surrogates > 0) {
|
||||
resultSize = nb_surrogates + 1;
|
||||
} else {
|
||||
resultSize = 6;
|
||||
}
|
||||
float *result = (float *) malloc(resultSize * sizeof(float));
|
||||
jidt_error_t ret;
|
||||
if (!reorderingsGiven) {
|
||||
ret = CMIKraskov_C(N, source, dimx, dest, dimy, cond, dimz,
|
||||
k, theiler, nb_surrogates, returnLocals, useMaxNorm,
|
||||
isAlgorithm1, result);
|
||||
|
||||
// } else {
|
||||
// ret = MIKraskovWithReorderings(N, source, dimx, dest, dimy, k, theiler,
|
||||
// nb_surrogates, returnLocals, useMaxNorm,
|
||||
// isAlgorithm1, result, reorderingsGiven,
|
||||
// reorderings);
|
||||
// }
|
||||
} else {
|
||||
ret = CMIKraskovWithReorderings(N, source, dimx, dest, dimy, cond, dimz, k, theiler,
|
||||
nb_surrogates, returnLocals, useMaxNorm,
|
||||
isAlgorithm1, result, reorderingsGiven,
|
||||
reorderings);
|
||||
}
|
||||
|
||||
// if (JIDT_ERROR == ret) {
|
||||
// jclass Exception = (*env)->FindClass(env, "java/lang/Exception");
|
||||
// (*env)->ThrowNew(env, Exception, "Error in GPU execution.");
|
||||
// }
|
||||
if (JIDT_ERROR == ret) {
|
||||
jclass Exception = (*env)->FindClass(env, "java/lang/Exception");
|
||||
(*env)->ThrowNew(env, Exception, "Error in GPU execution.");
|
||||
}
|
||||
|
||||
// // Free memory and return
|
||||
// // =========================
|
||||
// if (source) free(source);
|
||||
// if (dest) free(dest);
|
||||
// Free memory and return
|
||||
// =========================
|
||||
if (source) free(source);
|
||||
if (dest) free(dest);
|
||||
if (cond) free(cond);
|
||||
|
||||
// if (reorderingsGiven) {
|
||||
// for (int i = 0; i < nb_surrogates; i++) {
|
||||
// if (reorderings[i]) free(reorderings[i]);
|
||||
// }
|
||||
// if (reorderings) free (reorderings);
|
||||
// }
|
||||
if (reorderingsGiven) {
|
||||
for (int i = 0; i < nb_surrogates; i++) {
|
||||
if (reorderings[i]) free(reorderings[i]);
|
||||
}
|
||||
if (reorderings) free (reorderings);
|
||||
}
|
||||
|
||||
// jdouble outCArray[resultSize];
|
||||
// for (int i = 0; i < resultSize; i++) {
|
||||
// outCArray[i] = result[i];
|
||||
// }
|
||||
|
||||
int resultSize = 1;
|
||||
jdouble outCArray[resultSize];
|
||||
outCArray[0] = 0.0;
|
||||
for (int i = 0; i < resultSize; i++) {
|
||||
outCArray[i] = result[i];
|
||||
}
|
||||
|
||||
// Set Java array for return
|
||||
jdoubleArray outJNIArray = (*env)->NewDoubleArray(env, resultSize); // allocate
|
||||
|
|
@ -381,12 +374,12 @@ JNIEXPORT jdoubleArray JNICALL
|
|||
}
|
||||
(*env)->SetDoubleArrayRegion(env, outJNIArray, 0 , resultSize, outCArray); // copy
|
||||
|
||||
// if (result) { free(result); }
|
||||
if (result) { free(result); }
|
||||
|
||||
return outJNIArray;
|
||||
|
||||
|
||||
} // End of function MIKraskov
|
||||
} // End of function CMIKraskov
|
||||
|
||||
|
||||
#ifdef __cplusplus
|
||||
|
|
|
|||
Loading…
Reference in New Issue