update

mllib-dal/src/main/native/GPU.cpp

@@ -115,7 +115,7 @@ sycl::queue getQueue(const ComputeDevice device) {
 
 preview::spmd::communicator<preview::spmd::device_memory_access::usm>
 createDalCommunicator(const jint executorNum, const jint rank,
-                      const ccl::string ccl_ip_port) {
+                      const ccl::string ccl_ip_port, std::string breakdown_name) {
     auto gpus = get_gpus();
 
     auto t1 = std::chrono::high_resolution_clock::now();
@@ -127,23 +127,14 @@ createDalCommunicator(const jint executorNum, const jint rank,
         (float)std::chrono::duration_cast<std::chrono::milliseconds>(t2 - t1)
             .count();
 
-    logger::println(logger::INFO, "OneCCL singleton init took %f secs",
-                    duration / 1000);
-
-    t1 = std::chrono::high_resolution_clock::now();
+    logger::Logger::getInstance(breakdown_name).printLogToFile("rankID was %d, OneCCL singleton init took %f secs.", rank, duration / 1000 );
 
     auto kvs_attr = ccl::create_kvs_attr();
 
     kvs_attr.set<ccl::kvs_attr_id::ip_port>(ccl_ip_port);
 
     ccl::shared_ptr_class<ccl::kvs> kvs = ccl::create_main_kvs(kvs_attr);
 
-    t2 = std::chrono::high_resolution_clock::now();
-    duration =
-        (float)std::chrono::duration_cast<std::chrono::milliseconds>(t2 - t1)
-            .count();
-    logger::println(logger::INFO, "OneCCL (native): create kvs took %f secs",
-                    duration / 1000);
     sycl::queue queue{gpus[0]};
     t1 = std::chrono::high_resolution_clock::now();
     auto comm = preview::spmd::make_communicator<preview::spmd::backend::ccl>(
@@ -152,5 +143,6 @@ createDalCommunicator(const jint executorNum, const jint rank,
     duration =
         (float)std::chrono::duration_cast<std::chrono::milliseconds>(t2 - t1)
             .count();
+    logger::Logger::getInstance(c_breakdown_name).printLogToFile("rankID was %d, create communicator took %f secs.", rank, duration / 1000 );
     return comm;
 }