update

minmingzhu · Jul 30, 2024 · e458ba6 · e458ba6
1 parent dafd254
commit e458ba6
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/mllib-dal/src/main/scala/com/intel/oap/mllib/stat/CorrelationDALImpl.scala b/mllib-dal/src/main/scala/com/intel/oap/mllib/stat/CorrelationDALImpl.scala
@@ -128,9 +128,12 @@ class CorrelationDALImpl(
       } else {
         Iterator.empty
       }
-      OneCCL.cleanup()
+      if (useDevice == "CPU") {
+         OneCCL.cleanup()
+      }
       ret
     }.collect()
+    logInfo(s"CorrelationDAL compute end")
     corTimer.record("Training")
     corTimer.print()