natas_hw
/
jidt
mirror of https://github.com/jlizier/jidt
0
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Add new property for users to specify custom path to GPU library.

This commit is contained in:
Pedro Martinez Mediano 2017-05-19 06:01:45 +10:00
parent 1abd5a084e
commit 57aa4e95ee
1 changed files with 43 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

54
java/source/infodynamics/measures/continuous/kraskov/MutualInfoCalculatorMultiVariateKraskov.java
View File

@ -121,6 +121,13 @@ public abstract class MutualInfoCalculatorMultiVariateKraskov
* Property name for the flag to enable or disable the GPU module. * Property name for the flag to enable or disable the GPU module.
*/ */
public static final String PROP_USE_GPU = "USE_GPU"; public static final String PROP_USE_GPU = "USE_GPU";
/**
* Property name for the path to JIDT GPU library.
*
* Path must be full and contain the library filename.
* Example: /home/johndoe/myfolder/libKraskov.so
*/
public static final String PROP_GPU_LIBRARY_PATH = "GPU_LIBRARY_PATH";
/** /**
* Whether to normalise the incoming data * Whether to normalise the incoming data
@ -156,6 +163,10 @@ public abstract class MutualInfoCalculatorMultiVariateKraskov
* Whether to enable the GPU module * Whether to enable the GPU module
*/ */
protected boolean useGPU = false; protected boolean useGPU = false;
/**
* Path to JIDT GPU library
*/
protected String gpuLibraryPath = "";
/** /**
* Check whether C native code has been loaded * Check whether C native code has been loaded
*/ */
@ -270,6 +281,8 @@ public abstract class MutualInfoCalculatorMultiVariateKraskov
} }
} else if (propertyName.equalsIgnoreCase(PROP_USE_GPU)) { } else if (propertyName.equalsIgnoreCase(PROP_USE_GPU)) {
useGPU = Boolean.parseBoolean(propertyValue); useGPU = Boolean.parseBoolean(propertyValue);
} else if (propertyName.equalsIgnoreCase(PROP_GPU_LIBRARY_PATH)) {
gpuLibraryPath = propertyValue;
} else { } else {
// No property was set here // No property was set here
propertySet = false; propertySet = false;
@ -658,18 +671,37 @@ public abstract class MutualInfoCalculatorMultiVariateKraskov
if (!cudaLibraryLoaded) { if (!cudaLibraryLoaded) {
Path jarPath = Paths.get(MutualInfoCalculatorMultiVariateKraskov.class.getProtectionDomain().getCodeSource().getLocation().toURI()); String fullPath;
String fileSep = System.getProperty("file.separator", "");
if (fileSep.length() == 0) { if (gpuLibraryPath.length() < 1) {
throw new Exception("Unable to load GPU library."); // If user didn't provide a path, try the default
Path jarPath = Paths.get(MutualInfoCalculatorMultiVariateKraskov.class.getProtectionDomain().getCodeSource().getLocation().toURI());
String fileSep = System.getProperty("file.separator", "");
if (fileSep.length() == 0) {
throw new Exception("Unable to load GPU library.");
}
String jarFolder = jarPath.toString().substring(0, jarPath.toString().lastIndexOf(fileSep));
String relPath = fileSep + "bin" + fileSep +
"infodynamics" + fileSep + "measures" + fileSep +
"continuous" + fileSep + "kraskov" + fileSep + "cuda" +
fileSep + "libKraskov.so";
fullPath = jarFolder + relPath;
} else {
// Otherwise, use the provided path.
fullPath = gpuLibraryPath;
} }
String jarFolder = jarPath.toString().substring(0, jarPath.toString().lastIndexOf(fileSep));
String relPath = fileSep + "bin" + fileSep + try {
"infodynamics" + fileSep + "measures" + fileSep + System.load(fullPath);
"continuous" + fileSep + "kraskov" + fileSep + "cuda" + } catch (Throwable e) {
fileSep + "libKraskov.so"; if (gpuLibraryPath.length() > 1) {
String fullPath = jarFolder + relPath; System.err.println("GPU library was not found in the path provided. Provide full path including library file name.");
System.load(fullPath); System.err.println("Example: /home/johndoe/myfolder/libKraskov.so");
}
throw(e);
}
System.out.println("CUDA native library loaded.");
cudaLibraryLoaded = true; cudaLibraryLoaded = true;
} }