Então, basicamente, você deseja que seu código seja executado mais rapidamente. JNI é a resposta. Eu sei que você disse que não funcionou para você, mas deixe-me mostrar que você está errado.
Aqui está Dot.java
:
import java.nio.FloatBuffer;
import org.bytedeco.javacpp.*;
import org.bytedeco.javacpp.annotation.*;
@Platform(include = "Dot.h", compiler = "fastfpu")
public class Dot {
static { Loader.load(); }
static float[] a = new float[50], b = new float[50];
static float dot() {
float sum = 0;
for (int i = 0; i < 50; i++) {
sum += a[i]*b[i];
}
return sum;
}
static native @MemberGetter FloatPointer ac();
static native @MemberGetter FloatPointer bc();
static native @NoException float dotc();
public static void main(String[] args) {
FloatBuffer ab = ac().capacity(50).asBuffer();
FloatBuffer bb = bc().capacity(50).asBuffer();
for (int i = 0; i < 10000000; i++) {
a[i%50] = b[i%50] = dot();
float sum = dotc();
ab.put(i%50, sum);
bb.put(i%50, sum);
}
long t1 = System.nanoTime();
for (int i = 0; i < 10000000; i++) {
a[i%50] = b[i%50] = dot();
}
long t2 = System.nanoTime();
for (int i = 0; i < 10000000; i++) {
float sum = dotc();
ab.put(i%50, sum);
bb.put(i%50, sum);
}
long t3 = System.nanoTime();
System.out.println("dot(): " + (t2 - t1)/10000000 + " ns");
System.out.println("dotc(): " + (t3 - t2)/10000000 + " ns");
}
}
e Dot.h
:
float ac[50], bc[50];
inline float dotc() {
float sum = 0;
for (int i = 0; i < 50; i++) {
sum += ac[i]*bc[i];
}
return sum;
}
Podemos compilar e executar isso com JavaCPP usando este comando:
$ java -jar javacpp.jar Dot.java -exec
Com uma CPU Intel (R) Core (TM) i7-7700HQ a 2,80 GHz, Fedora 30, GCC 9.1.1 e OpenJDK 8 ou 11, recebo este tipo de resultado:
dot(): 39 ns
dotc(): 16 ns
Ou cerca de 2,4 vezes mais rápido. Precisamos usar buffers NIO diretos em vez de arrays, mas o HotSpot pode acessar os buffers NIO diretos tão rápido quanto os arrays . Por outro lado, desenrolar manualmente o loop não fornece um aumento mensurável no desempenho, neste caso.
-XX:+UnlockDiagnosticVMOptions -XX:+PrintAssembly -XX:+LogCompilation
. Você precisará de um programa que execute o método vetorizável vezes suficientes para torná-lo "quente".