丁云鹏 vor 5 Monaten
Ursprung
Commit
a47cead094

+ 3 - 3
recommend-model-jni/paddle/build.sh

@@ -3,9 +3,9 @@
 #
 
 mkdir build && cd build
-export library_path=$1
-export jni_path=$2
-export jni_sub_path=$3
+export library_path=../paddle_inference_c
+export jni_path=/usr/local/jdk1.8.0_431/include
+export jni_sub_path=/usr/local/jdk1.8.0_431/include/linux
 mkldnn_lib=$library_path"/third_party/install/mkldnn/lib"
 mklml_lib=$library_path"/third_party/install/mklml/lib"
 export paddle_inference_lib=$library_path"/paddle/lib"

+ 1 - 1
recommend-model-jni/src/main/c/com/baidu/paddle/inference/com_baidu_paddle_inference_Predictor.cpp

@@ -51,7 +51,7 @@ JNIEXPORT jlong JNICALL
 Java_com_baidu_paddle_inference_Predictor_clonePredictor(
     JNIEnv* env, jobject obj, jlong cppPaddlePredictorPointer){
   return (jlong)PD_PredictorClone(
-        reinterpret_cast<PD_Config*>(cppPaddlePredictorPointer));
+        reinterpret_cast<PD_Predictor*>(cppPaddlePredictorPointer));
 }
 
 JNIEXPORT jlong JNICALL Java_com_baidu_paddle_inference_Predictor_getInputNum(