Skip to content

Commit

Permalink
format cpp
Browse files Browse the repository at this point in the history
  • Loading branch information
KexinFeng committed Jun 27, 2022
1 parent 59bd30c commit 369b65a
Show file tree
Hide file tree
Showing 2 changed files with 12 additions and 11 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -116,14 +116,15 @@ JNIEXPORT jint JNICALL Java_ai_djl_fasttext_jni_FastTextLibrary_predictProba(
fasttext_ptr->predictLine(in, predictions, top_k, 0.0);

int size = predictions.size();
jclass java_lang_Float = static_cast<jclass>(env->NewGlobalRef(env->FindClass("java/lang/Float")));
jmethodID java_lang_Float_ = env->GetMethodID(java_lang_Float, "<init>", "(F)V");
jclass java_lang_Float = static_cast<jclass>(env->NewGlobalRef(env->FindClass("java/lang/Float")));
jmethodID java_lang_Float_ = env->GetMethodID(java_lang_Float, "<init>", "(F)V");
jclass java_util_ArrayList = static_cast<jclass>(env->NewGlobalRef(env->FindClass("java/util/ArrayList")));
jmethodID java_util_ArrayList_add = env->GetMethodID(java_util_ArrayList, "add", "(Ljava/lang/Object;)Z");
for (int i = 0; i < size; ++i) {
std::pair<real, std::string> pair = predictions[i];
env->CallBooleanMethod(jclasses, java_util_ArrayList_add, env->NewStringUTF(pair.second.c_str()));
env->CallBooleanMethod(jprob, java_util_ArrayList_add, env->NewObject(java_lang_Float, java_lang_Float_, pair.first));
env->CallBooleanMethod(jclasses, java_util_ArrayList_add, env->NewStringUTF(pair.second.c_str()));
env->CallBooleanMethod(
jprob, java_util_ArrayList_add, env->NewObject(java_lang_Float, java_lang_Float_, pair.first));
}

return size;
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -38,13 +38,13 @@ JNIEXPORT void JNICALL Java_ai_djl_sentencepiece_jni_SentencePieceLibrary_loadMo
}

JNIEXPORT void JNICALL Java_ai_djl_sentencepiece_jni_SentencePieceLibrary_loadModelFromBytes(
JNIEnv* env, jobject jthis, jlong jhandle, jbyteArray jserialized) {
auto* processor_ptr = reinterpret_cast<sentencepiece::SentencePieceProcessor*>(jhandle);
int length = env->GetArrayLength(jserialized);
std::vector<char> buff(length, 0);
env->GetByteArrayRegion(jserialized, 0, length, reinterpret_cast<jbyte*>(buff.data()));
std::string serialized(buff.data(), buff.size());
CheckStatus(env, processor_ptr->LoadFromSerializedProto(serialized));
JNIEnv* env, jobject jthis, jlong jhandle, jbyteArray jserialized) {
auto* processor_ptr = reinterpret_cast<sentencepiece::SentencePieceProcessor*>(jhandle);
int length = env->GetArrayLength(jserialized);
std::vector<char> buff(length, 0);
env->GetByteArrayRegion(jserialized, 0, length, reinterpret_cast<jbyte*>(buff.data()));
std::string serialized(buff.data(), buff.size());
CheckStatus(env, processor_ptr->LoadFromSerializedProto(serialized));
}

JNIEXPORT void JNICALL Java_ai_djl_sentencepiece_jni_SentencePieceLibrary_deleteSentencePieceProcessor(
Expand Down

0 comments on commit 369b65a

Please sign in to comment.