Отже, ви хочете, щоб ваш код працював швидше. 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
. Вам знадобиться програма, яка запускає векторизований метод достатньо разів, щоб зробити його «гарячим».