fix train lite demo bug

mindspore/lite/java/java/app/src/main/java/com/mindspore/lite/LiteSession.java

@@ -28,6 +28,7 @@ import com.mindspore.lite.config.MSConfig;
 public class LiteSession {
     static {
         System.loadLibrary("mindspore-lite-jni");
+        System.loadLibrary("mindspore-lite-train-jni");
     }
 
     private long sessionPtr = 0;

mindspore/lite/java/java/fl_client/src/main/java/com/mindspore/flclient/model/TrainLenet.java

@@ -108,6 +108,8 @@ public class TrainLenet extends TrainModel {
                 labelArray[j * NUM_OF_CLASS + labelByteArray[i]] = 1;
                 j++;
             }
+        } else {
+            labelArray = null;  // labelArray may be initialized from train
         }
 
         return trainSampleSize;

mindspore/lite/java/java/linux_x86/src/main/java/com.mindspore.lite/LiteSession.java

@@ -28,6 +28,7 @@ import com.mindspore.lite.config.MSConfig;
 public class LiteSession {
     static {
         System.loadLibrary("mindspore-lite-jni");
+        System.loadLibrary("mindspore-lite-train-jni");
     }
 
     private long sessionPtr = 0;

mindspore/lite/java/native/runtime/train_session.cpp

@@ -20,10 +20,8 @@
 #include "include/train/train_cfg.h"
 #include "include/errorcode.h"
 
-extern "C" JNIEXPORT jlong JNICALL Java_com_mindspore_lite_LiteSession_createTrainSession(JNIEnv *env, jobject thiz,
-                                                                                          jstring file_name,
-                                                                                          jlong ms_context_ptr,
-                                                                                          jboolean train_mode) {
+extern "C" JNIEXPORT jlong JNICALL Java_com_mindspore_lite_LiteSession_createTrainSession(
+  JNIEnv *env, jobject thiz, jstring file_name, jlong ms_context_ptr, jboolean train_mode, jlong train_config_ptr) {
   auto *pointer = reinterpret_cast<void *>(ms_context_ptr);
   if (pointer == nullptr) {
     MS_LOGE("Context pointer from java is nullptr");
@@ -39,5 +37,3 @@ extern "C" JNIEXPORT jlong JNICALL Java_com_mindspore_lite_LiteSession_createTra
   }
   return jlong(session);
 }
-
-

mindspore/lite/test/st/scripts/run_benchmark_x86.sh

@@ -237,7 +237,7 @@ function Run_x86_java() {
     cp ${x86_path}/avx/mindspore-lite-${version}-linux-x64.tar.gz ./java/ || exit 1
     cd ./java || exit 1
     tar -zxf mindspore-lite-${version}-linux-x64.tar.gz || exit 1
-    export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:${x86_path}/java/mindspore-lite-${version}-linux-x64/runtime/lib
+    export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:${x86_path}/java/mindspore-lite-${version}-linux-x64/runtime/lib:${x86_path}/java/mindspore-lite-${version}-linux-x64/runtime/third_party/libjpeg-turbo/lib
     # compile benchmark
     echo "javac -cp ${x86_path}/java/mindspore-lite-${version}-linux-x64/runtime/lib/mindspore-lite-java.jar ${basepath}/java/src/main/java/Benchmark.java -d ."
     javac -cp ${x86_path}/java/mindspore-lite-${version}-linux-x64/runtime/lib/mindspore-lite-java.jar ${basepath}/java/src/main/java/Benchmark.java -d .