update

minmingzhu · Jul 16, 2024 · 2959da4 · 2959da4
1 parent 204a759
commit 2959da4
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/mllib-dal/src/main/native/CorrelationSampleImpl.cpp b/mllib-dal/src/main/native/CorrelationSampleImpl.cpp
@@ -32,6 +32,7 @@
 #include "Communicator.hpp"
 #include "oneapi/dal/table/row_accessor.hpp"
 #include "oneapi/dal/table/common.hpp"
+#include "com_intel_oap_mllib_stat_CorrelationDALImpl.h"
 
 
 namespace dal = oneapi::dal;

diff --git a/mllib-dal/src/main/scala/com/intel/oap/mllib/stat/CorrelationSample.scala b/mllib-dal/src/main/scala/com/intel/oap/mllib/stat/CorrelationSample.scala
@@ -88,11 +88,10 @@ object CorrelationSample {
     val executorCores = Utils.sparkExecutorCores()
     logger.info(s"executorNum ${executorNum}")
     logger.info(s"executorCores ${executorCores}")
-
+    val cor = new CorrelationDALImpl(executorNum, executorCores)
     val kvsIPPort = getOneCCLIPPort(data)
     data.mapPartitionsWithIndex { (rank, iter) =>
-      new CorrelationDALImpl(executorNum, executorCores)
-        .cCorrelationSampleTrainDAL(rank, executorNum, kvsIPPort)
+      cor.cCorrelationSampleTrainDAL(rank, executorNum, kvsIPPort)
       Iterator.empty
     }.collect()
     spark.stop()