将 JNI Java 数组类型转换为 NEON

Question

我确信人们总是这样做，但我在这里遇到了困难。我将浮点数组传递给 JNI 函数，但随后我打算使用 ARM 的 NEON SIMD 功能对此数组执行一些操作。无论如何，我都不是 C 专家，所以我有点陷入困境。这是我正在尝试做的事情的草图。

JNIEXPORT jfloatArray JNICALL Java_com_blah (JNIEnv *env, jobject obj, jfloatArray input)
{
    jfloat * x;
    float32_t * y;

    x = (*env)->GetFloatArrayElements(env,input,0);

    // Assign x to y

    // Perform some stuff on y

    // Return y as a Java float array back to caller

}

我必须想出如何使用 C 来对整个数组进行类型转换。另外，也许有一种方法可以在从 Java 中提取数组条目时直接执行此操作（x = (*env)->GetFloatArrayElements(env,input,0)）。解决这个问题的正确方法是什么？

Answer 1

我认为您不必将整个数组分配给 y。如果您确定必须这样做，则可以使用 GetArrayLength 来查找浮点数组的大小。将元素一一复制。执行您的操作，然后创建一个新的浮点数组。
一定要调用ReleaseFloatArrayElements，否则会出现泄漏。