update

minmingzhu · Oct 15, 2024 · 79078f0 · 79078f0
1 parent f3d5e68
commit 79078f0
Show file tree

Hide file tree

Showing 9 changed files with 12 additions and 6 deletions.
diff --git a/mllib-dal/src/main/native/LinearRegressionImpl.cpp b/mllib-dal/src/main/native/LinearRegressionImpl.cpp
@@ -309,7 +309,7 @@ Java_com_intel_oap_mllib_regression_LinearRegressionDALImpl_cLinearRegressionTra
     jlong featureCols, jlong label, jlong labelCols, jboolean fitIntercept,
     jdouble regParam, jdouble elasticNetParam, jint executorNum,
     jint executorCores, jint computeDeviceOrdinal, jintArray gpuIdxArray, jstring ip_port, jstring breakdown_name,
-    jobject resultObj) {
+    jstring store_path, jobject resultObj) {
 
     logger::println(logger::INFO,
                     "oneDAL (native): use DPC++ kernels; device %s",

diff --git a/mllib-dal/src/main/native/PCAImpl.cpp b/mllib-dal/src/main/native/PCAImpl.cpp
@@ -266,7 +266,7 @@ JNIEXPORT jlong JNICALL
 Java_com_intel_oap_mllib_feature_PCADALImpl_cPCATrainDAL(
     JNIEnv *env, jobject obj, jint rank, jlong pNumTabData, jlong numRows, jlong numClos,
     jint executorNum, jint executorCores, jint computeDeviceOrdinal,
-    jintArray gpuIdxArray, jstring ip_port, jstring breakdown_name, jobject resultObj) {
+    jintArray gpuIdxArray, jstring ip_port, jstring breakdown_name, jstring store_path, jobject resultObj) {
     logger::println(logger::INFO,
                     "oneDAL (native): use DPC++ kernels; device %s",
                     ComputeDeviceString[computeDeviceOrdinal].c_str());

diff --git a/mllib-dal/src/main/native/SummarizerImpl.cpp b/mllib-dal/src/main/native/SummarizerImpl.cpp
@@ -292,7 +292,7 @@ JNIEXPORT jlong JNICALL
 Java_com_intel_oap_mllib_stat_SummarizerDALImpl_cSummarizerTrainDAL(
     JNIEnv *env, jobject obj, jint rank, jlong pNumTabData, jlong numRows, jlong numClos,
     jint executorNum, jint executorCores, jint computeDeviceOrdinal,
-    jintArray gpuIdxArray, jstring ip_port, jstring breakdown_name, jobject resultObj) {
+    jintArray gpuIdxArray, jstring ip_port, jstring breakdown_name, jstring store_path, jobject resultObj) {
     logger::println(logger::INFO,
                     "oneDAL (native): use DPC++ kernels; device %s",
                     ComputeDeviceString[computeDeviceOrdinal].c_str());

diff --git a/mllib-dal/src/main/native/javah/com_intel_oap_mllib_feature_PCADALImpl.h b/mllib-dal/src/main/native/javah/com_intel_oap_mllib_feature_PCADALImpl.h
diff --git a/mllib-dal/src/main/native/javah/com_intel_oap_mllib_regression_LinearRegressionDALImpl.h b/mllib-dal/src/main/native/javah/com_intel_oap_mllib_regression_LinearRegressionDALImpl.h
diff --git a/mllib-dal/src/main/native/javah/com_intel_oap_mllib_stat_SummarizerDALImpl.h b/mllib-dal/src/main/native/javah/com_intel_oap_mllib_stat_SummarizerDALImpl.h
diff --git a/mllib-dal/src/main/scala/com/intel/oap/mllib/feature/PCADALImpl.scala b/mllib-dal/src/main/scala/com/intel/oap/mllib/feature/PCADALImpl.scala
@@ -112,6 +112,7 @@ class PCADALImpl(val k: Int,
         gpuIndices,
         kvsIPPort,
         training_breakdown_name,
+        storePath,
         result
       )
 
@@ -251,5 +252,6 @@ class PCADALImpl(val k: Int,
                                    gpuIndices: Array[Int],
                                    kvsIPPort: String,
                                    training_breakdown_name: String,
+                                   storePath: String,
                                    result: PCAResult): Long
 }
diff --git a/mllib-dal/src/main/scala/com/intel/oap/mllib/regression/LinearRegressionDALImpl.scala b/mllib-dal/src/main/scala/com/intel/oap/mllib/regression/LinearRegressionDALImpl.scala
@@ -179,6 +179,7 @@ class LinearRegressionDALImpl( val fitIntercept: Boolean,
           gpuIndices,
           kvsIPPort,
           training_breakdown_name,
+          storePath,
           result
         )
 
@@ -227,6 +228,7 @@ class LinearRegressionDALImpl( val fitIntercept: Boolean,
                                   gpuIndices: Array[Int],
                                   kvsIPPort: String,
                                   training_breakdown_name: String,
+                                  storePath: String,
                                   result: LiRResult): Long
 
   }
diff --git a/mllib-dal/src/main/scala/com/intel/oap/mllib/stat/SummarizerDALImpl.scala b/mllib-dal/src/main/scala/com/intel/oap/mllib/stat/SummarizerDALImpl.scala
@@ -102,6 +102,7 @@ class SummarizerDALImpl(val executorNum: Int,
         gpuIndices,
         kvsIPPort,
         training_breakdown_name,
+        storePath,
         result
       )
 
@@ -188,5 +189,6 @@ class SummarizerDALImpl(val executorNum: Int,
                                           gpuIndices: Array[Int],
                                           kvsIPPort: String,
                                           training_breakdown_name: String,
+                                          storePath: String,
                                           result: SummarizerResult): Long
 }