mirror of https://github.com/jlizier/jidt
GPU tests and example now compute multiple surrogates.
This commit is contained in:
parent
c616ba55d4
commit
4908276333
|
|
@ -67,9 +67,10 @@ public class Example10GPUBenchmark {
|
|||
miCalc.setProperty("k", "4");
|
||||
miCalc.initialise(src[0].length, tgt[0].length);
|
||||
miCalc.setObservations(src, tgt);
|
||||
int nb_surrogates = 150;
|
||||
double[] timeAndValue = new double[2];
|
||||
long startTime = System.nanoTime();
|
||||
timeAndValue[1] = miCalc.computeAverageLocalOfObservations();
|
||||
timeAndValue[1] = miCalc.computeSignificance(nb_surrogates).actualValue;
|
||||
timeAndValue[0] = (System.nanoTime() - startTime)/1000000.0;
|
||||
return timeAndValue;
|
||||
}
|
||||
|
|
|
|||
|
|
@ -86,14 +86,14 @@ public class GPUPerformanceTester extends TestCase {
|
|||
startTime = Calendar.getInstance().getTimeInMillis();
|
||||
miCalc.initialise(1,1);
|
||||
miCalc.setObservations(source, dest);
|
||||
cpu_val = miCalc.computeSignificance(nb_surrogates).getMeanOfDistribution();
|
||||
cpu_val = miCalc.computeSignificance(nb_surrogates).actualValue;
|
||||
cpu_duration = Calendar.getInstance().getTimeInMillis() - startTime;
|
||||
|
||||
miCalc.setProperty("USE_GPU", "true");
|
||||
miCalc.initialise(1,1);
|
||||
startTime = Calendar.getInstance().getTimeInMillis();
|
||||
miCalc.setObservations(source, dest);
|
||||
gpu_val = miCalc.computeSignificance(nb_surrogates).getMeanOfDistribution();
|
||||
gpu_val = miCalc.computeSignificance(nb_surrogates).actualValue;
|
||||
gpu_duration = Calendar.getInstance().getTimeInMillis() - startTime;
|
||||
|
||||
assertEquals(cpu_val, gpu_val, 0.0001);
|
||||
|
|
@ -124,12 +124,12 @@ public class GPUPerformanceTester extends TestCase {
|
|||
return;
|
||||
}
|
||||
|
||||
int timeSteps = 10000;
|
||||
int timeSteps = 1000;
|
||||
RandomGenerator rg = new RandomGenerator();
|
||||
double[] source = rg.generateNormalData(timeSteps, 0, 1);
|
||||
double[] dest = rg.generateNormalData(timeSteps, 0, 1);
|
||||
|
||||
compareGPUPerformance(source, dest, "Random low-dimensional data");
|
||||
compareGPUPerformanceSurrogates(source, dest, 150, "Random low-dimensional data");
|
||||
|
||||
return;
|
||||
}
|
||||
|
|
|
|||
Loading…
Reference in New Issue