Итак, по сути, вы хотите, чтобы ваш код работал быстрее. JNI - это ответ. Я знаю, что вы сказали, что это не сработало для вас, но позвольте мне показать вам, что вы ошибаетесь.
Вот 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");
}
}
и 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;
}
Мы можем скомпилировать и запустить это с помощью JavaCPP, используя эту команду:
$ java -jar javacpp.jar Dot.java -exec
С процессором Intel (R) Core (TM) i7-7700HQ @ 2,80 ГГц, Fedora 30, GCC 9.1.1 и OpenJDK 8 или 11, я получаю такой вывод:
dot(): 39 ns
dotc(): 16 ns
Или примерно в 2,4 раза быстрее. Нам нужно использовать прямые буферы NIO вместо массивов, но HotSpot может обращаться к прямым буферам NIO так же быстро, как и к массивам . С другой стороны, ручное развертывание цикла в этом случае не дает ощутимого повышения производительности.
-XX:+UnlockDiagnosticVMOptions -XX:+PrintAssembly -XX:+LogCompilation
. Вам понадобится программа, которая запускает векторизуемый метод достаточное количество раз, чтобы сделать его «горячим».