mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-07-09 17:28:58 +00:00
896a963492
### Description This PR partially reverts changes introduced in https://github.com/microsoft/onnxruntime/pull/15643 We make two API return std::string always in UTF-8. We also move the entry points from OrtApiBase to OrtApi to make them versioned. ### Motivation and Context `GetVersionString` always returns x.y.z numbers that are not subject to internationalization. `GetBuildInfoString` can hold international chars, but UTF-8 should be fine to contain those. We prefix them with u8"" in case the compiler default charset is not UTF-8. Furthermore, creating platform dependent APIs is discouraged. `ORTCHAR_T` is platform dependent and was created for paths only. On non-unix platforms would still produce `std::string` that can only contain UTF-8 The API was introduced after the latest release, and can still be adjusted.
82 lines
3.2 KiB
C
82 lines
3.2 KiB
C
/*
|
|
* Copyright (c) 2019, 2023 Oracle and/or its affiliates. All rights reserved.
|
|
* Licensed under the MIT License.
|
|
*/
|
|
#include <jni.h>
|
|
#include <assert.h>
|
|
#include "onnxruntime/core/session/onnxruntime_c_api.h"
|
|
#include "ai_onnxruntime_OnnxRuntime.h"
|
|
#include "OrtJniUtil.h"
|
|
|
|
/*
|
|
* Class: ai_onnxruntime_OnnxRuntime
|
|
* Method: initialiseAPIBase
|
|
* Signature: (I)J
|
|
*/
|
|
JNIEXPORT jlong JNICALL Java_ai_onnxruntime_OnnxRuntime_initialiseAPIBase(JNIEnv* jniEnv, jclass clazz,
|
|
jint apiVersion) {
|
|
(void)jniEnv; (void)clazz; // required JNI parameters not needed by functions which don't call back into Java.
|
|
const OrtApi* ortPtr = OrtGetApiBase()->GetApi((uint32_t)apiVersion);
|
|
return (jlong) ortPtr;
|
|
}
|
|
/*
|
|
* Class: ai_onnxruntime_OnnxRuntime
|
|
* Method: initialiseTrainingAPIBase
|
|
* Signature: (JI)J
|
|
*/
|
|
JNIEXPORT jlong JNICALL Java_ai_onnxruntime_OnnxRuntime_initialiseTrainingAPIBase
|
|
(JNIEnv * jniEnv, jclass clazz, jlong apiHandle, jint trainingApiVersion) {
|
|
(void)jniEnv; (void)clazz; // required JNI parameters not needed by functions which don't call back into Java.
|
|
const OrtApi* api = (const OrtApi*)apiHandle;
|
|
const OrtTrainingApi* trainingApi = api->GetTrainingApi((uint32_t)trainingApiVersion);
|
|
return (jlong) trainingApi;
|
|
}
|
|
|
|
/*
|
|
* Class: ai_onnxruntime_OnnxRuntime
|
|
* Method: getAvailableProviders
|
|
* Signature: (J)[Ljava/lang/String;
|
|
*/
|
|
JNIEXPORT jobjectArray JNICALL Java_ai_onnxruntime_OnnxRuntime_getAvailableProviders(JNIEnv* jniEnv, jclass clazz,
|
|
jlong apiHandle) {
|
|
(void)jniEnv; (void)clazz; // required JNI parameters not needed by functions which don't call back into Java.
|
|
const OrtApi* api = (const OrtApi*)apiHandle;
|
|
|
|
char** providers = NULL;
|
|
int numProviders = 0;
|
|
|
|
// Extract the provider array
|
|
jobjectArray providerArray = NULL;
|
|
OrtErrorCode code = checkOrtStatus(jniEnv, api, api->GetAvailableProviders(&providers, &numProviders));
|
|
if (code == ORT_OK) {
|
|
// Convert to Java String Array
|
|
char* stringClassName = "java/lang/String";
|
|
jclass stringClazz = (*jniEnv)->FindClass(jniEnv, stringClassName);
|
|
providerArray = (*jniEnv)->NewObjectArray(jniEnv, numProviders, stringClazz, NULL);
|
|
|
|
for (int i = 0; i < numProviders; i++) {
|
|
// Read out the provider name and convert it to a java.lang.String
|
|
jstring provider = (*jniEnv)->NewStringUTF(jniEnv, providers[i]);
|
|
(*jniEnv)->SetObjectArrayElement(jniEnv, providerArray, i, provider);
|
|
}
|
|
|
|
// Release providers
|
|
// if this fails we return immediately anyway
|
|
checkOrtStatus(jniEnv, api, api->ReleaseAvailableProviders(providers, numProviders));
|
|
providers = NULL;
|
|
}
|
|
return providerArray;
|
|
}
|
|
|
|
/*
|
|
* Class: ai_onnxruntime_OnnxRuntime
|
|
* Method: initialiseVersion
|
|
* Signature: ()Ljava/lang/String;
|
|
*/
|
|
JNIEXPORT jstring JNICALL Java_ai_onnxruntime_OnnxRuntime_initialiseVersion
|
|
(JNIEnv * jniEnv, jclass clazz) {
|
|
(void)clazz; // required JNI parameter not needed by functions which don't access their host class.
|
|
const char* version = ORT_VERSION;
|
|
assert(version != NULL);
|
|
return (*jniEnv)->NewStringUTF(jniEnv, version);
|
|
}
|